PndHypDKalmanTask Class Reference

#include <PndHypDKalmanTask.h>

Inheritance diagram for PndHypDKalmanTask:

Public Member Functions
	PndHypDKalmanTask ()
	~PndHypDKalmanTask ()
void	SetTrackBranchName (const TString &name)
void	AddHitBranch (unsigned int detId, const TString &m)
void	SetPersistence (Bool_t opt=kTRUE)
void	SetSmooth (Bool_t opt=kTRUE)
void	UseMVD (bool mvd)
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)
void	WriteHistograms (const TString &filename)

Private Attributes
TString	fTrackBranchName
std::map< unsigned int, TString >	fHitBranchMap
TClonesArray *	fTrackArray
Bool_t	fPersistence
GFRecoHitFactory *	fTheRecoHitFactory
FairField *	fField
TH1D *	fPH
TH1D *	fChi2H
TH1D *	fXresH
TH1D *	fYresH
TH1D *	fXresFitH
TH1D *	fYresFitH
TH1D *	fPEnd
TH1D *	fPull
int	fTrackcount
Bool_t	fSmooth
int	fEvt
bool	fUseMVD

Detailed Description

Definition at line 35 of file PndHypDKalmanTask.h.

Constructor & Destructor Documentation

PndHypDKalmanTask::PndHypDKalmanTask

(

)

Definition at line 58 of file PndHypDKalmanTask.cxx.

References fTrackBranchName, and PndGeoHandling::Instance().

  : FairTask("Kalman Filter"), fPersistence(kFALSE),fSmooth(kFALSE),fUseMVD(false),fEvt(0)
{
  fTrackBranchName = "Track";
 PndGeoHandling::Instance();
}

PndHypDKalmanTask::~PndHypDKalmanTask

(

)

Definition at line 66 of file PndHypDKalmanTask.cxx.

References fChi2H, and fPH.

{
  if(fPH!=NULL)delete fPH;
  if(fChi2H!=NULL)delete fChi2H;
}

Member Function Documentation

void PndHypDKalmanTask::AddHitBranch ( unsigned int  detId, const TString &  m  )

inline

Definition at line 50 of file PndHypDKalmanTask.h.

References fHitBranchMap, and m.

{fHitBranchMap[detId]=m;};

void PndHypDKalmanTask::Exec ( Option_t *  opt )

virtual

Definition at line 159 of file PndHypDKalmanTask.cxx.

References GFTrack::addHitVector(), GFRecoHitFactory::createMany(), fChi2H, fEvt, fPEnd, fPH, fPull, fTheRecoHitFactory, fTrackArray, fTrackcount, GFTrack::getCand(), GFTrack::getCardinalRep(), GFTrack::getChiSqu(), GFAbsTrackRep::getMom(), GFTrack::getMom(), GFTrack::getNumHits(), GFTrack::getNumReps(), GFTrack::getPos(), GFTrackCand::getQoverPseed(), GFTrack::getRedChiSqu(), GFAbsTrackRep::getStatusFlag(), GFTrack::getTrackRep(), p, pos, GFAbsTrackRep::Print(), GFKalman::processTrack(), GFKalman::setNumIterations(), GeaneTrackRep::setPropDir(), trk, and GFException::what().

