#include "trace.h"
#include "acconf.h"

namespace ledger {

bool trace_mode;

void trace(const std::string& cat, const std::string& str)
{
  char buf[32];
  std::strftime(buf, 31, "%H:%M:%S", datetime_t::now.localtime());
  std::cerr << buf << " " << cat << ": " << str << std::endl;
}

void trace_push(const std::string& cat, const std::string& str,
		timing_t& timer)
{
  timer.start();
  trace(cat, str);
}

void trace_pop(const std::string& cat, const std::string& str,
	       timing_t& timer)
{
  timer.stop();
  std::ostringstream out;
  out << str << ": " << (double(timer.cumulative) / double(CLOCKS_PER_SEC)) << "s";
  trace(cat, out.str());
}

live_objects_map live_objects;
object_count_map ctor_count;
object_count_map object_count;
object_count_map live_count;

bool tracing_active = false;

bool trace_ctor(void * ptr, const std::string& name)
{
  if (! tracing_active)
    return true;

  DEBUG_PRINT("ledger.trace.debug", "trace_ctor " << ptr << " " << name);

  std::string::size_type pos = name.find_first_of('(');
  std::string cls_name(name, 0, pos);

  live_objects.insert(live_objects_pair(ptr, cls_name));

  object_count_map::iterator i = ctor_count.find(name);
  if (i != ctor_count.end()) {
    (*i).second++;
  } else {
    std::pair<object_count_map::iterator, bool> result
      = ctor_count.insert(object_count_pair(name, 1));
    if (! result.second) {
      tracing_active = false;
      return false;
    }
  }

  object_count_map::iterator j = object_count.find(cls_name);
  if (j != object_count.end()) {
    (*j).second++;
  } else {
    std::pair<object_count_map::iterator, bool> result
      = object_count.insert(object_count_pair(cls_name, 1));
    if (! result.second) {
      tracing_active = false;
      return false;
    }
  }

  object_count_map::iterator k = live_count.find(cls_name);
  if (k != live_count.end()) {
    (*k).second++;
  } else {
    std::pair<object_count_map::iterator, bool> result
      = live_count.insert(object_count_pair(cls_name, 1));
    if (! result.second) {
      tracing_active = false;
      return false;
    }
  }

  return true;
}

bool trace_dtor(void * ptr, const std::string& name)
{
  if (! tracing_active)
    return true;

  DEBUG_PRINT("ledger.trace.debug", "trace_dtor " << ptr << " " << name);

  live_objects_map::iterator i = live_objects.find(ptr);
  if (i == live_objects.end()) {
    std::cerr << "Destruction of unknown object " << name << " " << ptr
	      << std::endl;;
    tracing_active = false;
    return false;
  }

  std::string::size_type pos = name.find_first_of('(');
  std::string cls_name(name, 0, pos);

  int ptr_count = live_objects.count(ptr);
  for (int x = 0; x < ptr_count; x++) {
    if ((*i).second == cls_name) {
      live_objects.erase(i);
      break;
    } else {
      i++;
    }
  }

  object_count_map::iterator k = live_count.find(name);
  if (k == live_count.end()) {
    std::cerr << "Destruction of unregistered class " << name
	      << std::endl;;
    tracing_active = false;
    return false;
  }
  if (--(*k).second == 0)
    live_count.erase(k);

  return true;
}

void report_memory(std::ostream& out)
{
  if (live_count.size() > 0)
    out << "Live object counts:" << std::endl;

  for (object_count_map::iterator i = live_count.begin();
       i != live_count.end();
       i++) {
    out << "  ";
    out << std::right;
    out.width(5);
    out << (*i).second << "  " << (*i).first << std::endl;
  }

  DEBUG_IF("ledger.trace.verbose") {
    if (live_objects.size() > 0)
      out << "Live objects:" << std::endl;

    for (live_objects_map::iterator i = live_objects.begin();
	 i != live_objects.end();
	 i++) {
      out << "  ";
      out << std::right;
      out.width(5);
      out << (*i).first << "  " << (*i).second << std::endl;
    }
  }

  if (object_count.size() > 0)
    out << "Object counts:" << std::endl;

  for (object_count_map::iterator i = object_count.begin();
       i != object_count.end();
       i++) {
    out << "  ";
    out << std::right;
    out.width(5);
    out << (*i).second << "  " << (*i).first << std::endl;
  }

  if (ctor_count.size() > 0)
    out << "Constructor counts:" << std::endl;

  for (object_count_map::iterator i = ctor_count.begin();
       i != ctor_count.end();
       i++) {
    out << "  ";
    out << std::right;
    out.width(5);
    out << (*i).second << "  " << (*i).first << std::endl;
  }
}

} // namespace ledger