#include "TArtParameters.hh"

#include "TArtCore.hh"
#include "TArtStoreManager.hh"

#include <TDOMParser.h>
#include <TXMLNode.h>

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

using namespace std;
TArtParameters* TArtParameters::fParameters = 0;

//__________________________________________________________
TArtParameters::TArtParameters(const char* name, const char* title)
  : TNamed(name, title)
{
  TArtCore::Info(__FILE__,"Creating parameters...");
  fStoreManager = TArtStoreManager::Instance();
  fStoreManager->AddParameters(this);
}
//__________________________________________________________
TArtParameters::~TArtParameters() 
{
  fParameters = 0;
}
//__________________________________________________________
TArtParameters* TArtParameters::Instance(const char* name, const char* title)
{
  if(!fParameters) fParameters = new TArtParameters(name, title);
  return fParameters;
}    

//__________________________________________________________
void TArtParameters::Delete()
{
  delete fParameters;
  fParameters = 0;  
}
//__________________________________________________________
Bool_t TArtParameters::LoadParameter(const char *xmlfile)
{
  TArtCore::Info(__FILE__,"Load parameter from %s", xmlfile);
  TDOMParser domParser;
  domParser.SetValidate(false);
  Int_t parsecode = domParser.ParseFile(xmlfile);
  if(parsecode < 0){
    std::cerr << domParser.GetParseCodeMessage(parsecode) << std::endl;
    return false;
  }
  TXMLNode* node = domParser.GetXMLDocument()->GetRootNode();
  ParseParaList(node->GetChildren());
  return true;
}
//__________________________________________________________
void TArtParameters::ParseParaList(TXMLNode *node)
{
  for(; node; node = node->GetNextNode()){
    if(node->GetNodeType() != TXMLNode::kXMLElementNode) continue; // Element Node

    if(strcmp(node->GetNodeName(), "PARAMETER") == 0)
      ParsePara(node->GetChildren());
    else 
      TArtCore::Info(__FILE__,"Unknown Parameter %s.",node->GetNodeName());
  }
}
//__________________________________________________________
void TArtParameters::ParsePara(TXMLNode *node)
{
  TString name;
  TString type;
  TString svalue;

  for ( ; node; node = node->GetNextNode()) {
    if (node->GetNodeType() == TXMLNode::kXMLElementNode) { // Element Node

      if (strcmp(node->GetNodeName(), "name") == 0)
        name = node->GetText();
      else if (strcmp(node->GetNodeName(), "type") == 0)
        type = node->GetText();
      else if (strcmp(node->GetNodeName(), "value") == 0)
        svalue = node->GetText();
    }
  }

  if (type == "int" || type=="Int_t"){
    if (svalue.IsDigit())
      imap.insert(std::pair<TString,Int_t>(name,svalue.Atoi()) );
    else
      TArtCore::Info(__FILE__,"type %s is invalid for %s (value = %s).",type.Data(),name.Data(),svalue.Data());
  }

  else if (type == "double" || type=="Double_t"){
    if (svalue.IsFloat())
      dmap.insert(std::pair<TString,Double_t>(name,svalue.Atof()) );
    else
      TArtCore::Info(__FILE__,"type %s is invalid for %s (value = %s).",type.Data(),name.Data(),svalue.Data());
  }

  else if (type == "string" || type=="TString"){
    smap.insert(std::pair<TString,TString>(name,svalue) );
  }

  else
    TArtCore::Info(__FILE__,"Unknown parameter type %s for %s.",type.Data(),name.Data());

}
//_____________________________________________________________________
void TArtParameters::PrintPara(){

  std::map<TString, Int_t>::iterator i_itr;
  TArtCore::Info(__FILE__,"Parameters (Int_t)");
  for (i_itr=imap.begin();i_itr!=imap.end();i_itr++)
    std::cout<<i_itr->first<<" = "<<i_itr->second<<std::endl;

  std::map<TString, Double_t>::iterator d_itr;
  TArtCore::Info(__FILE__,"Parameters (Double_t)");
  for (d_itr=dmap.begin();d_itr!=dmap.end();d_itr++)
    std::cout<<d_itr->first<<" = "<<d_itr->second<<std::endl;

  std::map<TString, TString>::iterator s_itr;
  TArtCore::Info(__FILE__,"Parameters (TString)");
  for (s_itr=smap.begin();s_itr!=smap.end();s_itr++)
    std::cout<<s_itr->first<<" = "<<s_itr->second<<std::endl;
}
//_____________________________________________________________________
const Int_t TArtParameters::FindIntPara(const TString str) const
{
  std::map<TString, Int_t>::const_iterator itr = imap.find(str);
  if(itr != imap.end()) return itr->second;
  else return 0;
}
//_____________________________________________________________________
const Double_t TArtParameters::FindDoublePara(const TString str) const
{
  std::map<TString, Double_t>::const_iterator itr = dmap.find(str);
  if(itr != dmap.end()) return itr->second;
  else return 0;
}
//_____________________________________________________________________
const TString TArtParameters::FindStringPara(const TString str) const
{
  std::map<TString, TString>::const_iterator itr = smap.find(str);
  if(itr != smap.end()) return itr->second;
  else return "";
}
//_____________________________________________________________________