UtilTrack.cc

Go to the documentation of this file.

#include <cassert>
#include <geom_svc/GeomSvc.h>
#include <interface_main/SQTrackVector.h>
#include <interface_main/SQHitVector.h>
#include "UtilTrack.h"
using namespace std;
 
 
SQTrack* UtilTrack::FindTrackByID(const SQTrackVector* vec, const int id_trk, const bool do_assert)
{
  for (SQTrackVector::ConstIter it = vec->begin(); it != vec->end(); it++) {
    SQTrack* trk = *it;
    if (trk->get_track_id() == id_trk) {
      if (do_assert) assert(trk);
      return trk;
    }
  }
  return 0;
}
 
 
SQHitVector* UtilTrack::FindHitsOfTrack(const SQHitVector* vec_in, const int id_trk)
{
  SQHitVector* vec = vec_in->Clone();
  vec->clear();
  for (SQHitVector::ConstIter it = vec_in->begin(); it != vec_in->end(); it++) {
    SQHit* hit = *it;
    if (hit->get_track_id() == id_trk) vec->push_back(hit);
  }
  return vec;
}
 
 
SQHitVector* UtilTrack::FindDetectorHitsOfTrack(const SQHitVector* vec_in, const int id_trk, const std::string det_name)
{
  SQHitVector* vec = vec_in->Clone();
  vec->clear();
  for (SQHitVector::ConstIter it = vec_in->begin(); it != vec_in->end(); it++) {
    SQHit* hit = *it;
    if (hit->get_track_id() != id_trk) continue;
    string name = GeomSvc::instance()->getDetectorName( hit->get_detector_id() );
    if (name.substr(0, det_name.size()) != det_name) continue;
    vec->push_back(hit);
  }
  return vec;
}
 
SQHitVector* UtilTrack::FindDetectorHitsOfTrack(const SQHitVector* vec_in, const int id_trk, const char* det_name)
{
  return FindDetectorHitsOfTrack(vec_in, id_trk, (const string)det_name);
}
 
 
SQHitVector* UtilTrack::FindHodoHitsOfTrack(const SQHitVector* vec_in, const int id_trk)
{
  return FindDetectorHitsOfTrack(vec_in, id_trk, "H");
}