#include <PndPatternDBMerger.h>

Public Member Functions
	PndPatternDBMerger ()

virtual	~PndPatternDBMerger ()

void	SetInputFilename (TString filename)

void	SetInputTreename (TString treename)

void	SetOutputFilename (TString filename)

void	SetOutputTreename (TString treename)

void	Execute ()

Private Member Functions
bool	InitInputChain ()

void	InitOutputTree ()

IdenticalEntries	FindIdenticalEntries ()

std::vector< PndPattern >	GetPatternsForMerging (std::pair< int, std::set< int >> entryPair)

PndPattern	MergePatterns (std::vector< PndPattern > &entries)

void	FillOutputTree (PndPattern &mergedPattern)

Private Attributes
TChain *	fpatternTree

TFile *	foutputDBFile

TTree *	foutputTree

TString	finputFilename

TString	finputTreename

TString	foutputFilename

TString	foutputTreename

PndPattern *	bPattern

PndPattern *	bMergedPattern

Detailed Description

Definition at line 19 of file PndPatternDBMerger.h.

Constructor & Destructor Documentation

PndPatternDBMerger::PndPatternDBMerger ( )

Definition at line 16 of file PndPatternDBMerger.cxx.

                                        {
   fpatternTree = NULL;
   foutputDBFile = NULL;
   foutputTree = NULL;
 
   bPattern = NULL;
   bMergedPattern = NULL;
 
   finputFilename = "patternDB.root";
   finputTreename = "trackPatterns";
   foutputFilename = "patternDBclean.root";
   foutputTreename = finputTreename;
 }

PndPatternDBMerger::~PndPatternDBMerger ( )

virtual

Definition at line 30 of file PndPatternDBMerger.cxx.

30 {

31 }

Member Function Documentation

void PndPatternDBMerger::Execute ( )

Definition at line 33 of file PndPatternDBMerger.cxx.

References FillOutputTree(), FindIdenticalEntries(), foutputDBFile, foutputTree, GetPatternsForMerging(), InitInputChain(), InitOutputTree(), and MergePatterns().

Referenced by mergePatterns().

                                  {
   std::cout << "Executing PndPatternDBMerger" << std::endl;
 
   bool bInput = InitInputChain();
   if (bInput) {
     std::cout << "Input chain successfully initialised" << std::endl;
   }
 
   InitOutputTree();
 
   IdenticalEntries entriesToBeMerged;
   entriesToBeMerged = FindIdenticalEntries();
 
   std::cout << "Merged entries to be created: " << entriesToBeMerged.size() << std::endl;
   for (auto iter: boost::adaptors::reverse(entriesToBeMerged)) {
     std::vector<PndPattern> patternsToBeMerged = GetPatternsForMerging(iter);
     PndPattern mergedPattern = MergePatterns(patternsToBeMerged);
     FillOutputTree(mergedPattern);
   }
 
   std::cout << "Saving output tree" << std::endl;
   foutputTree->AutoSave();
   foutputDBFile->Close();
 }

void PndPatternDBMerger::FillOutputTree ( PndPattern & mergedPattern )

private

Definition at line 131 of file PndPatternDBMerger.cxx.

References bMergedPattern, and foutputTree.

Referenced by Execute().

                                                                  {
   bMergedPattern = &mergedPattern;
   foutputTree->Fill();
 }

IdenticalEntries PndPatternDBMerger::FindIdenticalEntries ( )

private

Definition at line 75 of file PndPatternDBMerger.cxx.

References bPattern, fpatternTree, and PndPattern::GetTubeIDs().

Referenced by Execute().

                                                           {
   std::cout << "Finding identical patterns in input chain" << std::endl;
 
 //  pattern, entries in file for corresponding pattern
   typedef std::map<std::set<int>, std::set<int>> PatternEntries;
   PatternEntries patternEntries;
   Long64_t nEntries = fpatternTree->GetEntries();
   std::cout << "Entries to process: " << nEntries << std::endl;
 
   for (Long64_t iEntry = 0; iEntry < nEntries; ++iEntry) {
     if (iEntry % 10000 == 0) {
       std::cout << "processing entry " << iEntry << " of " << nEntries << std::endl;
     }
     fpatternTree->GetEntry(iEntry);
     std::set<int> tubeIDs = bPattern->GetTubeIDs();
     patternEntries[tubeIDs].insert(iEntry);
   }
 
   std::cout << "Unique patterns found in input chain: " << patternEntries.size() << std::endl;
 
   std::cout << "Creating map with entries to be merged" << std::endl;
 //  number of entries, actual entries in file
   IdenticalEntries mapEntries;
   for (auto iter: patternEntries) {
     std::set<int> &vEntries = iter.second;
 //    set this to 1 to only consider recurring patterns
     if (vEntries.size() > 0) {
       mapEntries.insert(std::make_pair(vEntries.size(), vEntries));
     }
   }
 
   return mapEntries;
 }

std::vector< PndPattern > PndPatternDBMerger::GetPatternsForMerging ( std::pair< int, std::set< int >> entryPair )

private

Definition at line 108 of file PndPatternDBMerger.cxx.

References bPattern, and fpatternTree.

Referenced by Execute().

                                                                                                    {
   std::vector<PndPattern> patternsToBeMerged;
   for (auto const& entry: entryPair.second) {
     fpatternTree->GetEntry(entry);
     patternsToBeMerged.push_back(*bPattern);
   }
   return patternsToBeMerged;
 }

bool PndPatternDBMerger::InitInputChain ( )

private

Definition at line 58 of file PndPatternDBMerger.cxx.

References bPattern, finputFilename, finputTreename, and fpatternTree.

Referenced by Execute().

                                         {
   std::cout << "Initialising input chain with TTree" << finputTreename << std::endl;
   fpatternTree = new TChain(finputTreename);
   std::cout << "Adding input file(s) with name" << finputFilename << std::endl;
   fpatternTree->Add(finputFilename);
 
   fpatternTree->SetBranchAddress("pattern",&bPattern);
 
   if (!fpatternTree) return false;
   return true;
 }

void PndPatternDBMerger::InitOutputTree ( )

private

Definition at line 69 of file PndPatternDBMerger.cxx.

References bMergedPattern, foutputDBFile, foutputFilename, foutputTree, and foutputTreename.

Referenced by Execute().

                                         {
   std::cout << "Initialising output file and tree" << std::endl;
   foutputDBFile = new TFile(foutputFilename,"RECREATE");
   foutputTree = new TTree(foutputTreename,"Tree containing merged patterns");
   foutputTree->Branch("pattern",&bMergedPattern);
 }

PndPattern PndPatternDBMerger::MergePatterns ( std::vector< PndPattern > & entries )

private

Definition at line 116 of file PndPatternDBMerger.cxx.

References PndPattern::AddMomenta(), PndPattern::GetCount(), PndPattern::SetPatternCount(), PndPattern::SetSectorID(), and PndPattern::SetTubeIDs().

Referenced by Execute().

                                                                            {
   PndPattern merged;
   if (entries.size() == 0) return merged;
   merged.SetTubeIDs(entries.at(0).GetTubeIDs());
   merged.SetSectorID(entries.at(0).GetSectorID());
 
   for (auto const& pattern: entries) {
     int currentCount = merged.GetCount();
     int newCount = currentCount + pattern.GetCount();
     merged.SetPatternCount(newCount);
     std::vector<TVector3> momenta = pattern.GetMomenta();
     merged.AddMomenta(momenta);
   }
   return merged;
 }