hiro/bachelor_thesis
1
0
Fork 0
mirror of https://github.com/vale981/bachelor_thesis synced 2025-03-05 17:41:41 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

more elegant observable management

Browse source
This commit is contained in:
hiro98 2020-05-27 12:12:20 +02:00
parent 19b21b2f39
commit b9712b92d3
1 changed files with 55 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

82
prog/runcards/pp_phaeno/qqgg_proton/MC_DIPHOTON_PROTON.cc
View file

@ -11,6 +11,30 @@
namespace Rivet { namespace Rivet {
struct Observable {
double _min, _max;
int _bins = 50;
bool _log = false;
double _value = 0;
Histo1DPtr _hist = nullptr;
Observable(double min, double max, int bins = 50, bool log = false)
: _min{min}, _max{max}, _bins{bins}, _log{log} {};
void fill(const double value) {
_value = value;
if (_hist) {
_hist->fill(value);
}
}
operator double() const { return _value; };
// Observable(double min, double max, bool log = false) : _min{min},
// _max{max}, _log{log} {
// }
};
/// @brief Generate some simple histograms of the diphoton process. TODO: more /// @brief Generate some simple histograms of the diphoton process. TODO: more
class MC_DIPHOTON_PROTON : public Analysis { class MC_DIPHOTON_PROTON : public Analysis {
public: public:
@ -38,16 +62,24 @@ public:
double min_pT = 20; double min_pT = 20;
double eta = 2.5; double eta = 2.5;
_observables = {"pT", "eta", "cos_theta", "inv_m", _observables = {{"pT", {min_pT, energy, 50, true}},
"o_angle", "o_angle_cs", "total_pT"}; {"eta", {-eta, eta}},
{"cos_theta", {-1, 1}},
{"inv_m", {0.1, 2 * energy, 50, true}},
{"o_angle", {0, 1}},
{"o_angle_cs", {0, 1}},
{"total_pT", {.01, 2 * energy, 50, true}}};
book(_histos["pT"], "pT", logspace(50, min_pT, energy, true)); for (auto &[name, observable] : _observables) {
book(_histos["eta"], "eta", 50, -eta, eta); if (observable._log) {
book(_histos["cos_theta"], "cos_theta", 50, -1, 1); book(observable._hist, name,
book(_histos["inv_m"], "inv_m", logspace(50, 2 * min_pT, 2 * energy, true)); logspace(observable._bins, observable._min, observable._max));
book(_histos["o_angle"], "o_angle", 50, 0, 1); continue;
book(_histos["o_angle_cs"], "o_angle_cs", 50, 0, 1); }
book(_histos["total_pT"], "total_pT", logspace(50, .01, 2 * energy));
book(observable._hist, name, observable._bins, observable._min,
observable._max);
}
} }
/// Perform the per-event analysis /// Perform the per-event analysis
@ -90,30 +122,27 @@ public:
const auto &photon = photons.front(); const auto &photon = photons.front();
std::map<string, double> obs; _observables.at("pT").fill(photon.pT());
obs["pT"] = photon.pT(); _observables.at("eta").fill(photon.eta());
obs["eta"] = photon.eta(); _observables.at("cos_theta").fill(cos(photon.theta()));
obs["cos_theta"] = cos(photon.theta());
const auto &moms = photons.moms(); const auto &moms = photons.moms();
const auto total_momentum = const auto total_momentum =
std::accumulate(moms.begin(), moms.end(), FourMomentum(0, 0, 0, 0)); std::accumulate(moms.begin(), moms.end(), FourMomentum(0, 0, 0, 0));
obs["inv_m"] = total_momentum.mass(); _observables.at("inv_m").fill(total_momentum.mass());
obs["o_angle"] = std::abs(tanh((photons[1].eta() - photons[0].eta()) / 2)); _observables.at("o_angle").fill(
std::abs(tanh((photons[1].eta() - photons[0].eta()) / 2)));
// std::abs(photons[0].theta() + photons[1].theta()); // std::abs(photons[0].theta() + photons[1].theta());
obs["o_angle_cs"] = std::abs( _observables.at("o_angle_cs").fill(std::abs(
sinh((photons[0].eta() - photons[1].eta())) * 2.0 * photons[0].pT() * sinh((photons[0].eta() - photons[1].eta())) * 2.0 * photons[0].pT() *
photons[1].pT() / sqrt(sqr(obs["inv_m"]) + sqr(total_momentum.pT())) / photons[1].pT() /
obs["inv_m"]); sqrt(sqr(_observables.at("inv_m")._value) + sqr(total_momentum.pT())) /
_observables.at("inv_m")._value));
obs["total_pT"] = total_momentum.pT(); _observables.at("total_pT").fill(total_momentum.pT());
for (const auto &name : _observables) {
_histos[name]->fill(obs[name]);
}
} }
//@} //@}
@ -121,15 +150,14 @@ public:
void finalize() { void finalize() {
const double sf = crossSection() / (picobarn * sumOfWeights()); const double sf = crossSection() / (picobarn * sumOfWeights());
for (auto name : _observables) { for (auto const &[_, observable] : _observables) {
scale(_histos[name], sf); scale(observable._hist, sf);
} }
} }
/// @name Histograms /// @name Histograms
//@{ //@{
std::map<string, Histo1DPtr> _histos; std::map<string, Observable> _observables;
std::vector<string> _observables;
//@} //@}
}; };