SEnergyCorrelator.io.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file SEnergyCorrelator.io.h

// ----------------------------------------------------------------------------
// 'SEnergyCorrelator.io.h'
// Derek Anderson
// 01.27.2023
//
// A module to implement Peter Komiske's EEC library
// in the sPHENIX software stack for the Cold QCD
// Energy-Energy Correlator analysis.
// ----------------------------------------------------------------------------

#pragma once

using namespace std;



namespace SColdQcdCorrelatorAnalysis {

  // i/o methods --------------------------------------------------------------

  void SEnergyCorrelator::SetInputTree(const string &iTreeName, const bool isTruthTree, const bool isEmbedTree) {

    // print debug statemet
    if (m_inDebugMode) PrintDebug(18);
  
    m_inTreeName       = iTreeName;
    m_isInputTreeTruth = isTruthTree; 
    m_isInputTreeEmbed = isEmbedTree;
    return;

  }  // end 'SetInputTree(string&, bool)'



  void SEnergyCorrelator::SetJetParameters(const vector<pair<double, double>> &pTjetBins, const pair<double, double> etaJetRange) {

    // print debug statement
    if (m_inDebugMode) PrintDebug(20);

    m_etaJetRange = etaJetRange;
    m_nBinsJetPt  = pTjetBins.size();
    for (uint32_t iPtBin = 0; iPtBin < m_nBinsJetPt; iPtBin++) {
      const double               minPt = pTjetBins.at(iPtBin).first;
      const double               maxPt = pTjetBins.at(iPtBin).second;
      const pair<double, double> ptBin = {minPt, maxPt};
      m_ptJetBins.push_back(ptBin);
    }
    m_ptJetRange.first  = m_ptJetBins[0].first;
    m_ptJetRange.second = m_ptJetBins[m_nBinsJetPt - 1].second;

    // announce jet parameters
    if (m_inStandaloneMode) PrintMessage(6);
    return;

  }  // end 'SetJetParameters(vector<pair<double, double>>&, pair<double, double>)'



  void SEnergyCorrelator::SetConstituentParameters(const pair<double, double> momCstRange, const pair<double, double> drCstRange, const bool applyCstCuts) {

    // print debug statement
    if (m_inDebugMode) PrintDebug(24);

    m_momCstRange  = momCstRange;
    m_drCstRange   = drCstRange;
    m_applyCstCuts = applyCstCuts;

    // announce cst parameters
    if (m_inStandaloneMode) PrintMessage(12);
    return;

  }  // end 'SetConstituentParameters(pair<double, double>, pair<double, double>)'



  void SEnergyCorrelator::SetCorrelatorParameters(const uint32_t nPointCorr, const uint64_t nBinsDr, const pair<double, double> drBinRange) {

    // print debug statement
    if (m_inDebugMode) PrintDebug(19);

    m_nPointCorr = nPointCorr;
    m_nBinsDr    = nBinsDr;
    m_drBinRange = drBinRange;

    // announce correlator parameters
    if (m_inStandaloneMode) PrintMessage(5);
    return;

  }  // end 'SetCorrelatorParameters(uint32_t, uint64_t, pair<double, double>)'



  void SEnergyCorrelator::SetSubEventsToUse(const uint16_t subEvtOpt, const vector<int> vecSubEvtsToUse) {

    // print debug statement
    if (m_inDebugMode) PrintDebug(32);

    // parse options
    m_subEvtOpt = subEvtOpt;
    if (m_subEvtOpt != 0) {
      m_selectSubEvts = true;
    }

    // if vector isn't empty, load specific emebedding IDs and set flags accordingly
    if (vecSubEvtsToUse.size() > 0) {
      m_selectSubEvts = true;
      m_subEvtOpt     = 5;
      for (const int subEvtToUse : vecSubEvtsToUse) {
        m_subEvtsToUse.push_back(subEvtToUse);
      }
    }

    // announce sub-events being used
    if (m_inStandaloneMode) PrintMessage(15);
    return;

  }  // end 'SetSubEventsToUse(uint16_t, vector<int>)'



  void SEnergyCorrelator::GrabInputNode() {

    // print debug statement
    if (m_inDebugMode) PrintDebug(3);

    /* TODO method goes here */
    return;

  }  // end 'GrabInputNode()'



  void SEnergyCorrelator::OpenInputFiles() {

    // print debug statement
    if (m_inDebugMode) PrintDebug(11);

    // insantiate input chain
    m_inChain = new TChain(m_inTreeName.data());
    if (!m_inChain) {
      PrintError(6);
      assert(m_inChain);
    }

    // loop over provided input files
    bool     isFileGood = true;
    uint64_t bytes      = 0;
    for (const string& inFileName : m_inFileNames) {
      bytes      = m_inChain -> Add(inFileName.data(), 0);
      isFileGood = (bytes > 0);
      if (!isFileGood) {
        PrintError(7, 0, 0, inFileName);
        assert(isFileGood);
      }
    }  // end input file loop
    return;

  }  // end 'OpenInputFile()'



  void SEnergyCorrelator::OpenOutputFile() {

    // print debug statement
    if (m_inDebugMode) PrintDebug(15);

    // open file
    m_outFile = new TFile(m_outFileName.data(), "recreate");
    if (!m_outFile) {
      PrintError(11);
      assert(m_outFile);
    }
    return;

  }  // end 'OpenOutputFile()'



  void SEnergyCorrelator::SaveOutput() {

    // print debug statement
    if (m_inDebugMode) PrintDebug(9);

    m_outFile -> cd();
    for (size_t iPtBin = 0; iPtBin < m_nBinsJetPt; iPtBin++) {
      m_outHistVarDrAxis[iPtBin]   -> Write();
      m_outHistErrDrAxis[iPtBin]   -> Write();
      m_outHistVarLnDrAxis[iPtBin] -> Write();
      m_outHistErrLnDrAxis[iPtBin] -> Write();
    }

    // for weird cst check
    if (m_doSecondCstLoop) {
      m_outFile          -> cd();
      hCstPtOneVsDr      -> Write();
      hCstPtTwoVsDr      -> Write();
      hCstPtFracVsDr     -> Write();
      hCstPhiOneVsDr     -> Write();
      hCstPhiTwoVsDr     -> Write();
      hCstEtaOneVsDr     -> Write();
      hCstEtaTwoVsDr     -> Write();
      hDeltaPhiOneVsDr   -> Write();
      hDeltaPhiTwoVsDr   -> Write();
      hDeltaEtaOneVsDr   -> Write();
      hDeltaEtaTwoVsDr   -> Write();
      hJetPtFracOneVsDr  -> Write();
      hJetPtFracTwoVsDr  -> Write();
      hCstPairWeightVsDr -> Write();
    }

    // announce saving
    if (m_inStandaloneMode) PrintMessage(10);
    return;

  }  // end 'SaveOutput()'



  void SEnergyCorrelator::CloseOutputFile() {

    // print debug statement
    if (m_inDebugMode) PrintDebug(30);

    // close file
    m_outFile -> cd();
    m_outFile -> Close();
    return;

  }  // end 'CloseOutputFile()'

}  // end SColdQcdCorrelatorAnalysis namespace

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