/* -*- 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 "trace-resolver.h"

#include "value.h"

#include "log.h"

#include <vector>

#include <sstream>

NS_LOG_COMPONENT_DEFINE ("Object");

/*********************************************************************

 *         Helper code

 *********************************************************************/

namespace {

class IidManager

{

public:

  IidManager ();

  uint16_t AllocateUid (std::string name);

  void SetParent (uint16_t uid, uint16_t parent);

  void SetTypeName (uint16_t uid, std::string typeName);

  void SetGroupName (uint16_t uid, std::string groupName);

  void AddConstructor (uint16_t uid, ns3::CallbackBase callback, uint32_t nArguments);

  uint16_t GetUid (std::string name) const;

  std::string GetName (uint16_t uid) const;

  uint16_t GetParent (uint16_t uid) const;

  std::string GetTypeName (uint16_t uid) const;

  std::string GetGroupName (uint16_t uid) const;

  ns3::CallbackBase GetConstructor (uint16_t uid, uint32_t nArguments);

  bool HasConstructor (uint16_t uid);

  uint32_t GetRegisteredN (void);

  uint16_t GetRegistered (uint32_t i);

  void AddParameter (uint16_t uid, 

                     std::string name,

                     std::string help, 

                     uint32_t flags,

                     ns3::Ptr<const ns3::ParamSpec> spec);

  uint32_t GetParametersN (uint16_t uid) const;

  std::string GetParameterName (uint16_t uid, uint32_t i) const;

  uint32_t GetParameterFlags (uint16_t uid, uint32_t i) const;

  uint32_t GetParameterUid (uint16_t uid, uint32_t i) const;

  ns3::Ptr<const ns3::ParamSpec> GetParameterParamSpec (uint16_t uid, uint32_t i) const;

private:

  struct ConstructorInformation {

    ns3::CallbackBase cb;

    uint32_t nArguments;

  };

  struct ParameterInformation {

    std::string name;

    std::string help;

    uint32_t flags;

    uint32_t uid;

    ns3::Ptr<const ns3::ParamSpec> param;

  };

  struct IidInformation {

    std::string name;

    uint16_t parent;

    std::string typeName;

    std::string groupName;

    std::vector<struct ConstructorInformation> constructors;

    std::vector<struct ParameterInformation> parameters;

  };

  typedef std::vector<struct IidInformation>::const_iterator Iterator;

  struct IidManager::IidInformation *LookupInformation (uint16_t uid) const;

