156 lines
4.0 KiB
C++
156 lines
4.0 KiB
C++
#ifndef OBJECT_VECTOR_H
|
|
#define OBJECT_VECTOR_H
|
|
|
|
#include <vector>
|
|
#include "object.h"
|
|
#include "ptr.h"
|
|
#include "attribute.h"
|
|
#include "attribute-helper.h"
|
|
|
|
namespace ns3 {
|
|
|
|
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);
|
|
|
|
|
|
ATTRIBUTE_CHECKER_DEFINE (ObjectVector);
|
|
|
|
} // 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;
|
|
};
|
|
|
|
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);
|
|
}
|
|
|
|
} // namespace ns3
|
|
|
|
#endif /* OBJECT_VECTOR_H */
|