| 1 | /*
|
|---|
| 2 | * Delphes: a framework for fast simulation of a generic collider experiment
|
|---|
| 3 | * Copyright (C) 2012-2014 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 |
|
|---|
| 20 | #include "classes/DelphesLongFormula.h"
|
|---|
| 21 |
|
|---|
| 22 | #include "TString.h"
|
|---|
| 23 |
|
|---|
| 24 | #include <stdexcept>
|
|---|
| 25 |
|
|---|
| 26 | using namespace std;
|
|---|
| 27 |
|
|---|
| 28 | //------------------------------------------------------------------------------
|
|---|
| 29 |
|
|---|
| 30 | DelphesLongFormula::DelphesLongFormula() :
|
|---|
| 31 | TFormula()
|
|---|
| 32 | {
|
|---|
| 33 | }
|
|---|
| 34 |
|
|---|
| 35 | //------------------------------------------------------------------------------
|
|---|
| 36 |
|
|---|
| 37 | DelphesLongFormula::DelphesLongFormula(const char *name, const char *expression) :
|
|---|
| 38 | TFormula()
|
|---|
| 39 | {
|
|---|
| 40 | }
|
|---|
| 41 |
|
|---|
| 42 | //------------------------------------------------------------------------------
|
|---|
| 43 |
|
|---|
| 44 | DelphesLongFormula::~DelphesLongFormula()
|
|---|
| 45 | {
|
|---|
| 46 | }
|
|---|
| 47 |
|
|---|
| 48 | //------------------------------------------------------------------------------
|
|---|
| 49 |
|
|---|
| 50 | Int_t DelphesLongFormula::Compile(const char *expression)
|
|---|
| 51 | {
|
|---|
| 52 | TString buffer;
|
|---|
| 53 | const char *it;
|
|---|
| 54 | for(it = expression; *it; ++it)
|
|---|
| 55 | {
|
|---|
| 56 | if(*it == ' ' || *it == '\t' || *it == '\r' || *it == '\n' || *it == '\\' ) continue;
|
|---|
| 57 | buffer.Append(*it);
|
|---|
| 58 | }
|
|---|
| 59 |
|
|---|
| 60 | buffer.ReplaceAll("pt", "[pt]");
|
|---|
| 61 | buffer.ReplaceAll("eta", "[eta]");
|
|---|
| 62 | buffer.ReplaceAll("phi", "[phi]");
|
|---|
| 63 | buffer.ReplaceAll("energy", "[energy]");
|
|---|
| 64 | buffer.ReplaceAll("d0", "[d0]");
|
|---|
| 65 | buffer.ReplaceAll("dz", "[dz]");
|
|---|
| 66 | buffer.ReplaceAll("ctgTheta", "[ctgTheta]");
|
|---|
| 67 |
|
|---|
| 68 | #if ROOT_VERSION_CODE < ROOT_VERSION(6,0,0)
|
|---|
| 69 | TFormula::SetMaxima(100000,1000,1000000);
|
|---|
| 70 | #endif
|
|---|
| 71 |
|
|---|
| 72 | if(TFormula::Compile(buffer) != 0)
|
|---|
| 73 | {
|
|---|
| 74 | throw runtime_error("Invalid Long Formula.");
|
|---|
| 75 | }
|
|---|
| 76 |
|
|---|
| 77 | return 0;
|
|---|
| 78 | }
|
|---|
| 79 |
|
|---|
| 80 | //------------------------------------------------------------------------------
|
|---|
| 81 |
|
|---|
| 82 | Double_t DelphesLongFormula::Eval(Double_t pt,
|
|---|
| 83 | Double_t eta,
|
|---|
| 84 | Double_t phi,
|
|---|
| 85 | Double_t energy,
|
|---|
| 86 | Double_t d0,
|
|---|
| 87 | Double_t dz,
|
|---|
| 88 | Double_t ctgTheta
|
|---|
| 89 | )
|
|---|
| 90 | {
|
|---|
| 91 |
|
|---|
| 92 | TVarNameMap fVarNameMap;
|
|---|
| 93 | TVarValMap fVarValMap;
|
|---|
| 94 |
|
|---|
| 95 | fVarNameMap[this->GetParNumber("pt")]= "pt";
|
|---|
| 96 | fVarNameMap[this->GetParNumber("eta")]= "eta";
|
|---|
| 97 | fVarNameMap[this->GetParNumber("phi")]= "phi";
|
|---|
| 98 | fVarNameMap[this->GetParNumber("energy")]= "energy";
|
|---|
| 99 | fVarNameMap[this->GetParNumber("d0")]= "d0";
|
|---|
| 100 | fVarNameMap[this->GetParNumber("dz")]= "dz";
|
|---|
| 101 | fVarNameMap[this->GetParNumber("ctgTheta")]= "ctgTheta";
|
|---|
| 102 |
|
|---|
| 103 | fVarValMap["pt"]= pt;
|
|---|
| 104 | fVarValMap["eta"]= eta;
|
|---|
| 105 | fVarValMap["phi"]= phi;
|
|---|
| 106 | fVarValMap["energy"]= energy;
|
|---|
| 107 | fVarValMap["d0"]= d0;
|
|---|
| 108 | fVarValMap["dz"]= dz;
|
|---|
| 109 | fVarValMap["ctgTheta"]= ctgTheta;
|
|---|
| 110 |
|
|---|
| 111 | Double_t vals[7];
|
|---|
| 112 |
|
|---|
| 113 | Int_t j = 0;
|
|---|
| 114 | for (Int_t i=0; i != 7; i++)
|
|---|
| 115 | {
|
|---|
| 116 | if ( fVarNameMap.find(i) != fVarNameMap.end() )
|
|---|
| 117 | {
|
|---|
| 118 | TString var_name = fVarNameMap[i];
|
|---|
| 119 | vals[i] = fVarValMap[var_name];
|
|---|
| 120 | }
|
|---|
| 121 | else
|
|---|
| 122 | vals[i] = 0.;
|
|---|
| 123 | }
|
|---|
| 124 | return EvalPar(nullptr, vals);
|
|---|
| 125 | }
|
|---|
| 126 |
|
|---|
| 127 | //------------------------------------------------------------------------------
|
|---|
