GFTrack.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 GFTRACK_H 
#define GFTRACK_H

#include"assert.h"
#include<map>

#include "GFAbsTrackRep.h"
#include "GFAbsRecoHit.h"

#include "TClonesArray.h"
#include "TObjArray.h"

#include "GFTrackCand.h"
#include "GFBookkeeping.h"

class TVirtualGeoTrack;

class GFTrack : public TObject {   
private:
  
  
  TObjArray* fTrackReps; //->

  std::vector<GFAbsRecoHit*>      fHits;
  
  std::vector< GFBookkeeping* > fBookkeeping;

  std::vector<int> fRepAtHit;

  GFTrackCand fCand; // list of hits
    
  static const int fDefNumTrackReps = 10; 
  unsigned int fCardinal_rep; // THE selected rep, default=0;

  unsigned int fNextHitToFit;

  bool fSmooth;
  
public:
  
  GFTrack(); 

  GFTrack(const GFTrack&); 

  GFTrack& operator=(const GFTrack&);

  GFTrack(GFAbsTrackRep*, bool smooth = false); 
  
  virtual ~GFTrack();

  // -----------------------
  // Accessors
  // -----------------------

  void reset();  // deletes the RecoHits!

  int getFailedHits(int repId=-1){
        int theRep;
        if(repId==-1) theRep=fCardinal_rep;
        else theRep = repId;
        return fBookkeeping.at(theRep)->getNumFailed();
  }

  std::vector<GFAbsRecoHit*> getHits() {return fHits;}

  const GFTrackCand& getCand() const {return fCand;}

  GFAbsRecoHit* getHit(int id) const {
    return fHits.at(id);
  }

  unsigned int getNumHits() const {
    return fHits.size();
  }

  void mergeHits(GFTrack* trk); 

  void releaseHits(){fHits.clear();} 

  void releaseTrackReps(){ fTrackReps->SetOwner(kFALSE); fTrackReps->Clear();} 

  unsigned int getNextHitToFit() const {return fNextHitToFit;}

  void setNextHitToFit(unsigned int i) {fNextHitToFit=i;}

  GFAbsTrackRep* getTrackRep(int id) const {
    return reinterpret_cast<GFAbsTrackRep*>(fTrackReps->At(id));
  }

  unsigned int getNumReps() const {
    return fTrackReps->GetEntriesFast();
  }

  GFAbsTrackRep* getCardinalRep() const {return ((GFAbsTrackRep*)fTrackReps->At(fCardinal_rep));}
  

  TVector3 getMom() const {return getCardinalRep()->getMom();}

  TVector3 getMom(const GFDetPlane& pl) const {return getCardinalRep()->getMom(pl);}

  TVector3 getPos() const {return getCardinalRep()->getPos();}

  TVector3 getPos(const GFDetPlane& pl) const {return getCardinalRep()->getPos(pl);}

  void getPosMomCov(TVector3& pos,TVector3& mom,TMatrixT<double>& cov){
    getCardinalRep()->getPosMomCov(pos,mom,cov);
  }

  void getPosMomCov(const GFDetPlane& pl,TVector3& pos,TVector3& mom,TMatrixT<double>& cov){
    getCardinalRep()->getPosMomCov(pl,pos,mom,cov);
  }

  double getChiSqu() const {return getCardinalRep()->getChiSqu();}

  unsigned int getNDF() const {return getCardinalRep()->getNDF();}

  double getRedChiSqu() const {return getCardinalRep()->getRedChiSqu();}

  double getCharge() const {return getCardinalRep()->getCharge();}

  void fillGeoTrack(TVirtualGeoTrack* tr) const {fillGeoTrack(tr,fCardinal_rep);} 

  void fillGeoTrack(TVirtualGeoTrack* tr,unsigned int repid) const;
  
  // ---------------------
  // Modifiers
  // ---------------------

  void addFailedHit(unsigned int irep,unsigned int id){
    assert(irep<fBookkeeping.size());
    fBookkeeping.at(irep)->addFailedHit(id);
  }

  inline void addHit(GFAbsRecoHit* theHit) { 
    fHits.push_back(theHit);
  }
  
  void addHit(GFAbsRecoHit* theHit, 
              unsigned int detId,
              unsigned int hitId,
              double rho=0.,
              unsigned int planeId=0){
    fHits.push_back(theHit);
    fCand.addHit(detId,hitId,rho,planeId);
  }

  void addHitVector(std::vector<GFAbsRecoHit*> hits) {
    fHits = hits;
  }

  void addTrackRep(GFAbsTrackRep* theTrackRep) {
    if(fTrackReps==NULL)fTrackReps=new TObjArray(fDefNumTrackReps);
    fTrackReps->Add(theTrackRep);
    fBookkeeping.push_back( new GFBookkeeping() );
    fRepAtHit.push_back(-1);
  }

  GFBookkeeping* getBK(int index=-1){
    if(index==-1) return fBookkeeping.at(fCardinal_rep);
    assert((unsigned int)index<getNumReps());
    return fBookkeeping.at(index);
  }
    
  void setCandidate(const GFTrackCand& cand, bool doreset=false);
  
  void setCardinalRep(unsigned int r){if((int)r<fTrackReps->GetEntriesFast())fCardinal_rep=r;}
  

  void getResiduals(unsigned int detId, // which detector?
                    unsigned int dim,   // which projection?
                    unsigned int rep,   // which trackrep ?
                    std::vector<double>& result);
                    

  void setRepAtHit(unsigned int irep,int ihit){
    assert(irep<getNumReps());
    fRepAtHit.at(irep) = ihit;
  }

  int getRepAtHit(unsigned int irep){
    assert(irep<getNumReps());
    return fRepAtHit.at(irep);
  }

  void clearRepAtHit(){
    for(unsigned int i=0;i<getNumReps();++i){
      fRepAtHit.at(i)=-1;
    }
  }

  void printBookkeeping();

  void Print();

  void clearBookkeeping(){
    for(unsigned int i=0;i<getNumReps();++i){
      fBookkeeping.at(i)->clearAll();
    }
  }

  void clearFailedHits(){
    for(unsigned int i=0;i<getNumReps();++i){
      fBookkeeping.at(i)->clearFailedHits();
    }
  }

  void getHitsByPlane(std::vector<std::vector<int>*>& retVal);

  void setSmoothing(bool smooth = true) { fSmooth = smooth; }

  bool getSmoothing() { return fSmooth; }

  void blowUpCovs(double blowUpFactor);


public:
  ClassDef(GFTrack,1)
};

#endif