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)
