PndPatternDBMerger.h

Go to the documentation of this file.

/*
 * PndPatternDBMerger.h
 *
 *  Created on: Nov 15, 2017
 *      Author: vagrant
 */

#ifndef PNDTRACKERS_PNDPATTERNMATCHER_PNDPATTERNDBMERGER_H_
#define PNDTRACKERS_PNDPATTERNMATCHER_PNDPATTERNDBMERGER_H_
#include <PndPattern.h>
#include <TChain.h>
#include <TFile.h>
#include <TString.h>
#include <TTree.h>

// number of identical entries, actual entries in file
typedef std::multimap<int, std::set<int>> IdenticalEntries;

class PndPatternDBMerger {
public:
  PndPatternDBMerger();
  virtual ~PndPatternDBMerger();

  void SetInputFilename(TString filename) {finputFilename = filename;}
  void SetInputTreename(TString treename) {finputTreename = treename;}
  void SetOutputFilename(TString filename) {foutputFilename = filename;}
  void SetOutputTreename(TString treename) {foutputTreename = treename;}

  void Execute();

private:
  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);

  TChain *fpatternTree;
  TFile *foutputDBFile;
  TTree *foutputTree;

  TString finputFilename;
  TString finputTreename;
  TString foutputFilename;
  TString foutputTreename;

  PndPattern *bPattern;
  PndPattern *bMergedPattern;


  ClassDef(PndPatternDBMerger,1)
};

#endif /* PNDTRACKERS_PNDPATTERNMATCHER_PNDPATTERNDBMERGER_H_ */