source: trunk/modules/MadGraphParticleClassifier.h@ 21

Last change on this file since 21 was 2, checked in by Pavel Demin, 16 years ago

first commit

File size: 823 bytes
Line 
1#ifndef MadGraphParticleClassifier_h
2#define MadGraphParticleClassifier_h
3
4#include "ExRootAnalysis/ExRootClassifier.h"
5
6#include <map>
7#include <set>
8#include <deque>
9
10#include "TString.h"
11
12class TObject;
13
14class MadGraphParticleClassifier : public ExRootClassifier
15{
16public:
17
18 MadGraphParticleClassifier();
19
20 void InsertParticleStatus(Int_t status);
21 void InsertClassPID(const TString &className, Int_t pid);
22 void SetExtendable(Bool_t extendable);
23
24 Int_t GetCategory(TObject *object);
25 Int_t GetMaxCategories() const;
26 TString GetCategoryClassName(Int_t category) const;
27
28private:
29
30 std::map< Int_t, Int_t > fParticleIDMap;
31 std::map< TString, Int_t > fClassNameMap;
32 std::deque< TString > fClassNameArray;
33 std::set< Int_t > fParticleStatusSet;
34
35 Int_t fMaxCategories;
36
37 Bool_t fIsExtendable;
38};
39
40#endif
Note: See TracBrowser for help on using the repository browser.