/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */

/*

 * Copyright (c) 2007 INRIA, Gustavo Carneiro

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License version 2 as

 * published by the Free Software Foundation;

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 *

 * Authors: Gustavo Carneiro <gjcarneiro@gmail.com>,

 *          Mathieu Lacage <mathieu.lacage@sophia.inria.fr>

 */

#include "object.h"

#include "assert.h"

#include "singleton.h"

#include "uid-manager.h"

#include <vector>

namespace {

class IidManager : public ns3::UidManager

{};

class IidTree

{

public:

  void SetParent (uint16_t child, const uint16_t *parent);

  uint16_t LookupParent (uint16_t child);

private:

  std::vector<const uint16_t *> m_parents;

};

void 

IidTree::SetParent (uint16_t child, const uint16_t *parent)

{

  m_parents.resize (child+1);

  m_parents[child] = parent;

}

uint16_t 

IidTree::LookupParent (uint16_t child)

{

  NS_ASSERT (child < m_parents.size ());

  return *(m_parents[child]);

}

} // anonymous namespace

namespace ns3 {

class InterfaceIdTraceResolver : public TraceResolver

{

public:

  InterfaceIdTraceResolver (Ptr<Object> aggregate);

  virtual void Connect (std::string path, CallbackBase const &cb, const TraceContext &context);

  virtual void Disconnect (std::string path, CallbackBase const &cb);

private:

  Ptr<Object> ParseForInterface (std::string path);

  Ptr<Object> m_aggregate;

};

InterfaceIdTraceResolver::InterfaceIdTraceResolver (Ptr<Object> aggregate)

  : m_aggregate (aggregate)

{}

Ptr<Object>

InterfaceIdTraceResolver::ParseForInterface (std::string path)

{

  std::string element = GetElement (path);

  std::string::size_type dollar_pos = element.find ("$");

  if (dollar_pos != 0)

    {

      return 0;

    }

  std::string interfaceName = element.substr (1, std::string::npos);

  InterfaceId interfaceId = InterfaceId::LookupByName (interfaceName);

  Ptr<Object> interface = m_aggregate->QueryInterface<Object> (interfaceId);

  return interface;

}

void 

InterfaceIdTraceResolver::Connect (std::string path, CallbackBase const &cb, const TraceContext &context)

{

  Ptr<Object> interface = ParseForInterface (path);

  if (interface != 0)

    {

      interface->TraceConnect (GetSubpath (path), cb, context);

    }

}

void 

InterfaceIdTraceResolver::Disconnect (std::string path, CallbackBase const &cb)

{

  Ptr<Object> interface = ParseForInterface (path);

  if (interface != 0)

    {

      interface->TraceDisconnect (GetSubpath (path), cb);

    }

}

InterfaceId::InterfaceId (uint16_t iid)

  : m_iid (iid)

{}

InterfaceId::~InterfaceId ()

{}

InterfaceId 

InterfaceId::LookupByName (std::string name)

{

  uint32_t uid = Singleton<IidManager>::Get ()->LookupByName (name);

  NS_ASSERT (uid != 0 && uid <= 0xffff);

  return InterfaceId (uid);

}

InterfaceId 

InterfaceId::LookupParent (InterfaceId iid)

{

  return Singleton<IidTree>::Get ()->LookupParent (iid.m_iid);

}

bool operator == (const InterfaceId &a, const InterfaceId &b)

{

  return a.m_iid == b.m_iid;

}

bool operator != (const InterfaceId &a, const InterfaceId &b)

{

  return a.m_iid != b.m_iid;

}

InterfaceId

MakeInterfaceId (std::string name, const InterfaceId &parent)

{

  uint32_t uid = Singleton<IidManager>::Get ()->Allocate (name);

  NS_ASSERT (uid <= 0xffff);

  InterfaceId iid = uid;

  Singleton<IidTree>::Get ()->SetParent (iid.m_iid, &parent.m_iid);

  return iid;

}

InterfaceId

MakeObjectInterfaceId (void)

{

  InterfaceId iid = Singleton<IidManager>::Get ()->Allocate ("Object");

  Singleton<IidTree>::Get ()->SetParent (iid.m_iid, &iid.m_iid);

  return iid;

}

const InterfaceId Object::iid = MakeObjectInterfaceId ();

Object::Object ()

