Fork me on GitHub

source: git/classes/DelphesHepMC3Reader.h@ 0799cd1

Last change on this file since 0799cd1 was a26a130, checked in by Pavel Demin <pavel.demin@…>, 3 years ago

move arrays to FinalizeParticles in DelphesHepMC3Reader

  • Property mode set to 100644
File size: 2.8 KB
Line 
1/*
2 * Delphes: a framework for fast simulation of a generic collider experiment
3 * Copyright (C) 2012-2021 Universite catholique de Louvain (UCL), Belgium
4 *
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation, either version 3 of the License, or
8 * (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 */
18
19#ifndef DelphesHepMC3Reader_h
20#define DelphesHepMC3Reader_h
21
22/** \class DelphesHepMC3Reader
23 *
24 * Reads HepMC file
25 *
26 * \author P. Demin - UCL, Louvain-la-Neuve
27 *
28 */
29
30#include <map>
31#include <vector>
32
33#include <stdio.h>
34
35class TObjArray;
36class TStopwatch;
37class TDatabasePDG;
38class TLorentzVector;
39class ExRootTreeBranch;
40class DelphesFactory;
41class Candidate;
42
43class DelphesHepMC3Reader
44{
45public:
46 DelphesHepMC3Reader();
47 ~DelphesHepMC3Reader();
48
49 void SetInputFile(FILE *inputFile);
50
51 void Clear();
52 bool EventReady();
53
54 bool ReadBlock(DelphesFactory *factory,
55 TObjArray *allParticleOutputArray,
56 TObjArray *stableParticleOutputArray,
57 TObjArray *partonOutputArray);
58
59 void AnalyzeEvent(ExRootTreeBranch *branch, long long eventNumber,
60 TStopwatch *readStopWatch, TStopwatch *procStopWatch);
61
62 void AnalyzeWeight(ExRootTreeBranch *branch);
63
64private:
65 void AnalyzeVertex(DelphesFactory *factory, int code, Candidate *candidate = 0);
66
67 void AnalyzeParticle(DelphesFactory *factory);
68
69 void FinalizeParticles(TObjArray *allParticleOutputArray,
70 TObjArray *stableParticleOutputArray,
71 TObjArray *partonOutputArray);
72
73 FILE *fInputFile;
74
75 char *fBuffer;
76
77 TDatabasePDG *fPDG;
78
79 int fEventNumber, fMPI, fProcessID, fSignalCode, fVertexCounter, fParticleCounter;
80 double fScale, fAlphaQCD, fAlphaQED;
81
82 double fMomentumCoefficient, fPositionCoefficient;
83
84 std::vector<double> fWeights;
85
86 double fCrossSection, fCrossSectionError;
87
88 int fID1, fID2;
89 double fX1, fX2, fScalePDF, fPDF1, fPDF2;
90
91 int fVertexCode, fVertexStatus;
92 double fX, fY, fZ, fT;
93
94 int fParticleCode, fPID, fParticleStatus, fOutVertexCode;
95 double fPx, fPy, fPz, fE, fMass;
96
97 std::vector<std::pair<TLorentzVector *, TObjArray *> > fVertices;
98 std::vector<int> fParticles;
99
100 std::map<int, int> fInVertexMap;
101 std::map<int, int> fOutVertexMap;
102
103 std::map<int, std::pair<int, int> > fMotherMap;
104 std::map<int, std::pair<int, int> > fDaughterMap;
105};
106
107#endif // DelphesHepMC3Reader_h
Note: See TracBrowser for help on using the repository browser.