harpoanalysis.cxx

Go to the documentation of this file.

//
// Analysis program holding all the available analyses
// activate/deactivate in the config file
//

#include <stdlib.h>
#include <stdio.h>
#include <iostream>

#include "HarpoConfig.h"
#include "HarpoReader.h"
#include "HarpoReaderRoot.h"
#include "HarpoRunHeader.h"
#include "HarpoAnalyseVertex.h"
#include "HarpoAnalysePedestalShift.h"
#include "HarpoAnalyseRates.h"
#include "HarpoAnalyseXtalk.h"
#include "HarpoAnalyseVdrift.h"
#include "HarpoAnalyseMatching.h"
#include "HarpoEventSelector.h"
#include "HarpoSelectCosmics.h"
#include "HarpoExcludeBadEvents.h"
#include "HarpoAnalyseTrackPairs.h"
#include "HarpoSelectorStraightTracks.h"
#include "HarpoVertexing.h"
#include "HarpoAnalyseMultiplicityAget.h"
#include "HarpoSelectorPileUp.h"
#include "HarpoClusteringBlocs.h"
#include "HarpoAnalyseBertrand.h"
#include "HarpoAnalyseVertex3D.h"
#include "HarpoAnalyseVertex3Dsim.h"
#include "HarpoMatchingVertex.h"
#include "HarpoSelectorVertex.h"
#include "HarpoAnalyseEtrack.h"
#include "HarpoAnalysePattern.h"
#include "HarpoAnalyseTemplate.h"
#include "HarpoAnalyseGainStudy.h"
#include "HarpoAnalysePrf.h"
#include "HarpoAnalyseBaselineFluct.h"
#include "HarpoAnalyseBasic.h"
#include "HarpoAnalyseRunNoZS.h"
#include "HarpoAnalyseResolution.h"
#include "HarpoAnalyseWriter.h"
#include "HarpoAnalyseNoiseSuppression.h"
#include "HarpoClustering.h"
#include "HarpoHoughTracking.h"
#include "HarpoTrackingPh.h"
#include "HarpoKalmanTracking.h"
#include "HarpoKalmanFitting.h"
#include "HarpoKalmanNew.h"
#include "HarpoMatching.h"
#include "HarpoAnalyseTriggerTime.h"
#include "HarpoAnalyseMonitorNew.h"

