#ifndef OBJECT_VECTOR_H
#define OBJECT_VECTOR_H
#include <vector>
#include "object.h"
#include "ptr.h"
#include "attribute.h"
namespace ns3 {
/**
* \brief contain a vector of ns3::Object pointers.
*
* This class it used to get attribute access to an array of
* ns3::Object pointers.
*/
class ObjectVector
{
public:
typedef std::vector<Ptr<Object> >::const_iterator Iterator;
ObjectVector ();
Iterator Begin (void) const;
Iterator End (void) const;
uint32_t GetN (void) const;
Ptr<Object> Get (uint32_t i) const;
ObjectVector (Attribute value);
operator Attribute () const;
private:
friend class ObjectVectorAccessor;
std::vector<Ptr<Object> > m_objects;
};
template <typename T, typename U>
Ptr<const AttributeAccessor>
MakeObjectVectorAccessor (U T::*memberVector);
template <typename T, typename U, typename INDEX>
Ptr<const AttributeAccessor>
MakeObjectVectorAccessor (Ptr<U> (T::*get) (INDEX) const,
INDEX (T::*getN) (void) const);
template <typename T, typename U, typename INDEX>
Ptr<const AttributeAccessor>
MakeObjectVectorAccessor (INDEX (T::*getN) (void) const,
Ptr<U> (T::*get) (INDEX) const);
class ObjectVectorChecker : public AttributeChecker
{
public:
virtual TypeId GetItemTypeId (void) const = 0;
};
template <typename T>
Ptr<const AttributeChecker> MakeObjectVectorChecker (void);
} // namespace ns3
namespace ns3 {
class ObjectVectorValue : public AttributeValue
{
public:
ObjectVectorValue ();
ObjectVectorValue (const ObjectVector &vector);
ObjectVector Get (void) const;
virtual Attribute Copy (void) const;
virtual std::string SerializeToString (Ptr<const AttributeChecker> checker) const;
virtual bool DeserializeFromString (std::string value, Ptr<const AttributeChecker> checker);
private:
friend class ObjectVectorAccessor;
ObjectVector m_vector;
};
namespace internal {
template <typename T>
class AnObjectVectorChecker : public ObjectVectorChecker
{
public:
virtual TypeId GetItemTypeId (void) const {
return T::GetTypeId ();
}
virtual bool Check (Attribute value) const {
return value.DynCast<const ObjectVectorValue *> () != 0;
}
virtual std::string GetType (void) const {
return "ns3::ObjectVector";
}
virtual bool HasTypeConstraints (void) const {
return true;
}
virtual std::string GetTypeConstraints (void) const {
return T::GetTypeId ().GetName ();
}
virtual Attribute Create (void) const {
return Attribute::Create<ObjectVectorValue> ();
}
};
} // namespace internal
class ObjectVectorAccessor : public AttributeAccessor
{
public:
virtual bool Set (ObjectBase * object, Attribute value) const;
virtual bool Get (const ObjectBase * object, Attribute value) const;
private:
virtual bool DoGetN (const ObjectBase *object, uint32_t *n) const = 0;
virtual Ptr<Object> DoGet (const ObjectBase *object, uint32_t i) const = 0;
};
template <typename T, typename U>
Ptr<const AttributeAccessor>
MakeObjectVectorAccessor (U T::*memberVector)
{
struct MemberStdContainer : public ObjectVectorAccessor
{
virtual bool DoGetN (const ObjectBase *object, uint32_t *n) const {
const T *obj = dynamic_cast<const T *> (object);
if (obj == 0)
{
return false;
}
*n = (obj->*m_memberVector).size ();
return true;
}
virtual Ptr<Object> DoGet (const ObjectBase *object, uint32_t i) const {
const T *obj = static_cast<const T *> (object);
typename U::const_iterator begin = (obj->*m_memberVector).begin ();
typename U::const_iterator end = (obj->*m_memberVector).end ();
uint32_t k = 0;
for (typename U::const_iterator j = begin; j != end; j++, k++)
{
if (k == i)
{
return *j;
break;
}
}
NS_ASSERT (false);
// quiet compiler.
return 0;
}
U T::*m_memberVector;
} *spec = new MemberStdContainer ();
spec->m_memberVector = memberVector;
return Ptr<const AttributeAccessor> (spec, false);
}
template <typename T, typename U, typename INDEX>
Ptr<const AttributeAccessor>
MakeObjectVectorAccessor (Ptr<U> (T::*get) (INDEX) const,
INDEX (T::*getN) (void) const)
{
struct MemberGetters : public ObjectVectorAccessor
{
virtual bool DoGetN (const ObjectBase *object, uint32_t *n) const {
const T *obj = dynamic_cast<const T *> (object);
if (obj == 0)
{
return false;
}
*n = (obj->*m_getN) ();
return true;
}
virtual Ptr<Object> DoGet (const ObjectBase *object, uint32_t i) const {
const T *obj = static_cast<const T *> (object);
return (obj->*m_get) (i);
}
Ptr<U> (T::*m_get) (INDEX) const;
INDEX (T::*m_getN) (void) const;
} *spec = new MemberGetters ();
spec->m_get = get;
spec->m_getN = getN;
return Ptr<const AttributeAccessor> (spec, false);
}
template <typename T, typename U, typename INDEX>
Ptr<const AttributeAccessor>
MakeObjectVectorAccessor (INDEX (T::*getN) (void) const,
Ptr<U> (T::*get) (INDEX) const)
{
return MakeObjectVectorAccessor (get, getN);
}
template <typename T>
Ptr<const AttributeChecker> MakeObjectVectorChecker (void)
{
return Create<internal::AnObjectVectorChecker<T> > ();
}
} // namespace ns3
#endif /* OBJECT_VECTOR_H */