/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/*
* Copyright (c) 2006,2007 INRIA
*
* 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
*
* Author: Mathieu Lacage <mathieu.lacage@sophia.inria.fr>
*/
#ifdef NS3_LOG_ENABLE
#include <list>
#include <utility>
#include <iostream>
#include "log.h"
#include "assert.h"
#include "ns3/core-config.h"
#include "fatal-error.h"
#ifdef HAVE_STDLIB_H
#include <stdlib.h>
#endif
namespace ns3 {
LogTimePrinter g_logTimePrinter = 0;
typedef std::list<std::pair <std::string, LogComponent *> > ComponentList;
typedef std::list<std::pair <std::string, LogComponent *> >::iterator ComponentListI;
static class PrintList
{
public:
PrintList ();
} g_printList;
static
ComponentList *GetComponentList (void)
{
static ComponentList components;
return &components;
}
PrintList::PrintList ()
{
#ifdef HAVE_GETENV
char *envVar = getenv("NS_LOG");
if (envVar == 0)
{
return;
}
std::string env = envVar;
std::string::size_type cur = 0;
std::string::size_type next = 0;
while (next != std::string::npos)
{
next = env.find_first_of (":", cur);
std::string tmp = std::string (env, cur, next-cur);
if (tmp == "print-list")
{
LogComponentPrintList ();
exit (0);
break;
}
cur = next + 1;
}
#endif
}
LogComponent::LogComponent (char const * name)
: m_levels (0), m_name (name)
{
EnvVarCheck (name);
ComponentList *components = GetComponentList ();
for (ComponentListI i = components->begin ();
i != components->end ();
i++)
{
if (i->first == name)
{
NS_FATAL_ERROR ("Log component \""<<name<<"\" has already been registered once.");
}
}
components->push_back (std::make_pair (name, this));
}
void
LogComponent::EnvVarCheck (char const * name)
{
#ifdef HAVE_GETENV
char *envVar = getenv("NS_LOG");
if (envVar == 0)
{
return;
}
std::string env = envVar;
std::string myName = name;
std::string::size_type cur = 0;
std::string::size_type next = 0;
while (next != std::string::npos)
{
next = env.find_first_of (":", cur);
std::string tmp = std::string (env, cur, next-cur);
std::string::size_type equal = tmp.find ("=");
std::string component;
if (equal == std::string::npos)
{
component = tmp;
if (component == myName || component == "*")
{
int level = LOG_ALL | LOG_PREFIX_TIME | LOG_PREFIX_FUNC;
Enable ((enum LogLevel)level);
return;
}
}
else
{
component = tmp.substr (0, equal);
if (component == myName || component == "*")
{
int level = 0;
std::string::size_type cur_lev;
std::string::size_type next_lev = equal;
do
{
cur_lev = next_lev + 1;
next_lev = tmp.find ("|", cur_lev);
std::string lev = tmp.substr (cur_lev, next_lev - cur_lev);
if (lev == "error")
{
level |= LOG_ERROR;
}
else if (lev == "warn")
{
level |= LOG_WARN;
}
else if (lev == "debug")
{
level |= LOG_DEBUG;
}
else if (lev == "info")
{
level |= LOG_INFO;
}
else if (lev == "function")
{
level |= LOG_FUNCTION;
}
else if (lev == "logic")
{
level |= LOG_LOGIC;
}
else if (lev == "all")
{
level |= LOG_ALL;
}
else if (lev == "prefix_func")
{
level |= LOG_PREFIX_FUNC;
}
else if (lev == "prefix_time")
{
level |= LOG_PREFIX_TIME;
}
else if (lev == "level_error")
{
level |= LOG_LEVEL_ERROR;
}
else if (lev == "level_warn")
{
level |= LOG_LEVEL_WARN;
}
else if (lev == "level_debug")
{
level |= LOG_LEVEL_DEBUG;
}
else if (lev == "level_info")
{
level |= LOG_LEVEL_INFO;
}
else if (lev == "level_function")
{
level |= LOG_LEVEL_FUNCTION;
}
else if (lev == "level_logic")
{
level |= LOG_LEVEL_LOGIC;
}
else if (lev == "level_all")
{
level |= LOG_LEVEL_ALL;
}
} while (next_lev != std::string::npos);
Enable ((enum LogLevel)level);
}
}
cur = next + 1;
}
#endif
}
bool
LogComponent::IsEnabled (enum LogLevel level) const
{
// LogComponentEnableEnvVar ();
return (level & m_levels) ? 1 : 0;
}
bool
LogComponent::IsNoneEnabled (void) const
{
return m_levels == 0;
}
void
LogComponent::Enable (enum LogLevel level)
{
m_levels |= level;
}
void
LogComponent::Disable (enum LogLevel level)
{
m_levels &= ~level;
}
char const *
LogComponent::Name (void) const
{
return m_name;
}
void
LogComponentEnable (char const *name, enum LogLevel level)
{
ComponentList *components = GetComponentList ();
for (ComponentListI i = components->begin ();
i != components->end ();
i++)
{
if (i->first.compare (name) == 0)
{
i->second->Enable (level);
break;
}
}
}
void
LogComponentEnableAll (enum LogLevel level)
{
ComponentList *components = GetComponentList ();
for (ComponentListI i = components->begin ();
i != components->end ();
i++)
{
i->second->Enable (level);
}
}
void
LogComponentDisable (char const *name, enum LogLevel level)
{
ComponentList *components = GetComponentList ();
for (ComponentListI i = components->begin ();
i != components->end ();
i++)
{
if (i->first.compare (name) == 0)
{
i->second->Disable (level);
break;
}
}
}
void
LogComponentDisableAll (enum LogLevel level)
{
ComponentList *components = GetComponentList ();
for (ComponentListI i = components->begin ();
i != components->end ();
i++)
{
i->second->Disable (level);
}
}
void
LogComponentPrintList (void)
{
ComponentList *components = GetComponentList ();
for (ComponentListI i = components->begin ();
i != components->end ();
i++)
{
std::cout << i->first << "=";
if (i->second->IsNoneEnabled ())
{
std::cout << "0" << std::endl;
continue;
}
if (i->second->IsEnabled (LOG_ERROR))
{
std::cout << "error";
}
if (i->second->IsEnabled (LOG_WARN))
{
std::cout << "|warn";
}
if (i->second->IsEnabled (LOG_DEBUG))
{
std::cout << "|debug";
}
if (i->second->IsEnabled (LOG_INFO))
{
std::cout << "|info";
}
if (i->second->IsEnabled (LOG_FUNCTION))
{
std::cout << "|function";
}
if (i->second->IsEnabled (LOG_LOGIC))
{
std::cout << "|logic";
}
if (i->second->IsEnabled (LOG_ALL))
{
std::cout << "|all";
}
std::cout << std::endl;
}
}
void LogSetTimePrinter (LogTimePrinter printer)
{
g_logTimePrinter = printer;
}
LogTimePrinter LogGetTimePrinter(void)
{
return g_logTimePrinter;
}
ParameterLogger::ParameterLogger (std::ostream &os)
: m_itemNumber (0),
m_os (os)
{}
} // namespace ns3
#endif // NS3_LOG_ENABLE