PndPatternDBGenerator Class Reference

#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

(

)

Definition at line 19 of file PndPatternDBGenerator.cxx.

                                             {
  fSttParameters = NULL;
  fEventHeader = NULL;
  fTubeArray = NULL;
  fSttHitArray = NULL;
  fMCTrackArray = NULL;
  trackCands = NULL;
  sttBranchID = -1;
  mcTrackID = -1;

  foutputFile = NULL;
  fsectorPatternTree = NULL;
  ftrackPatternTree = NULL;
  bPattern = NULL;
  foutputFilename = "patternDB.root";

  nTotalTracks = 0;
  nMultipleMCTrackLinks = 0;
}

PndPatternDBGenerator::~PndPatternDBGenerator ( )

virtual

Definition at line 39 of file PndPatternDBGenerator.cxx.

                                              {
}

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.

{foutputFilename = filename;}

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

Member Data Documentation

PndPattern* PndPatternDBGenerator::bPattern

private

Definition at line 52 of file PndPatternDBGenerator.h.

Referenced by AddPatternsToTree(), and Init().

TClonesArray* PndPatternDBGenerator::fEventHeader

private

Definition at line 42 of file PndPatternDBGenerator.h.

Referenced by Init().

TClonesArray* PndPatternDBGenerator::fMCTrackArray

private

Definition at line 45 of file PndPatternDBGenerator.h.

Referenced by Init().

TFile* PndPatternDBGenerator::foutputFile

private

Definition at line 48 of file PndPatternDBGenerator.h.

Referenced by FinishTask(), and Init().

TString PndPatternDBGenerator::foutputFilename

private

Definition at line 56 of file PndPatternDBGenerator.h.

Referenced by Init(), and SetOutputFilename().

TTree* PndPatternDBGenerator::fsectorPatternTree

private

Definition at line 49 of file PndPatternDBGenerator.h.

Referenced by FinishTask(), GenerateTrackPatterns(), and Init().

TClonesArray* PndPatternDBGenerator::fSttHitArray

private

Definition at line 44 of file PndPatternDBGenerator.h.

Referenced by Init().

PndGeoSttPar* PndPatternDBGenerator::fSttParameters

private

Definition at line 41 of file PndPatternDBGenerator.h.

Referenced by Init(), and SetParContainers().

TTree* PndPatternDBGenerator::ftrackPatternTree

private

Definition at line 50 of file PndPatternDBGenerator.h.

Referenced by FinishTask(), GenerateTrackPatterns(), and Init().

TClonesArray* PndPatternDBGenerator::fTubeArray

private

Definition at line 43 of file PndPatternDBGenerator.h.

Referenced by FillSectorHitMap(), and Init().

int PndPatternDBGenerator::mcTrackID

private

Definition at line 55 of file PndPatternDBGenerator.h.

Referenced by GenerateTrackPatterns(), and Init().

int PndPatternDBGenerator::nMultipleMCTrackLinks

private

Definition at line 59 of file PndPatternDBGenerator.h.

Referenced by FinishTask(), and GenerateTrackPatterns().

int PndPatternDBGenerator::nTotalTracks

private

Definition at line 58 of file PndPatternDBGenerator.h.

Referenced by FinishTask(), and GenerateTrackPatterns().

int PndPatternDBGenerator::sttBranchID

private

Definition at line 54 of file PndPatternDBGenerator.h.

Referenced by GenerateTrackPatterns(), and Init().

TClonesArray* PndPatternDBGenerator::trackCands

private

Definition at line 46 of file PndPatternDBGenerator.h.

Referenced by GenerateTrackPatterns(), and Init().

---

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

• PndPatternDBGenerator.h
• PndPatternDBGenerator.cxx