KVFAZIARawDataReconstructor.cpp

//Created by KVClassFactory on Fri Jan 23 14:17:23 2015
//Author: ,,,
 
#include "KVFAZIARawDataReconstructor.h"
#include "KVDataAnalyser.h"
#include "KVDataSet.h"
#include "KVFAZIADB.h"
#include "KVDataRepositoryManager.h"
#include "KVDataRepository.h"
#include "RVersion.h"
#include "KVFAZIADetector.h"
#include "KVFAZIA.h"
#include "KVSignal.h"
#include "TSystem.h"
 
ClassImp(KVFAZIARawDataReconstructor)
 
// BEGIN_HTML <!--
/* -->
<h2>KVFAZIARawDataReconstructor</h2>
<h4>Handle reconstruction of FAZIA events</h4>
<!-- */
// --> END_HTML
 
 
 
KVFAZIARawDataReconstructor::KVFAZIARawDataReconstructor()
   : taskname("Reconstruction"), datatype("recon")
{
   //Default constructor
   file = 0;
   tree = 0;
   recev = 0;
   nb_recon = 0;
}
 
 
 
 
KVFAZIARawDataReconstructor::~KVFAZIARawDataReconstructor()
{
   // Destructor
   SafeDelete(recev);
}
 
 
 
 
 
void KVFAZIARawDataReconstructor::InitRun()
{
   // Creates new ROOT file with TTree for reconstructed events.
   // By default this file will be written in the same data repository as the raw data file we are reading.
   // This can be changed by setting the environment variable(s):
   //
   //     Reconstruction.DataAnalysisTask.OutputRepository:     [name of repository]
   //     [name of dataset].Reconstruction.DataAnalysisTask.OutputRepository:         [name of repository]
   //
   // If no value is set for the current dataset (second variable), the value of the
   // first variable will be used. If neither is defined, the new file will be written in the same repository as
   // the raw file (if possible, i.e. if repository is not remote).
 
   // Create new KVReconstructedEvent filled with KVFAZIAReconNuc object
   // used to reconstruct & store events
 
   if (!recev) recev = new KVReconstructedEvent(50);
 
   // get dataset to which we must associate new run
   KVDataSet* OutputDataset =
      gDataRepositoryManager->GetDataSet(gDataSet->GetOutputRepository(taskname), gDataSet->GetName());
 
   file = OutputDataset->NewRunfile(datatype.Data(), GetCurrentRunNumber());
 
   std::cout << "Writing \"" << datatype.Data() << "\" events in ROOT file " << file->GetName() << std::endl;
 
   //tree for reconstructed events
   tree = new TTree("ReconstructedEvents", Form("%s : %s : %s events created from raw data",
                    gFaziaDB->GetRun(GetCurrentRunNumber())->GetName(),
                    gFaziaDB->GetRun(GetCurrentRunNumber())->GetTitle(),
                    datatype.Data())
                   );
 
   //leaves for reconstructed events
   KVEvent::MakeEventBranch(tree, "FAZIAReconEvent", recev);
 
   Info("InitRun", "Created reconstructed data tree %s : %s", tree->GetName(), tree->GetTitle());
   nb_recon = 0;
 
}
 
 
 
 
 
Bool_t KVFAZIARawDataReconstructor::Analysis()
{
   // Analysis of event.
   // RawData TTree is filled with values of parameters for event.
   // If event has INDRA trigger information (IsIndraEvent()==kTRUE)
   // then
   //    *) event reconstruction is performed for 'Physics' events
   //    *) or the GeneTree is filled with pulser/laser data for 'Gene' events
 
 
   recev->SetNumber(GetEvent()->GetNumber());
   gFazia->ReconstructEvent(recev, GetDetectorEvent());
 
   ExtraProcessing();
 
   nb_recon++;
   tree->Fill();
 
   recev->Clear();
 
   return kTRUE;
}
 
 
 
 
 
void KVFAZIARawDataReconstructor::ExtraProcessing()
{
   KVString label = "";
 
   KVFAZIADetector* det = 0;
   KVSignal* sig = 0;
   KVReconstructedNucleus* recnuc = 0;
   while ((recnuc = recev->GetNextParticle())) {
      recnuc->GetReconstructionTrajectory()->IterateFrom();
      KVGeoDetectorNode* _node;
      while ((_node = recnuc->GetReconstructionTrajectory()->GetNextNode())) {
         auto det = dynamic_cast<KVFAZIADetector*>(_node->GetDetector());
         TIter next_s(det->GetListOfSignals());
         while ((sig = (KVSignal*)next_s())) {
            if (sig->HasFPGA()) {
               for (Int_t ii = 0; ii < sig->GetNFPGAValues(); ii += 1) {
                  //SI2-T3-Q1-B003.Q2.RawAmplitude=14
                  if (ii == 0) label = "FPGAEnergy";
                  if (ii == 1) label = "FastFPGAEnergy"; //only for CsI Q3
                  TString ene = GetEvent()->GetFPGAEnergy(
                                   det->GetBlockNumber(),
                                   det->GetQuartetNumber(),
                                   det->GetTelescopeNumber(),
                                   sig->GetType(),
                                   ii
                                );
 
                  recnuc->GetParameters()->SetValue(
                     Form("%s.%s.%s", det->GetName(), sig->GetName(), label.Data()),
                     ene.Data()
                  );
               }
            }
            if (!sig->PSAHasBeenComputed()) {
               sig->TreateSignal();
            }
 
            sig->GetPSAResult(det);
//            if (psa.GetNpar()) {
//               *(recnuc->GetParameters()) += psa;
//            }
         }
      }
   }
}
 
 
 
 
 
void KVFAZIARawDataReconstructor::EndRun()
{
   SafeDelete(recev);
 
   std::cout << std::endl << " *** Number of reconstructed FAZIA events : "
             << nb_recon << " ***" << std::endl << std::endl;
   file->cd();
   gDataAnalyser->WriteBatchInfo(tree);
   tree->Write();//write tree to file
 
   // get dataset to which we must associate new run
   KVDataSet* OutputDataset =
      gDataRepositoryManager->GetDataSet(gDataSet->GetOutputRepository(taskname), gDataSet->GetName());
   //add new file to repository
   OutputDataset->CommitRunfile(datatype.Data(), GetCurrentRunNumber(), file);
 
   ProcInfo_t pid;
   if (gSystem->GetProcInfo(&pid) == 0) {
      std::cout << "     ------------- Process infos -------------" << std::endl;
      printf(" CpuSys = %f  s.    CpuUser = %f s.    ResMem = %f MB   VirtMem = %f MB\n",
             pid.fCpuSys, pid.fCpuUser, pid.fMemResident / 1024., pid.fMemVirtual / 1024.);
   }
 
}