PndHypKalmanTask Class Reference

#include <PndHypKalmanTask.h>

Inheritance diagram for PndHypKalmanTask:

Public Member Functions
	PndHypKalmanTask ()
	~PndHypKalmanTask ()
void	SetTrackBranchName (const TString &name)
void	AddHitBranch (unsigned int detId, const TString &m)
void	SetPersistence (Bool_t opt=kTRUE)
virtual InitStatus	Init ()
virtual void	Exec (Option_t *opt)
void	WriteHistograms (const TString &filename)
	ClassDef (PndHypKalmanTask, 2)

Private Attributes
TClonesArray *	fTrackArray
TClonesArray *	fTrArray
TString	fTrackBranchName
std::map< unsigned int, TString >	fHitBranchMap
Bool_t	fPersistence
GFRecoHitFactory *	fTheRecoHitFactory
FairField *	fField
TH1D *	fPH
TH1D *	fChi2H
TH1D *	fMassV0
TH1D *	fMassETAC
TH1D *	fMasses
Int_t	fTrackcount
Int_t	fEvent
TGeant3 *	gMC3
FairGeanePro *	fPro

Detailed Description

Definition at line 35 of file PndHypKalmanTask.h.

Constructor & Destructor Documentation

PndHypKalmanTask::PndHypKalmanTask

(

)

Definition at line 67 of file PndHypKalmanTask.cxx.

References fTrackBranchName.

  : FairTask("Kalman Filter"), fPersistence(kFALSE)
{
  fTrackBranchName = "HypTrackCand";
  //PndGeoHandling::Instance();
}

PndHypKalmanTask::~PndHypKalmanTask

(

)

Definition at line 75 of file PndHypKalmanTask.cxx.

References fChi2H, and fPH.

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

Member Function Documentation

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

inline

Definition at line 50 of file PndHypKalmanTask.h.

References fHitBranchMap, and m.

{fHitBranchMap[detId]=m;};

PndHypKalmanTask::ClassDef	(	PndHypKalmanTask	,
		2
	)

void PndHypKalmanTask::Exec ( Option_t *  opt )

virtual

Definition at line 154 of file PndHypKalmanTask.cxx.

References b, GFRecoHitFactory::createMany(), Double_t, fb, fChi2H, fField, fParticle, fPH, fTheRecoHitFactory, fTrackArray, fTrArray, GFTrackCand::getDirSeed(), GFTrackCand::getNHits(), GFTrackCand::getPdgCode(), GFTrackCand::getPosSeed(), GFTrackCand::getQoverPseed(), p, GFKalman::processTrack(), GFTrack::setCandidate(), trk, and GFException::what().

{
  std::cout<<"PndHypKalmanTask::Exec"<<std::endl;
  // Reset output Array
  if(fTrArray==0) Fatal("Kalman::Exec)","No TrackArray");
   fTrArray->Delete();

  Int_t ntracks=fTrackArray->GetEntriesFast();
 // Detailed output
 
 
  
  if(ntracks>20){
    std::cout<<"ntracks="<<ntracks<<" Evil Event! skipping"<<std::endl;
    return;
  }

  // Fitting ---------------- can go to another task!
  GFKalman fitter;

  // std::vector<TLorentzVector*> particles;
  //std::vector<Int_t> signs;
  
  for(Int_t itr=0;itr<ntracks;++itr){

    std::cout<<"starting track"<<itr<<std::endl;
    GFTrackCand* trcnd = (GFTrackCand*)fTrackArray->At(itr);
    unsigned int detid=12345, index=12345;
    
    // setting the magnetic field properly
     fField= FairRunAna::Instance()->GetField();
     PndFieldAdaptor* b= new PndFieldAdaptor(fField);
     GFFieldManager * fb;
     //fb->getInstance();
     //fb->init(b);

     GFAbsTrackRep* rep = 0;
     
     std::cout<<trcnd->getNHits()<<std::endl;
   // Starting values for guessing

    Int_t PDGCode= trcnd->getPdgCode();//2212;
    TVector3 StartPos    = trcnd->getPosSeed();//TVector3 (1.0,0.0,0.0);//cmn
    TVector3 StartPosErr = TVector3(0,0,0);
    TVector3 StartMom    = trcnd->getDirSeed();//TVector3 (1.,0.,1.);
    //StartMom.SetMagThetaPhi(1.05 , 70.*TMath::Pi()/360. , 0.);
    //   StartMom.SetMag(1.1);
    TVector3 StartMomErr = TVector3(0,0,0);
    TDatabasePDG *fdbPDG= TDatabasePDG::Instance();
    TParticlePDG *fParticle= fdbPDG->GetParticle(PDGCode);
    Double_t  fCharge= fParticle->Charge();
     // calc momentum projections
       TVector3 dir=StartMom.Unit();
       double dxdz=dir.X()/dir.Z();
       double dydz=dir.Y()/dir.Z();
    
       rep=new LSLTrackRep(StartPos.Z(),StartPos.X(),StartPos.Y(),dxdz,dydz,trcnd->getQoverPseed(),
                           StartPosErr.X(),StartPosErr.Y(),0.1,0.1,0.1,b);//NULL instead of b field

    // what to guess here?
       /* TVector3 U(1.,0.,0.);
    TVector3 V(0.,1.,0.);
    
    GFDetPlane start_pl(StartPos,U,V);
    GFAbsTrackRep* rep = new GeaneTrackRep(fPro,
    start_pl,StartMom,
                                         StartPosErr,StartMomErr,
                                         fCharge,PDGCode);*/
   
       //GFTrack* trk= new GFTrack(rep);
       GFTrack* trk=new((*fTrArray)[fTrArray->GetEntriesFast()]) GFTrack(rep);

    if(trk!=NULL)trk->setCandidate(*(GFTrackCand*)fTrackArray->At(itr));
   
    else {
      std::cout<<" "<< "caca de vaca "<<std::endl;
      continue;}
    
    //std::cout<<" "<< trk->getTrackRep(0)->Print()<<std::endl;
    //cout<<" "<<trk->Print()<<endl;

    //GFTrack* trk=(GFTrack*)fTrackArray->At(itr);
    // Load RecoHits 
        try {
          trk->addHitVector(fTheRecoHitFactory->createMany(*trcnd));
         
    }
        catch(GFException& e) {
          std::cout << e.what();
          throw e;
        }
    // Start Fitter
    try{
      std::cout<<"starting fit"<<std::endl;
      fitter.processTrack(trk);
    }
    catch (GFException e){
      std::cout<<e.what()<<std::endl;
    }

    // Print Track Parameters after fit
    if(trk->getTrackRep(0)->getStatusFlag()==0){
      //trk->getTrackRep(0)->Print();
      GFDetPlane plane(TVector3(0,0,0.1),TVector3(1,0,0),TVector3(0,1,0));
      TVector3 p3=trk->getTrackRep(0)->getMom(plane);
      Double_t p=trk->getMom().Mag();
      std::cout<<" oye"<<p<<std::endl;
      
      fPH->Fill(p);
      
    

      Double_t chi2=trk->getChiSqu();
      fChi2H->Fill(chi2);

    }

  }

  std::cout<<"Fitting done"<<std::endl;



  return;
}

