KFParticle_sPHENIX.h

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

/*
 * This file is part of KFParticle package
 * Copyright (C) 2007-2019 FIAS Frankfurt Institute for Advanced Studies
 *               2007-2019 Goethe University of Frankfurt
 *               2007-2019 Ivan Kisel <I.Kisel@compeng.uni-frankfurt.de>
 *               2007-2019 Maksym Zyzak
 *
 * KFParticle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * KFParticle is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

#ifndef KFPARTICLESPHENIX_KFPARTICLESPHENIX_H
#define KFPARTICLESPHENIX_KFPARTICLESPHENIX_H

#include "KFParticle_eventReconstruction.h"
#include "KFParticle_nTuple.h"
#include "KFParticle_DST.h"

//sPHENIX stuff
#include <fun4all/SubsysReco.h>

//KFParticle stuff
#include <KFParticle.h>

#include <map>
#include <string>

class TFile;

class KFParticle_sPHENIX : public SubsysReco, public KFParticle_nTuple, public KFParticle_DST, protected KFParticle_eventReconstruction
{
 public:
  KFParticle_sPHENIX();

  explicit KFParticle_sPHENIX(const std::string &name);

  virtual ~KFParticle_sPHENIX(){}

  int Init(PHCompositeNode *topNode);

  int process_event(PHCompositeNode *topNode);

  int End(PHCompositeNode *topNode);

  void printParticles(KFParticle motherParticle,
                      KFParticle chosenVertex,
                      std::vector<KFParticle> daughterParticles,
                      std::vector<KFParticle> intermediateParticles,
                      int numPVs, int numTracks);

  int parseDecayDescriptor();

  bool findParticle(std::string particle);


  void setDecayDescriptor(const std::string &decayDescriptor) { m_decayDescriptor = decayDescriptor; }

  static const int max_particles = 99;

  void setMotherName(const std::string& mother_name)
  {
    m_mother_name = mother_name;
    m_mother_name_Tools = mother_name;
  }

  void useIntermediateName(bool use_intermediate_name) { m_use_intermediate_name = use_intermediate_name; }

  void hasIntermediateStates(bool has_intermediates)
  {
    m_has_intermediates = has_intermediates;
    m_has_intermediates_nTuple = has_intermediates;
    m_has_intermediates_sPHENIX = has_intermediates;
    m_has_intermediates_DST = has_intermediates;
  }

  void setNumberOfTracks(int num_tracks)
  {
    m_num_tracks = num_tracks;
    m_num_tracks_nTuple = num_tracks;
  }

  void setNumberTracksFromIntermeditateState(const std::vector<int> & num_tracks)
  {
    for (unsigned int i = 0; i < num_tracks.size(); ++i) 
    {
      m_num_tracks_from_intermediate.push_back(num_tracks[i]);
      m_num_tracks_from_intermediate_nTuple.push_back(num_tracks[i]);
    }
  }

  void setNumberOfIntermediateStates(int n_intermediates)
  {
    m_num_intermediate_states = n_intermediates;
    m_num_intermediate_states_nTuple = n_intermediates;
  }

  void getChargeConjugate(bool get_charge_conjugate)
  {
    m_get_charge_conjugate_nTuple = get_charge_conjugate;
    m_get_charge_conjugate = get_charge_conjugate;
  }

  void setDaughters(std::vector<std::pair<std::string, int>> daughter_list)
  {
    for (unsigned int i = 0; i < daughter_list.size(); ++i)
    {
      m_daughter_name.push_back(daughter_list[i].first);
      m_daughter_charge.push_back(daughter_list[i].second);
    }
  }

  void setIntermediateStates(std::vector<std::pair<std::string, int>> intermediate_list)
  {
    for (unsigned int i = 0; i < intermediate_list.size(); ++i)
    {
      m_intermediate_name_ntuple.push_back(intermediate_list[i].first);
      m_intermediate_name.push_back(intermediate_list[i].first);
      m_intermediate_charge.push_back(intermediate_list[i].second);
    }
  }

  void setMinimumMass(float min_mass) { m_min_mass = min_mass; }

  void setMaximumMass(float max_mass) { m_max_mass = max_mass; }

  void setDecayTimeRange(float min_decayTime, float max_decayTime)
  {
    m_min_decayTime = min_decayTime; 
    m_max_decayTime = max_decayTime; 
  }

  void setDecayLengthRange(float min_decayLength, float max_decayLength)
  {
    m_min_decayLength = min_decayLength; 
    m_max_decayLength = max_decayLength; 
  }

  void setMinimumTrackPT(float pt) { m_track_pt = pt; }

  void setMaximumTrackPTchi2(float ptchi2) { m_track_ptchi2 = ptchi2; }

  void setMinimumTrackIP(float ip) { m_track_ip = ip; }

  void setMinimumTrackIPchi2(float ipchi2) { m_track_ipchi2 = ipchi2; }

  void setMaximumTrackchi2nDOF(float trackchi2ndof) { m_track_chi2ndof = trackchi2ndof; }

  void setMaximumDaughterDCA(float dca) { m_comb_DCA = dca; }

  void setMaximumVertexchi2nDOF(float vertexchi2nDOF) { m_vertex_chi2ndof = vertexchi2nDOF; }

  void setFlightDistancechi2(float fdchi2) { m_fdchi2 = fdchi2; }

  void setMinDIRA(float dira_min) { m_dira_min = dira_min; }

  void setMaxDIRA(float dira_max) { m_dira_max = dira_max; }

  void setMotherPT(float mother_pt) { m_mother_pt = mother_pt; }

  void setMotherIPchi2(float mother_ipchi2) { m_mother_ipchi2 = mother_ipchi2; }

  void constrainToPrimaryVertex(bool constrain_to_vertex)
  {
    m_constrain_to_vertex = constrain_to_vertex;
    m_constrain_to_vertex_nTuple = constrain_to_vertex;
    m_constrain_to_vertex_sPHENIX = constrain_to_vertex;
  }

  void useFakePrimaryVertex(bool use_fake) { m_use_fake_pv = use_fake; }

  void allowZeroMassTracks(bool allow) { m_allowZeroMassTracks = allow; }

  void constrainIntermediateMasses(bool constrain_int_mass) { m_constrain_int_mass = constrain_int_mass; }

  void setIntermediateMassRange(std::vector<std::pair<float, float>> intermediate_mass_range)
  {
    for (unsigned int i = 0; i < intermediate_mass_range.size(); ++i) m_intermediate_mass_range.push_back(intermediate_mass_range[i]);
  }

  void setIntermediateMinPT(std::vector<float> intermediate_min_pt)
  {
    m_intermediate_min_pt = intermediate_min_pt;
  }

  void setIntermediateMinIP(std::vector<float> intermediate_min_IP)
  {
    for (unsigned int i = 0; i < intermediate_min_IP.size(); ++i) m_intermediate_min_ip.push_back(intermediate_min_IP[i]);
  }

  void setIntermediateIPRange(std::vector<std::pair<float, float>> intermediate_IP_range)
  {
    for (unsigned int i = 0; i < intermediate_IP_range.size(); ++i)
    {
       m_intermediate_min_ip.push_back(intermediate_IP_range[i].first);
       m_intermediate_max_ip.push_back(intermediate_IP_range[i].second);
    }
  }

  void setIntermediateMinIPchi2(std::vector<float> intermediate_min_IPchi2)
  {
    for (unsigned int i = 0; i < intermediate_min_IPchi2.size(); ++i) m_intermediate_min_ipchi2.push_back(intermediate_min_IPchi2[i]);
  }

  void setIntermediateIPchi2Range(std::vector<std::pair<float, float>> intermediate_IPchi2_range)
  {
    for (unsigned int i = 0; i < intermediate_IPchi2_range.size(); ++i)
    {
       m_intermediate_min_ipchi2.push_back(intermediate_IPchi2_range[i].first);
       m_intermediate_max_ipchi2.push_back(intermediate_IPchi2_range[i].second);
    }
  }

  void setIntermediateMinDIRA(std::vector<float> intermediate_min_DIRA)
  {
    for (unsigned int i = 0; i < intermediate_min_DIRA.size(); ++i) m_intermediate_min_dira.push_back(intermediate_min_DIRA[i]);
  }

  void setIntermediateMinFDchi2(std::vector<float> intermediate_min_FDchi2)
  {
    for (unsigned int i = 0; i < intermediate_min_FDchi2.size(); ++i) m_intermediate_min_fdchi2.push_back(intermediate_min_FDchi2[i]);
  }

  void useMVA(bool require_mva) { m_require_mva = require_mva; }

  void setNumMVAPars(unsigned int nPars) { m_nPars = nPars; }

  void setMVAVarList(std::vector<std::string> mva_variable_list)
  {
    for (unsigned int i = 0; i < mva_variable_list.size(); ++i) m_mva_variable_list.push_back(mva_variable_list[i]);
  }

  void setMVAType(const std::string& mva_type) { m_mva_type = mva_type; }

  void setMVAWeightsPath(const std::string& mva_weights_path) { m_mva_path = mva_weights_path; }

  void setMVACutValue(float cut_value) { m_mva_cut_value = cut_value; }

  void saveDST(bool save) { m_save_dst = save; }

  void saveTrackContainer(bool save) { m_write_track_container = save; }

  void saveParticleContainer(bool save) { m_write_particle_container = save; }

  void setContainerName(const std::string& name) { m_container_name = name; }

  void saveOutput(bool save) { m_save_output = save; }

  void setOutputName(const std::string& name) { m_outfile_name = name; }

  void doTruthMatching(bool truth) { m_truth_matching = truth; }

  void getDetectorInfo(bool detinfo) { m_detector_info = detinfo; }

  void getCaloInfo(bool caloinfo) { m_calo_info = caloinfo; }

  void getAllPVInfo(bool pvinfo) { m_get_all_PVs = pvinfo; }

  void setVertexMapNodeName(const std::string& vtx_map_node_name) { m_vtx_map_node_name = m_vtx_map_node_name_nTuple = vtx_map_node_name; }

  void setTrackMapNodeName(const std::string& trk_map_node_name) { m_trk_map_node_name = m_trk_map_node_name_nTuple = trk_map_node_name; }

 private:
  bool m_has_intermediates_sPHENIX;
  bool m_constrain_to_vertex_sPHENIX;
  bool m_require_mva;
  bool m_save_dst;
  bool m_save_output;
  std::string m_outfile_name;
  TFile *m_outfile;
  std::string m_decayDescriptor;
};

#endif  //KFPARTICLESPHENIX_KFPARTICLESPHENIX_H