/*++
Copyright (c) 2011 Microsoft Corporation

Module Name:

    duality_profiling.cpp

Abstract:

   collection performance information for duality

Author:

    Ken McMillan (kenmcmil)

Revision History:


--*/


#include <map>
#include <iostream>
#include <string>
#include <string.h>
#include <stdlib.h>

#ifdef _WINDOWS
#pragma warning(disable:4996)
#pragma warning(disable:4800)
#pragma warning(disable:4267)
#endif

#include "duality_wrapper.h"
#include "iz3profiling.h"

namespace Duality {
  
  void show_time(){
    output_time(std::cout,current_time());
    std::cout << "\n";
  }

  typedef std::map<const char*, struct node> nmap;
  
  struct node {
    std::string name;
    clock_t time;
    clock_t start_time;
    nmap sub;
    struct node *parent;
    
    node();
  } top;
  
  node::node(){
    time =  0;
    parent = 0;
  }
  
  struct node *current;
  
  struct init {
    init(){
      top.name = "TOTAL";
      current = &top;
    }
  } initializer;
  
  struct time_entry {
    clock_t t;
    time_entry(){t = 0;};
    void add(clock_t incr){t += incr;}
  };
  
  struct ltstr
  {
    bool operator()(const char* s1, const char* s2) const
    {
      return strcmp(s1, s2) < 0;
    }
  };
  
  typedef  std::map<const char*, time_entry, ltstr> tmap;

  static std::ostream *pfs;

  void print_node(node &top, int indent, tmap &totals){
    for(int i = 0; i < indent; i++) (*pfs) << "  ";
    (*pfs) << top.name;
    int dots = 70 - 2 * indent - top.name.size();
    for(int i = 0; i <dots; i++) (*pfs) << ".";
    output_time(*pfs, top.time);
    (*pfs) << std::endl;
    if(indent != 0)totals[top.name.c_str()].add(top.time);
    for(nmap::iterator it = top.sub.begin(); it != top.sub.end(); it++)
      print_node(it->second,indent+1,totals);
  }
  
  void print_profile(std::ostream &os) {
    pfs = &os;
    top.time = 0;
    for(nmap::iterator it = top.sub.begin(); it != top.sub.end(); it++)
      top.time += it->second.time;
    tmap totals;
    print_node(top,0,totals);
    (*pfs) << "TOTALS:" << std::endl;
    for(tmap::iterator it = totals.begin(); it != totals.end(); it++){
      (*pfs) << (it->first) << " ";
      output_time(*pfs, it->second.t);
      (*pfs) << std::endl;
    }
    profiling::print(os); // print the interpolation stats
  }
  
  void timer_start(const char *name){
    node &child = current->sub[name];
    if(child.name.empty()){ // a new node
      child.parent = current;
      child.name = name;
    }
    child.start_time = current_time();
    current = &child;
  }

  void timer_stop(const char *name){
      if (current->name != name || !current->parent) {
#if 0
          std::cerr << "imbalanced timer_start and timer_stop";
          exit(1);
#endif
          // in case we lost a timer stop due to an exception
          while (current->name != name && current->parent)
              current = current->parent;
          if (current->parent) {
              current->time += (current_time() - current->start_time);
              current = current->parent;
          }
          return;
      }
    current->time += (current_time() - current->start_time);
    current = current->parent;
  }
}