FillData.cxx

Go to the documentation of this file.

// $Id: FillData.cxx,v 1.2 2007/12/01 22:56:20 rustem Exp $

// C/C++
#include <iomanip>
#include <sstream>

// ROOT
#include "TFile.h"
#include "TTree.h"

// Local
#include "PhysicsNtuple/Default.h"
#include "PhysicsNtuple/Factory.h"
#include "FillData.h"

REGISTER_ANP_OBJECT(AlgSnarl,FillData)

using namespace std;

//----------------------------------------------------------------------
Anp::FillData::FillData()
   :fDir(0),
    fFile(0),
    fDirName("fill")
{   
}

//----------------------------------------------------------------------
Anp::FillData::~FillData()
{
}

//----------------------------------------------------------------------
bool Anp::FillData::Run(Record &record)
{
   if(!fDir)
   {
      return true;
   }

   for(EventIter ievent = record.EventBeg(); ievent != record.EventEnd(); ++ievent)
   {
      Fill(ievent -> DataBeg(), ievent -> DataEnd(), "event");
   }
   for(TrackIter itrack = record.TrackBeg(); itrack != record.TrackEnd(); ++itrack)
   {
      Fill(itrack -> DataBeg(), itrack -> DataEnd(), "track");
   }  

   return true;
}

//----------------------------------------------------------------------
void Anp::FillData::Fill(DataIter ibeg, DataIter iend, const string &key)
{   
   //
   // Fill (create first time) tree and branches
   //
   if(!fDir || ibeg == iend || key.empty())
   {
      return;
   }

   for(DataIter dit = ibeg; dit != iend; ++dit)
   {
      stringstream tname;
      tname << key << "_" << setw(5) << setfill('0') << dit -> Key();

      //
      // Find (create first time) tree 
      //
      TreeMap::iterator itree = fTree.find(tname.str());
      if(itree == fTree.end())
      {
         itree = fTree.insert(TreeMap::value_type(tname.str(), FillData::Tree())).first;

         Tree &tree = itree -> second;
         tree.tree_ptr = new TTree(tname.str().c_str(), tname.str().c_str());
         tree.tree_ptr -> SetDirectory(fDir);
         tree.branch_ptr = tree.tree_ptr -> Branch("data", &tree.data, "data/F");       
      }
  
      (itree -> second).data = dit -> Data();
      (itree -> second).tree_ptr -> Fill();
   }
}

//----------------------------------------------------------------------
void Anp::FillData::Set(TDirectory *dir)
{
   //
   // If not ROOT file exist, then use outside TDirectory pointer if it is valid
   //
   if(!fDir)
   {
      fDir = Anp::GetDir(fDirName, dir);
   }
}

//----------------------------------------------------------------------
void Anp::FillData::Config(const Registry &reg)
{
   //
   // create an output file if a path is given
   //

   const char *value_char = 0;
   if(reg.Get("FillDataDirName", value_char) && value_char)
   {
      fDirName = value_char;
   }

   const char *path_char = 0;
   if(reg.Get("FillDataPath", path_char) && path_char)
   {
      fFile = new TFile(path_char, "RECREATE");   
      if(!fFile || !fFile -> IsOpen())
      {
         cerr << "FillData::Config - failed to create ROOT file:\n   " << path_char << endl;
         fFile = 0;
      }
      else
      {
         cout << "FillData::Config - recreated ROOT file:\n   " << path_char << endl;
      }
   
      fDir = Anp::GetDir(fDirName, fFile);
   }

   if(reg.KeyExists("PrintConfig"))
   {
      cout << "FillData::Config" << endl
           << "   DirName = " << fDirName << endl;
      
      if(path_char)
      {
         cout << "   Path = " << path_char << endl;
      }
   }
}

//----------------------------------------------------------------------
void Anp::FillData::End(const DataBlock &)
{
   if(fFile)
   {
      fFile -> Write();
      fFile -> Close();
   }
}

---