makeEEmcTreePart1.C

/* 
 * Created May 2012, by S. Gliske
 *
 * Macro to write "Part 1" of the EEmcTree, by reading in a MuDst
 * and applying the calibration.
 *
 * Note: much of this script is based on Alice's runABClusterMaker.C 
 * and Weihong's runEEmcPi0.C
 * 
 */

// forward declarations
class StChain;
class St_db_Maker;
class StEEmcDbMaker;
class StMuDstMaker;
class StEEmcA2EMaker;
class StSpinDbMaker;
class StEEmcSlowMaker;

class StSpinInfoMaker_t;
class StEEmcEnergyMaker_t;
class StEEmcTreeMaker_t;
class StMcEEmcTreeMaker_t;
class StTrigCounter;

//
// some variables that others tend to make global
//
StChain             *analysisChain    = 0;
StMuDstMaker        *muDstMaker       = 0;
St_db_Maker         *starDatabase     = 0;
StEEmcDbMaker       *eemcDbMaker      = 0;
StSpinDbMaker       *spinDb           = 0;
StEEmcA2EMaker      *a2EMakerPtr      = 0;
StEEmcEnergyMaker_t *energyMakerPtr   = 0;
StEEmcTreeMaker_t   *treeMakerPtr     = 0;
StMcEEmcTreeMaker_t *mcTreeMakerPtr   = 0;
StSpinInfoMaker_t   *spinInfoMakerPtr = 0;

