PndCATracks.h

Go to the documentation of this file.

//-*- Mode: C++ -*-
// *****************************************************************************
//                                                                             *
// @Autors: I.Kulakov; M.Zyzak; I.Kisel                                        *
// @e-mail: I.Kulakov@gsi.de; M.Zyzak@gsi.de; I.Kisel@compeng.uni-frankfurt.de *
//                                                                             *
// *****************************************************************************

#ifndef PNDCATRACKS_H
#define PNDCATRACKS_H

#include "PndCAHits.h"
#include "PndCAFunctionality.h"

class PndCATrack {
 public:
  PndCATrack() : fChi2(0), fNDF(-5), fIHits(), fLevel(0) { fIHits.reserve(PndCAParameters::MaxNStations); };

  int NHits() const { return fIHits.size(); }

  vector<PndCATES> &IHits() { return fIHits; }
  const vector<PndCATES> &IHits() const { return fIHits; }

  void AddHit(char iS, int iH)
  {
    fIHits.push_back(PndCATES(iS, iH));
    fNDF++;
  }
  void AddHit(const PndCATES &iH)
  {
    fIHits.push_back(iH);
    fNDF++;
  }

  float &Chi2() { return fChi2; }
  const float &Chi2() const { return fChi2; }
  char &NDF() { return fNDF; }
  const char &NDF() const { return fNDF; }
  char &Level() { return fLevel; }
  const char &Level() const { return fLevel; }

  static bool Compare(const PndCATrack *a, const PndCATrack *b)
  {
    // return (a->fIHits.size() > b->fIHits.size()) ||
    //        ( (a->fIHits.size() == b->fIHits.size()) &&
    //          ( (a->fIHits[0].s < b->fIHits[0].s) ||
    //            ( (a->fIHits[0].s == b->fIHits[0].s) && (a->fChi2 < b->fChi2)
    //            )
    //          )
    //        );
    return (a->fIHits.size() > b->fIHits.size()) || ((a->fIHits.size() == b->fIHits.size()) && (a->fChi2 < b->fChi2));
  }

  int NUsedHits(const PndCAHits &hits) const
  {
    int nused = 0;
    for (unsigned int iH = 0; iH < fIHits.size(); ++iH) {
      const PndCAHit &h = hits[fIHits[iH]];
      if (h.IsUsed())
        nused++;
    }
    return nused;
  }

  void SetHitsAsUsed(PndCAHits &hits) const
  {
    for (unsigned int iH = 0; iH < fIHits.size(); ++iH) {
      PndCAHit &h = hits[fIHits[iH]];
      h.SetAsUsed();
    }
  }

  const PndCATrackParam
  Fit(const PndCAHits &hits, const PndCATarget &target, const PndCAParam &caParam, bool dir = true); // dir = true - get outer parameters, dir = false - get inner ones

 private:
  float fChi2;
  char fNDF;
  vector<PndCATES> fIHits; // index in PndCATracks::fHits array

  char fLevel; // n of track segments
};

// typedef vector<PndCATrack> PndCATracks;

class PndCATracks;

class PndCATracks : public vector<PndCATrack> {
 public:
  PndCATracks(PndCAHits *hits) : fHitsRef(hits){};

  const PndCAHit &Hit(int iH, int iT) const { return (*fHitsRef)[(*this)[iT].IHits()[iH]]; }

  // selects tracks and marks all hits as used
  void SelectAndSaveTracks(PndCATracks &tracks);

  PndCAHits *HitsRef() { return fHitsRef; }

 private:
  PndCAHits *fHitsRef;
};

inline void PndCATracks::SelectAndSaveTracks(PndCATracks &tracks)
{
  vector<PndCATrack *> vptrackcandidate; // vptrackcandidate - array of pointers to vtrackcandidate
  vptrackcandidate.resize(size());

  for (unsigned int iC = 0; iC < size(); ++iC) {
    vptrackcandidate[iC] = &((*this)[iC]);
  }

  sort(vptrackcandidate.begin(), vptrackcandidate.end(), PndCATrack::Compare);

  for (vector<PndCATrack *>::iterator trIt = vptrackcandidate.begin(); trIt != vptrackcandidate.end(); ++trIt) {
    PndCATrack *tr = *trIt;

    if (tr->NUsedHits(*fHitsRef) > 0)
      continue; // don't allow tracks have shared hits. Track will be shorter, leave it for the next iteration

    tr->SetHitsAsUsed(*fHitsRef);

    tracks.push_back(*tr);
  } // i_trackCandidate
}

inline const PndCATrackParam PndCATrack::Fit(const PndCAHits &hits, const PndCATarget &target0, const PndCAParam &caParam, bool dir)
{

  vector<PndCATESV> iHits(NHits());
  const int NTHits = iHits.size();

  for (unsigned short ihit = 0; ihit < NTHits; ihit++) {
    iHits[ihit] = IHits()[ihit];
  }
  PndCATrackParamVector param;
  PndCAFunctionality::Fit(caParam, hits, param, iHits, target0, dir);
  PndCATrackParam paramS(param, 0);
  fChi2 = paramS.Chi2();
  fNDF = paramS.NDF();
  return paramS;
}

#endif