KVDataQualityAuditSelector.cpp

#include "KVDataQualityAuditSelector.h"
#include "KVDataSet.h"
 
ClassImp(KVDataQualityAuditSelector)
 
 
 
 
void KVDataQualityAuditSelector::InitAnalysis(void)
{
   // Create audit objects if not already done
   //   + PROOF: only on slave workers
   Info("InitAnalyis", "Welcome to data quality auditing");
   if(!fDisableCreateTreeFile)
   {
      Info("InitAnalysis", "Creating audits");
      audit_reac = ::new KVDataQualityAudit("audit_reac");
      audit_sys = ::new KVDataQualityAudit("audit_sys");
      audit_ds = ::new KVDataQualityAudit(gDataSet->GetName(), gDataSet->GetTitle());
      GetOutputList()->Add(audit_reac);
      GetOutputList()->Add(audit_sys);
      GetOutputList()->Add(audit_ds);
   }
}
 
 
 
 
void KVDataQualityAuditSelector::InitRun()
{
   // only now do we know what system we are analysing
   auto sys = gDataAnalyser->GetAnalysedSystem();
   if(audit_reac)
   {
      Info("InitRun", "Setting audit names & titles");
      audit_reac->SetNameTitle(sys->GetBatchName(), sys->GetName());
   }
   if(audit_sys)
      audit_sys->SetNameTitle(sys->GetBatchNameWithoutEnergy(), sys->GetReactionNameWithoutEnergy());
}
 
 
 
 
Bool_t KVDataQualityAuditSelector::Analysis(void)
{
   for (auto& particle : ReconEventOKIterator(GetEvent()))
   {
      audit_reac->Add(particle);
      audit_sys->Add(particle);
      audit_ds->Add(particle);
   }
 
   return kTRUE;
}
 
 
 
 
void KVDataQualityAuditSelector::EndAnalysis()
{
   TString filename;
   if (gDataAnalyser->GetBatchSystem())
      filename.Form("%s.root", gDataAnalyser->GetBatchSystem()->GetJobName());
   else
      filename = "DataQualityAudit.root";
   // in case of PROOF
   Info("EndAnalysis", "proof mode=%d audit_reac=%p",gDataAnalyser->GetProofMode(),audit_reac);
   GetOutputList()->ls();
   if(gDataAnalyser->GetProofMode()==KVDataAnalyser::EProofMode::Lite)
   {
      auto sys = gDataAnalyser->GetAnalysedSystem();
      audit_reac = (KVDataQualityAudit*)GetOutputList()->FindObject(sys->GetBatchName());
      Info("EndAnalysis", "audit_reac=%p",audit_reac);
      audit_sys  = (KVDataQualityAudit*)GetOutputList()->FindObject(sys->GetBatchNameWithoutEnergy());
      audit_ds   = (KVDataQualityAudit*)GetOutputList()->FindObject(gDataSet->GetName());
      TFile output(filename, "recreate");
      audit_reac->Write();
      audit_sys->Write();
      audit_ds->Write();
      output.Write();
   }
   else if(gDataAnalyser->GetProofMode()==KVDataAnalyser::EProofMode::None)
   {
      TFile output(filename, "recreate");
      audit_reac->Write();
      audit_sys->Write();
      audit_ds->Write();
      output.Write();
   }
}