analysis/TreeMaker/run_macro.C

0001 int run_macro(
0002           std::string infile = "G4sPHENIX.root",
0003           std::string outfile = "test.root"
0004               )
0005 {
0006
0007   gSystem->Load("libTreeMaker.so");
0008
0009   Fun4AllServer *se = Fun4AllServer::instance();
0010   se->Verbosity( 0 );
0011   // just if we set some flags somewhere in this macro
0012   recoConsts *rc = recoConsts::instance();
0013
0014   Fun4AllInputManager *hitsin = new Fun4AllDstInputManager("DSTin");
0015   //hitsin->fileopen("");
0016   //hitsin->fileopen("$SPHENIX/blank/macros/macros/g4simulations/output/outputDijetTrig_35.root");
0017   hitsin->fileopen( infile );
0018   se->registerInputManager(hitsin);
0019
0020   JetReco *truthjetreco = new JetReco();
0021   TruthJetInput *tji = new TruthJetInput(Jet::PARTICLE);
0022   tji->add_embedding_flag( 2 );
0023   truthjetreco->add_input(tji);
0024   truthjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.2),"AntiKt_Truth_r02");
0025   truthjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.3),"AntiKt_Truth_r03");
0026   truthjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.4),"AntiKt_Truth_r04");
0027   truthjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.5),"AntiKt_Truth_r05");
0028   truthjetreco->set_algo_node("ANTIKT");
0029   truthjetreco->set_input_node("TRUTH");
0030   truthjetreco->Verbosity(0);
0031   se->registerSubsystem(truthjetreco);
0032
0033   JetReco *towerjetreco = new JetReco();
0034   towerjetreco->add_input(new TowerJetInput(Jet::CEMC_TOWER_RETOWER));
0035   towerjetreco->add_input(new TowerJetInput(Jet::HCALIN_TOWER));
0036   towerjetreco->add_input(new TowerJetInput(Jet::HCALOUT_TOWER));
0037   towerjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.2),"AntiKt_Tower_HIRecoSeedsRaw_r02");
0038   towerjetreco->set_algo_node("ANTIKT");
0039   towerjetreco->set_input_node("TOWER");
0040   towerjetreco->Verbosity(0);
0041   se->registerSubsystem(towerjetreco);
0042
0043   JetReco *towerjetreco = new JetReco();
0044   // --- added by Ron (is this right?)
0045   towerjetreco->add_input(new TowerJetInput(Jet::CEMC_TOWER));
0046   towerjetreco->add_input(new TowerJetInput(Jet::HCALIN_TOWER));
0047   towerjetreco->add_input(new TowerJetInput(Jet::HCALOUT_TOWER));
0048   towerjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.2),"AntiKt_Tower_r02");
0049   towerjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.3),"AntiKt_Tower_r03");
0050   towerjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.4),"AntiKt_Tower_r04");
0051   towerjetreco->add_algo(new FastJetAlgo(Jet::ANTIKT,0.5),"AntiKt_Tower_r05");
0052   // ---
0053   towerjetreco->set_algo_node("ANTIKT");
0054   towerjetreco->set_input_node("TOWER");
0055   towerjetreco->Verbosity( 0 );
0056   se->registerSubsystem(towerjetreco);
0057
0058   TreeMaker *tt = new TreeMaker( outfile );
0059   tt->Verbosity(2);
0060   se->registerSubsystem( tt );
0061
0062   se->run( 1 );
0063
0064   se->End();
0065   std::cout << "All done" << std::endl;
0066   delete se;
0067
0068   gSystem->Exit(0);
0069 }