int main(int argc, char **argv)
{
  Long_t processed = -1;  // Processed events counter , negative on error
  
  HarpoRunHeader *hdr;
  //HarpoEvent *hevent;
  HarpoConfig::gHReaderType = hRootReader;

  HarpoConfig *cfg = new HarpoConfig(argc,argv,0x17);
  cfg->Init();

  cfg->print();

  HarpoReader *rdr = new HarpoReader( cfg );
  //  HarpoReaderRoot *rdr = new HarpoReaderRoot( cfg );
  // rdr->SetRootFile(new TString("test.root"));
  // hNoCMPTime,  // Read without timeshump comparision
  // hFirstTime, // return only plane which smaller time stamp
  // hSyncTime   // skip plane event with smaller timeshatp, return only

  //  if ( rdr->Init() ) {
    //  if ( rdr->Init(hNoCMPTime) ) {
    //if ( rdr->Init(hFirstTime) ) {
  if ( rdr->Init(hSyncTime) ) {
    hdr = rdr->GetRunHeader();
    hdr->print();
  }
  else
  {
    printf ("Int ERROR exiting ...\n");
    return 1;
  }

  printf ("Initialized\n");
 

  //  TString* analyses = new TString();
  TString* analyses = gHConfig->GetAnalyses();
  //  cfg->Lookup("Analyses",analyses);
  if(analyses == NULL) return 0;
  std::cout << analyses->Data() << std::endl;
  TObjArray* list = analyses->Tokenize(" ");
  std::cout << list->GetEntries() << std::endl;
  HarpoAnalyse* hana[30];
  Int_t i = 0;
  for(Int_t k = 0; k<list->GetEntries(); k++){
    TString str = ((TObjString*)(list->At(k)))->String();
    if(str=="HarpoAnalyseVertex"){
      hana[i] = new HarpoAnalyseVertex();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalysePedestalShift"){
      hana[i] = new HarpoAnalysePedestalShift();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseRates"){
      hana[i] = new HarpoAnalyseRates();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseXtalk"){
      hana[i] = new HarpoAnalyseXtalk();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseVdrift"){
      hana[i] = new HarpoAnalyseVdrift();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseMatching"){
      hana[i] = new HarpoAnalyseMatching();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoEventSelector"){
      hana[i] = new HarpoEventSelector();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoSelectCosmics"){
      hana[i] = new HarpoSelectCosmics();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoExcludeBadEvents"){
      hana[i] = new HarpoExcludeBadEvents();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseTrackPairs"){
      hana[i] = new HarpoAnalyseTrackPairs();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoSelectorStraightTracks"){
      hana[i] = new HarpoSelectorStraightTracks();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoVertexing"){
      hana[i] = new HarpoVertexing();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseMultiplicityAget"){
      hana[i] = new HarpoAnalyseMultiplicityAget();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoSelectorPileUp"){
      hana[i] = new HarpoSelectorPileUp();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoClusteringBlocs"){
      hana[i] = new HarpoClusteringBlocs();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseBertrand"){
      hana[i] = new HarpoAnalyseBertrand();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseVertex3D"){
      hana[i] = new HarpoAnalyseVertex3D();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseVertex3Dsim"){
      hana[i] = new HarpoAnalyseVertex3Dsim();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoMatchingVertex"){
      hana[i] = new HarpoMatchingVertex();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoSelectorVertex"){
      hana[i] = new HarpoSelectorVertex();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseEtrack"){
      hana[i] = new HarpoAnalyseEtrack();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalysePattern"){
      hana[i] = new HarpoAnalysePattern();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseTemplate"){
      hana[i] = new HarpoAnalyseTemplate();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseResolution"){
      hana[i] = new HarpoAnalyseResolution();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseNoiseSuppression"){
      hana[i] = new HarpoAnalyseNoiseSuppression();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoClustering"){
      hana[i] = new HarpoClustering();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoHoughTracking"){
      hana[i] = new HarpoHoughTracking();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoTrackingPh"){
      hana[i] = new HarpoTrackingPh();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoKalmanTracking"){
      hana[i] = new HarpoKalmanTracking();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoKalmanNew"){
      hana[i] = new HarpoKalmanNew();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoKalmanFitting"){
      hana[i] = new HarpoKalmanFitting();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoMatching"){
      hana[i] = new HarpoMatching();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseGainStudy"){
      hana[i] = new HarpoAnalyseGainStudy();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalysePrf"){
      hana[i] = new HarpoAnalysePrf();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseBaselineFluct"){
      hana[i] = new HarpoAnalyseBaselineFluct();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseBasic"){
      hana[i] = new HarpoAnalyseBasic();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseRunNoZS"){
      hana[i] = new HarpoAnalyseRunNoZS();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseWriter"){
      hana[i] = new HarpoAnalyseWriter();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseTriggerTime"){
      hana[i] = new HarpoAnalyseTriggerTime();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }
    if(str=="HarpoAnalyseMonitorNew"){
      hana[i] = new HarpoAnalyseMonitorNew();
      std::cout << "Adding " << str.Data() << " " << i << " " << hana[i] << std::endl;
      i++;
      continue;
    }

    std::cout << "Unknown analysis " << str.Data() << std::endl;
  }

  if(i == 0){
    std::cout << "No valid analysis" << std::endl;
    return 0;
  } else
    std::cout << "** N of analysis = " << i << std::endl;
  //  return 0;

  for(Int_t k =0; k<i; k++){
    hana[k]->Init();
    rdr->AddAnalyseFunction(hana[k]);
  }

  HarpoRunConfig * rcfg = hdr->GetRunConfig();
  if (rcfg != NULL ) rcfg->print();
  
  processed = rdr->LoopAll();

  for(Int_t k = i-1; k>=0; k--){
    hana[k]->Save();
  }
  HarpoHistManager::GetPointer()->CloseHistFile();

  delete rdr;
  delete cfg;
  for(Int_t k = i-1; k>=0; k--)
    delete hana[k];
  printf("\n Done processed Events %li\n",processed);
  return 0;
}