{
  std::cout<<"PndHypDKalmanTask::Exec Event "<<fEvt++<<std::endl;
  //Reset output Array
  //if(_trackArray==0) Fatal("PndHypDKalman::Exec)","No TrackArray");
  // _trackArray->Delete();

  Int_t ntracks=fTrackArray->GetEntriesFast();
  

  

  // Fitting ---------------- can go to another task!
  GFKalman fitter;
  //fitter.setLazy(1); // tell the fitter to skip hits if error occurs  
  fitter.setNumIterations(1);
  for(Int_t itr=0;itr<ntracks;++itr){
    GFTrack* trk=(GFTrack*)fTrackArray->At(itr);
    GFTrackCand trcnd = trk->getCand();

    // Load RecoHits 
    try {
      trk->addHitVector(fTheRecoHitFactory->createMany(trk->getCand()));
      std::cout<<trk->getNumHits()<<" hits in track "
               <<itr<<std::endl;
    }
   
    catch(GFException& e) {
      std::cout << e.what() << std::endl;
      throw e;
    }
    
    /*
    // fill tpc residuals
    std::vector<double> res;
    trk->getResiduals(2,0,0,res);
    for(int i=0;i<res.size();++i){
      _xresH->Fill(res[i]);
    }
    res.clear();
    trk->getResiduals(2,1,0,res);
    for(int i=0;i<res.size();++i){
      _yresH->Fill(res[i]);
    }
    res.clear();
    */
    // Start Fitter
    try{
      fitter.processTrack(trk);
      
    }
    catch (GFException e){
      std::cout<<"*** FITTER EXCEPTION ***"<<std::endl;
      std::cout<<e.what()<<std::endl;
      
    }

    std::cout<<"SUCESSFULL FIT!"<<std::endl;

    // Fill some histos after fit
    if(trk->getTrackRep(0)->getStatusFlag()==0){
      // propagate backwards
      GeaneTrackRep* gtrk=dynamic_cast<GeaneTrackRep*>(trk->getTrackRep(0));
      if(gtrk!=NULL)gtrk->setPropDir(-1);

      trk->getCardinalRep()->Print();
      
      // ------- Propagation to prim vertex ---------
        
      GFDetPlane pl(TVector3(0,0,-55.0),TVector3(1,0,0),TVector3(0,1,0));

      //TVector3 p3=trk->getTrackRep(0)->getMom(pl);
      //--------------------------------------------
      
      double p=trk->getTrackRep(0)->getMom().Mag();

      std::cout<<" momentum "<<" "<<trk->getMom().Mag()<<std::endl;
      std::cout<<" "<<" N reps "<<trk->getNumReps()<<" chi2red "<<trk->getRedChiSqu()<<std::endl;
      
      //INFO: Changing the EPSIL param to 
      //0.05 (HYPsilicon, HYPdiamond, HYPcarbon--> materials in geo  file)
      //propagation to prim vertex works smoothly for 500 ev.
      //no dedx modification from media file is needed

      fPH->Fill(p);

      fPEnd->Fill(-1/(trcnd.getQoverPseed()));

      fPull->Fill((-1/(trcnd.getQoverPseed()))-p);

      TVector3 pos=trk->getPos();
      //fPEnd->Fill(pos.X(),pos.Z());

      double chi2=trk->getChiSqu();
       fChi2H->Fill(chi2);
      ++fTrackcount;


      
     //  // fill tpc residuals
//       trk->getResiduals(2,0,0,res);
//       for(int i=0;i<res.size();++i){
//      _xresFitH->Fill(res[i]);
//       }
//       res.clear();
//       trk->getResiduals(2,1,0,res);
//       for(int i=0;i<res.size();++i){
//      _yresFitH->Fill(res[i]);
//       }
//       res.clear();
      
    }
  }
  
  return;
}

InitStatus PndHypDKalmanTask::Init ( )

virtual

Definition at line 73 of file PndHypDKalmanTask.cxx.

References GFRecoHitFactory::addProducer(), fChi2H, fHitBranchMap, fPEnd, fPH, fPull, fTheRecoHitFactory, fTrackArray, fTrackBranchName, fUseMVD, fXresFitH, fXresH, fYresFitH, fYresH, and gGeoManager.

