NcDataStreamStats.cxx

Go to the documentation of this file.

 
#include "NcDataStreamStats.h"
#include "Riostream.h"
 
ClassImp(NcDataStreamStats); // Class implementation to enable ROOT I/O
 
NcDataStreamStats::NcDataStreamStats(const char* name,const char* title) : NcTaggingBase(name,title)
{
 
 fData=0;
 fBranchname="none";
 fMaxevt=0;
 fPfreq=0;
}

NcDataStreamStats::~NcDataStreamStats()
{
 
 if (fData)
 {
  delete fData;
  fData=0;
 }
}

NcDataStreamStats::NcDataStreamStats(const NcDataStreamStats& q) : NcTaggingBase(q)
{
 
 fData=0;
 if (q.fData) fData=(TChain*)(q.fData->Clone());
 fBranchname=q.fBranchname;
 fMaxevt=q.fMaxevt;
 fPfreq=q.fPfreq;
}

void NcDataStreamStats::Exec(Option_t* opt)
{
 
 if (!fData)
 {
  cout << endl;
  cout << " *NcDataStreamStats::Exec* No input data provided." << endl;
  return;
 }
 
 if (fDevname=="")
 {
  cout << endl;
  cout << " *NcDataStreamStats::Exec* No device name for the event tags provided." << endl;
  return;
 }
 
 cout << endl;
 cout << " *** Device and indicator identifiers ***" << endl;
 cout << " Tag device name : " << fDevname << endl;
 cout << " Pass indicator  : " << fPassname << endl;
 cout << " Write indicator : " << fWritename << endl;
 
 // Branch in the tree for the event input
 fData->SetBranchAddress(fBranchname.Data(),&fEvt);
 
 Int_t nen=(int)fData->GetEntries();
 cout << endl;
 cout << " Number of entries available : " << nen << endl;
 cout << endl;
 
 if (fMaxevt && fMaxevt<nen) nen=fMaxevt;
 cout << " *** Number of entries to be processed : " << nen << endl;
 
 Int_t nact=0;
 Int_t ndeact=0;
 if (fAct) nact=fAct->GetEntries();
 if (fDeact) ndeact=fDeact->GetEntries();
 for (Int_t i=0; i<nact; i++)
 {
  if (!i)
  {
   cout << endl;
   cout << " === Tag names (*=wildcard) that are activated by the user for re-tagged event sample studies ===" << endl;
   cout << " The non-activated tags are flagged as \"dead\" in the combined \"Passing*Writing\" listing." << endl;
   cout << " However, they are still shown in the corresponding tag matrix to identify the missed tags." << endl; 
   cout << endl;
  }
  TObjString* tagx=(TObjString*)fAct->At(i);
  if (!tagx) continue;
 
  cout << " ... *" << tagx->GetString() << "*" << endl;
 }
 for (Int_t i=0; i<ndeact; i++)
 {
  if (!i)
  {
   cout << endl;
   cout << " === Tag names (*=wildcard) that are de-activated by the user for re-tagged background reduction studies ===" << endl;
   cout << " The de-activated tags are flagged as \"dead\" in the combined \"Passing*Writing\" listing." << endl;
   cout << " However, they are still shown in the corresponding tag matrix to identify the missed tags." << endl; 
   cout << endl;
  }
  TObjString* tagx=(TObjString*)fDeact->At(i);
  if (!tagx) continue;
 
  cout << " ... *" << tagx->GetString() << "*" << endl;
 }
 cout << endl;
 
 fNevt=0;     // Number of events that were tagged
 fNevtComb=0; // Number of user reduced events via selected re-tagging
 
 TString title;
 title=fPassname;
 if (title=="*") title="Passing";
 title+=" statistics";
 fPass.SetNameTitle(fDevname.Data(),title.Data());
 fApass.Set(fMaxtag);
 fMpass.ResizeTo(fMaxtag,fMaxtag);
 fMpass.Zero();
 title=fWritename;
 if (title=="*") title="Writing";
 title+=" statistics";
 fWrites.SetNameTitle(fDevname.Data(),title.Data());
 fAwrites.Set(fMaxtag);
 fMwrites.ResizeTo(fMaxtag,fMaxtag);
 fMwrites.Zero();
 title="Combined ";
 if (nact || ndeact) title="User reduced combined ";
 if (fPassname!="*" && fWritename!="*")
 {
  title+=fPassname;
  title+="*";
  title+=fWritename;
 }
 else
 {
  title+="Passing*Writing";
 }
 title+=" statistics"; 
 fCombis.SetNameTitle(fDevname.Data(),title.Data());
 fAcombis.Set(fMaxtag);
 fMcombis.ResizeTo(fMaxtag,fMaxtag);
 fMcombis.Zero();
 
 Int_t select=0;
 Int_t ntags=0;
 Int_t jrun=0;
 Int_t jevt=0;
 Int_t perc=0;
 TString name;
 Int_t ipass=0;
 Int_t iwrite=0;
 Int_t combflag=0;
 Int_t idx=0;
 Int_t ival1=0;
 Int_t ival2=0;
 Int_t ival3=0;
 Int_t ival4=0;
 
 TString namex;
 Int_t jdeact=0;
 
 for (Int_t ient=0; ient<nen; ient++)
 {
  fData->GetEntry(ient);
  if (!fEvt) continue;
 
  // Do not process rejected events
  select=fEvt->GetSelectLevel();
  if (select<0) continue;
 
  ntags=0;
  NcDevice* tagdev=(NcDevice*)fEvt->GetDevice(fDevname);
  if (!tagdev) continue;
 
  ntags=tagdev->GetNhits();
  if (ntags>fMaxtag) ntags=fMaxtag;
 
  fApass.Reset();
  fAwrites.Reset();
  fAcombis.Reset();
 
  jrun=fEvt->GetRunNumber();
  jevt=fEvt->GetEventNumber();
 
  // Determine the first and last recorded event
  if (!fEvtFirst) fEvtFirst=new NcEvent(*fEvt);
  if (!fEvtLast) fEvtLast=new NcEvent(*fEvt);
  if (fEvtFirst->GetDifference(fEvt,"s")<0)
  {
   delete fEvtFirst;
   fEvtFirst=new NcEvent(*fEvt);
  }
  if (fEvtLast->GetDifference(fEvt,"s")>0)
  {
   delete fEvtLast;
   fEvtLast=new NcEvent(*fEvt);
  }
 
  fNevt++;
 
  combflag=0;
  for (Int_t itag=1; itag<=ntags; itag++)
  {
   NcSignal* sx=tagdev->GetHit(itag);
   if (!sx) continue;
   name=sx->GetName();
   if (sx->GetSlotIndex(fPassname))
   {
    ipass=int(sx->GetSignal(fPassname)+0.1); // Flag to indicate if the tag passing criteria are satisfied (1) or not (0)
   }
   else
   {
    ipass=1;
   }
   if (sx->GetSlotIndex(fWritename))
   {
    iwrite=int(sx->GetSignal(fWritename)+0.1);   // Flag to indicate if this tag would write the event (1) or not (0)
   }
   else
   {
    iwrite=1;
   }
 
   // Update stats for this tag name
   fPass.AddNamedSlot(name);
   fWrites.AddNamedSlot(name);
   fCombis.AddNamedSlot(name);
   fPass.AddSignal(ipass,name);
   idx=fPass.GetSlotIndex(name);
   if (idx && idx<=fMaxtag) fApass.AddAt(ipass,idx-1);
   fWrites.AddSignal(iwrite,name);
   idx=fWrites.GetSlotIndex(name);
   if (idx && idx<=fMaxtag) fAwrites.AddAt(iwrite,idx-1);
   fCombis.AddSignal(ipass*iwrite,name);
   idx=fCombis.GetSlotIndex(name);
   if (idx && idx<=fMaxtag) fAcombis.AddAt(ipass*iwrite,idx-1);

   // (De)activate some tags for the combined pass*write statistics      //
   // which will enable to study event samples or background reduction   //
 
   namex="";
   jdeact=0;
   if (nact) jdeact=1; // The user has specified the tags to be activated
 
   // Look for a user selection of activated tags
   for (Int_t i=0; i<nact; i++)
   {
    TObjString* sox=(TObjString*)fAct->At(i);
    if (!sox) continue;
    namex=sox->GetString();
    if (name.Contains(namex.Data()))
    {
     jdeact=0;
     break;
    }
   }
   // Look for a user selection of de-activated tags
   // This may de-activate a previously activated tag
   for (Int_t i=0; i<ndeact; i++)
   {
    TObjString* sox=(TObjString*)fDeact->At(i);
    if (!sox) continue;
    namex=sox->GetString();
    if (name.Contains(namex.Data()))
    {
     jdeact=1;
     break;
    }
   }
 
   if (jdeact)
   {
    fCombis.SetDead(name);
   }
   else
   {
    if (ipass*iwrite) combflag=1;
   }
  } // End of tag loop
 
  if (combflag) fNevtComb++;
 
  // Update the various tag matrices for this event
  ntags=fPass.GetNslots();
  if (ntags>fMaxtag) ntags=fMaxtag;
 
  for (Int_t irow=0; irow<ntags; irow++)
  {
   ival1=fApass.At(irow);
   ival2=fAwrites.At(irow);
   ival3=fAcombis.At(irow);
   if (!ival1 && !ival2 && !ival3) continue;
   for (Int_t icol=0; icol<ntags; icol++)
   {
    ival4=fApass.At(icol);
    if (ival1 && ival4) fMpass(irow,icol)+=float(ival4);
    ival4=fAwrites.At(icol);
    if (ival2 && ival4) fMwrites(irow,icol)+=float(ival4);
    ival4=fAcombis.At(icol);
    if (ival3 && ival4) fMcombis(irow,icol)+=float(ival4);
   }
  }  
 
  if (fPfreq>0 && (!(ient%fPfreq) || (ient+1)==nen))
  {
   perc=100*(ient+1)/nen;
   cout << " *** Processed input entry : " << ient << " run : " << jrun << " event : " << jevt << " (" << perc << "%)" << endl;
  }
 
 } // End of event loop
 
 // List the final statistics overview
 ListStatistics();
}

void NcDataStreamStats::AddInputFile(TString file,TString tree,TString branch)
{
 
 // Expand the path name of the provided input file
 file=gSystem->ExpandPathName(file.Data());
 
 // Initialise the master data Chain and the Branch name if needed
 if (!fData)
 {
  fData=new TChain("Data");
  fBranchname=branch;
 }
 
 TChain input(tree.Data());
 input.Add(file.Data());
 
 // Add the file(s) to the master data chain
 fData->Add(&input);
}

void NcDataStreamStats::SetMaxEvt(Int_t n)
{
 
 fMaxevt=n;
}

void NcDataStreamStats::SetPrintFrequency(Int_t m)
{
 
 fPfreq=m;
}

void NcDataStreamStats::ListInputFiles(Option_t* opt) const
{
 
 if (fData)
 {
  cout << endl;
  fData->Print(opt);
 }
}

TObject* NcDataStreamStats::Clone(const char* name) const
{
 
 NcDataStreamStats* fstat=new NcDataStreamStats(*this);
 if (name)
 {
  if (strlen(name)) fstat->SetName(name);
 }
 return fstat;
}