#include "ns3/object.h"

using namespace ns3;

class AnObject : public Object

{

public:

  static const InterfaceId iid;

  AnObject ();

protected:

  virtual void DoDispose (void);

};

const InterfaceId AnObject::iid = MakeInterfaceId ("AnObject", Object::iid);

AnObject::AnObject ()

{

  // enable our interface

  SetInterfaceId (AnObject::iid);

}

void

AnObject::DoDispose (void)

{

  // Do your work here.

  // chain up

  Object::DoDispose ();

}

class AnotherObject : public Object

{

public:

  static const InterfaceId iid;

  AnotherObject (int a);

private:

  virtual void DoDispose (void);

};

const InterfaceId AnotherObject::iid = MakeInterfaceId ("AnotherObject", Object::iid);

AnotherObject::AnotherObject (int a)

{

  // enable our interface

  SetInterfaceId (AnotherObject::iid);

}

void

AnotherObject::DoDispose (void)

{

  // Do your work here.

  // chain up

  Object::DoDispose ();

}

class YetAnotherObject : public Object

{

public:

  static const InterfaceId iid;

  YetAnotherObject (int a);

private:

  virtual void DoDispose (void);

};

const InterfaceId YetAnotherObject::iid = MakeInterfaceId ("YetAnotherObject", Object::iid);

YetAnotherObject::YetAnotherObject (int a)

{

  // enable our interface

  SetInterfaceId (YetAnotherObject::iid);

  // aggregated directly to another object.

  AddInterface (Create<AnObject> ());

}

void

YetAnotherObject::DoDispose (void)

{

  // Do your work here.

  // chain up

  Object::DoDispose ();

}

int main (int argc, char *argv[])

{

  Ptr<Object> p;

  Ptr<AnObject> anObject;

  Ptr<AnotherObject> anotherObject;

  Ptr<YetAnotherObject> yetAnotherObject;

  p = Create<AnObject> ();

  // p gives you access to AnObject's interface

  anObject = p->QueryInterface<AnObject> (AnObject::iid);

  NS_ASSERT (anObject != 0);

  // p does not give you access to AnotherObject's interface

  anotherObject = p->QueryInterface<AnotherObject> (AnotherObject::iid);

  NS_ASSERT (anotherObject == 0);

  anotherObject = Create<AnotherObject> (1);

  // AnotherObject does not give you access to AnObject's interface

  anObject = anotherObject->QueryInterface<AnObject> (AnObject::iid);

  NS_ASSERT (anObject == 0);

  // aggregate the two objects

  p->AddInterface (anotherObject);

  // p gives you access to AnObject's interface

  anObject = p->QueryInterface<AnObject> (AnObject::iid);

  NS_ASSERT (anObject != 0);

  // p gives you access to AnotherObject's interface

  anotherObject = p->QueryInterface<AnotherObject> (AnotherObject::iid);

  NS_ASSERT (anotherObject != 0);

  yetAnotherObject = Create<YetAnotherObject> (2);

  // gives you acess to AnObject interface too.

  anObject = yetAnotherObject->QueryInterface<AnObject> (AnObject::iid);

  NS_ASSERT (anObject != 0);

  return 0;

}