  : m_count (1),

    m_iid (Object::iid),

    m_disposed (false),

    m_next (this)

{}

Object::~Object () 

{

  m_next = 0;

}

Ptr<Object>

Object::DoQueryInterface (InterfaceId iid) const

{

  NS_ASSERT (Check ());

  const Object *currentObject = this;

  do {

    NS_ASSERT (currentObject != 0);

    InterfaceId cur = currentObject->m_iid;

    while (cur != iid && cur != Object::iid)

      {

        cur = InterfaceId::LookupParent (cur);

      }

    if (cur == iid)

      {

        return const_cast<Object *> (currentObject);

      }

    currentObject = currentObject->m_next;

  } while (currentObject != this);

  return 0;

}

void 

Object::Dispose (void)

{

  Object *current = this;

  do {

    NS_ASSERT (current != 0);

    NS_ASSERT (!current->m_disposed);

    current->DoDispose ();

    current->m_disposed = true;

    current = current->m_next;

  } while (current != this);

}

void 

Object::AddInterface (Ptr<Object> o)

{

  NS_ASSERT (!m_disposed);

  NS_ASSERT (!o->m_disposed);

  NS_ASSERT (Check ());

  NS_ASSERT (o->Check ());

  Object *other = PeekPointer (o);

  Object *next = m_next;

  m_next = other->m_next;

  other->m_next = next;

  NS_ASSERT (Check ());

  NS_ASSERT (o->Check ());

}

void 

Object::TraceConnect (std::string path, const CallbackBase &cb)

{

  TraceConnect (path, cb, TraceContext ());

}

void 

Object::TraceConnect (std::string path, const CallbackBase &cb, const TraceContext &context)

{

  GetTraceResolver ()->Connect (path, cb, context);

}

void 

Object::TraceDisconnect (std::string path, const CallbackBase &cb)

{

  GetTraceResolver ()->Disconnect (path, cb);

}

void 

Object::SetInterfaceId (InterfaceId iid)

{

  NS_ASSERT (Check ());

  m_iid = iid;

}

void

Object::DoDispose (void)

{

  NS_ASSERT (!m_disposed);

}

Ptr<TraceResolver>

Object::GetTraceResolver (void)

{

  Ptr<InterfaceIdTraceResolver> resolver =

    Create<InterfaceIdTraceResolver> (this);

  return resolver;

}

bool 

Object::Check (void) const

{

  return (m_count > 0);

}

void

Object::MaybeDelete (void) const

{

  // First, check if any of the attached

  // Object has a non-zero count.

  const Object *current = this;

  do {

    NS_ASSERT (current != 0);

    if (current->m_count != 0)

      {

        return;

      }

    current = current->m_next;

  } while (current != this);

  // all attached objects have a zero count so, 

  // we can delete all attached objects.

  current = this;

  const Object *end = this;

  do {

    NS_ASSERT (current != 0);

    Object *next = current->m_next;

    delete current;

    current = next;

  } while (current != end);

}

} // namespace ns3

#ifdef RUN_SELF_TESTS

#include "test.h"

namespace {

class BaseA : public ns3::Object

{

public:

  static const ns3::InterfaceId iid;

  BaseA ()

  {

    SetInterfaceId (BaseA::iid);

  }

  virtual void Dispose (void) {}

};

class DerivedA : public BaseA

{

public:

  static const ns3::InterfaceId iid;

  DerivedA (int v)

  {

    SetInterfaceId (DerivedA::iid);

  }

