From a3d594b5083a6952aabb48698f473ef073d4fd8b Mon Sep 17 00:00:00 2001
From: maxiaotian <maxiaotian@beslogin012.ihep.ac.cn>
Date: Thu, 26 Dec 2024 20:20:50 +0800
Subject: [PATCH 1/2] debug TofRecAlg.cpp and add AnalysisPIDAlg
---
Analysis/AnalysisPID/AnalysisPID.py | 41 +++
Analysis/AnalysisPID/CMakeLists.txt | 27 ++
Analysis/AnalysisPID/src/AnalysisPIDAlg.cpp | 339 ++++++++++++++++++++
Analysis/AnalysisPID/src/AnalysisPIDAlg.h | 116 +++++++
Reconstruction/ParticleID/src/TofRecAlg.cpp | 7 +-
5 files changed, 526 insertions(+), 4 deletions(-)
create mode 100644 Analysis/AnalysisPID/AnalysisPID.py
create mode 100644 Analysis/AnalysisPID/CMakeLists.txt
create mode 100644 Analysis/AnalysisPID/src/AnalysisPIDAlg.cpp
create mode 100644 Analysis/AnalysisPID/src/AnalysisPIDAlg.h
diff --git a/Analysis/AnalysisPID/AnalysisPID.py b/Analysis/AnalysisPID/AnalysisPID.py
new file mode 100644
index 00000000..b832435b
--- /dev/null
+++ b/Analysis/AnalysisPID/AnalysisPID.py
@@ -0,0 +1,41 @@
+import os, sys
+from Gaudi.Configuration import *
+
+########### k4DataSvc ####################
+from Configurables import k4DataSvc
+podioevent = k4DataSvc("EventDataSvc", input="track.root")
+##########################################
+
+########## CEPCSWData #################
+cepcswdatatop ="/cvmfs/cepcsw.ihep.ac.cn/prototype/releases/data/latest"
+#######################################
+
+
+########## Podio Input ###################
+from Configurables import PodioInput
+inp = PodioInput("InputReader")
+inp.collections = [ "CompleteTracks",
+ "CompleteTracksParticleAssociation",
+ "RecTofCollection",
+ "DndxTracks" ]
+##########################################
+
+from Configurables import AnalysisPIDAlg
+anaPID = AnalysisPIDAlg("AnalysisPIDAlg")
+anaPID.OutputFile = "./pid.root"
+
+##############################################################################
+# POD I/O
+##############################################################################
+
+
+########################################
+
+from Configurables import ApplicationMgr
+ApplicationMgr(
+ TopAlg=[inp, anaPID ],
+ EvtSel="NONE",
+ EvtMax=-1,
+ ExtSvc=[podioevent],
+ #OutputLevel=DEBUG
+)
diff --git a/Analysis/AnalysisPID/CMakeLists.txt b/Analysis/AnalysisPID/CMakeLists.txt
new file mode 100644
index 00000000..626b122f
--- /dev/null
+++ b/Analysis/AnalysisPID/CMakeLists.txt
@@ -0,0 +1,27 @@
+# gaudi_add_header_only_library(AnalysisPIDLib)
+
+# Modules
+gaudi_add_module(AnalysisPID
+ SOURCES src/AnalysisPIDAlg.cpp
+ LINK Gaudi::GaudiAlgLib
+ Gaudi::GaudiKernel
+ DataHelperLib
+ DetSegmentation
+ DetInterface
+ ${GSL_LIBRARIES}
+ ${GEAR_LIBRARIES}
+ ${LCIO_LIBRARIES}
+ EDM4CEPC::edm4cepc EDM4CEPC::edm4cepcDict
+ EDM4HEP::edm4hep EDM4HEP::edm4hepDict
+ k4FWCore::k4FWCore
+)
+
+target_include_directories(AnalysisPID PUBLIC
+ $<BUILD_INTERFACE:${CMAKE_CURRENT_LIST_DIR}>/include
+ $<INSTALL_INTERFACE:${CMAKE_INSTALL_INCLUDEDIR}>)
+
+install(TARGETS AnalysisPID
+ EXPORT CEPCSWTargets
+ RUNTIME DESTINATION "${CMAKE_INSTALL_BINDIR}" COMPONENT bin
+ LIBRARY DESTINATION "${CMAKE_INSTALL_LIBDIR}" COMPONENT shlib
+ COMPONENT dev)
diff --git a/Analysis/AnalysisPID/src/AnalysisPIDAlg.cpp b/Analysis/AnalysisPID/src/AnalysisPIDAlg.cpp
new file mode 100644
index 00000000..ed9ea3e8
--- /dev/null
+++ b/Analysis/AnalysisPID/src/AnalysisPIDAlg.cpp
@@ -0,0 +1,339 @@
+#include "AnalysisPIDAlg.h"
+
+#include "GaudiKernel/DataObject.h"
+#include "GaudiKernel/IHistogramSvc.h"
+#include "GaudiKernel/MsgStream.h"
+#include "GaudiKernel/SmartDataPtr.h"
+
+#include "DetInterface/IGeomSvc.h"
+#include "DataHelper/HelixClass.h"
+
+#include "DD4hep/Detector.h"
+#include "DD4hep/DD4hepUnits.h"
+#include "CLHEP/Units/SystemOfUnits.h"
+#include <cmath>
+#include "UTIL/ILDConf.h"
+#include "DetIdentifier/CEPCConf.h"
+#include "UTIL/CellIDEncoder.h"
+
+using namespace edm4hep;
+
+DECLARE_COMPONENT( AnalysisPIDAlg )
+
+//------------------------------------------------------------------------------
+AnalysisPIDAlg::AnalysisPIDAlg( const std::string& name, ISvcLocator* pSvcLocator )
+ : Algorithm( name, pSvcLocator ) {
+ declareProperty("CompleteTracks", _FultrkCol, "handler of the input complete track collection");
+ declareProperty("CompleteTracksParticleAssociation", _FultrkParAssCol, "handler of the input track particle association collection");
+ declareProperty("DndxTracks", _inDndxColHdl, "handler of the collection of dN/dx tracks");
+ declareProperty("RecTofCollection", _inTofColHdl, "handler of the collection of tof tracks");
+ // output
+ declareProperty("OutputFile", m_outputFile = "pid.root", "output file name");
+}
+
+//------------------------------------------------------------------------------
+StatusCode AnalysisPIDAlg::initialize(){
+
+ info() << "Booking Ntuple" << endmsg;
+ m_file = new TFile(m_outputFile.value().c_str(),"RECREATE");
+ m_tree = new TTree("pid","pid");
+
+ m_tree->Branch("Nevt",&_nEvt,"Nevt/I");
+ m_tree->Branch("Ndndxtrk",&Ndndxtrk,"Ndndxtrk/I");
+ m_tree->Branch("Ntoftrk",&Ntoftrk,"Ntoftrk/I");
+ m_tree->Branch("Nfullass",&Nfullass,"Nfullass/I");
+ m_tree->Branch("Nfulltrk",&Nfulltrk,"Nfulltrk/I");
+ m_tree->Branch("tpcidx",&tpcidx);
+ m_tree->Branch("tofidx",&tofidx);
+ m_tree->Branch("matchedtpc",&matchedtpc);
+ m_tree->Branch("matchedtof",&matchedtof);
+ m_tree->Branch("truthidx",&truthidx);
+
+ m_tree->Branch("tof_chi2s",&tof_chi2s);
+ m_tree->Branch("tof_chis",&tof_chis);
+ m_tree->Branch("tof_expt",&tof_expt);
+ m_tree->Branch("tof_meast",&tof_meast);
+ m_tree->Branch("tof_measterr",&tof_measterr);
+
+ m_tree->Branch("tpc_chi2s",&tpc_chi2s);
+ m_tree->Branch("tpc_chis",&tpc_chis);
+ m_tree->Branch("tpc_expdndxs",&tpc_expdndxs);
+ m_tree->Branch("tpc_measdndx",&tpc_measdndx);
+ m_tree->Branch("tpc_measdndxerr",&tpc_measdndxerr);
+
+ m_tree->Branch("tot_chi2s",&tot_chi2s);
+ m_tree->Branch("recoPDG",&recoPDG);
+ m_tree->Branch("tpcrecoPDG",&tpcrecoPDG);
+ m_tree->Branch("tofrecoPDG",&tofrecoPDG);
+
+ //gen trk parameters
+ m_tree->Branch("genpx",&genpx);
+ m_tree->Branch("genpy",&genpy);
+ m_tree->Branch("genpz",&genpz);
+ m_tree->Branch("genE",&genE);
+ m_tree->Branch("genp",&genp);
+ m_tree->Branch("genM",&genM);
+ m_tree->Branch("genphi",&genphi);
+ m_tree->Branch("gentheta",&gentheta);
+ m_tree->Branch("endx",&endx);
+ m_tree->Branch("endy",&endy);
+ m_tree->Branch("endz",&endz);
+ m_tree->Branch("endr",&endr);
+ m_tree->Branch("PDG",&PDG);
+ m_tree->Branch("genstatus",&genstatus);
+ m_tree->Branch("simstatus",&simstatus);
+ m_tree->Branch("isdecayintrker",&isdecayintrker);
+ m_tree->Branch("iscreatedinsim",&iscreatedinsim);
+ m_tree->Branch("isbackscatter",&isbackscatter);
+ m_tree->Branch("isstopped",&isstopped);
+
+ _nEvt = 0;
+ return StatusCode::SUCCESS;
+}
+
+//------------------------------------------------------------------------------
+StatusCode AnalysisPIDAlg::execute(){
+
+ const edm4hep::TrackCollection* trkCol = nullptr;
+ const edm4hep::RecDqdxCollection* dndxCols = nullptr;
+ const edm4hep::RecTofCollection* tofCols = nullptr;
+ const edm4hep::MCRecoTrackParticleAssociationCollection* fultrkparassCols = nullptr;
+
+ ClearVars();
+
+ try {
+ trkCol = _FultrkCol.get();
+ }
+ catch ( GaudiException &e ) {
+ debug() << "Complete track collection " << _FultrkCol.fullKey() << " is unavailable in event " << _nEvt << endmsg;
+ Nfulltrk = -1;
+ }
+ if ( trkCol->size() == 0 ) {
+ debug() << "No full track found in event " << _nEvt << endmsg;
+ Nfulltrk = 0;
+ }
+ else{
+ Nfulltrk = trkCol->size();
+ }
+
+ try {
+ fultrkparassCols = _FultrkParAssCol.get();
+ }
+ catch ( GaudiException &e ) {
+ debug() << "Complete track particle association collection " << _FultrkParAssCol.fullKey() << " is unavailable in event " << _nEvt << endmsg;
+ Nfullass = -1;
+ m_tree->Fill();
+ _nEvt++;
+ return StatusCode::SUCCESS;
+ }
+
+ try {
+ dndxCols = _inDndxColHdl.get();
+ }
+ catch ( GaudiException &e ) {
+ debug() << "DndxTrack collection " << _inDndxColHdl.fullKey() << " is unavailable in event " << _nEvt << endmsg;
+ Ndndxtrk = -1;
+ }
+
+ if ( dndxCols->size() == 0 ) {
+ debug() << "No dndx track found in event " << _nEvt << endmsg;
+ Ndndxtrk = 0;
+ }
+ else{
+ Ndndxtrk = dndxCols->size();
+ }
+
+ try {
+ tofCols = _inTofColHdl.get();
+ }
+ catch ( GaudiException &e ) {
+ debug() << "TofTrack collection " << _inTofColHdl.fullKey() << " is unavailable in event " << _nEvt << endmsg;
+ Ntoftrk = -1;
+ }
+
+ if ( tofCols->size() == 0 ) {
+ debug() << "No tof track found in event " << _nEvt << endmsg;
+ Ntoftrk = 0;
+ }
+ else{
+ Ntoftrk = tofCols->size();
+ }
+
+ if ( fultrkparassCols->size() == 0 ) {
+ debug() << "No full track particle association found in event " << _nEvt << endmsg;
+ Nfullass = 0;
+ m_tree->Fill();
+ _nEvt++;
+ return StatusCode::SUCCESS;
+ }
+ else{
+ Nfullass = fultrkparassCols->size();
+ }
+
+ info() << "normal eventID: " << _nEvt << endmsg;
+ for(auto track : *trkCol){
+ //truth association match
+ max_weight = -999.;
+ max_weight_idx = -1;
+ ass_idx = 0;
+ for (auto ass : *fultrkparassCols) {
+ if (ass.getRec() == track) {
+ weight = ass.getWeight();
+ if (weight > max_weight) {
+ max_weight = weight;
+ max_weight_idx = ass_idx;
+ }
+ }
+ ass_idx++;
+ }
+ truthidx.push_back(max_weight_idx);
+// if (max_weight_idx < 0) continue;
+ p1=fultrkparassCols->at(max_weight_idx).getSim().getMomentum()[0];
+ p2=fultrkparassCols->at(max_weight_idx).getSim().getMomentum()[1];
+ p3=fultrkparassCols->at(max_weight_idx).getSim().getMomentum()[2];
+ genpx.push_back(p1);
+ genpy.push_back(p2);
+ genpz.push_back(p3);
+ genp.push_back(std::sqrt(p1*p1 + p2*p2 + p3*p3));
+ genE.push_back(fultrkparassCols->at(max_weight_idx).getSim().getEnergy());
+ genM.push_back(fultrkparassCols->at(max_weight_idx).getSim().getMass());
+ genphi.push_back(std::atan2(p2,p1));
+ gentheta.push_back(std::acos(p3/std::sqrt(p1*p1 + p2*p2 + p3*p3)));
+ x1=fultrkparassCols->at(max_weight_idx).getSim().getEndpoint()[0];
+ y1=fultrkparassCols->at(max_weight_idx).getSim().getEndpoint()[1];
+ endx.push_back(x1);
+ endy.push_back(y1);
+ endz.push_back(fultrkparassCols->at(max_weight_idx).getSim().getEndpoint()[2]);
+ endr.push_back(std::sqrt(x1*x1 + y1*y1));
+ PDG.push_back(fultrkparassCols->at(max_weight_idx).getSim().getPDG());
+ genstatus.push_back(fultrkparassCols->at(max_weight_idx).getSim().getGeneratorStatus());
+ simstatus.push_back(fultrkparassCols->at(max_weight_idx).getSim().getSimulatorStatus());
+ isdecayintrker.push_back(fultrkparassCols->at(max_weight_idx).getSim().isDecayedInTracker());//getSimulatorStatus().isDecayInTracker();
+ iscreatedinsim.push_back(fultrkparassCols->at(max_weight_idx).getSim().isCreatedInSimulation());
+ isbackscatter.push_back(fultrkparassCols->at(max_weight_idx).getSim().isBackscatter());
+ isstopped.push_back(fultrkparassCols->at(max_weight_idx).getSim().isStopped());
+
+ //find corresponding dndx track
+ edm4hep::RecDqdx dndxtrk;
+ dndx_index = -1;
+ matched1 = false;
+ for(int i=0; i<Ndndxtrk; i++){
+ if ( dndxCols->at(i).getTrack() == track ){
+ dndxtrk = dndxCols->at(i);
+ dndx_index = i;
+ matched1 = true;
+ break;
+ }
+ }
+ tpcidx.push_back(dndx_index);
+ tpc_chi2s_1.clear();tpc_expdndxs_1.clear();tpc_chis_1.clear();
+ tpcdndx=-1;tpcdndxerr=-1;
+ if( matched1 ) {
+ tpcdndx = dndxtrk.getDQdx().value;
+ tpcdndxerr = dndxtrk.getDQdx().error;
+ debug()<<"tpc_measdndx = "<<dndxtrk.getDQdx().value<<endmsg;
+ for (int idx=0;idx<5;idx++) {
+ double tpc_chi2 = dndxtrk.getHypotheses(idx).chi2;
+ double tpc_expdndx = dndxtrk.getHypotheses(idx).expected;
+ double tpc_chi = ( tpcdndx - tpc_expdndx ) / tpcdndxerr;
+ tpc_chi2s_1.push_back(tpc_chi2);
+ tpc_chis_1.push_back(tpc_chi);
+ tpc_expdndxs_1.push_back(tpc_expdndx);
+ debug()<<" idx : "<< idx <<" tpc_chi2 : "<< tpc_chi2 <<endmsg;
+ }
+ }
+ else{
+ tpc_chi2s_1={-999,-999,-999,-999,-999};
+ tpc_chis_1={-999,-999,-999,-999,-999};
+ tpc_expdndxs_1={-1,-1,-1,-1,-1};
+ }
+ tpc_measdndx.push_back(tpcdndx);
+ tpc_measdndxerr.push_back(tpcdndxerr);
+ matchedtpc.push_back(matched1);
+ tpc_chi2s.push_back(tpc_chi2s_1);
+ tpc_chis.push_back(tpc_chis_1);
+ tpc_expdndxs.push_back(tpc_expdndxs_1);
+
+ //find corresponding tof track
+ edm4hep::RecTof toftrk;
+ tof_index = -1;
+ matched2 = false;
+ for(int i=0; i<Ntoftrk; i++){
+ if ( tofCols->at(i).getTrack() == track ){
+ toftrk = tofCols->at(i);
+ tof_index = i;
+ matched2 = true;
+ break;
+ }
+ }
+ tofidx.push_back(tof_index);
+ tof_chi2s_1.clear();tof_expt_1.clear();tof_chis_1.clear();
+ toft=-1;tofterr=-1;
+ if ( matched2 ) {
+ toft = toftrk.getTime();
+ std::array<float, 5> tofexpts = toftrk.getTimeExp();
+ tofterr = toftrk.getSigma();
+ debug()<<"tof_meast = "<<toftrk.getTime()<<endmsg;
+ for (int idx=0;idx<5;idx++){
+ double tof_chi = ( toft - tofexpts[idx] ) / tofterr;
+ double tof_chi2 = tof_chi*tof_chi;
+ tof_chi2s_1.push_back(tof_chi2);
+ tof_chis_1.push_back(tof_chi);
+ tof_expt_1.push_back(tofexpts[idx]);
+ debug()<<" idx : "<< idx <<" tof_chi2 : "<< tof_chi2 <<endmsg;
+ }//end loop over masses
+ }
+ else{
+ tof_chi2s_1={-999,-999,-999,-999,-999};
+ tof_chis_1={-999,-999,-999,-999,-999};
+ tof_expt_1={-1,-1,-1,-1,-1};
+ }
+ tof_meast.push_back(toft);
+ tof_measterr.push_back(tofterr);
+ matchedtof.push_back(matched2);
+ tof_chi2s.push_back(tof_chi2s_1);
+ tof_chis.push_back(tof_chis_1);
+ tof_expt.push_back(tof_expt_1);
+
+ tot_chi2s_1.clear();
+ recpdg = 9999;tpcrecpdg = 9999;tofrecpdg = 9999;
+ if(matched1){
+ minchi2idx = std::distance(tpc_chi2s_1.begin(), std::min_element(tpc_chi2s_1.begin(), tpc_chi2s_1.end()));
+ tpcrecpdg = PDGIDs.at(minchi2idx);
+ }
+ if(matched2){
+ minchi2idx = std::distance(tof_chi2s_1.begin(), std::min_element(tof_chi2s_1.begin(), tof_chi2s_1.end()));
+ tofrecpdg = PDGIDs.at(minchi2idx);
+ }
+ if(matched1 || matched2){
+ if(matched1 && matched2){
+ for(int i=0;i<5;i++){
+ tot_chi2s_1.push_back(tpc_chi2s_1[i]+tof_chi2s_1[i]);
+ }
+ }
+ if(matched1 && !matched2) tot_chi2s_1=tpc_chi2s_1;
+ if(!matched1 && matched2) tot_chi2s_1=tof_chi2s_1;
+ minchi2idx = std::distance(tot_chi2s_1.begin(), std::min_element(tot_chi2s_1.begin(), tot_chi2s_1.end()));
+ recpdg = PDGIDs.at(minchi2idx);
+ }
+ else{
+ tot_chi2s_1={-999,-999,-999,-999,-999};
+ }
+ int charge = track.getTrackStates(1).omega/fabs(track.getTrackStates(1).omega);
+ tot_chi2s.push_back(tot_chi2s_1);
+ recoPDG.push_back(charge*recpdg);
+ tpcrecoPDG.push_back(charge*tpcrecpdg);
+ tofrecoPDG.push_back(charge*tofrecpdg);
+ }
+m_tree->Fill();
+_nEvt++;
+return StatusCode::SUCCESS;
+}// end execute
+
+//------------------------------------------------------------------------------
+StatusCode AnalysisPIDAlg::finalize(){
+ debug() << "Finalizing..." << endmsg;
+ m_file->cd();
+ m_tree->Write();
+ return StatusCode::SUCCESS;
+}
diff --git a/Analysis/AnalysisPID/src/AnalysisPIDAlg.h b/Analysis/AnalysisPID/src/AnalysisPIDAlg.h
new file mode 100644
index 00000000..2c665feb
--- /dev/null
+++ b/Analysis/AnalysisPID/src/AnalysisPIDAlg.h
@@ -0,0 +1,116 @@
+#ifndef AnalysisPIDAlg_h
+#define AnalysisPIDAlg_h 1
+
+#include "k4FWCore/DataHandle.h"
+#include "GaudiKernel/Algorithm.h"
+
+#include "edm4hep/MCParticleCollection.h"
+#include "edm4hep/SimTrackerHitCollection.h"
+
+#include "edm4hep/TrackerHit.h"
+#include "edm4hep/TrackCollection.h"
+#include "edm4hep/TrackerHitCollection.h"
+#include "edm4hep/MCRecoTrackerAssociationCollection.h"
+#include "edm4hep/MCRecoTrackParticleAssociationCollection.h"
+#include "edm4hep/RecDqdx.h"
+#include "edm4hep/RecDqdxCollection.h"
+#include "edm4cepc/RecTofCollection.h"
+#include "edm4hep/TrackState.h"
+#include "edm4hep/Vector3d.h"
+
+#include "TFile.h"
+#include "TTree.h"
+#include <random>
+#include "GaudiKernel/NTuple.h"
+
+class AnalysisPIDAlg : public Algorithm {
+ public:
+ // Constructor of this form must be provided
+ AnalysisPIDAlg( const std::string& name, ISvcLocator* pSvcLocator );
+
+ // Three mandatory member functions of any algorithm
+ StatusCode initialize() override;
+ StatusCode execute() override;
+ StatusCode finalize() override;
+
+ private:
+ DataHandle<edm4hep::TrackCollection> _FultrkCol{"CompleteTracks", Gaudi::DataHandle::Reader, this};
+ DataHandle<edm4hep::MCRecoTrackParticleAssociationCollection> _FultrkParAssCol{"CompleteTracksParticleAssociation", Gaudi::DataHandle::Reader, this};
+ DataHandle<edm4hep::RecDqdxCollection> _inDndxColHdl{"DndxTracks", Gaudi::DataHandle::Reader, this};
+ DataHandle<edm4hep::RecTofCollection> _inTofColHdl{"RecTofCollection", Gaudi::DataHandle::Reader, this};
+
+ Gaudi::Property<std::string> m_outputFile{this, "OutputFile", "pid.root"};
+
+ std::vector<double> genpx, genpy, genpz, genE, genp, genM, gentheta, genphi, endx, endy, endz, endr;
+ std::vector<int> PDG, genstatus, simstatus, recoPDG, tpcrecoPDG, tofrecoPDG;
+ std::vector<bool> isdecayintrker, iscreatedinsim, isbackscatter, isstopped;
+ std::vector<bool> matchedtpc, matchedtof;
+ std::vector<int> truthidx, tpcidx, tofidx;
+
+ std::vector<std::vector<double>> tof_chi2s, tof_expt;
+ std::vector<std::vector<double>> tpc_chi2s, tpc_expdndxs;
+ std::vector<std::vector<double>> tof_chis, tpc_chis, tot_chi2s;
+ std::vector<double> tof_meast, tof_measterr;
+ std::vector<double> tpc_measdndx, tpc_measdndxerr;
+ std::vector<double> tof_chi2s_1, tof_expt_1;
+ std::vector<double> tpc_chi2s_1, tpc_expdndxs_1;
+ std::vector<double> tof_chis_1, tpc_chis_1, tot_chi2s_1;
+
+ int _nEvt;
+ const std::map<int, int> PDGIDs = {
+ {0, -11},
+ {1, -13},
+ {2, 211},
+ {3, 321},
+ {4, 2212},
+ };
+
+ double tpcdndx, tpcdndxerr, toft, tofterr;
+ int Ndndxtrk, Ntoftrk, Nfulltrk, Nfullass;
+ double max_weight, weight;
+ int max_weight_idx, ass_idx, dndx_index, tof_index;
+ double p1, p2, p3, x1, y1;
+ bool matched1, matched2;
+ int recpdg, tpcrecpdg, tofrecpdg, minchi2idx;
+
+ TFile* m_file;
+ TTree* m_tree;
+
+ void ClearVars(){
+
+ Ndndxtrk = 0;
+ Ntoftrk = 0;
+ Nfullass = 0;
+ Nfulltrk = 0;
+
+ genpx.clear(); genpy.clear(); genpz.clear(); genE.clear();
+ genp.clear(); genM.clear(); gentheta.clear(); genphi.clear();
+ endx.clear(); endy.clear(); endz.clear(); endr.clear();
+ PDG.clear(); genstatus.clear(); simstatus.clear();
+ recoPDG.clear(); tpcrecoPDG.clear(); tofrecoPDG.clear();
+ isdecayintrker.clear(); iscreatedinsim.clear();
+ isbackscatter.clear(); isstopped.clear();
+
+ tof_chi2s.clear();
+ tof_chis.clear();
+ tof_meast.clear();
+ tof_measterr.clear();
+ tof_expt.clear();
+
+ tpc_chi2s.clear();
+ tpc_chis.clear();
+ tpc_expdndxs.clear();
+ tpc_measdndx.clear();
+ tpc_measdndxerr.clear();
+
+ tot_chi2s.clear();
+
+ matchedtpc.clear();
+ matchedtof.clear();
+ truthidx.clear();
+ tpcidx.clear();
+ tofidx.clear();
+ }
+
+};
+#endif
diff --git a/Reconstruction/ParticleID/src/TofRecAlg.cpp b/Reconstruction/ParticleID/src/TofRecAlg.cpp
index f7509088..de202d0f 100644
--- a/Reconstruction/ParticleID/src/TofRecAlg.cpp
+++ b/Reconstruction/ParticleID/src/TofRecAlg.cpp
@@ -57,10 +57,9 @@ StatusCode TofRecAlg::execute(){
return StatusCode::SUCCESS;
}
- hasFTDHit = false;
- hasSETHit = false;
-
for(auto track : *fultrkcol){
+ hasFTDHit = false;
+ hasSETHit = false;
FindToFHits( track, hasFTDHit, hasSETHit, Toft, Tofx, Tofy, Tofz );
@@ -157,4 +156,4 @@ void TofRecAlg::FindToFHits(const edm4hep::Track& _track, bool& _hasFTDHit, bool
}//find ftd hit
}//find ftd hit
}//end track hits loop
-}
\ No newline at end of file
+}
--
GitLab
From 52370c93974c98c191580622bc39a6429c03e85a Mon Sep 17 00:00:00 2001
From: maxiaotian <maxiaotian@beslogin012.ihep.ac.cn>
Date: Thu, 26 Dec 2024 20:25:27 +0800
Subject: [PATCH 2/2] add Analysis/CMakeLists.txt
---
Analysis/CMakeLists.txt | 1 +
1 file changed, 1 insertion(+)
diff --git a/Analysis/CMakeLists.txt b/Analysis/CMakeLists.txt
index 6fc2f919..86707404 100644
--- a/Analysis/CMakeLists.txt
+++ b/Analysis/CMakeLists.txt
@@ -5,3 +5,4 @@ add_subdirectory(DumpEvent)
add_subdirectory(ReadDigi)
add_subdirectory(JetClustering)
add_subdirectory(GenMatch)
+add_subdirectory(AnalysisPID)
--
GitLab