#include <PndPatternDBGenerator.h>

Inheritance diagram for PndPatternDBGenerator:

Public Member Functions
	PndPatternDBGenerator ()

virtual	~PndPatternDBGenerator ()

void	SetOutputFilename (TString filename="patternDB.root")

Protected Member Functions
virtual void	SetParContainers ()

virtual InitStatus	Init ()

virtual void	Exec (Option_t *opt)

virtual void	FinishTask ()

Private Member Functions
void	GenerateTrackPatterns ()

HitSectorMap	FillSectorHitMap (HitArray sttHitArray)

PndPatterns	FillSectorPatterns (HitArray hitArray, PndMCTrack *mcTrack)

PndPatterns	FillTrackPatterns (HitArray hitArray, PndMCTrack *mcTrack)

void	AddPatternsToTree (PndPatterns patterns, TTree *tree)

Private Attributes
PndGeoSttPar *	fSttParameters

TClonesArray *	fEventHeader

TClonesArray *	fTubeArray

TClonesArray *	fSttHitArray

TClonesArray *	fMCTrackArray

TClonesArray *	trackCands

TFile *	foutputFile

TTree *	fsectorPatternTree

TTree *	ftrackPatternTree

PndPattern *	bPattern

int	sttBranchID

int	mcTrackID

TString	foutputFilename

int	nTotalTracks

int	nMultipleMCTrackLinks

Detailed Description

Definition at line 21 of file PndPatternDBGenerator.h.

Constructor & Destructor Documentation

PndPatternDBGenerator::PndPatternDBGenerator ( )

PndPatternDBGenerator::~PndPatternDBGenerator ( )

virtual

Definition at line 39 of file PndPatternDBGenerator.cxx.

39 {

40 }

Member Function Documentation

void PndPatternDBGenerator::AddPatternsToTree	(	PndPatterns	patterns,
		TTree *	tree
	)

private

Definition at line 191 of file PndPatternDBGenerator.cxx.

References bPattern.

Referenced by GenerateTrackPatterns().

                                                                                {
   for (auto pattern: patterns) {
     bPattern = &pattern;
     tree->Fill();
   }
 }

void PndPatternDBGenerator::Exec ( Option_t * opt )

protectedvirtual

Definition at line 86 of file PndPatternDBGenerator.cxx.

References GenerateTrackPatterns().

                                             { //[R.K. 9/2018] unused
   GenerateTrackPatterns();
 }

HitSectorMap PndPatternDBGenerator::FillSectorHitMap ( HitArray sttHitArray )

private

Definition at line 143 of file PndPatternDBGenerator.cxx.

References fTubeArray, and PndSttTube::GetSectorID().

Referenced by FillSectorPatterns().

                                                                          {
   HitSectorMap hitMap;
   for (auto const& sttHit : sttHitArray) {
     short tubeID = sttHit->GetTubeID();
     PndSttTube *tube = (PndSttTube*) fTubeArray->At(tubeID);
     short sectorID = tube->GetSectorID();
     hitMap.insert(std::make_pair(sectorID, tubeID));
   }
   return hitMap;
 }

PndPatterns PndPatternDBGenerator::FillSectorPatterns	(	HitArray	hitArray,
		PndMCTrack *	mcTrack
	)

private

Definition at line 153 of file PndPatternDBGenerator.cxx.

References PndPattern::AddMomentum(), PndPattern::AddTubeID(), FillSectorHitMap(), PndMCTrack::GetMomentum(), PndPattern::IsEmpty(), PndPattern::RaisePatternCount(), and PndPattern::SetSectorID().

Referenced by GenerateTrackPatterns().

                                                                                                {
   PndPatterns patterns;
   HitSectorMap hitMap = FillSectorHitMap(sttHitArray);
   TVector3 momentum = mcTrack->GetMomentum();
   for (int iSector = 0; iSector < 6; ++iSector) {
     auto hitMapSectorRange = hitMap.equal_range(iSector);
     PndPattern pattern;
     pattern.SetSectorID(iSector);
     pattern.AddMomentum(momentum);
     for (auto iter = hitMapSectorRange.first; iter != hitMapSectorRange.second; ++iter) {
       int tubeID = iter->second;
       pattern.AddTubeID(tubeID);
     }
     pattern.RaisePatternCount();
     if (!pattern.IsEmpty()) {
       patterns.push_back(pattern);
     }
   }
 
   return patterns;
 }

