CbmLitTrackSelectionSharedHits.cxx

Go to the documentation of this file.

/* Copyright (C) 2008-2013 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#include "selection/CbmLitTrackSelectionSharedHits.h"
 
#include "data/CbmLitTrack.h"
#include "utils/CbmLitComparators.h"
 
#include <algorithm>
#include <set>
 
CbmLitTrackSelectionSharedHits::CbmLitTrackSelectionSharedHits() : fNofSharedHits(0) {}
 
CbmLitTrackSelectionSharedHits::~CbmLitTrackSelectionSharedHits() {}
 
LitStatus CbmLitTrackSelectionSharedHits::DoSelect(TrackPtrIterator itBegin, TrackPtrIterator itEnd)
{
  if (itBegin == itEnd) {
    return kLITSUCCESS;
  }
 
  CbmLitQualitySort::DoSortNofHits(itBegin, itEnd);
  CheckSharedHits(itBegin, itEnd);
 
  return kLITSUCCESS;
}
 
LitStatus CbmLitTrackSelectionSharedHits::DoSelect(TrackPtrVector& tracks)
{
  return DoSelect(tracks.begin(), tracks.end());
}
 
void CbmLitTrackSelectionSharedHits::CheckSharedHits(TrackPtrIterator itBegin, TrackPtrIterator itEnd)
{
  std::set<int> hitsId;
  const int STRIPSTART = 100000;
  const int TRDSTART   = 1000000;
 
  for (TrackPtrIterator iTrack = itBegin; iTrack != itEnd; iTrack++) {
    CbmLitTrack* track = *iTrack;
 
    if (track->GetQuality() == kLITBAD) {
      continue;
    }
 
    int nofSharedHits = 0;
    int nofHits       = track->GetNofHits();
 
    for (int iHit = 0; iHit < nofHits; iHit++) {
      int hitId         = track->GetHit(iHit)->GetRefId();
      LitHitType type   = track->GetHit(iHit)->GetType();
      LitSystemId sysId = track->GetHit(iHit)->GetSystem();
      if (type == kLITSTRIPHIT) {
        hitId += STRIPSTART;
      }
      if (sysId == kLITTRD) {
        hitId += TRDSTART;
      }
      if (hitsId.find(hitId) != hitsId.end()) {
        nofSharedHits++;
        if (nofSharedHits > fNofSharedHits) {
          track->SetQuality(kLITBAD);
          break;
        }
      }
    }
 
    if (track->GetQuality() == kLITBAD) {
      continue;
    }
 
    for (int iHit = 0; iHit < nofHits; iHit++) {
      int hitId         = track->GetHit(iHit)->GetRefId();
      LitHitType type   = track->GetHit(iHit)->GetType();
      LitSystemId detId = track->GetHit(iHit)->GetSystem();
      if (type == kLITSTRIPHIT) {
        hitId += STRIPSTART;
      }
      if (detId == kLITTRD) {
        hitId += TRDSTART;
      }
      hitsId.insert(hitId);
    }
  }
  hitsId.clear();
}