16 import modularAnalysis
as ma
17 import variables
as va
21 my_path = b2.create_path()
24 ma.inputMdst(environmentType=
'default',
25 filename=b2.find_file(
'B2pi0D_D2hh_D2hhh_B2munu.root',
'examples',
False),
30 ma.fillParticleListFromMC(decayString=
'pi+:gen', cut=
'', path=my_path)
33 weight_table_id =
"ParticleReweighting:TestMomentum"
35 if not os.getenv(
'BELLE2_EXAMPLES_DATA_DIR'):
36 b2.B2FATAL(
"You need the example data installed. Run `b2install-data example` in terminal for it.")
38 db_location = os.getenv(
'BELLE2_EXAMPLES_DATA_DIR') +
'/database/'
39 b2.use_local_database(db_location +
'database.txt',
40 directory=db_location,
45 va.variables.addAlias(
'Weight',
'extraInfo(' + weight_table_id +
'_Weight)')
46 va.variables.addAlias(
'StatErr',
'extraInfo(' + weight_table_id +
'_StatErr)')
47 va.variables.addAlias(
'SystErr',
'extraInfo(' + weight_table_id +
'_SystErr)')
48 va.variables.addAlias(
'binID',
'extraInfo(' + weight_table_id +
'_binID)')
52 reweighter = b2.register_module(
'ParticleWeighting')
53 reweighter.param(
'tableName', weight_table_id)
54 reweighter.param(
'particleList',
'pi+:gen')
55 my_path.add_module(reweighter)
58 pivars = [
'p',
'pz',
'Weight',
'StatErr',
'SystErr',
'binID']
61 output_file =
'B2A905-ApplyWeightsToTracks.root'
62 ma.variablesToNtuple(decayString=
'pi+:gen',