GFTrackCand.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GFTrackCand.h

/* 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>
#include <TMatrixD.h>
#include <TDatabasePDG.h>

#include <cmath>

class GFTrackCand : public TObject {
public:

  // Constructors/Destructors ---------
  GFTrackCand();
  ~GFTrackCand();

  /* @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 {
    std::cerr << "the method GFTrackCand::getCurv() is deprecated. Use the setter/getter for the 6D state or pos,mom instead instead\n";
    return fCurv;}
  double getDip() const {
    std::cerr << "the method GFTrackCand::getDip() is deprecated. Use the setter/getter for the 6D state or pos,mom instead instead\n";
    return fDip;}
  //bool inverted() const {return fInv;} //nobody seems to use it so I commented it out
  std::vector<unsigned int> getHitIDs(int detId = -1) const;
  std::vector<unsigned int> GetHitIDs(int detId = -1) const;
  std::vector<unsigned int> getDetIDs() const {return fDetId;}
  std::vector<unsigned int> GetDetIDs() const { std::cerr << "the method GFTrackCand::GetDetIDs() is deprecated. Use GFTrackCand::getDetIDs() instead\n"; return fDetId;}
  std::vector<double>       getRhos() const {return fRho;}
  std::vector<double>       GetRhos() const {
    std::cerr << "the method GFTrackCand::GetRhos() is deprecated. Use GFTrackCand::getRhos() instead\n";
    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 {
    TVector3 posSeed(fState6D(0,0), fState6D(1,0), fState6D(2,0));
    return posSeed;
  }
  TVector3 getMomSeed() const {
    TVector3 momSeed(fState6D(3,0), fState6D(4,0), fState6D(5,0));
    return momSeed;
  }
  TVector3 getDirSeed() const {
    std::cerr << "the method GFTrackCand::getDirSeed() is deprecated. Use GFTrackCand::getStateSeed() or GFTrackCand::getMomSeed() instead\n";
    double p = std::sqrt(fState6D(3,0) * fState6D(3,0) + fState6D(4,0) * fState6D(4,0) + fState6D(5,0) * fState6D(5,0));
    TVector3 dirSeed(fState6D(3,0)/p, fState6D(4,0)/p, fState6D(5,0)/p);
    return dirSeed;
  }
  double getQoverPseed() const {
    std::cerr << "the method GFTrackCand::getQoverPseed() is deprecated. Use GFTrackCand::getStateSeed() or GFTrackCand::getMomSeed()  and/or getChargeSeed() instead\n";
    double p = std::sqrt(fState6D(3,0) * fState6D(3,0) + fState6D(4,0) * fState6D(4,0) + fState6D(5,0) * fState6D(5,0));
    return fQ / p;
  }
  TMatrixD getStateSeed() const {
    return fState6D;
  }
  TMatrixD getCovSeed() const {
    return fCov6D;
  }
  double getChargeSeed()const {
    return fQ;
  }
  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) {
    std::cerr << "the method GFTrackCand::setCurv is deprecated. Use GFTrackCand::set6DSeed instead\n";
    fCurv = c;
  }
  void setDip(double d) {
    std::cerr << "the method GFTrackCand::setDip is deprecated. Use GFTrackCand::set6DSeed instead\n";
    fDip = d;
  }
  //void setInverted(bool f=true) {fInv=f;} //nobody seems to use it so I commented it out
  void setMcTrackId(int i) {fMcTrackId = i;}
  bool hitInTrack(unsigned int detId, unsigned int hitId) const;

  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;
    TParticlePDG* part = TDatabasePDG::Instance()->GetParticle(fPdg);
    fQ = part->Charge() / (3.);
  }
  void append(const GFTrackCand&);

  void sortHits();

  void sortHits(std::vector<unsigned int> indices);

  // Operations ----------------------
  void reset();
  void Print(const Option_t* = "") const ;
  void set6DSeed(const TMatrixD& state6D, const double charge, TMatrixD cov6D = -1.0 * TMatrixD(TMatrixD::kUnit, TMatrixD(6, 6))) {
    fQ = charge;
    fState6D = state6D;
    fCov6D = cov6D;
  }
  void set6DSeedAndPdgCode(const TMatrixD& state6D, const int pdgCode, TMatrixD cov6D = -1.0 * TMatrixD(TMatrixD::kUnit, TMatrixD(6, 6))) {
    setPdgCode(pdgCode);
    fState6D = state6D;
    fCov6D = cov6D;
  }
  void setPosMomSeed(const TVector3& pos, const TVector3& mom, const double charge, TMatrixD cov6D = -1.0 * TMatrixD(TMatrixD::kUnit, TMatrixD(6, 6))) {
    fQ = charge;
    fState6D(0,0) = pos[0];
    fState6D(1,0) = pos[1];
    fState6D(2,0) = pos[2];
    fState6D(3,0) = mom[0];
    fState6D(4,0) = mom[1];
    fState6D(5,0) = mom[2];
    fCov6D = cov6D;
  }
  void setPosMomSeedAndPdgCode(const TVector3& pos, const TVector3& mom, const int pdgCode, TMatrixD cov6D = -1.0 * TMatrixD(TMatrixD::kUnit, TMatrixD(6, 6))) {
    setPdgCode(pdgCode);
    fState6D(0,0) = pos[0];
    fState6D(1,0) = pos[1];
    fState6D(2,0) = pos[2];
    fState6D(3,0) = mom[0];
    fState6D(4,0) = mom[1];
    fState6D(5,0) = mom[2];
    fCov6D = cov6D;
  }
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. Deprecated! Will be deleted soon.
  double fDip;  // dip angle from pattern reco. Deprecated! Will be deleted soon.

  int fMcTrackId; 
  int fPdg; 
  TMatrixD fState6D; 
  TMatrixD fCov6D; 
  double fQ; 
  // Private Methods -----------------

public:
  ClassDef(GFTrackCand, 9)
};

#endif