Skip to content

Commit 337dd6a

Browse files
committed
* make swiches for SV Tree and Track Tree.
1 parent b8f9ea3 commit 337dd6a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

PWGJE/Tasks/bjetTreeCreator.cxx

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -245,6 +245,7 @@ struct BJetTreeCreator {
245245

246246
Configurable<std::vector<double>> jetRadii{"jetRadii", std::vector<double>{0.4}, "jet resolution parameters"};
247247

248+
Configurable<bool> produceSVTree{"produceSVTree", true, "produce the SV-correlated jet TTree”"};
248249
Configurable<bool> produceTree{"produceTree", true, "produce the jet TTree"};
249250

250251
Configurable<float> vtxRes{"vtxRes", 0.01, "Vertex position resolution (cluster size) for GNN vertex predictions (cm)"};
@@ -415,7 +416,7 @@ struct BJetTreeCreator {
415416
double energySV = candSV.e();
416417

417418
if (svIndices.size() < (svReductionFactor * myJet.template tracks_as<AnyTracks>().size()) && svIndices.size() < maxConstSV) {
418-
if (produceTree) {
419+
if (produceSVTree) {
419420
bjetSVParamsTable(bjetParamsTable.lastIndex() + 1, candSV.pt(), deltaRJetSV, massSV, energySV / myJet.energy(), candSV.impactParameterXY(), candSV.cpa(), candSV.chi2PCA(), candSV.dispersion(), candSV.decayLengthXY(), candSV.errorDecayLengthXY(), candSV.decayLength(), candSV.errorDecayLength());
420421
}
421422
svIndices.push_back(bjetSVParamsTable.lastIndex());
@@ -755,6 +756,7 @@ struct BJetTreeCreator {
755756
}
756757

757758
std::vector<int> indicesTracks;
759+
std::vector<int> indicesSVs;
758760

759761
int16_t jetFlavor = analysisJet.origin();
760762

@@ -791,7 +793,6 @@ struct BJetTreeCreator {
791793
}
792794

793795
if (produceTree) {
794-
std::vector<int> indicesSVs;
795796
bjetConstituentsTable(bjetParamsTable.lastIndex() + 1, indicesTracks, indicesSVs);
796797
bjetParamsTable(analysisJet.pt(), analysisJet.eta(), analysisJet.phi(), indicesTracks.size(), nVertices, analysisJet.mass(), jetFlavor, analysisJet.r());
797798
}

0 commit comments

Comments
 (0)