Skip to content
Snippets Groups Projects

Reco QA - primary vertex

Merged Alexandru Bercuci requested to merge a.bercuci/cbmroot:reco_qa into master
Files
9
/* Copyright (C) 2005-2020 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
/* Copyright (C) 2005-2024 GSI Helmholtzzentrum fuer Schwerionenforschung, Darmstadt
SPDX-License-Identifier: GPL-3.0-only
Authors: Volker Friese, Florian Uhlig, Denis Bertini [committer] */
Authors: Volker Friese, Florian Uhlig, Alex Bercuci, Denis Bertini [committer] */
// -------------------------------------------------------------------------
// ----- CbmVertex source file -----
@@ -178,5 +178,45 @@ string CbmVertex::ToString() const
}
// -------------------------------------------------------------------------
// -------------------------------------------------------------------------
int32_t CbmVertex::GetTrackIndex(int32_t iTrack) const
{
if (iTrack < 0 || iTrack >= fNTracks) {
LOG(warning) << GetName() << "::GetTrackIndex(" << iTrack << ") : outside range.";
return -1;
}
if (!fTrkIdx.size() || size_t(iTrack) >= fTrkIdx.size()) return -1;
return fTrkIdx[iTrack];
}
// -------------------------------------------------------------------------
// -------------------------------------------------------------------------
bool CbmVertex::FindTrackByIndex(uint32_t iTrack) const
{
auto idx = find_if(fTrkIdx.begin(), fTrkIdx.end(), [iTrack](uint32_t p) { return p == iTrack; });
if (idx != fTrkIdx.end()) return true;
return false;
}
// -------------------------------------------------------------------------
// -------------------------------------------------------------------------
bool CbmVertex::SetTracks(std::vector<uint32_t>& indexVector)
{
fTrkIdx = indexVector;
if (!fNTracks)
fNTracks = fTrkIdx.size();
else {
if (size_t(fNTracks) != fTrkIdx.size()) {
LOG(error) << GetName()
<< "::SetTracks() : fNTracks does not match fTrkIdx info. This might point to a problem !";
fNTracks = fTrkIdx.size();
return false;
}
}
return true;
}
// -------------------------------------------------------------------------
ClassImp(CbmVertex)
Loading