PndPatterns PndPatternDBGenerator::FillTrackPatterns	(	HitArray	hitArray,
		PndMCTrack *	mcTrack
	)

private

Definition at line 174 of file PndPatternDBGenerator.cxx.

References PndPattern::AddMomentum(), PndPattern::AddTubeID(), PndMCTrack::GetMomentum(), PndPattern::IsEmpty(), PndPattern::RaisePatternCount(), and PndPattern::SetSectorID().

Referenced by GenerateTrackPatterns().

                                                                                               {
   PndPatterns patterns;
   TVector3 momentum = mcTrack->GetMomentum();
   PndPattern pattern;
   pattern.SetSectorID(-1); // sectorID undefined for whole track patterns
   pattern.AddMomentum(momentum);
   for (auto const& sttHit: sttHitArray) {
     int tubeID = sttHit->GetTubeID();
     pattern.AddTubeID(tubeID);
   }
   pattern.RaisePatternCount();
   if (!pattern.IsEmpty()) {
     patterns.push_back(pattern);
   }
 
   return patterns;
 }

void PndPatternDBGenerator::FinishTask ( )

protectedvirtual

Definition at line 89 of file PndPatternDBGenerator.cxx.

References foutputFile, fsectorPatternTree, ftrackPatternTree, nMultipleMCTrackLinks, and nTotalTracks.

                                        {
   fsectorPatternTree->AutoSave();
   ftrackPatternTree->AutoSave();
   foutputFile->Close();
 
   std::cout << "Total number of tracks found: " << nTotalTracks << std::endl;
   std::cout << "Number of track with multiple MCTrack links: " << nMultipleMCTrackLinks << std::endl;
   float ratio = 100. * (float)nMultipleMCTrackLinks / (float)nTotalTracks;
   std::cout << "Ratio: " << ratio << "%" << std::endl;
 }

void PndPatternDBGenerator::GenerateTrackPatterns ( )

private

Definition at line 100 of file PndPatternDBGenerator.cxx.

References AddPatternsToTree(), FillSectorPatterns(), FillTrackPatterns(), fsectorPatternTree, ftrackPatternTree, mcTrackID, nMultipleMCTrackLinks, nTotalTracks, sttBranchID, and trackCands.

Referenced by Exec().

                                                   {
 //  This method generates hit patterns for each track
 
 //  Get FairRootManager instance to access objects through FairLinks
   FairRootManager *ioman = FairRootManager::Instance();
 
 //  Loop over all track candidates
   short nTrackCands = trackCands->GetEntriesFast();
   for (int iTrackCand = 0; iTrackCand < nTrackCands; ++iTrackCand) {
 //    Get the next track candidate
     PndTrackCand *cand = (PndTrackCand*) trackCands->At(iTrackCand);
 //    Get the FairLinks from the candidate pointing at STTHit branch
     FairMultiLinkedData sttLinks = cand->GetLinksWithType(sttBranchID);
 //    Construct sttHitArray for each track candidate,
 //    taking into account all STTHits of the corresponding track
     HitArray sttHitArray;
     for (int iLink = 0; iLink < sttLinks.GetNLinks(); ++iLink) {
       FairLink sttLink = sttLinks.GetLink(iLink);
       PndSttHit *sttHit = (PndSttHit*) ioman->GetCloneOfLinkData(sttLink);
       sttHitArray.push_back(sttHit);
     }
 //    Get link to corresponding MC track
     FairMultiLinkedData mcTrackLinks = cand->GetLinksWithType(mcTrackID);
 //    Check if only one MC track was found
     nTotalTracks++;
     if (mcTrackLinks.GetNLinks() == 0) {
       std::cout << "WARNING: PndPatternDBGenerator::GenerateTrackPatterns: No MCTrackfound, skipping event!" << std::endl;
       continue;
     }
     if (mcTrackLinks.GetNLinks() > 1) {
       nMultipleMCTrackLinks++;
       std::cout << "WARNING: PndPatternDBGenerator::GenerateTrackPatterns: Found more than one MCTrack link!!!" << std::endl;
     }
 //    Get only first link since only one should be associated with track candidate
     FairLink mcTrackLink = mcTrackLinks.GetLink(0);
     PndMCTrack *mcTrack = (PndMCTrack*) ioman->GetCloneOfLinkData(mcTrackLink);
 
     PndPatterns sectorPatterns = FillSectorPatterns(sttHitArray, mcTrack);
     AddPatternsToTree(sectorPatterns,fsectorPatternTree);
     PndPatterns trackPatterns = FillTrackPatterns(sttHitArray, mcTrack);
     AddPatternsToTree(trackPatterns,ftrackPatternTree);
   }
 }

