import sys import os from P2VV.ToyMCUtils import Toy toy = Toy() parser = toy.parser() (options, args) = toy.configure() from itertools import product from P2VV.RooFitWrappers import * from P2VV.Load import P2VVLibrary from ROOT import RooCBShape as CrystalBall from P2VV.Parameterizations.GeneralUtils import valid_combinations obj = RooObject(workspace="w") w = obj.ws() from math import pi t = RealVar("time", Title="decay time", Unit="ps", Observable=True, MinMax=(0.3, 14)) m = RealVar("mass", Title="B mass", Unit="MeV", Observable=True, MinMax=(5250, 5550)) nPV = RealVar("nPV", Title="nPV", Observable=True, MinMax=(0, 15)) mpsi = RealVar("mdau1", Title="J/psi mass", Unit="MeV", Observable=True, MinMax=(3030, 3150)) st = RealVar("sigmat", Title="#sigma(t)", Unit="ps", Observable=True, MinMax=(0.0001, 0.12)) # Categories hlt1_biased = Category("hlt1_biased", States={"biased": 1, "not_biased": 0}, Observable=True) hlt1_unbiased = Category("hlt1_unbiased", States={"unbiased": 1, "not_unbiased": 0}, Observable=True) hlt1_excl_biased = Category("hlt1_excl_biased", States={"excl_biased": 1, "unbiased": 0}, Observable=True) hlt2_biased = Category("hlt2_biased", States={"biased": 1, "not_biased": 0}, Observable=True) hlt2_unbiased = Category("hlt2_unbiased", States={"unbiased": 1, "not_unbiased": 0}, Observable=True)