KVEnv.cpp

//Created by KVClassFactory on Mon Jan 14 12:23:45 2013
//Author: bonnet
 
#include "KVEnv.h"
#include "TNamed.h"
#include "THashList.h"
#include "KVString.h"
#include "TSystem.h"
 
ClassImp(KVEnv)
 
 
 
KVEnv::KVEnv(const KVString& name) : TEnv(name)
{
   // The TEnv constructor has a major bug: its behaviour changes depending on whether the
   // current working directory is the user's home directory or not.
   // If the two are *not* the same, i.e. outside the home directory, the constructor will
   // try to read the file using just the name as given, after failing to find it in the home
   // directory: in this case, when an absolute filename is given, the TEnv constructor
   // succeeds in finding and opening the file. However, if the same code is executed while
   // in the user's home directory (e.g. user launches 'kaliveda', or 'KaliVedaGUI' or 'KaliVedaSim'
   // in her home directory), this does not work!
   //
   // Here we try to correct this madness.
 
   if (gSystem->IsAbsoluteFileName(name) && !gSystem->AccessPathName(name)
         && (!GetTable() || !GetTable()->GetEntries()))
      ReadFile(name, kEnvUser);
}
 
 
 
 
KVEnv::KVEnv(TEnv& other)
   : TEnv()
{
   // Copy constructor
   //
   // Copies all key-value pairs from other to this
 
   CopyTable(other);
}
 
 
 
 
void KVEnv::CopyTable(TEnv& env)
{
   //Copy table of env to this
   THashList* hl = 0;
   if ((hl = env.GetTable())) {
      TIter next(hl);
      TEnvRec* env_rec = 0;
      while ((env_rec = (TEnvRec*)next.Next())) {
         SetValue(env_rec->GetName(), env_rec->GetValue());
      }
   }
 
}
 
 
 
 
void KVEnv::AddCommentLine(const Char_t* line)
{
   //Add a comment for the current TEnv
   //the line will started with "# " prefix and will
   //be ended with "\n" postfix
   //
   fComments.Add(new TNamed(Form("# %s\n", line), ""));
 
}
 
 
 
 
void KVEnv::AddComments(const Char_t* comments)
{
   //Add comments
   //the comments can contains "\n" character
   //each line contained in the comments will
   //begin with "# " prefix and will
   //be ended with "\n" postfix
 
   KVString st(comments);
   st.Begin("\n");
   while (!st.End()) {
      AddCommentLine(st.Next().Data());
   }
}
 
 
 
 
void KVEnv::ClearComments()
{
   fComments.Clear();
}
 
 
 
 
void KVEnv::PrintComments()
{
   //print comments as they will be written
   //in the file
   if (fComments.GetEntries() == 0) {
      printf("No comments defined\n");
      return;
   }
   for (Int_t ii = 0; ii < fComments.GetEntries(); ii += 1) {
      printf("%s", fComments.At(ii)->GetName());
   }
}
 
 
 
 
Int_t KVEnv::WriteFile(const char* fname, EEnvLevel level)
{
   // Write first comments associated to the current TEnv
   // Write resourse records to file fname for a certain level. Use
   // level kEnvAll to write all resources. Returns -1 on case of error,
   // 0 in case of success.
 
   if (!fname || !strlen(fname)) {
      Error("WriteFile", "no file name specified");
      return -1;
   }
 
   if (!GetTable()) {
      Error("WriteFile", "TEnv table is empty");
      return -1;
   }
 
   FILE* ofp;
   if ((ofp = fopen(fname, "w"))) {
      for (Int_t ii = 0; ii < fComments.GetEntries(); ii += 1) {
         fprintf(ofp, "%s", fComments.At(ii)->GetName());
      }
 
      //for (Int_t jj=0;jj<GetTable()->GetEntries();jj+=1){
      // TEnvRec *er = (TEnvRec*) GetTable()->At(jj);
      TIter next(GetTable());
      TEnvRec* er;
      while ((er = (TEnvRec*) next()))
         if (er->GetLevel() == level || level == kEnvAll)
            fprintf(ofp, "%-40s %s\n", Form("%s:", er->GetName()),
                    er->GetValue());
      //}
      fclose(ofp);
      return 0;
   }
 
   Error("WriteFile", "cannot open %s for writing", fname);
   return -1;
}