DoStandaloneCorrelatorCalculation.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file DoStandaloneCorrelatorCalculation.C

// ----------------------------------------------------------------------------
// 'DoStandaloneCorrelatorCalculation.C'
// Derek Anderson
// 02.02.2023
//
// Use this to run the SEnergyCorrelator
// class in standalone mode.
//
// NOTE: subEvtOpt controls which sub-event
// to use in MC
//   0 -- use everything
//   1 -- use only signal subevent
//   2 -- use only background subevents
//        (pileup and underlying event)
//   3 -- use only primary background
//        (i.e. underlying event)
//   4 -- use only pileup subevents
//   5 -- use specific set of subevents
// ----------------------------------------------------------------------------

#ifndef DOSTANDALONECORRELATORCALCULATION_C
#define DOSTANDALONECORRELATORCALCULATION_C

// standard c includes
#include <string>
#include <vector>
#include <cstdlib>
#include <utility>
// user includes
#include "/sphenix/user/danderson/install/include/senergycorrelator/SEnergyCorrelator.h"

// load libraries
R__LOAD_LIBRARY(/sphenix/user/danderson/install/lib/libsenergycorrelator.so)

using namespace std;
using namespace SColdQcdCorrelatorAnalysis;



void DoStandaloneCorrelatorCalculation() {

  // io parameters
  const vector<vector<string>> inFile = {
    {"../SCorrelatorJetTree/output/condor/final_merge/correlatorJetTree.pp200py8jet10run8_trksWithRoughCuts.d26m9y2023.root"},
    {"../SCorrelatorJetTree/output/condor/final_merge/correlatorJetTree.pp200py8jet10run8_trksWithRoughCuts.d26m9y2023.root"}
  };
  const vector<string> inTree = {
    "RecoJetTree",
    "TruthJetTree"
  };
  const vector<string> outFile = {
    "twoPoint.pp200py8jet10run8.testCstPtUpperLimit_pt10100_roughCutsWithGoodJetEta_reco.d25m10y2023.root",
    "twoPoint.pp200py8jet10run8.testCstPtUpperLimit_pt10100_roughCutsWithGoodJetEta_true.d25m10y2023.root"
  };

  // module parameters
  const vector<string> moduleName = {
    "SRecoEnergyCorrelator",
    "STrueEnergyCorrelator"
  };
  const vector<bool> isTruth = {
    false,
    true
  };

  // correlator parameters
  const uint32_t             nPointCorr = 2;
  const uint64_t             nBinsDr    = 75;
  const pair<double, double> binRangeDr = {1e-5, 1.};

  // jet/cst parameters
  const pair<double, double> etaJetRange = {-0.7, 0.7};
  const pair<double, double> momCstRange = {10., 100.};
  const pair<double, double> drCstRange  = {0.,  100.};

  // jet pT bins
  const vector<pair<double, double>> ptJetBins = {{5., 10.}, {10., 15.}, {15., 20.}, {20., 30.}, {30., 50.}};

  // embedding-specific options
  const int  subEvtOpt = 0;
  const bool isEmbed   = false;

  // misc parameters
  const int  verbosity  = 0;
  const bool isComplex  = false;
  const bool doCstCuts  = true;
  const bool doDebug    = false;
  const bool doCstLoop2 = false;
  const bool inBatch    = false;

  // do correlator calculation on reco jets
  SEnergyCorrelator* recoCorrelator = new SEnergyCorrelator(moduleName[0], isComplex, doDebug, inBatch);
  recoCorrelator -> SetVerbosity(verbosity);
  recoCorrelator -> SetInputFiles(inFile[0]);
  recoCorrelator -> SetInputTree(inTree[0], isTruth[0]);
  recoCorrelator -> SetOutputFile(outFile[0]);
  recoCorrelator -> SetJetParameters(ptJetBins, etaJetRange);
  recoCorrelator -> SetConstituentParameters(momCstRange, drCstRange, doCstCuts);
  recoCorrelator -> SetCorrelatorParameters(nPointCorr, nBinsDr, binRangeDr);
  recoCorrelator -> SetDoSecondCstLoop(doCstLoop2);
  recoCorrelator -> Init();
  recoCorrelator -> Analyze();
  recoCorrelator -> End();

  // do correlator calculation on truth jets
  SEnergyCorrelator* trueCorrelator = new SEnergyCorrelator(moduleName[1], isComplex, doDebug, inBatch);
  trueCorrelator -> SetVerbosity(verbosity);
  trueCorrelator -> SetInputFiles(inFile[1]);
  trueCorrelator -> SetInputTree(inTree[1], isTruth[1], isEmbed);
  trueCorrelator -> SetOutputFile(outFile[1]);
  trueCorrelator -> SetJetParameters(ptJetBins, etaJetRange);
  trueCorrelator -> SetConstituentParameters(momCstRange, drCstRange, doCstCuts);
  trueCorrelator -> SetCorrelatorParameters(nPointCorr, nBinsDr, binRangeDr);
  if (isEmbed) {
    trueCorrelator -> SetSubEventsToUse(subEvtOpt);
  }
  trueCorrelator -> SetDoSecondCstLoop(doCstLoop2);
  trueCorrelator -> Init();
  trueCorrelator -> Analyze();
  trueCorrelator -> End();
  return;

}

#endif

// end ------------------------------------------------------------------------