  std::vector<struct IidInformation> m_information;

};

IidManager::IidManager ()

{}

uint16_t 

IidManager::AllocateUid (std::string name)

{

  uint16_t j = 1;

  for (Iterator i = m_information.begin (); i != m_information.end (); i++)

    {

      if (i->name == name)

        {

          NS_FATAL_ERROR ("Trying to allocate twice the same uid: " << name);

          return 0;

        }

      j++;

    }

  struct IidInformation information;

  information.name = name;

  information.parent = 0;

  information.typeName = "";

  information.groupName = "";

  m_information.push_back (information);

  uint32_t uid = m_information.size ();

  NS_ASSERT (uid <= 0xffff);

  return uid;

}

struct IidManager::IidInformation *

IidManager::LookupInformation (uint16_t uid) const

{

  NS_ASSERT (uid <= m_information.size ());

  return const_cast<struct IidInformation *> (&m_information[uid-1]);

}

void 

IidManager::SetParent (uint16_t uid, uint16_t parent)

{

  NS_ASSERT (parent <= m_information.size ());

  struct IidInformation *information = LookupInformation (uid);

  information->parent = parent;

}

void 

IidManager::SetTypeName (uint16_t uid, std::string typeName)

{

  struct IidInformation *information = LookupInformation (uid);

  information->typeName = typeName;

}

void 

IidManager::SetGroupName (uint16_t uid, std::string groupName)

{

  struct IidInformation *information = LookupInformation (uid);

  information->groupName = groupName;

}

void 

IidManager::AddConstructor (uint16_t uid, ns3::CallbackBase callback, uint32_t nArguments)

{

  struct IidInformation *information = LookupInformation (uid);

  struct ConstructorInformation constructor;

  constructor.cb = callback;

  constructor.nArguments = nArguments;

  for (std::vector<struct ConstructorInformation>::const_iterator i = information->constructors.begin ();

       i != information->constructors.end (); i++)

    {

      if (i->nArguments == nArguments)

        {

          NS_FATAL_ERROR ("registered two constructors on the same type with the same number of arguments.");

          break;

        }

    }

  information->constructors.push_back (constructor);

}

uint16_t 

IidManager::GetUid (std::string name) const

{

  uint32_t j = 1;

  for (Iterator i = m_information.begin (); i != m_information.end (); i++)

    {

      if (i->name == name)

        {

          NS_ASSERT (j <= 0xffff);

          return j;

        }

      j++;

    }

  return 0;

}

std::string 

IidManager::GetName (uint16_t uid) const

{

  struct IidInformation *information = LookupInformation (uid);

  return information->name;

}

uint16_t 

IidManager::GetParent (uint16_t uid) const

{

  struct IidInformation *information = LookupInformation (uid);

  return information->parent;

}

std::string 

IidManager::GetTypeName (uint16_t uid) const

{

  struct IidInformation *information = LookupInformation (uid);

  return information->typeName;

}

std::string 

IidManager::GetGroupName (uint16_t uid) const

{

  struct IidInformation *information = LookupInformation (uid);

  return information->groupName;

}

ns3::CallbackBase 

IidManager::GetConstructor (uint16_t uid, uint32_t nArguments)

{

  struct IidInformation *information = LookupInformation (uid);

  for (std::vector<struct ConstructorInformation>::const_iterator i = information->constructors.begin ();

       i != information->constructors.end (); i++)

    {

      if (i->nArguments == nArguments)

        {

          return i->cb;

        } 

    }

  NS_FATAL_ERROR ("Requested constructor with "<<nArguments<<" arguments not found");

  return ns3::CallbackBase ();

}

bool 

IidManager::HasConstructor (uint16_t uid)

{

  struct IidInformation *information = LookupInformation (uid);

  return !information->constructors.empty ();

}

uint32_t 

IidManager::GetRegisteredN (void)

{

  return m_information.size ();

}

uint16_t 

IidManager::GetRegistered (uint32_t i)

{

  return i + 1;

}

void 

IidManager::AddParameter (uint16_t uid, 

                          std::string name,

                          std::string help, 

                          uint32_t flags,

                          ns3::Ptr<const ns3::ParamSpec> spec)

{

  struct IidInformation *information = LookupInformation (uid);

  for (std::vector<struct ParameterInformation>::const_iterator j = information->parameters.begin ();

       j != information->parameters.end (); j++)

    {

      if (j->name == name)

        {

          NS_FATAL_ERROR ("Registered the same parameter twice name=\""<<name<<"\" in TypeId=\""<<information->name<<"\"");

          return;

        }

    }

  struct ParameterInformation param;

  param.name = name;

  param.help = help;

  param.flags = flags;

  param.param = spec;

  information->parameters.push_back (param);

}

uint32_t 

IidManager::GetParametersN (uint16_t uid) const

{

  struct IidInformation *information = LookupInformation (uid);

  return information->parameters.size ();

}

std::string 

IidManager::GetParameterName (uint16_t uid, uint32_t i) const

{

  struct IidInformation *information = LookupInformation (uid);

  NS_ASSERT (i < information->parameters.size ());

  return information->parameters[i].name;

}

uint32_t

IidManager::GetParameterFlags (uint16_t uid, uint32_t i) const

{

  struct IidInformation *information = LookupInformation (uid);

  NS_ASSERT (i < information->parameters.size ());

  return information->parameters[i].flags;

}

ns3::Ptr<const ns3::ParamSpec>

IidManager::GetParameterParamSpec (uint16_t uid, uint32_t i) const

{

  struct IidInformation *information = LookupInformation (uid);

  NS_ASSERT (i < information->parameters.size ());

  return information->parameters[i].param;

}

} // anonymous namespace

/*********************************************************************

 *         The TypeId TraceResolver

 *********************************************************************/

namespace ns3 {

class TypeIdTraceResolver : public TraceResolver

{

public:

  TypeIdTraceResolver (Ptr<const Object> aggregate);

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

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

  virtual void CollectSources (std::string path, const TraceContext &context, 

                               SourceCollection *collection);

  virtual void TraceAll (std::ostream &os, const TraceContext &context);

private:

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

