Last change
on this file since 1382 was 1082, checked in by Pavel Demin, 11 years ago |
fix formatting and add include for TString
|
File size:
1.8 KB
|
Line | |
---|
1 |
|
---|
2 | #include "classes/DelphesFormula.h"
|
---|
3 |
|
---|
4 | #include "TString.h"
|
---|
5 |
|
---|
6 | #include <stdexcept>
|
---|
7 | #include <string>
|
---|
8 |
|
---|
9 | using namespace std;
|
---|
10 |
|
---|
11 | //------------------------------------------------------------------------------
|
---|
12 |
|
---|
13 | DelphesFormula::DelphesFormula() :
|
---|
14 | TFormula()
|
---|
15 | {
|
---|
16 | }
|
---|
17 |
|
---|
18 | //------------------------------------------------------------------------------
|
---|
19 |
|
---|
20 | DelphesFormula::DelphesFormula(const char *name, const char *expression) :
|
---|
21 | TFormula()
|
---|
22 | {
|
---|
23 | }
|
---|
24 |
|
---|
25 | //------------------------------------------------------------------------------
|
---|
26 |
|
---|
27 | DelphesFormula::~DelphesFormula()
|
---|
28 | {
|
---|
29 | }
|
---|
30 |
|
---|
31 | //------------------------------------------------------------------------------
|
---|
32 |
|
---|
33 | Int_t DelphesFormula::Compile(const char *expression)
|
---|
34 | {
|
---|
35 | string buffer;
|
---|
36 | const char *it;
|
---|
37 | for(it = expression; *it; ++it)
|
---|
38 | {
|
---|
39 | if(*it == ' ' || *it == '\t' || *it == '\r' || *it == '\n' || *it == '\\' ) continue;
|
---|
40 | buffer.push_back(*it);
|
---|
41 | }
|
---|
42 | if(TFormula::Compile(buffer.c_str()) != 0)
|
---|
43 | {
|
---|
44 | throw runtime_error("Invalid formula.");
|
---|
45 | }
|
---|
46 | return 0;
|
---|
47 | }
|
---|
48 |
|
---|
49 | //------------------------------------------------------------------------------
|
---|
50 |
|
---|
51 | Double_t DelphesFormula::Eval(Double_t pt, Double_t eta, Double_t phi, Double_t energy)
|
---|
52 | {
|
---|
53 | Double_t x[4] = {pt, eta, phi, energy};
|
---|
54 | return EvalPar(x);
|
---|
55 | }
|
---|
56 |
|
---|
57 | //------------------------------------------------------------------------------
|
---|
58 |
|
---|
59 | Int_t DelphesFormula::DefinedVariable(TString &chaine, Int_t &action)
|
---|
60 | {
|
---|
61 | action = kVariable;
|
---|
62 | if(chaine == "pt")
|
---|
63 | {
|
---|
64 | if(fNdim < 1) fNdim = 1;
|
---|
65 | return 0;
|
---|
66 | }
|
---|
67 | else if(chaine == "eta")
|
---|
68 | {
|
---|
69 | if(fNdim < 2) fNdim = 2;
|
---|
70 | return 1;
|
---|
71 | }
|
---|
72 | else if(chaine == "phi")
|
---|
73 | {
|
---|
74 | if(fNdim < 3) fNdim = 3;
|
---|
75 | return 2;
|
---|
76 | }
|
---|
77 | else if(chaine == "energy")
|
---|
78 | {
|
---|
79 | if(fNdim < 4) fNdim = 4;
|
---|
80 | return 3;
|
---|
81 | }
|
---|
82 | return -1;
|
---|
83 | }
|
---|
84 |
|
---|
85 | //------------------------------------------------------------------------------
|
---|
Note:
See
TracBrowser
for help on using the repository browser.