// Macro savecond.C searches all condition objects
// in a file, the analysis folders, or the Go4 GUI Analysis folders
// It writes a macro for each found condition setting the current values.
// The generated macro has the name pref_conditionname.C
// It can execute only in analysis environment
// Running in ROOT (needs .rootmap or Go4 libraries loaded)
//     savecond.C(rootfile,wildcard,prefix)
// Running in Go4 GUI or analysis
//     savecond.C(wildcard,prefix)
// Wildcard applies only to condition name without folders
// The generated file can be edited, if needed.
// The generated function has three boolean arguments:
// flags, counters, reset
// to control if flags and counters have to be restored.
// H.G.Essel July 2006
// Implement macro using TGo4Condition::SavePrimitive() methods
// S.Linev October 2009

#ifndef __GO4MACRO__
#ifndef __GO4ANAMACRO__
#define __NOGO4MACRO__
#endif
#endif

#include <iostream>
using namespace std;

// Recursive iterator to build a TList of all found objects
// In case of CINT full names are in the TList (needed for Get)
// File must be open
// In case of GO4 GUI parameter names are in TList
// In case of analysis the object pointers are in the TList

void conditer(TDirectory *dir, const char *wildcard, TList *found)
{
#ifdef __GO4MACRO__
   TRegexp wild(wildcard, kTRUE);
   TGo4Iter iter(go4->Browser()->BrowserSlot("Analysis"), kFALSE);
   while(iter.next()) {
      if(!iter.isfolder() && (TString(iter.getname()).Index(wild) != kNPOS)) {
         TString itemname = "Analysis/";
         itemname.Append(iter.getfullname());
         TClass *cc = go4->Browser()->ItemClass(itemname.Data());
         if(cc && cc->InheritsFrom("TGo4Condition")) {
            go4->FetchItem(itemname.Data(), 1000);
            TObject *obj = go4->GetObject(itemname.Data());
            if (obj) found->Add(obj);
         }
      }
   }
#endif
#ifdef __GO4ANAMACRO__
   Bool_t reset = kTRUE;
   TObject *obj = nullptr;
   while((obj = go4->NextMatchingObject(wildcard,"Go4",reset)) != nullptr) {
      reset = kFALSE;
      if (obj->InheritsFrom("TGo4Condition")) found->Add(obj);
   }
#endif
#ifdef __NOGO4MACRO__
   if (!dir) return;
   // found->SetOwner(kTRUE);
   TRegexp wild(wildcard, kTRUE);
   TIter next(dir->GetListOfKeys());
   while(auto key = (TKey *)next()) {
      if(strcmp(key->GetClassName(),"TDirectoryFile") == 0)
         conditer(dir->GetDirectory(key->GetName()), wildcard, found);
      else
         if (TString(key->GetName()).Index(wild) != kNPOS) {
            TClass *cl = TClass::GetClass(key->GetClassName());
            if (cl && cl->InheritsFrom("TGo4Condition")) {
                TObject *obj = dir->Get(key->GetName());
                if (obj) found->Add(obj);
            }
         }
   }
#endif
}

TString MakeCondFuncName(const char *main, const char *objname)
{
   TString subfunc;
   subfunc.Form("%s_%s", main, objname);
   subfunc.ReplaceAll("#","_");
   subfunc.ReplaceAll("(","_");
   subfunc.ReplaceAll(")","_");
   subfunc.ReplaceAll("*","_");
   subfunc.ReplaceAll("@","_");
   subfunc.ReplaceAll("[","_");
   subfunc.ReplaceAll("]","_");
   return subfunc;
}

// Function to process one condition
// outside Go4 get condition from file (1st arg)
Bool_t save1cond(TObject *obj, const char *prefix)
{
  if(!obj || !obj->InheritsFrom("TGo4Condition")) return kFALSE;
  TGo4Condition *cond = (TGo4Condition *) obj;

  TString funcname = MakeCondFuncName(prefix, cond->GetName());

  std::cout << "Write macro " << funcname << std::endl;
  std::ofstream xout(funcname+".C");

  xout << "// written by macro savecond.C at " << TDatime().AsString() << std::endl;
  xout << TString::Format("void %s(Bool_t flags = kTRUE, Bool_t counters = kFALSE, Bool_t reset = kFALSE)", funcname.Data()) << std::endl;
  xout << "{" << std::endl;
  xout << "#ifndef __GO4ANAMACRO__" << std::endl;
  xout << TString::Format("   std::cout << \"Macro %s can execute only in analysis\" << std::endl;", funcname.Data()) << std::endl;
  xout << "   return;" << std::endl;
  xout << "#endif" << std::endl;

  cond->SavePrimitive(xout, "savemacro");

  xout << std::endl;
  xout << "}" << std::endl;
  xout.close();

  return kTRUE;
}

#ifdef __NOGO4MACRO__
// Get objects from ROOT file
void savecond(const char *file, const char *wildcard="*", const char *prefix="set")
{
   TFile *f = TFile::Open(file,"r");
#else
void savecond(const char *wildcard="*", const char *prefix="set")
{
   TFile *f = nullptr;
   const char *file = nullptr;
#endif

   TList lst;
   conditer(f, wildcard, &lst);
   if (f) { delete f; f = nullptr; }

   TIter next(&lst);

   TObject *obj = nullptr;
   while((obj = next()) != nullptr)
      save1cond(obj, prefix);

   lst.Clear();
}