{
  //Get ROOT Manager
  FairRootManager* ioman= FairRootManager::Instance();

  if(ioman==0)
    {
      Error("PndHypDKalmanTask::Init","RootManager not instantiated!");
      return kERROR;
    }
  
  // Get input collection
  fTrackArray=(TClonesArray*) ioman->GetObject(fTrackBranchName);
  
  //if(_trackArray==0)
  //{
      //   Error("PndHypDKalmanTask::Init","track-array not found!");
      //return kERROR;
  //}
  
  // create and register output array
  //_trackArray = new TClonesArray("GFTrack"); 
  //ioman->Register("TrackPreFit","GenFit",_trackArray,_persistence);
 
  // Build hit factory -----------------------------
  fTheRecoHitFactory = new GFRecoHitFactory();

  std::map<unsigned int,TString>::iterator iter=fHitBranchMap.begin();
  while(iter!=fHitBranchMap.end()){

    TClonesArray* har=(TClonesArray*) ioman->GetObject("HypHit");
   
    //cout<<" "<<iter->second<<endl;

    if(har==0){
      
      Error("PndHypKalmanTask::Init","Hit array not found");
    }
    else{ 
      
      if(iter->first==2)fTheRecoHitFactory->addProducer(iter->first,new GFRecoHitProducer<PndHypHit,PndHypRecoSPHit>(har));
     
    }

     if(fUseMVD==true){
       TClonesArray* sar=(TClonesArray*) ioman->GetObject("MVDHit");
       if(sar==0){
         
         Error("PndHypKalmanTask::Init","Hit array not found");
       }
       else{ 
         
         if(iter->first==3)fTheRecoHitFactory->addProducer(iter->first,new GFRecoHitProducer<PndSdsHit,PndSdsRecoHit>(sar));
       }
     }

    // TClonesArray* ar=(TClonesArray*) ioman->GetObject(iter->second);
    // if(ar==0){
    //   Error("PndHypDKalmanTask::Init","point-array %s not found!",iter->second.Data());
    // }
    // else{ 
    //   // the next lines is not general because it will work only for CmMCPoints!
    //   fTheRecoHitFactory->addProducer(iter->first,new GFRecoHitProducer<PndHypHit,PndHypRecoSPHit>(ar));
    //   //FairHit,PndHypDSPHit>(ar));PndHypPoint,PndHypRecoHit
    // }
    
    
    ++iter;
  }//end loops over hit types
  
  // setup histograms
   fPH=new TH1D("pH","p",500,0.02,0.7);
  fChi2H=new TH1D("chi2H","chi2",100,0,20);
  fXresH=new TH1D("xres","xres",100,-5,5);
  fYresH=new TH1D("yres","yres",100,-5,5);
  fXresFitH=new TH1D("xresfit","xres after fit",100,-5,5);
  fYresFitH=new TH1D("yresfit","yres after fit",100,-5,5);
  fPEnd=new TH1D("pPre","Endpoint",500,0.02,0.7);
  fPull=new TH1D("pPull","Pull",500,-0.3,0.3);
 std::cout << "-I- gGeoManager = "<<gGeoManager << std::endl;

  return kSUCCESS;
}

void PndHypDKalmanTask::SetPersistence ( Bool_t  opt = kTRUE )

inline

Definition at line 51 of file PndHypDKalmanTask.h.

References fPersistence.

{fPersistence=opt;}

void PndHypDKalmanTask::SetSmooth ( Bool_t  opt = kTRUE )

inline

Definition at line 52 of file PndHypDKalmanTask.h.

References fSmooth.

{fSmooth=opt;}

void PndHypDKalmanTask::SetTrackBranchName ( const TString &  name )

inline

Definition at line 49 of file PndHypDKalmanTask.h.

References fTrackBranchName, and name.

{fTrackBranchName=name;}

void PndHypDKalmanTask::UseMVD ( bool  mvd )

inline

Definition at line 53 of file PndHypDKalmanTask.h.

References fUseMVD.

{fUseMVD=mvd;}

void PndHypDKalmanTask::WriteHistograms

(

const TString &

filename

)

Definition at line 277 of file PndHypDKalmanTask.cxx.

