Hyperiso 1.0.3
Modular flavour-physics calculations, Wilson coefficients and statistical inference
Loading...
Searching...
No Matches
test_uncertainties_FL.cpp
Go to the documentation of this file.
1#include <chrono>
2#include <fstream>
3#include <iomanip>
4#include <iostream>
5#include <map>
6#include <memory>
7#include <set>
8#include <string>
9#include <utility>
10#include <vector>
11
12#include "StatisticManager.h"
13#include "ObservableInterface.h"
14#include "ObservableInterfaceProxy.h"
15#include "StatCorrelationProxy.h"
16#include "StatParameterProxy.h"
17#include "StatParamSourcesProxy.h"
18#include "StatDependencyPruner.h"
19#include "FitAbstraction.h"
20#include "NuisanceReader.h"
21#include "DefaultNuisancePathsProvider.h"
22
23
24int main() {
25 HyperisoMaster hyp;
26 HyperisoConfig config_hyp;
27 config_hyp.model = Model::SM;
28 hyp.init("lha/si_input.flha", config_hyp);
29
30 auto oint = std::make_shared<ObservableInterface>();
31
32 std::vector<std::pair<double, double>> lhcb_bins = {
33 {0.06, 0.98},
34 {1.1, 2.5},
35 {2.5, 4.0},
36 {4.0, 6.0},
37 {6.0, 8.0},
38 {15.0, 17.0},
39 {17.0, 19.0}
40 };
41
42 // double q2_min = 0.05;
43 // double q2_max = 8.20;
44 // int N = 500;
45 // double bin_width = (q2_max - q2_min) / N;
46 // double bin_low = q2_min;
47
48 // for (size_t i = 0; i < N; i++) {
49 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::DGAMMA_DQ2_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
50 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::F_L_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
51 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_1_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
52 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_2_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
53 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_3_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
54 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_4_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
55 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_5_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
56 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_6_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
57 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_8_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
58 // bin_low += bin_width;
59 // }
60
61 // q2_min = 14.70;
62 // q2_max = 19.00;
63 // N = 250;
64 // bin_width = (q2_max - q2_min) / N;
65 // bin_low = q2_min;
66
67 // for (size_t i = 0; i < N; i++) {
68 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::DGAMMA_DQ2_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
69 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::F_L_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
70 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_1_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
71 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_2_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
72 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_3_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
73 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_4_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
74 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_5_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
75 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_6_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
76 // oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_8_B0__KSTAR0_MU_MU), {bin_low, bin_low + bin_width}}, QCDOrder::NNLO);
77 // bin_low += bin_width;
78 // }
79
80 for (auto& bin: lhcb_bins) {
81 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::DGAMMA_DQ2_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
82 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::F_L_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
83 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_1_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
84 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_2_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
85 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_3_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
86 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_4_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
87 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_5_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
88 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_6_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
89 oint->add_observable(BinnedObservableId{ObservableMapper::to_id(Observables::P_PRIME_8_B0__KSTAR0_MU_MU), bin}, QCDOrder::NNLO);
90 }
91
92 BKstarllConfig cfg;
93 cfg.ff_src = BV_FF_Src::GRvDV;
94 oint->set_decay_config(Decays::B__Kstar_l_l, cfg);
95 oint->set_bkstarll_threads(25);
96 std::shared_ptr<IStatParamOptimizerProxy> spop = std::make_shared<StatParamOptimizerProxy>();
97
98 auto model = std::make_shared<ObservableInterfaceProxy>(oint, spop);
99
100 StatisticConfig config;
101 config.MC_draws = 1000;
102 config.advanced.nuisance_relevance_cutoff = 1e-10;
103 config.advanced.override_nuisance_marginals = {
104 {ParamId{ParameterType::WILSON, "EW_SCALE", 1}, MarginalType::FLAT},
105 {ParamId{ParameterType::WILSON, "B_SCALE", 1}, MarginalType::FLAT},
106 };
107
108 std::shared_ptr<INuisancePathsProvider> npp = std::make_shared<DefaultNuisancePathsProvider>();
109
110 StatisticManager stat(
111 config,
112 model,
113 std::make_shared<StatCorrelationProxy>(),
114 std::make_shared<StatParameterProxy>(),
115 std::make_shared<StatParamSourcesProxy>(),
116 std::make_shared<StatDependencyPruner>(),
117 std::make_shared<NuisanceReader>(npp),
118 spop
119 );
120
121 // stat.select_experiment("CMS");
122
123 auto start = std::chrono::steady_clock::now();
124 auto pred_with_u = stat.compute_uncertainties();
125 auto stop = std::chrono::steady_clock::now();
126 LOG_INFO("Uncertainty computation took", std::chrono::duration_cast<std::chrono::seconds>(stop - start).count(), "s");
127
128 std::ofstream fs;
129 std::map<std::string, Observables> file_names = {
130 {"dGamma_dq2", Observables::DGAMMA_DQ2_B0__KSTAR0_MU_MU},
131 {"F_L", Observables::F_L_B0__KSTAR0_MU_MU},
132 {"P_1", Observables::P_1_B0__KSTAR0_MU_MU},
133 {"P_2", Observables::P_2_B0__KSTAR0_MU_MU},
134 {"P_3", Observables::P_3_B0__KSTAR0_MU_MU},
135 {"Pp_4", Observables::P_PRIME_4_B0__KSTAR0_MU_MU},
136 {"Pp_5", Observables::P_PRIME_5_B0__KSTAR0_MU_MU},
137 {"Pp_6", Observables::P_PRIME_6_B0__KSTAR0_MU_MU},
138 {"Pp_8", Observables::P_PRIME_8_B0__KSTAR0_MU_MU}
139 };
140
141 for (auto &&[name, obs] : file_names) {
142 fs.open(name + "_LHCB_bins.csv");
143 fs << "bin_low,bin_high,value,u_low,u_high,u_sym\n";
144
145 for (auto &&[k, v] : pred_with_u) {
146 if (k.s != ObservableMapper::to_id(obs)) continue;
147 fs << k.p.first << "," << k.p.second << "," << v.mu << "," << v.sigma_m << "," << v.sigma_p << "," << v.sigma << '\n';
148 }
149 fs.close();
150 }
151
152 return 0;
153}
DefaultNuisancePathsProvider.h
Default nuisance-configuration path provider.
Observables::P_2_B0__KSTAR0_MU_MU
@ P_2_B0__KSTAR0_MU_MU
Observables::P_PRIME_8_B0__KSTAR0_MU_MU
@ P_PRIME_8_B0__KSTAR0_MU_MU
Observables::P_PRIME_5_B0__KSTAR0_MU_MU
@ P_PRIME_5_B0__KSTAR0_MU_MU
Observables::DGAMMA_DQ2_B0__KSTAR0_MU_MU
@ DGAMMA_DQ2_B0__KSTAR0_MU_MU
Observables::F_L_B0__KSTAR0_MU_MU
@ F_L_B0__KSTAR0_MU_MU
Observables::P_1_B0__KSTAR0_MU_MU
@ P_1_B0__KSTAR0_MU_MU
Observables::P_3_B0__KSTAR0_MU_MU
@ P_3_B0__KSTAR0_MU_MU
Observables::P_PRIME_4_B0__KSTAR0_MU_MU
@ P_PRIME_4_B0__KSTAR0_MU_MU
Observables::P_PRIME_6_B0__KSTAR0_MU_MU
@ P_PRIME_6_B0__KSTAR0_MU_MU
Decays::B__Kstar_l_l
@ B__Kstar_l_l
LOG_INFO
#define LOG_INFO(...)
Macro for logging informational messages.
Definition Logger.h:39
MarginalType::FLAT
@ FLAT
Uniform (flat) marginal on a finite interval.
NuisanceReader.h
Concrete reader for nuisance-parameter definition files.
ObservableInterfaceProxy.h
Adapter from ObservableInterface to the statistical model interface.
ObservableInterface.h
High-level, user-facing entry point to compute flavor observables.
StatCorrelationProxy.h
Concrete statistical proxy forwarding correlation queries to CorrelationProvider.
StatDependencyPruner.h
Statistics-layer adapter over the core DependencyPruner service.
StatParamSourcesProxy.h
Statistics-layer adapter for retrieving leaf parameter sources.
StatParameterProxy.h
Statistics-layer proxy for read-only access to parameters and observables.
StatisticManager.h
High-level orchestration of statistical uncertainty propagation, likelihood construction and fit scan...
GenericMapperWithExt< ObservableTag, Observables, LhaID, std::hash< LhaID >, observable_mapping, observable_flha_mapping >::to_id
static IdOf< ObservableTag > to_id(Observables e)
Converts an enum value to an IdOf<Tag>.
HyperisoMaster
High-level interface to initialize and monitor the main framework configuration.
Definition HyperisoMaster.h:61
HyperisoMaster::init
void init(const std::string &lhaFile, HyperisoConfig config)
Initializes Hyperiso using a LHA file and a full Config object.
Definition HyperisoMaster.cpp:185
StatisticManager
Coordinates statistical inputs, nuisance distributions, MLE fits and contour/scan computations.
Definition StatisticManager.h:210
StatisticManager::compute_uncertainties
std::map< BinnedObservableId, GaussianSummary > compute_uncertainties()
Computes Gaussian summaries for MC-propagated observable uncertainties.
Definition StatisticManager.cpp:642
AdvancedStatisticConfig::nuisance_relevance_cutoff
double nuisance_relevance_cutoff
Relative-uncertainty cutoff for the first nuisance preselection pass.
Definition StatisticManager.h:81
AdvancedStatisticConfig::override_nuisance_marginals
std::map< ParamId, MarginalType > override_nuisance_marginals
Per-parameter overrides for nuisance marginal laws.
Definition StatisticManager.h:69
BKstarllConfig::ff_src
BV_FF_Src ff_src
Definition BKsllDecay.h:19
BinnedObservableId
Identifies an observable together with a numerical bin.
Definition BinnedObservableId.h:285
HyperisoConfig
Configuration object controlling model, input flags and optional MARTY resources.
Definition Config.h:24
HyperisoConfig::model
Model model
Current model.
Definition Config.h:33
ParamId
Composite identifier for a single parameter.
Definition ParamID.h:57
StatisticConfig::advanced
AdvancedStatisticConfig advanced
Advanced fit/pruning/covariance configuration.
Definition StatisticManager.h:145
StatisticConfig::MC_draws
std::size_t MC_draws
Number of accepted MC draws used for uncertainty propagation.
Definition StatisticManager.h:122
main
int main()
Definition test_uncertainties_FL.cpp:24