GFTrackCand.h

Go to the documentation of this file.

/* Copyright 2008-2010, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert

   This file is part of GENFIT.

   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef GFTRACKCAND_H
#define GFTRACKCAND_H

#include <vector>
#include <set>
#include <iostream>
#include "assert.h"

#include "TObject.h"
#include "TVector3.h"

class GFTrackCand : public TObject {
 public:
  // Constructors/Destructors ---------
  GFTrackCand();
  ~GFTrackCand();

  GFTrackCand(double curv, double dip, double inv, std::vector<unsigned int> detIDs, std::vector<unsigned int> hitIDs);
  /* @brief same as previous ctor, but with ordering parameters */
  GFTrackCand(double curv, double dip, double inv, std::vector<unsigned int> detIDs, std::vector<unsigned int> hitIDs, std::vector<double> rhos);

  /* @brief == operator does not check for rho */
  friend bool operator==(const GFTrackCand &lhs, const GFTrackCand &rhs);

  // Accessors -----------------------
  void getHit(unsigned int i, unsigned int &detId, unsigned int &hitId) const
  {
    assert(i < getNHits());
    detId = fDetId.at(i);
    hitId = fHitId.at(i);
  }
  void getHit(unsigned int i, unsigned int &detId, unsigned int &hitId, double &rho) const
  {
    assert(i < getNHits());
    detId = fDetId.at(i);
    hitId = fHitId.at(i);
    rho = fRho.at(i);
  }
  void getHitWithPlane(unsigned int i, unsigned int &detId, unsigned int &hitId, unsigned int &planeId) const
  {
    assert(i < getNHits());
    detId = fDetId.at(i);
    hitId = fHitId.at(i);
    planeId = fPlaneId.at(i);
  }

  unsigned int getNHits() const { return fDetId.size(); }
  double getCurv() const { return fCurv; }
  double getDip() const { return fDip; }
  bool inverted() const { return fInv; }
  std::vector<unsigned int> GetHitIDs(int detId = -1);
  std::vector<unsigned int> GetDetIDs() const { return fDetId; }
  std::vector<double> GetRhos() const { return fRho; }
  std::set<unsigned int> GetUniqueDetIDs() const
  {
    std::set<unsigned int> retVal;
    for (unsigned int i = 0; i < fDetId.size(); ++i) {
      retVal.insert(fDetId.at(i));
    }
    return retVal;
  }
  int getMcTrackId() const { return fMcTrackId; }
  TVector3 getPosSeed() const { return fPosSeed; }
  TVector3 getDirSeed() const { return fDirSeed; }
  double getQoverPseed() const { return fQoverpSeed; }
  TVector3 getPosError() const { return fPosError; }
  TVector3 getDirError() const { return fDirError; }
  int getPdgCode() const { return fPdg; }
  // Modifiers -----------------------
  void addHit(unsigned int detId, unsigned int hitId, double rho = 0., unsigned int planeId = 0);
  void setCurv(double c) { fCurv = c; }
  void setDip(double d) { fDip = d; }
  void setInverted(bool f = true) { fInv = f; }
  void setMcTrackId(int i) { fMcTrackId = i; }
  bool HitInTrack(unsigned int detId, unsigned int hitId);
  void setTrackSeed(const TVector3 &pos, const TVector3 &direction, const double qop)
  {
    fPosSeed = pos;
    fDirSeed = direction;
    fQoverpSeed = qop;
  }
  void setComplTrackSeed(const TVector3 &pos, const TVector3 &mom, const int pdgCode, TVector3 posError = TVector3(1.0, 1.0, 1.0), TVector3 dirError = TVector3(1.0, 1.0, 1.0));
  void setPdgCode(int pdgCode) { fPdg = pdgCode; }
  void append(const GFTrackCand &);

  // Operations ----------------------
  void reset();
  void Print() const;

 private:
  // Private Data Members ------------
  std::vector<unsigned int> fDetId;
  std::vector<unsigned int> fHitId;
  std::vector<unsigned int> fPlaneId;
  std::vector<double> fRho;

  double fCurv; // curvature from pattern reco
  double fDip;  // dip angle from pattern reco
  bool fInv;    // true if inverted track

  TVector3 fPosSeed;  // seed value for the track: pos
  TVector3 fDirSeed;  // direction
  TVector3 fPosError; // error on position seed given as a standard deviation
  TVector3 fDirError; // error on direction seed given as a standard deviation
  double fQoverpSeed; // q/p
  int fMcTrackId;     // if MC simulation, store the mct track id here
  int fPdg;           // particle data groupe's id for a particle

  // Private Methods -----------------

 public:
  ClassDef(GFTrackCand, 5)
};

#endif