  Ptr<const Object> m_aggregate;

};

TypeIdTraceResolver::TypeIdTraceResolver (Ptr<const Object> aggregate)

  : m_aggregate (aggregate)

{}

Ptr<const Object>

TypeIdTraceResolver::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);

  TypeId interfaceId = TypeId::LookupByName (interfaceName);

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

  return interface;

}

void  

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

{

  Ptr<const Object> interface = ParseForInterface (path);

  if (interface != 0)

    {

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

    }

}

void 

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

{

  Ptr<const Object> interface = ParseForInterface (path);

  if (interface != 0)

    {

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

    }

}

void 

TypeIdTraceResolver::CollectSources (std::string path, const TraceContext &context, 

                                          SourceCollection *collection)

{

  m_aggregate->DoCollectSources (path, context, collection);

}

void 

TypeIdTraceResolver::TraceAll (std::ostream &os, const TraceContext &context)

{

  m_aggregate->DoTraceAll (os, context);

}

/*********************************************************************

 *         The TypeId class

 *********************************************************************/

TypeId::TypeId ()

  : m_tid (0)

{}

TypeId::TypeId (std::string name)

{

  uint16_t uid = Singleton<IidManager>::Get ()->AllocateUid (name);

  NS_ASSERT (uid != 0);

  m_tid = uid;

}

TypeId::TypeId (uint16_t tid)

  : m_tid (tid)

{}

TypeId::~TypeId ()

{}

TypeId 

TypeId::LookupByName (std::string name)

{

  uint16_t uid = Singleton<IidManager>::Get ()->GetUid (name);

  NS_ASSERT (uid != 0);

  return TypeId (uid);

}

bool

TypeId::LookupParameterByFullName (std::string fullName, struct TypeId::ParameterInfo *info)

{

  std::string::size_type pos = fullName.find ("::");

  if (pos == std::string::npos)

    {

      return 0;

    }

  std::string tidName = fullName.substr (0, pos);

  std::string paramName = fullName.substr (pos+2, fullName.size () - (pos+2));

  TypeId tid = LookupByName (tidName);

  return tid.LookupParameterByName (paramName, info);

}

uint32_t 

TypeId::GetRegisteredN (void)

{

  return Singleton<IidManager>::Get ()->GetRegisteredN ();

}

TypeId 

TypeId::GetRegistered (uint32_t i)

{

  return TypeId (Singleton<IidManager>::Get ()->GetRegistered (i));

}

bool

TypeId::LookupParameterByName (std::string name, struct TypeId::ParameterInfo *info) const

{

  TypeId tid = TypeId (0);

  TypeId nextTid = *this;

  do {

    tid = nextTid;

    for (uint32_t i = 0; i < GetParametersN (); i++)

      {

        std::string paramName = GetParameterName (i);

        if (paramName == name)

          {

            info->spec = GetParameterParamSpec (i);

            info->flags = GetParameterFlags (i);

            return true;

          }

      }

    nextTid = tid.GetParent ();

  } while (nextTid != tid);

  return false;

}

bool

TypeId::LookupParameterByPosition (uint32_t i, struct TypeId::ParameterInfo *info) const

{

  uint32_t cur = 0;

  TypeId tid = TypeId (0);

  TypeId nextTid = *this;

  do {

    tid = nextTid;

    for (uint32_t j = 0; j < tid.GetParametersN (); j++)

      {

        if (cur == i)

          {

            info->spec = tid.GetParameterParamSpec (j);

            info->flags = tid.GetParameterFlags (j);

            return true;

          }

        cur++;

      }

    nextTid = tid.GetParent ();

  } while (nextTid != tid);

  return false;

}

TypeId 

TypeId::SetParent (TypeId tid)

{

  Singleton<IidManager>::Get ()->SetParent (m_tid, tid.m_tid);

  return *this;

}

TypeId 

TypeId::SetGroupName (std::string groupName)

{

  Singleton<IidManager>::Get ()->SetGroupName (m_tid, groupName);

  return *this;

}

TypeId 

TypeId::SetTypeName (std::string typeName)

{

  Singleton<IidManager>::Get ()->SetTypeName (m_tid, typeName);

  return *this;

}

TypeId 

TypeId::GetParent (void) const

{

  uint16_t parent = Singleton<IidManager>::Get ()->GetParent (m_tid);

  return TypeId (parent);

}