| [0e2f49b] | 1 | #ifndef VertexFinder_h
|
|---|
| 2 | #define VertexFinder_h
|
|---|
| 3 |
|
|---|
| 4 | /** \class VertexFinder
|
|---|
| 5 | *
|
|---|
| [2600216] | 6 | * Cluster vertices from tracks
|
|---|
| [0e2f49b] | 7 | *
|
|---|
| 8 | * \author M. Selvaggi - UCL, Louvain-la-Neuve
|
|---|
| 9 | *
|
|---|
| 10 | */
|
|---|
| 11 |
|
|---|
| 12 | #include "classes/DelphesModule.h"
|
|---|
| 13 | #include "classes/DelphesClasses.h"
|
|---|
| 14 | #include <utility>
|
|---|
| 15 | #include <algorithm>
|
|---|
| 16 | #include <stdexcept>
|
|---|
| 17 | #include <iostream>
|
|---|
| 18 | #include <sstream>
|
|---|
| 19 | #include <vector>
|
|---|
| 20 |
|
|---|
| 21 | using namespace std;
|
|---|
| 22 |
|
|---|
| 23 | class TObjArray;
|
|---|
| 24 | class Candidate;
|
|---|
| 25 | class TVector3;
|
|---|
| 26 |
|
|---|
| 27 | class VertexFinder: public DelphesModule
|
|---|
| 28 | {
|
|---|
| 29 | public:
|
|---|
| 30 |
|
|---|
| 31 | VertexFinder();
|
|---|
| 32 | ~VertexFinder();
|
|---|
| 33 |
|
|---|
| 34 | void Init();
|
|---|
| 35 | void Process();
|
|---|
| 36 | void Finish();
|
|---|
| 37 |
|
|---|
| 38 | static bool secondDescending (pair<unsigned, double>, pair<unsigned, double>);
|
|---|
| 39 | static bool secondAscending (pair<unsigned, double>, pair<unsigned, double>);
|
|---|
| 40 |
|
|---|
| 41 | private:
|
|---|
| 42 |
|
|---|
| 43 | void createSeeds ();
|
|---|
| 44 | void growCluster (const unsigned);
|
|---|
| 45 | double weight (const unsigned);
|
|---|
| 46 | void addTrackToCluster (const unsigned, const unsigned);
|
|---|
| 47 | void removeTrackFromCluster (const unsigned, const unsigned);
|
|---|
| 48 |
|
|---|
| 49 | Double_t fSigma;
|
|---|
| 50 | Double_t fMinPT;
|
|---|
| 51 | Double_t fMaxEta;
|
|---|
| 52 | Double_t fSeedMinPT;
|
|---|
| 53 | Int_t fMinNDF;
|
|---|
| 54 | Int_t fGrowSeeds;
|
|---|
| 55 |
|
|---|
| 56 | TObjArray *fInputArray;
|
|---|
| 57 | TIterator *fItInputArray;
|
|---|
| 58 |
|
|---|
| 59 | TObjArray *fOutputArray;
|
|---|
| [2600216] | 60 | TObjArray *fVertexOutputArray;
|
|---|
| [0e2f49b] | 61 |
|
|---|
| 62 | map<unsigned, map<string, double> > trackIDToDouble;
|
|---|
| 63 | map<unsigned, map<string, int> > trackIDToInt;
|
|---|
| 64 | map<unsigned, map<string, bool> > trackIDToBool;
|
|---|
| 65 |
|
|---|
| 66 | map<unsigned, map<string, double> > clusterIDToDouble;
|
|---|
| 67 | map<unsigned, map<string, int> > clusterIDToInt;
|
|---|
| 68 | map<unsigned, map<string, bool> > clusterIDToBool;
|
|---|
| 69 | vector<pair<unsigned, double> > trackPT;
|
|---|
| 70 | vector<pair<unsigned, double> > clusterSumPT2;
|
|---|
| 71 |
|
|---|
| 72 | ClassDef(VertexFinder, 1)
|
|---|
| 73 | };
|
|---|
| 74 |
|
|---|
| 75 | #endif
|
|---|
