OpenSceneGraph/include/osgIntrospection/CustomAttributeProvider

#ifndef OSGINTROSPECTION_CUSTOMATTRIBUTEPROVIDER_
#define OSGINTROSPECTION_CUSTOMATTRIBUTEPROVIDER_

#include <osgIntrospection/Export>
#include <osgIntrospection/CustomAttribute>

#include <vector>
#include <typeinfo>

namespace osgIntrospection
{

	// forward declarations
	class Type;
	class CustomAttribute;
	class CustomAttributeProvider;

	// vector of attributes
	typedef std::vector<const CustomAttribute *> CustomAttributeList;

	// vector of attribute providers
	typedef std::vector<const CustomAttributeProvider *> CustomAttributeProviderList;


	/// This is the base class for custom attribute providers, that is objects
	/// that can be assigned a list of custom attributes. Methods defined in
	/// this class provide the means for adding, retrieving and searching for
	/// attributes.
	class OSGINTROSPECTION_EXPORT CustomAttributeProvider
	{
	public:
		/// Returns the const list of custom attributes.
		inline const CustomAttributeList &getCustomAttributes() const
		{
			return attribs_;
		}

		/// Returns the list of custom attributes.
		inline CustomAttributeList &getCustomAttributes() 
		{
			return attribs_;
		}

		/// Adds a new attribute to the list.
		inline CustomAttributeProvider *addAttribute(const CustomAttribute *attr)
		{
			attribs_.push_back(attr);
			return this;
		}

		/// Returns whether at least one attribute of the given type is
		/// present in the attribute list. If the inherit parameter is
		/// set to true, the search is forwarded to base types.
		bool isDefined(const Type &type, bool inherit) const;

		/// Returns whether at least one attribute of the given type is
		/// present in the attribute list. If the inherit parameter is
		/// set to true, the search is forwarded to base types.
		/// [template version]
		template<typename T> inline bool isDefined(bool inherit) const
		{
			for (CustomAttributeList::const_iterator i=attribs_.begin(); i!=attribs_.end(); ++i)
				if (typeid(**i) == typeid(T)) return true;
			if (inherit)
			{
				CustomAttributeProviderList providers;
				getInheritedProviders(providers);
				for (CustomAttributeProviderList::const_iterator i=providers.begin(); i!=providers.end(); ++i)
				{
					if ((*i)->isDefined<T>(true)) return true;
				}
			}
			return false;
		}

		/// Searchs for an attribute of the given type and returns a pointer
		/// to it if found, a null pointer otherwise. If the inherit parameter
		/// is set to true, the search is forwarded to base types.
		const CustomAttribute *getAttribute(const Type &type, bool inherit) const;

		/// Searchs for an attribute of the given type and returns a pointer
		/// to it if found, a null pointer otherwise. If the inherit parameter
		/// is set to true, the search is forwarded to base types.
		/// [template version]
		template<typename T> inline const T *getAttribute(bool inherit) const
		{
			for (CustomAttributeList::const_iterator i=attribs_.begin(); i!=attribs_.end(); ++i)
				if (typeid(**i) == typeid(T)) return static_cast<const T *>(*i);
			if (inherit)
			{
				CustomAttributeProviderList providers;
				getInheritedProviders(providers);
				for (CustomAttributeProviderList::const_iterator i=providers.begin(); i!=providers.end(); ++i)
				{
					const T *ca = (*i)->getAttribute<T>(true);
					if (ca) return ca;
				}
			}
			return 0;
		}

	protected:
		virtual void getInheritedProviders(CustomAttributeProviderList &providers) const = 0;
		virtual ~CustomAttributeProvider() {}

	private:
		CustomAttributeList attribs_;
	};

}

#endif