def main():
# build a simple dummy analysis to test the composition of Morphisms
ana = Analysis("ana")
ana_t = Analysis("ana_t")
m0 = DummyMorphism("m0") # always put a dummy morphism at the beginning
m1 = ExpGeneratorMorphism("m1", pars = {"scale": 1.0})
m2 = ConvMorphism("m2", pars = {"loc": 1.0, "scale": 1.6}, kern = np.random.normal)
m3 = TransfMorphism("m3", transf = lambda x: x - 0.05 * x**2 + 0.01 * x**3)
m4 = DummyMorphism("m4")
m5 = DummyMorphism("m5")
# this is the modified morphism
m1_t = ExpGeneratorMorphism("m1_t", pars = {"scale": 2.0})
ana.add_morphisms([m0, m1, m2, m3, m4, m5])
ana_t.add_morphisms([m0, m1_t, m2, m3, m4, m5])
nsamp = 1000000
dummy_data = pd.DataFrame.from_dict({"datacol": np.random.rand(nsamp)})
# run both analyses on the same data
out = ana.run(dummy_data)
out_t = ana_t.run(dummy_data)
# compute the resulting distributions
binning = np.linspace(-2, 20, 30)
hist = Histogram.from_data(name = r'$p(x_3|\theta_1,\theta_2,\theta_3)$', data = out, binning = binning)
hist_t = Histogram.from_data(name = r'$p(x_3|\tilde{\theta_1},\theta_2,\theta_3)$', data = out_t, binning = binning)
# play a bit with the likelihood ratio that should be propagated
source_hist = Histogram.from_data(r'$p(x_1|\theta_1)$', data = m1(dummy_data), binning = binning)
source_hist_t = Histogram.from_data(r'$p(x_1|\tilde{\theta_1})$', data = m1_t(dummy_data), binning = binning)
HistogramPlotter.plot_histograms([source_hist, source_hist_t], color = ["black", "salmon"], outfile = "/home/philipp/OX/thesis/FastProp/run/source.pdf", xlabel = r'$x_1$', ylabel = "events")
# also get the weights and compare the result
# first, create the profile of this analysis
prof = Profiler(ana)
prof.profile(dummy_data)
est = HistogramLREstimator(nbins = 30)
est.build_estimate(data_num = m1_t(dummy_data), data_den = m1(dummy_data))
prop = Propagator("prop")
prop.generate_propagator("m1", "m5", prof, est)
hist_rw = HistogramReweighter.reweight_to(hist, prop)
hist_rw.name = r'$p(x_3|\theta_1,\theta_2,\theta_3)\cdot R(x_3; \tilde{\theta_1}, \theta_1)$'
# plot the final histograms
HistogramPlotter.plot_histograms([hist, hist_t, hist_rw], show_ratio = True, color = ["black", "salmon", "mediumseagreen"], ratio_reference = hist_t, xlabel = r'$x_3$', ylabel = "events", outfile = "/home/philipp/OX/thesis/FastProp/run/target_rw.pdf")
HistogramPlotter.plot_histograms([hist, hist_t], show_ratio = True, color = ["black", "salmon"], ratio_reference = hist_t, xlabel = r'$x_3$', ylabel = "events", outfile = "/home/philipp/OX/thesis/FastProp/run/target.pdf")
# plot both the original reweighting factor, and the final one
finebinning = np.linspace(-2, 10, 300)
finebinning = np.expand_dims(finebinning, axis = 1)
source_reweighting = est.evaluate(finebinning)
target_reweighting = prop.predict(finebinning)
FunctionPlotter.plot_function(finebinning, source_reweighting, outfile = "/home/philipp/OX/thesis/FastProp/run/source_reweighting.pdf", xlabel = r'$x_1$', ylabel = r'$R(x_1; \tilde{\theta_1}, \theta_1)$', color = "black")
FunctionPlotter.plot_function(finebinning, target_reweighting, outfile = "/home/philipp/OX/thesis/FastProp/run/target_reweighting.pdf", xlabel = r'$x_3$', ylabel = r'$R(x_3; \tilde{\theta_1}, \theta_1)$', color = "black")