  virtual void Dispose (void) {

    BaseA::Dispose ();

  }

};

const ns3::InterfaceId BaseA::iid = 

  ns3::MakeInterfaceId ("BaseA", Object::iid);

const ns3::InterfaceId DerivedA::iid = 

  ns3::MakeInterfaceId ("DerivedA", BaseA::iid);;

class BaseB : public ns3::Object

{

public:

  static const ns3::InterfaceId iid;

  BaseB ()

  {

    SetInterfaceId (BaseB::iid);

  }

  virtual void Dispose (void) {}

};

class DerivedB : public BaseB

{

public:

  static const ns3::InterfaceId iid;

  DerivedB (int v)

  {

    SetInterfaceId (DerivedB::iid);

  }

  virtual void Dispose (void) {

    BaseB::Dispose ();

  }

};

const ns3::InterfaceId BaseB::iid = 

  ns3::MakeInterfaceId ("BaseB", Object::iid);

const ns3::InterfaceId DerivedB::iid = 

  ns3::MakeInterfaceId ("DerivedB", BaseB::iid);;

} // namespace anonymous

namespace ns3 {

class ObjectTest : public Test

{

public:

  ObjectTest ();

  virtual bool RunTests (void);

};

ObjectTest::ObjectTest ()

  : Test ("Object")

{}

bool 

ObjectTest::RunTests (void)

{

  bool result = true;

  Ptr<BaseA> baseA = Create<BaseA> ();

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<BaseA> (BaseA::iid), baseA);

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<BaseA> (DerivedA::iid), 0);

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<DerivedA> (DerivedA::iid), 0);

  baseA = Create<DerivedA> (10);

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<BaseA> (BaseA::iid), baseA);

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<BaseA> (DerivedA::iid), baseA);

  NS_TEST_ASSERT_UNEQUAL (baseA->QueryInterface<DerivedA> (DerivedA::iid), 0);

  baseA = Create<BaseA> ();

  Ptr<BaseB> baseB = Create<BaseB> ();

  Ptr<BaseB> baseBCopy = baseB;

  baseA->AddInterface (baseB);

  NS_TEST_ASSERT_UNEQUAL (baseA->QueryInterface<BaseA> (BaseA::iid), 0);

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<DerivedA> (DerivedA::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseA->QueryInterface<BaseB> (BaseB::iid), 0);

  NS_TEST_ASSERT_EQUAL (baseA->QueryInterface<DerivedB> (DerivedB::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseB->QueryInterface<BaseB> (BaseB::iid), 0);

  NS_TEST_ASSERT_EQUAL (baseB->QueryInterface<DerivedB> (DerivedB::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseB->QueryInterface<BaseA> (BaseA::iid), 0);

  NS_TEST_ASSERT_EQUAL (baseB->QueryInterface<DerivedA> (DerivedA::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseBCopy->QueryInterface<BaseA> (BaseA::iid), 0);

  baseA = Create<DerivedA> (1);

  baseB = Create<DerivedB> (1);

  baseBCopy = baseB;

  baseA->AddInterface (baseB);

  NS_TEST_ASSERT_UNEQUAL (baseA->QueryInterface<DerivedB> (DerivedB::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseA->QueryInterface<BaseB> (BaseB::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseB->QueryInterface<DerivedA> (DerivedA::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseB->QueryInterface<BaseA> (BaseA::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseBCopy->QueryInterface<DerivedA> (DerivedA::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseBCopy->QueryInterface<BaseA> (BaseA::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseB->QueryInterface<DerivedB> (DerivedB::iid), 0);

  NS_TEST_ASSERT_UNEQUAL (baseB->QueryInterface<BaseB> (BaseB::iid), 0)

  baseA = Create<BaseA> ();

  baseB = Create<BaseB> ();

  baseA->AddInterface (baseB);

  baseA = 0;

  baseA = baseB->QueryInterface<BaseA> (BaseA::iid);