Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add decisions to NTUPLE maker #26

Merged
merged 1 commit into from
May 20, 2024
Merged

Add decisions to NTUPLE maker #26

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions reconstruction/io/RootStreamBuilder/python/RootStreamNTUPLEMaker.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ def __init__( self, name,
InputCellsKey : str,
InputClusterKey : str,
InputRingerKey : str,
InputElectronKey : str,
InputFile : str,
OutputLevel : int=0,
NtupleName : str="CollectionTree",
Expand All @@ -30,6 +31,7 @@ def __init__( self, name,
self.setProperty( "InputCellsKey" , InputCellsKey )
self.setProperty( "InputClusterKey" , InputClusterKey )
self.setProperty( "InputRingerKey" , InputRingerKey )
self.setProperty( "InputElectronKey", InputElectronKey)
self.setProperty( "InputFile" , InputFile )
self.setProperty( "SecondLambdaCuts", SecondLambdaCuts )
self.setProperty( "LateralMomCuts" , LateralMomCuts)
Expand Down
173 changes: 110 additions & 63 deletions reconstruction/io/RootStreamBuilder/src/RootStreamNTUPLEMaker.cxx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,14 @@
#include "TruthParticle/TruthParticleContainer.h"
#include "CaloCluster/CaloClusterContainer.h"
#include "CaloRings/CaloRingsContainer.h"
#include "Particle/ElectronContainer.h"
#include "CaloCell/CaloCellConverter.h"
#include "CaloCell/CaloDetDescriptorConverter.h"
#include "EventInfo/EventInfoConverter.h"
#include "TruthParticle/TruthParticleConverter.h"
#include "CaloCluster/CaloClusterConverter.h"
#include "CaloRings/CaloRingsConverter.h"
#include "Particle/ElectronConverter.h"
#include "RootStreamNTUPLEMaker.h"
#include "GaugiKernel/EDM.h"

Expand All @@ -28,6 +30,7 @@ RootStreamNTUPLEMaker::RootStreamNTUPLEMaker( std::string name ) :
declareProperty( "InputCellsKey" , m_cellsKey="Cells" );
declareProperty( "InputClusterKey" , m_clusterKey="Clusters" );
declareProperty( "InputRingerKey" , m_ringerKey="Rings" );
declareProperty( "InputElectronKey" , m_electronKey="Electrons" );
declareProperty( "OutputLevel" , m_outputLevel=1 );
declareProperty( "NtupleName" , m_ntupleName="physics" );
declareProperty( "OutputNtupleName" , m_outputNtupleName="events" );
Expand Down Expand Up @@ -103,6 +106,10 @@ StatusCode RootStreamNTUPLEMaker::bookHistograms( EventContext &ctx ) const
float f3 = 0;
float weta2 = 0;
std::vector<float>*rings = nullptr;
bool el_tight = false;
bool el_medium = false;
bool el_loose = false;
bool el_vloose = false;
float secondR = 0;
float lambdaCenter = 0;
float secondLambda = 0;
Expand Down Expand Up @@ -146,6 +153,10 @@ StatusCode RootStreamNTUPLEMaker::bookHistograms( EventContext &ctx ) const
outputTree->Branch("cluster_f3" , &f3);
outputTree->Branch("cluster_weta2" , &weta2);
outputTree->Branch("rings" , &rings);
outputTree->Branch("el_tight" , &el_tight);
outputTree->Branch("el_medium" , &el_medium);
outputTree->Branch("el_loose" , &el_loose);
outputTree->Branch("el_vloose" , &el_vloose);
outputTree->Branch("cluster_secondR", &secondR);
outputTree->Branch("cluster_lambdaCenter", &lambdaCenter);
outputTree->Branch("cluster_secondLambda", &secondLambda);
Expand Down Expand Up @@ -205,6 +216,7 @@ StatusCode RootStreamNTUPLEMaker::deserialize( int evt, EventContext &ctx ) cons
std::vector<xAOD::TruthParticle_t > *collection_truth = nullptr;
std::vector<xAOD::CaloCluster_t > *collection_cluster = nullptr;
std::vector<xAOD::CaloRings_t > *collection_rings = nullptr;
std::vector<xAOD::Electron_t > *collection_electrons = nullptr;

MSG_DEBUG( "Link all branches..." );

Expand All @@ -218,6 +230,7 @@ StatusCode RootStreamNTUPLEMaker::deserialize( int evt, EventContext &ctx ) cons
InitBranch( tree, ("CaloDetDescriptorContainer_" + m_cellsKey).c_str() , &collection_descriptor);
InitBranch( tree, ("CaloClusterContainer_" + m_clusterKey).c_str(), &collection_cluster);
InitBranch( tree, ("CaloRingsContainer_" + m_ringerKey).c_str(), &collection_rings);
InitBranch( tree, ("ElectronContainer_" + m_electronKey).c_str(), &collection_electrons);

tree->GetEntry( evt );

Expand Down Expand Up @@ -255,6 +268,10 @@ StatusCode RootStreamNTUPLEMaker::deserialize( int evt, EventContext &ctx ) cons
float f3 = 0;
float weta2 = 0;
std::vector<float> *rings = nullptr;
bool el_tight = false;
bool el_medium = false;
bool el_loose = false;
bool el_vloose = false;
float secondR = 0;
float lambdaCenter = 0;
float secondLambda = 0;
Expand Down Expand Up @@ -295,6 +312,11 @@ StatusCode RootStreamNTUPLEMaker::deserialize( int evt, EventContext &ctx ) cons
InitBranch( outputTree,"cluster_f2" , &f2);
InitBranch( outputTree,"cluster_f3" , &f3);
InitBranch( outputTree,"cluster_weta2" , &weta2);
InitBranch( outputTree,"el_tight" , &el_tight);
InitBranch( outputTree,"el_medium" , &el_medium);
InitBranch( outputTree,"el_loose" , &el_loose);
InitBranch( outputTree,"el_vloose" , &el_vloose);

InitBranch( outputTree,"cluster_secondR" , &secondR);
InitBranch( outputTree,"cluster_lambdaCenter" , &lambdaCenter);
InitBranch( outputTree,"cluster_secondLambda" , &secondLambda);
Expand All @@ -306,79 +328,99 @@ StatusCode RootStreamNTUPLEMaker::deserialize( int evt, EventContext &ctx ) cons
InitBranch( outputTree,"el_fwdLoose", &el_fwdLoose);


{ //main loop (from cluster to truth)
{ //main loop (from electron to rings)

InitBranch(outputTree, "rings", &rings);
SG::WriteHandle<xAOD::CaloRingsContainer> ringerContainer(m_ringerKey, ctx);
ringerContainer.record( std::unique_ptr<xAOD::CaloRingsContainer>(new xAOD::CaloRingsContainer()));

SG::WriteHandle<xAOD::CaloClusterContainer> clusterContainer(m_clusterKey, ctx);
clusterContainer.record( std::unique_ptr<xAOD::CaloClusterContainer>(new xAOD::CaloClusterContainer()));

SG::WriteHandle<xAOD::ElectronContainer> electronContainer(m_electronKey, ctx);
electronContainer.record( std::unique_ptr<xAOD::ElectronContainer>(new xAOD::ElectronContainer()));

int cluster_link = 0;
for (auto &cluster_t: *collection_cluster){
for (auto &ringer_t : *collection_rings ){
if (ringer_t.cluster_link == cluster_link){
MSG_DEBUG("Match between cluster and ringer");
xAOD::CaloRings *ringer = nullptr;
xAOD::CaloRingsConverter cnv;
cnv.convert(ringer_t, ringer); // alloc memory
for (auto ring : ringer->rings()) rings->push_back(ring);
xAOD::CaloCluster *cluster = nullptr;
xAOD::CaloClusterConverter clusterCnv;
clusterCnv.convert(cluster_t, cluster);
eta = cluster->eta();
phi = cluster->phi();
e = cluster->e();
et = cluster->et();
deta = cluster->deltaEta();
dphi = cluster->deltaPhi();
e0 = cluster->e0();
e1 = cluster->e1();
e2 = cluster->e2();
e3 = cluster->e3();
ehad1 = cluster->ehad1();
ehad2 = cluster->ehad2();
ehad3 = cluster->ehad3();
etot = cluster->etot();
e233 = cluster->e233();
e237 = cluster->e237();
e277 = cluster->e277();
emaxs1 = cluster->emaxs1();
emaxs2 = cluster->emaxs2();
e2tsts1 = cluster->e2tsts1();
reta = cluster->reta();
rphi = cluster->rphi();
rhad = cluster->rhad();
rhad1 = cluster->rhad1();
eratio = cluster->eratio();
f0 = cluster->f0();
f1 = cluster->f1();
f2 = cluster->f2();
f3 = cluster->f3();
weta2 = cluster->weta2();

secondR = cluster->secondR();
lambdaCenter = cluster->lambdaCenter();
secondLambda = cluster->secondLambda();
fracMax = cluster->fracMax();
lateralMom = cluster->lateralMom();
longitudinalMom = cluster->longitudinalMom();
el_fwdTight = computeForwardDecision(cluster,"tight");
el_fwdMedium = computeForwardDecision(cluster,"medium");
el_fwdLoose = computeForwardDecision(cluster,"loose");

}
for (auto &electron_t: *collection_electrons){
for (auto &cluster_t: *collection_cluster){
for (auto &ringer_t : *collection_rings ){
if (ringer_t.cluster_link == cluster_link && electron_t.cluster_link == cluster_link){
MSG_DEBUG("Match between cluster and ringer");

xAOD::CaloRings *ringer = nullptr;
xAOD::CaloRingsConverter cnv;
cnv.convert(ringer_t, ringer); // alloc memory
for (auto ring : ringer->rings()) rings->push_back(ring);

xAOD::CaloCluster *cluster = nullptr;
xAOD::CaloClusterConverter clusterCnv;
clusterCnv.convert(cluster_t, cluster);

xAOD::Electron *electron = nullptr;
xAOD::ElectronConverter electronCnv;
electronCnv.convert(electron_t, electron);

eta = cluster->eta();
phi = cluster->phi();
e = cluster->e();
et = cluster->et();
deta = cluster->deltaEta();
dphi = cluster->deltaPhi();
e0 = cluster->e0();
e1 = cluster->e1();
e2 = cluster->e2();
e3 = cluster->e3();
ehad1 = cluster->ehad1();
ehad2 = cluster->ehad2();
ehad3 = cluster->ehad3();
etot = cluster->etot();
e233 = cluster->e233();
e237 = cluster->e237();
e277 = cluster->e277();
emaxs1 = cluster->emaxs1();
emaxs2 = cluster->emaxs2();
e2tsts1 = cluster->e2tsts1();
reta = cluster->reta();
rphi = cluster->rphi();
rhad = cluster->rhad();
rhad1 = cluster->rhad1();
eratio = cluster->eratio();
f0 = cluster->f0();
f1 = cluster->f1();
f2 = cluster->f2();
f3 = cluster->f3();
weta2 = cluster->weta2();

secondR = cluster->secondR();
lambdaCenter = cluster->lambdaCenter();
secondLambda = cluster->secondLambda();
fracMax = cluster->fracMax();
lateralMom = cluster->lateralMom();
longitudinalMom = cluster->longitudinalMom();
el_fwdTight = computeForwardDecision(cluster,"tight");
el_fwdMedium = computeForwardDecision(cluster,"medium");
el_fwdLoose = computeForwardDecision(cluster,"loose");

el_tight = electron->getDecisions().at(0);
el_medium = electron->getDecisions().at(1);
el_loose = electron->getDecisions().at(2);
el_vloose = electron->getDecisions().at(3);


}
break;
}
outputTree->Fill(); //each cluster should be 1 entry on ntulpe file
cluster_link++;
} //end loop of cluster
}

outputTree->Fill(); //each cluster should be 1 entry on ntulpe file
cluster_link++;
}
}

eta = 0;
phi = 0 ;
e = 0 ;
et = 0 ;
eta = 0;
phi = 0;
e = 0;
et = 0;
deta = 0;
dphi = 0;
e0 = 0;
Expand Down Expand Up @@ -411,16 +453,21 @@ StatusCode RootStreamNTUPLEMaker::deserialize( int evt, EventContext &ctx ) cons
fracMax = 0;
lateralMom = 0;
longitudinalMom = 0;
el_fwdTight = false;
el_tight = false;
el_medium = false;
el_loose = false;
el_vloose = false;
el_fwdTight = false;
el_fwdMedium = false;
el_fwdLoose = false;
el_fwdLoose = false;
delete rings ;

delete collection_descriptor;
delete collection_cells ;
delete collection_event ;
delete collection_truth ;
delete collection_cluster ;
delete collection_electrons ;

return StatusCode::SUCCESS;

Expand Down
2 changes: 2 additions & 0 deletions reconstruction/io/RootStreamBuilder/src/RootStreamNTUPLEMaker.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
#include "TruthParticle/TruthParticle.h"
#include "CaloCluster/CaloCluster.h"
#include "CaloRings/CaloRings.h"
#include "Particle/Electron.h"



Expand Down Expand Up @@ -49,6 +50,7 @@ class RootStreamNTUPLEMaker : public Gaugi::Algorithm
std::string m_truthKey;
std::string m_clusterKey;
std::string m_ringerKey;
std::string m_electronKey;
std::string m_inputFile;
std::string m_ntupleName;
std::string m_outputNtupleName;
Expand Down
1 change: 1 addition & 0 deletions scripts/ntuple_trf.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -52,6 +52,7 @@
InputTruthKey = recordable("Particles"),
InputClusterKey = recordable("Clusters"),
InputRingerKey = recordable("Rings"),
InputElectronKey = recordable("Electrons"),
NtupleName = "CollectionTree",
OutputLevel = outputLevel,
OutputNtupleName = "events",
Expand Down