InitStatus PndHypKalmanTask::Init ( )

virtual

Definition at line 82 of file PndHypKalmanTask.cxx.

References GFRecoHitFactory::addProducer(), fChi2H, fHitBranchMap, fPersistence, fPH, fTheRecoHitFactory, fTrackArray, fTrackBranchName, and fTrArray.

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

  if(ioman==0)
    {
      Error("PndHypKalmanTask::Init","RootManager not instantiated!");
      return kERROR;
    }
  
  // Get input collection
  fTrackArray=(TClonesArray*) ioman->GetObject(fTrackBranchName);
  
  if(fTrackArray==0)
    {
      Error("PndHypKalmanTask::Init","track-array not found!");
      return kERROR;
    }
  

  // 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));
     
    }

    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));
    }

   ++iter;
  }//end loops over hit types

// create and register output array
  fTrArray = new TClonesArray("GFTrack"); 
  ioman->Register("Track","GenFit",fTrArray,fPersistence);

   //fPro = new FairGeanePro();

  // setup histograms
  fPH=new TH1D("pH","p",100,0.4,0.6);
  fChi2H=new TH1D("chi2H","chi2",100,0,20);
  

  return kSUCCESS;
}

void PndHypKalmanTask::SetPersistence ( Bool_t  opt = kTRUE )

inline

Definition at line 51 of file PndHypKalmanTask.h.

References fPersistence.

{fPersistence=opt;}

void PndHypKalmanTask::SetTrackBranchName ( const TString &  name )

inline

Definition at line 49 of file PndHypKalmanTask.h.

References fTrackBranchName, and name.

{fTrackBranchName=name;}

void PndHypKalmanTask::WriteHistograms

(

const TString &

filename

)

Definition at line 281 of file PndHypKalmanTask.cxx.

References fChi2H, file, and fPH.

                                                        {
  TFile* file = new TFile(filename,"UPDATE");
//   if(file->cd("Kalman")==false) file->mkdir("Kalman");
//   file->cd("Kalman");

  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;

  

  file->Close();
  delete file;
}

Member Data Documentation

TH1D* PndHypKalmanTask::fChi2H

private

Definition at line 75 of file PndHypKalmanTask.h.

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

Int_t PndHypKalmanTask::fEvent

private

Definition at line 81 of file PndHypKalmanTask.h.

FairField* PndHypKalmanTask::fField

private

Definition at line 72 of file PndHypKalmanTask.h.

Referenced by Exec().

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

private

Definition at line 68 of file PndHypKalmanTask.h.

Referenced by AddHitBranch(), and Init().

TH1D* PndHypKalmanTask::fMasses

private

Definition at line 78 of file PndHypKalmanTask.h.

TH1D* PndHypKalmanTask::fMassETAC

private

Definition at line 77 of file PndHypKalmanTask.h.

TH1D* PndHypKalmanTask::fMassV0

private

Definition at line 76 of file PndHypKalmanTask.h.

Bool_t PndHypKalmanTask::fPersistence

private

Definition at line 69 of file PndHypKalmanTask.h.

Referenced by Init(), and SetPersistence().

TH1D* PndHypKalmanTask::fPH

private

Definition at line 74 of file PndHypKalmanTask.h.

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

FairGeanePro* PndHypKalmanTask::fPro

private

Definition at line 83 of file PndHypKalmanTask.h.

GFRecoHitFactory* PndHypKalmanTask::fTheRecoHitFactory

private

Definition at line 71 of file PndHypKalmanTask.h.

Referenced by Exec(), and Init().

TClonesArray* PndHypKalmanTask::fTrackArray

private

Definition at line 65 of file PndHypKalmanTask.h.

Referenced by Exec(), and Init().

TString PndHypKalmanTask::fTrackBranchName

private

Definition at line 67 of file PndHypKalmanTask.h.

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

Int_t PndHypKalmanTask::fTrackcount

private

Definition at line 80 of file PndHypKalmanTask.h.

TClonesArray* PndHypKalmanTask::fTrArray

private

Definition at line 66 of file PndHypKalmanTask.h.

Referenced by Exec(), and Init().

TGeant3* PndHypKalmanTask::gMC3

private

Definition at line 82 of file PndHypKalmanTask.h.

---

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

• PndHypKalmanTask.h
• PndHypKalmanTask.cxx