InitStatus PndPatternDBGenerator::Init ( )

protectedvirtual

Definition at line 46 of file PndPatternDBGenerator.cxx.

References bPattern, fEventHeader, PndSttMapCreator::FillTubeArray(), fMCTrackArray, foutputFile, foutputFilename, fsectorPatternTree, fSttHitArray, fSttParameters, ftrackPatternTree, fTubeArray, mcTrackID, sttBranchID, and trackCands.

                                        {
   std::cout << "Filename for pattern DB: " << foutputFilename << std::endl;
   foutputFile = new TFile(foutputFilename,"RECREATE");
   fsectorPatternTree = new TTree("sectorPatterns","Tree containing sector patterns");
   fsectorPatternTree->Branch("pattern",&bPattern);
   ftrackPatternTree = new TTree("trackPatterns","Tree containing track patterns");
   ftrackPatternTree->Branch("pattern",&bPattern);
 
 //  Get instance of the FairRootManager to access tree branches
   FairRootManager *ioman = FairRootManager::Instance();
   if(!ioman) {
     std::cout << "-E- PndPatternDBGenerator::Init: FairRootManager not instantiated!" << std::endl;
     return kFATAL;
   }
 
 //  Get the EventHeader
   fEventHeader = (TClonesArray*) ioman->GetObject("EventHeader.");
   if(!fEventHeader) {
     std::cout << "-E- PndPatternDBGenerator::Init: EventHeader not loaded!" << std::endl;
     return kFATAL;
   }
 
 //  Access STTHit branch and determine its branch ID
   fSttHitArray = (TClonesArray*) ioman->GetObject("STTHit");
   sttBranchID = ioman->GetBranchId("STTHit");
 
 //  Initialise STT map and get array of all tubes
   PndSttMapCreator *mapper = new PndSttMapCreator(fSttParameters);
   fTubeArray = mapper->FillTubeArray();
 
 //  Access MCTrack branch and determine its branch ID
   fMCTrackArray = (TClonesArray*) ioman->GetObject("MCTrack");
   mcTrackID = ioman->GetBranchId("MCTrack");
 
 //  Get track candidates from ideal track finder
   trackCands = (TClonesArray*) ioman->GetObject("SttMvdGemIdealTrackCand");
 
   std::cout << "-I- PndPatternDBGenerator: Initialisation successful" << std::endl;
   return kSUCCESS;
 }

void PndPatternDBGenerator::SetOutputFilename ( TString filename = "patternDB.root" )

inline

Definition at line 26 of file PndPatternDBGenerator.h.

References filename, and foutputFilename.

26 {foutputFilename = filename;}

PndPatternDBGenerator::foutputFilename

TString foutputFilename

Definition: PndPatternDBGenerator.h:56

filename

const string filename

Definition: Modify_root_geometry.C:23

void PndPatternDBGenerator::SetParContainers ( )

protectedvirtual

Definition at line 42 of file PndPatternDBGenerator.cxx.

References fSttParameters, and rtdb.

                                              {
   FairRuntimeDb *rtdb = FairRunAna::Instance()->GetRuntimeDb();
   fSttParameters = (PndGeoSttPar*) rtdb->getContainer("PndGeoSttPar");
 }