source: svn/trunk/Examples/Trigger_Only.cpp@ 262

/***********************************************************************
**                                                                    **
**          /----------------------------------------------\          **
**         |  Delphes, a framework for the fast simulation  |         **
**         |       of a  generic collider experiment        |         **
**          \----------------------------------------------/          **
**                                                                    **
**                                                                    **
**   This package uses:                                               **
**   ------------------                                               **
**   FastJet algorithm: Phys. Lett. B641 (2006) [hep-ph/0512210]      ** 
**   Hector: JINST 2:P09005 (2007) [physics.acc-ph:0707.1198v2]       ** 
**   FROG: [hep-ex/0901.2718v1]                                       **
**                                                                    **
** ------------------------------------------------------------------ **
**                                                                    **
**   Main authors:                                                    **
**   -------------                                                    **
**                                                                    **
**                Severine Ovyn                Xavier Rouby           ** 
**          severine.ovyn@uclouvain.be      xavier.rouby@cern         ** 
**                                                                    ** 
**         Center for Particle Physics and Phenomenology (CP3)        ** 
**                Universite catholique de Louvain (UCL)              **       
**                     Louvain-la-Neuve, Belgium                      **            
**                                                                    **
**                      Copyright (C) 2008-2009,                      **
**                        All rights reserved.                        **  
**                                                                    **
***********************************************************************/

/// \file Trigger_Only.cpp
/// \brief executable for the Trigger running

#include "TChain.h"
#include "TApplication.h"

#include "ExRootTreeReader.h"
#include "ExRootTreeWriter.h"
#include "ExRootTreeBranch.h"
#include "TriggerUtil.h"
#include "BlockClasses.h"

#include <iostream>
#include <vector>
#include <fstream>
#include <iomanip>


using namespace std;

//------------------------------------------------------------------------------
void todo(string filename) {
  ifstream infile(filename.c_str());
  cout << "** TODO list ..." << endl;
  while(infile.good()) {
        string temp;
        getline(infile,temp);
        cout << "*" << temp << endl;
  }
  cout << "** done...\n"; 
}

//------------------------------------------------------------------------------

int main(int argc, char *argv[])
{
  int appargc = 2;
  char *appName = "Trigger_Only";
  char *appargv[] = {appName, "-b"};
  TApplication app(appName, &appargc, appargv);

  if(argc != 2 && argc !=3 ) {
      cout << " Usage: " << argv[0] << " input_file output_file [detector_card] [trigger_card] " << endl;
      cout << " input_file - file in Delphe root format," << endl;
      cout << " trigger_card - Datacard containing the trigger algorithms (optional) "<<endl;
      exit(1);
  }

  srand (time (NULL));         /* Initialisation du générateur */
  
  //read the input TROOT file
  string inputFile(argv[1]);
  
  //create output log-file name
  string forLog = inputFile;
  string LogName = forLog.erase(forLog.find(".root"));
  LogName = LogName+"_run.log";

  string line,buffer;
  TChain chain("Analysis");
  chain.Add(inputFile.c_str());
  ExRootTreeReader *treeReaderT = new ExRootTreeReader(&chain);

  //read the datacard input file
  string TrigDatacard("");
  if(argc==3)  TrigDatacard =argv[2];

  //Trigger information
  TriggerTable *TRIGT = new TriggerTable();
  TRIGT->TriggerCardReader(TrigDatacard.c_str());
  TRIGT->PrintTriggerTable(LogName);

  TClonesArray *branchElecTrig = treeReaderT->UseBranch("Electron");
  TClonesArray *branchMuonTrig = treeReaderT->UseBranch("Muon");
  TClonesArray *branchJetTrig = treeReaderT->UseBranch("Jet");
  TClonesArray *branchTauJetTrig = treeReaderT->UseBranch("TauJet");
  TClonesArray *branchPhotonTrig = treeReaderT->UseBranch("Photon");
  TClonesArray *branchETmisTrig = treeReaderT->UseBranch("ETmis");
      
  ExRootTreeWriter *treeWriterT = new ExRootTreeWriter(inputFile, "Trigger");
  ExRootTreeBranch *branchTrigger = treeWriterT->NewBranch("TrigResult", TRootTrigger::Class());
      
  Long64_t entryT, allEntriesT = treeReaderT->GetEntries();
  cout << "** Chain contains " << allEntriesT << " events" << endl;
  for(entryT = 0; entryT < allEntriesT; ++entryT)
    {
      treeWriterT->Clear();
      treeReaderT->ReadEntry(entryT);
      TRIGT->GetGlobalResult(branchElecTrig, branchMuonTrig,branchJetTrig, branchTauJetTrig,branchPhotonTrig, branchETmisTrig,branchTrigger);
      treeWriterT->Fill();
    }

  treeWriterT->Write();
  delete treeWriterT;

  cout << "** Exiting..." << endl;
  
  delete TRIGT;
 
}