Skip to content
Snippets Groups Projects

Analysis tree kfpf interface

Closed Oleksii Lubynets requested to merge o.lubynets/cbmroot:AnalysisTreeKfpfInterface into master
12 files
+ 472
16
Compare changes
  • Side-by-side
  • Inline

Files

@@ -98,8 +98,9 @@ void CbmStsTracksConverter::ReadVertexTracks() {
auto* track = vtx_tracks_->AddChannel();
track->Init(out_config_->GetBranchConfig(vtx_tracks_->GetId()));
int pdg = GetMcPid(
sts_track, (CbmTrackMatchNew*) cbm_sts_match_->At(i_track), track);
int pdg = GetMcPid((CbmTrackMatchNew*) cbm_sts_match_->At(i_track), track);
bool is_good_track = IsGoodCovMatrix(sts_track);
float chi2_vertex = ExtrapolateToVertex(sts_track, track, pdg);
const FairTrackParam* trackParamFirst = sts_track->GetParamFirst();
@@ -123,12 +124,13 @@ void CbmStsTracksConverter::ReadVertexTracks() {
out_indexes_map_.insert(std::make_pair(i_track, track->GetId()));
if (is_write_kfinfo_) { WriteKFInfo(track, sts_track); }
if (is_write_kfinfo_) { WriteKFInfo(track, sts_track, is_good_track); }
}
}
void CbmStsTracksConverter::WriteKFInfo(AnalysisTree::Track* track,
const CbmStsTrack* sts_track) const {
const CbmStsTrack* sts_track,
bool is_good_track) const {
assert(track && sts_track);
const FairTrackParam* trackParamFirst = sts_track->GetParamFirst();
@@ -145,7 +147,7 @@ void CbmStsTracksConverter::WriteKFInfo(AnalysisTree::Track* track,
icov_ + iCov);
}
}
track->SetField(IsGoodCovMatrix(sts_track), ipasscuts_);
track->SetField(is_good_track, ipasscuts_);
}
bool CbmStsTracksConverter::IsGoodCovMatrix(
@@ -185,8 +187,7 @@ bool CbmStsTracksConverter::IsGoodCovMatrix(
return ok;
}
int CbmStsTracksConverter::GetMcPid(const CbmStsTrack* /*sts_track*/,
const CbmTrackMatchNew* match,
int CbmStsTracksConverter::GetMcPid(const CbmTrackMatchNew* match,
AnalysisTree::Track* track) const {
if (!is_write_kfinfo_) { return -2; }
@@ -319,7 +320,7 @@ void CbmStsTracksConverter::MapTracks() {
}
auto sim_tracks_map = it->second;
assert(!out_indexes_map_.empty() || !sim_tracks_map.empty());
if (out_indexes_map_.empty() || sim_tracks_map.empty()) return;
CbmTrackMatchNew* match {nullptr};
for (const auto& track_id : out_indexes_map_) {
Loading