References fChi2H, file, fPEnd, fPH, fPull, fXresFitH, fXresH, fYresFitH, and fYresH.

                                                         {
  TFile* file = FairRootManager::Instance()->GetOutFile();
  file->cd();
  file->mkdir("PndHypDKalman");
  file->cd("PndHypDKalman");
  
  // TFile* file = new TFile(filename,"UPDATE");
  // if(file->cd("Kalman")) file->Delete("Kalman;*");
  // file->mkdir("Kalman");
  // file->cd("Kalman");
  
  fPH->Write();
  delete fPH;
  fPH=NULL;
  
  fChi2H->Write();
  delete fChi2H;
  fChi2H=NULL;

  fXresH->Write();
  delete fXresH;
  fXresH=NULL;

  fYresH->Write();
  delete fYresH;
  fYresH=NULL;

  fXresFitH->Write();
  delete fXresFitH;
  fXresFitH=NULL;

  fYresFitH->Write();
  delete fYresFitH;
  fYresFitH=NULL;

 fPEnd->Write();
  delete fPEnd;
  fPEnd=NULL;

 fPull->Write();
  delete fPull;
  fPull=NULL;

  file->Close();delete file;

}

Member Data Documentation

TH1D* PndHypDKalmanTask::fChi2H

private

Definition at line 76 of file PndHypDKalmanTask.h.

Referenced by Exec(), Init(), WriteHistograms(), and ~PndHypDKalmanTask().

int PndHypDKalmanTask::fEvt

private

Definition at line 86 of file PndHypDKalmanTask.h.

Referenced by Exec().

FairField* PndHypDKalmanTask::fField

private

Definition at line 73 of file PndHypDKalmanTask.h.

std::map<unsigned int,TString> PndHypDKalmanTask::fHitBranchMap

private

Definition at line 67 of file PndHypDKalmanTask.h.

Referenced by AddHitBranch(), and Init().

TH1D* PndHypDKalmanTask::fPEnd

private

Definition at line 81 of file PndHypDKalmanTask.h.

Referenced by Exec(), Init(), and WriteHistograms().

Bool_t PndHypDKalmanTask::fPersistence

private

Definition at line 70 of file PndHypDKalmanTask.h.

Referenced by SetPersistence().

TH1D* PndHypDKalmanTask::fPH

private

Definition at line 75 of file PndHypDKalmanTask.h.

Referenced by Exec(), Init(), WriteHistograms(), and ~PndHypDKalmanTask().

TH1D* PndHypDKalmanTask::fPull

private

Definition at line 82 of file PndHypDKalmanTask.h.

Referenced by Exec(), Init(), and WriteHistograms().

Bool_t PndHypDKalmanTask::fSmooth

private

Definition at line 85 of file PndHypDKalmanTask.h.

Referenced by SetSmooth().

GFRecoHitFactory* PndHypDKalmanTask::fTheRecoHitFactory

private

Definition at line 72 of file PndHypDKalmanTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndHypDKalmanTask::fTrackArray

private

Definition at line 68 of file PndHypDKalmanTask.h.

Referenced by Exec(), and Init().

TString PndHypDKalmanTask::fTrackBranchName

private

Definition at line 66 of file PndHypDKalmanTask.h.

Referenced by Init(), PndHypDKalmanTask(), and SetTrackBranchName().

int PndHypDKalmanTask::fTrackcount

private

Definition at line 84 of file PndHypDKalmanTask.h.

Referenced by Exec().

bool PndHypDKalmanTask::fUseMVD

private

Definition at line 87 of file PndHypDKalmanTask.h.

Referenced by Init(), and UseMVD().

TH1D* PndHypDKalmanTask::fXresFitH

private

Definition at line 79 of file PndHypDKalmanTask.h.

Referenced by Init(), and WriteHistograms().

TH1D* PndHypDKalmanTask::fXresH

private

Definition at line 77 of file PndHypDKalmanTask.h.

Referenced by Init(), and WriteHistograms().

TH1D* PndHypDKalmanTask::fYresFitH

private

Definition at line 80 of file PndHypDKalmanTask.h.

Referenced by Init(), and WriteHistograms().

TH1D* PndHypDKalmanTask::fYresH

private

Definition at line 78 of file PndHypDKalmanTask.h.

Referenced by Init(), and WriteHistograms().

---

The documentation for this class was generated from the following files:

• PndHypDKalmanTask.h
• PndHypDKalmanTask.cxx