//
// the main routine
//
void makeEEmcTreePart1( Long_t neventsIn = -1, 
                        Long_t neventsOut = -1, 
                        const Char_t *inputFileName = "",
                        const Char_t *outputFileName = "McEEmcTree.root",
                        Int_t displayFreq = 1000,
                        Char_t trigVer = 'f',
                        Char_t* eemcSetupPath = "/star/u/sgliske/Share/StarTrigSimuSetup/",  // make sure this path ends with a '/' and has a subdirectory for the year of interest
                        Bool_t isMC = 0,
                        Bool_t noMCdateTime = 0,
                        Bool_t skipMcTreeMaker = 0 ){

   std::cout << "***** Loading libraries *****" << endl;

   gROOT->LoadMacro("$STAR/StRoot/StMuDSTMaker/COMMON/macros/loadSharedLibraries.C");
   loadSharedLibraries();
   assert( !gSystem->Load("StDetectorDbMaker"));
   assert( !gSystem->Load("StTpcDb"));
   assert( !gSystem->Load("StDbUtilities"));
   assert( !gSystem->Load("StDbBroker"));
   assert( !gSystem->Load("St_db_Maker"));
   assert( !gSystem->Load("StEEmcUtil")); // needed by eemcDb
   assert( !gSystem->Load("StEEmcDbMaker"));
   assert( !gSystem->Load("StEEmcA2EMaker"));
   assert( !gSystem->Load("StDaqLib")); // needed by bemcDb
   assert( !gSystem->Load("StEmcRawMaker"));
   assert( !gSystem->Load("StEmcADCtoEMaker"));
   if (isMC) {
      assert( !gSystem->Load("StMcEvent"));
      assert( !gSystem->Load("StMcEventMaker"));
      assert( !gSystem->Load("StEmcSimulatorMaker"));
      assert( !gSystem->Load("StEEmcSimulatorMaker"));
      assert( !gSystem->Load("StEpcMaker"));
   }
   assert( !gSystem->Load("StSpinDbMaker") );
   assert( !gSystem->Load("StTriggerUtilities"));
   assert( !gSystem->Load("StEEmcPoolEEmcTreeContainers") );
   assert( !gSystem->Load("StEEmcTreeMaker") );

   //gDebug=5;

   std::cout << "***** Done loading libraries *****" << endl;
   std::cout << "***** Instanciating all the classes *****" << endl;

   //
   // Create the analysis chain
   //
   analysisChain = new StChain("eemcAnalysisChain");

   //
   // IO Maker to read the geant file
   //
   if( isMC ){
      ioMaker = new StIOMaker();

      TString geantFileName = inputFileName;
      geantFileName.ReplaceAll("MuDst","geant");
      ioMaker->SetFile(geantFileName);

      ioMaker->SetIOMode("r");
      ioMaker->SetBranch("*",0,"1");             //deactivate all branches
      ioMaker->SetBranch("geantBranch",0,"r");   //activate geant Branch
      ioMaker->SetBranch("minimcBranch",0,"r");   //activate geant Branch
   };

   //
   // MuDst maker for reading input
   //
   Int_t nfiles = 10000;
   muDstMaker = new StMuDstMaker( 0, 0, "", inputFileName, "MuDst", nfiles );
   muDstMaker->SetStatus("*",0);
   muDstMaker->SetStatus("*Event*",1);
   muDstMaker->SetStatus("PrimaryVertices",1);
   muDstMaker->SetStatus("EmcAll",1);

   // just to make sure StEvent is found for the trig simu
   StMuDst2StEventMaker *muDst2StEvent = new StMuDst2StEventMaker();

   //
   // Maker for StMcEvent
   //
   if( isMC ){
      StMcEventMaker *mcEventMaker = new StMcEventMaker();
      mcEventMaker->doPrintEventInfo = false;
      mcEventMaker->doPrintMemoryInfo = false;
   };


   //
   // Connect to the STAR databse
   //
   starDatabase = new St_db_Maker("StarDb","MySQL:StarDb","MySQL:StarDb","$STAR/StarDb");
   starDatabase->SetAttr("blacklist", "fgt");
   starDatabase->SetAttr("blacklist", "svt");
   starDatabase->SetAttr("blacklist", "tpc");
   starDatabase->SetAttr("blacklist", "ftpc");

   if( isMC && noMCdateTime )
      starDatabase->SetDateTime(20060512,43500);

   //
   // Initialize EEMC database
   //
   eemcDbMaker = new StEEmcDbMaker("eemcDb");

   // this is the interface to the STAR logger
   // SwitchOn("D") turns on debugging
   // SwitchOff("D") turns off debuggin
   // "I" is info and "W" is warn
   gMessMgr->SwitchOff("D");
   gMessMgr->SwitchOff("I");
   //gMessMgr->SetLevel(2);

   // note: moving the slow simu to here doesn't seem to make any difference

   //
   // Trigger counter
   //
   StTrigCounter* trigCounter = new StTrigCounter( "trigCounter", 117001 );

   //
   // Energy to ADC maker
   //
   a2EMakerPtr = new StEEmcA2EMaker("EEmcA2EMaker");
   a2EMakerPtr->database("eemcDb");          // sets db connection
   a2EMakerPtr->source("MuDst",1);           // sets mudst as input
   a2EMakerPtr->threshold(3.0,0);            // tower threshold
   a2EMakerPtr->threshold(3.0,1);            // pre1 threshold 
   a2EMakerPtr->threshold(3.0,2);            // pre2 threshold
   a2EMakerPtr->threshold(3.0,3);            // post threshold
   a2EMakerPtr->threshold(3.0,4);            // smdu threshold
   a2EMakerPtr->threshold(3.0,5);            // smdv threshold

   //
   // Now start things particular to the StEEmcTreeMaker
   //


   //
   // create spin DB maker and info maker
   //
   if( !isMC ){
      spinDb = new StSpinDbMaker("spinDb");
      spinInfoMakerPtr = new StSpinInfoMaker_t( "SpinInfoMaker" );
   };

   // Energy Structure Maker
   energyMakerPtr = new StEEmcEnergyMaker_t( "energyMkr", "EEmcA2EMaker" );
   energyMakerPtr->setStripThres( isMC ? 0.0 : 0.001 );
   energyMakerPtr->setTowerThres( isMC ? 0.0 : 1.0 );

   // I'm told one must use slow simulator to get pedestals correct for L2,
   // but I'll put it after the A2Emaker and Energy maker, so it doesn't mess them up

   // Note: this will actually add the pedestal to the adc value, so
   // it seems this is the wrong thing to do if one already has the
   // pedestals in the adc's in the MuDst.
   if( 0 && isMC==1 ){
      StEEmcSlowMaker *slowSim = new StEEmcSlowMaker("slowSim");

      // note: changing the sampling fraction does not effect the
      // L0 trigger decision.  Also, it is probably better to use the value
      // in the slow simulator, so that it matches data. (S. Gliske,
      // Sept 2012)

      // This 1.3 comes from the iron/air MC, where the sampling
      // fraction was believed to be near 4%, but slow/fast simulator
      // used 5%.

      // Now that the slow/fast simulator uses 0.048, it is a 1.25 scaling factor

      // slowSim->setSamplingFraction(0.0384); // effectively scale all Tower energies with a factor of 1.3 (added by: Ilya Selyuzhenkov; April 11, 2008)

      slowSim->setAddPed(true);
      slowSim->setSmearPed(true);
   };

   //
   // Trigger emulator
   //
   StTriggerSimuMaker *simuTrig = new StTriggerSimuMaker("StarTrigSimu");
   simuTrig->setMC(isMC); // must be before individual detectors, to be passed
   simuTrig->useBbc();
   simuTrig->useEemc(0); //default=0:just process ADC, 1,2:comp w/trgData,see .
   simuTrig->eemc->setSetupPath(eemcSetupPath);

   // Collect all output histograms 
   TObjArray* HList=new TObjArray; 
   simuTrig->setHList(HList);

   // DSM settings
   int eemcDsmSetup[20]; // see StEemcTriggerSimu::initRun() for definition
   memset(eemcDsmSetup, 0,sizeof(eemcDsmSetup));// clear all, may be a bad default

   if( trigVer == 'd' ){
      // valid runs 7130037-7132029
      eemcDsmSetup[0]=6;  // HTthr0
      eemcDsmSetup[1]=12; // HTthr1
      eemcDsmSetup[2]=22; // HTthr2
      eemcDsmSetup[3]=1;  // TPthr0
      eemcDsmSetup[4]=20; // TPthr1
      eemcDsmSetup[5]=31; // TPthr2
   } else if ( trigVer == 'e' ){
      // valid runs 7132045-7133051
      eemcDsmSetup[0]=6;  // HTthr0
      eemcDsmSetup[1]=17; // HTthr1
      eemcDsmSetup[2]=22; // HTthr2
      eemcDsmSetup[3]=1;  // TPthr0
      eemcDsmSetup[4]=20; // TPthr1
      eemcDsmSetup[5]=31; // TPthr2
   } else if ( trigVer == 'f' ){
      // valid runs 7133052-7156040
      eemcDsmSetup[0]=6;  // HTthr0
      eemcDsmSetup[1]=16; // HTthr1
      eemcDsmSetup[2]=22; // HTthr2
      eemcDsmSetup[3]=1;  // TPthr0
      eemcDsmSetup[4]=20; // TPthr1
      eemcDsmSetup[5]=31; // TPthr2
   } else if ( trigVer == 'g' ){
      // test trigger, ht1 & tp2 are 1 DSM (roughly 5%) higher than 'e'
      eemcDsmSetup[0]=6;  // HTthr0
      eemcDsmSetup[1]=18; // HTthr1
      eemcDsmSetup[2]=22; // HTthr2
      eemcDsmSetup[3]=1;  // TPthr0
      eemcDsmSetup[4]=21; // TPthr1
      eemcDsmSetup[5]=31; // TPthr2
   } else if ( trigVer == 'h' ){
      // test trigger, ht1 & tp2 are 2 DSM (roughly 10%) higher than 'e'
      eemcDsmSetup[0]=6;  // HTthr0
      eemcDsmSetup[1]=19; // HTthr1
      eemcDsmSetup[2]=22; // HTthr2
      eemcDsmSetup[3]=1;  // TPthr0
      eemcDsmSetup[4]=22; // TPthr1
      eemcDsmSetup[5]=31; // TPthr2
   } else {
      cerr << "Invalid trigger version" << endl;
      return;
   };
   eemcDsmSetup[10]=2; //HTTPthrSelc, 2=use_thres_#1
   simuTrig->eemc->setDsmSetup(eemcDsmSetup);    

   // so far, just 2006 in this macro
   StGenericL2Emulator* simL2Mk = new StL2_2006EmulatorMaker;
   simL2Mk->setSetupPath(eemcSetupPath);
   
   if (isMC) simL2Mk->setMC();
   simuTrig->useL2(simL2Mk);

   //
   // Tree Maker
   //
   treeMakerPtr = new StEEmcTreeMaker_t( "EEmcTreeMkr" );
   treeMakerPtr->setTreeStatus( StEEmcTreeMaker_t::PART_1, StEEmcTreeMaker_t::WRITE,  outputFileName );
   treeMakerPtr->setTreeStatus( StEEmcTreeMaker_t::PART_2, StEEmcTreeMaker_t::IGNORE, "" );
   treeMakerPtr->setTreeStatus( StEEmcTreeMaker_t::PART_3, StEEmcTreeMaker_t::IGNORE, "" );
   treeMakerPtr->setMaxNumEvents( neventsIn );
   treeMakerPtr->setEEmcEnergyMkr( energyMakerPtr );
   treeMakerPtr->doSpinInfoIO( !isMC );
   treeMakerPtr->doEvtHddrIO( 1 );
   treeMakerPtr->setHTTPthres( 0, 0 );
 


   if( !isMC )
      treeMakerPtr->setSpinInfoMkr( spinInfoMakerPtr );

   // McTree Maker 
   if( isMC && !skipMcTreeMaker ){
      // determine the McTree filename from the outputFileName
      TString filename = outputFileName;
      if( filename.Contains("EEmcTree") )
         filename.ReplaceAll("EEmcTree", "McEEmcTree" );
      else if ( filename.EndsWith(".root") )
         filename.ReplaceAll(".root", ".McEEmcTree.root");
      else 
         filename += ".McEEmcTree.root";
      filename.ReplaceAll("_Part1","");
      filename.ReplaceAll(".Part1","");

      mcTreeMakerPtr = new StMcEEmcTreeMaker_t( "McEEmcTreeMkr" );
      mcTreeMakerPtr->setEnergyThreshold( 0.0 );
      mcTreeMakerPtr->addTrigger( -999 );
      mcTreeMakerPtr->setTreeStatus( StMcEEmcTreeMaker_t::WRITE, filename.Data() );
   };

   // debugging info
   std::cout << "***** Done instanciating all the classes *****" << endl;
   analysisChain->ls(3);

   //
   // Initialize all makers
   //

   std::cout << "***** Initializing all makers in the analysis chain *****" << std::endl;

   analysisChain->Init();

   std::cout << "***** Initialization done *****" << std::endl;

   //
   // Finally ready to loop over the events 
   //

   // If neventsIn/Out is negative, reset to a large value
   // for an Int_t
   if( neventsIn < 0 )
      neventsIn = 1<<30-1;
   if( neventsOut < 0 )
      neventsOut = 1<<30-1;

   Int_t ierr  = kStOK;  // err flag
   Long_t nevents = 1;    // cumulative number of events in
   for( ; nevents <= neventsIn && treeMakerPtr->getNumPart1EventsWritten() < neventsOut && !ierr; ++nevents ){
      // clear
      analysisChain->Clear();

      // make
      ierr = analysisChain->Make();

      // Print every so many events
      if( (nevents+1) % displayFreq == 1 ){
         trigCounter->printStatus();
         std::cout << "***** finished event number " << nevents << ", " << treeMakerPtr->getNumPart1EventsWritten() << " *****" << std::endl;
      };

      if( ierr ){
         std::cout << "***** ERROR FLAG " << ierr << " on event number " << nevents << " *****" << endl;
      };
   };
 
   //---------------------------------------------------------------


   //
   // Calls the ::Finish() method on all makers
   //
   analysisChain->Finish(); 

   //
   // Delete the chain
   //
   // analysisChain->Delete();

   return;
};

/*
 * $Id: makeEEmcTreePart1.C,v 1.1 2012/12/17 20:01:28 sgliske Exp $
 * $Log: makeEEmcTreePart1.C,v $
 * Revision 1.1  2012/12/17 20:01:28  sgliske
 * moved from offline/users/sgliske/StRoot/StEEmcPool/StEEmcTreeMaker/macros
 *
 *
 */