Use Analysis package
V.S.Vorobev (Talk | contribs) |
V.S.Vorobev (Talk | contribs) |
||
| Line 198: | Line 198: | ||
am.add_analysis(cfg=anaysisCfg) | am.add_analysis(cfg=anaysisCfg) | ||
am.run(evtmax=10**4) | am.run(evtmax=10**4) | ||
| − | |||
| − | |||
[[Category:Not_public]] | [[Category:Not_public]] | ||
Revision as of 12:59, 10 February 2021
Contents |
Introduction
The Analysis module implements tools needed for:
- Access the reconstructed final-state-particles
- Reconstruction arbitrary decay trees
- Imposing selection criteria
- Applying kinematic fit to the decay tree (TODO)
- Saving a flat ntuple to a ROOT TTree for further analysis
The decay language
The Analysis package supports an easy-to-read string description of particle decays. The EvtGen particle naming scheme is used. The following strings are valid decay expressions:
"D0" "D0 -> K- pi+" "D0 -> [rho0 -> pi+ pi-] pi0"
A decay string may or may not contain the arrow and right-hand side. Spaces around the arrow are optional. Nested decays are expressed with square brackets.
A particle in the decay string can be labeled:
"pi+:lowpt"
"lowpt" is a label. Labels allow working with different particle lists of the same type. For example:
"D*+ -> [D0 -> K- pi+] pi+:lowpt"
A particle in the decay string can be selected using the "^" symbol:
"D0 -> ^K- ^pi+"
"K-" and "pi+" are selected here. We will consider the use cases for these features below.
The cuts language
The Analysis package contains large set of predefined variables. Most of these variables can be calculated for a given particle. Selection criteria are imposed with string expressions like:
* "M < 0.12" # the mass less than 0.12 GeV * "charge == 0" # zero electric charge * "1.8 < M < 1.9" # the mass is between 1.8 GeV and 1.9 GeV * "charge == 0 and M < 0.12" * "charge == 0 and [M < 0.12 or pt > 0.1]"
Square brackets are used to manage the order of logical operations. The following operators are available: ">", ">=", "<", "<=", "==", "!=", "and", "or".
AuroraMaster interface
An analysis procedure can be connected to the AuroraMaster instance with the add_analysis method. An AuroraConfig object with analysis configuration must be passed as the cfg parameter:
am = AuroraMaster('analysis', 'info')
am.add_analysis(cfg=anaysisCfg)
An analysis configuration should contain three items
anaysisCfg = AuroraConfig({
'EventLoader' : [...],
'Combiners' : [...],
'Tuples' : [...]
})
EventLoader
The EventLoader algorithm specifies lists of the final-state-particles and corresponding cuts. Its configuration is a list of dictionaries of the following format:
[
{
'decstr': 'pi+ cc',
'cutstr': 'pt > 0.10',
},
{
'decstr': 'K+ cc',
'cutstr': 'pt > 0.10',
},
{
'decstr': 'pi+:lowpt cc',
'cutstr': 'pt < 0.20',
},
{
'decstr': 'gamma',
'cutstr': 'E > 0.07',
},
...
]
The decstr key correspond to the decay string. "cc" at the end of the decay string means that list of anti-particles will be created, too. The configuration above leads seven lists: "pi+", "pi-", "K+", "K-", "pi+:lowpt", "pi-:lowpt", and "gamma". These lists are available for the further analysis.
Particle combiner
ParticleCombiner algorithm is used to construct candidates for intermediate unstable particles like D0 or Lambda_c. An analysis job usually includes several ParticleCombiner algorithms. Combines are configured with the following parameters:
[
{
'label': 'Dkpi',
'decstr': 'D0 -> pi+ K-',
'cutstr': '1.82 < M < 1.90',
'selfconj': False
},
{
'label': 'Dkk',
'decstr': 'D0:kk -> K+ K-',
'cutstr': '1.82 < M < 1.90',
'selfconj': True
},
{
'label': 'DstDkpi',
'decstr': 'D*+ -> D0 pi+:lowpt',
'cutstr': 'deltaM < 0.10',
'selfconj': False
},
]
The left side of the decay string defines the list of particles to create, and the right side must refer to existing particle lists (created by EventLoader or other ParticleCombiner). If selfconj us false then the charge conjugated list will be created, too ('anti-D0 -> pi- K+' and `D*- -> anti-D0 pi-:lowpt` will be created for the configuration above).
Ntuple configuration
The last step of the analysis is saving necessary variables in a flat ntuple. Each ntuple algorithm corresponds to a particle list. There can be several ntuple algorithms corresponding to different particle lists. An example configuration is
[
{
'label': 'd0tup',
'root': 'D0',
'ofile': 'tup.root',
'vars' : [
{
'selector': 'root',
'varlist': ['momentumVars', 'E', 'M'],
},
{
'selector': 'D0 -> ^pi+ ^K-',
'varlist': ['momentumVars', 'pidVars', 'matchVars', 'charge']
}
]
},
{
'label': 'dsttup',
'root': 'D*+',
'ofile': 'tup.root',
'vars' : [
{
'selector': 'root',
'varlist': ['M', 'deltaM'],
},
{
'selector': 'D*+ -> D0 ^pi+:lowpt',
'varlist': ['momentumVars', 'pidVars', 'charge']
}
]
},
]
The "label" value defines the name of TTree in the output file. The "root" value must correspond to an existing particle list. The "selector" value is a decay string specifying particles for which variables will be calculated and saved. The "varlist" is a list of named variables. Note that "momentumVars" and "pidVars" names denote groups of variables. Here is the complete list of names variable groups (defined in analysis.py):
VARSETS = {
'momentumVars': ['px', 'py', 'pz', 'p', 'pt'],
'pidVars': ['pidkpi', 'pidmupi', 'pidkp', 'pide'],
'matchVars': ['pdgid_mc', 'px_mc', 'py_mc', 'pz_mc'],
}
Convenience methods
A set of convenience methods are defined in the python Analysis module. These methods ease configuration of the analysis. The example below shows how to define the same configuration as we considered above:
from AuroraMaster.auroramaster import AuroraMaster, AuroraConfig
from AuroraMaster.auroramaster import Analysis as A
am = AuroraMaster('analysis', 'info')
edminputCfg = AuroraConfig({
'filename': './parsim.root', # should be in your run directory
'collections': ['Particles', 'allGenParticles'],
})
am.add_edmi(cfg=edminputCfg)
anaysisCfg = A.analysis(
fsps=[
A.fspList('pi+ cc'),
A.fspList('K+ cc'),
A.fspList('pi+:lowpt cc', 'pt < 0.20),
A.fspList('gamma'),
],
combiners=[
A.combiner('Dkpi', 'D0 -> pi+ K-', '1.8 < M < 1.9'),
A.combiner('Dkk', 'D0 -> K+ K-', '1.8 < M < 1.9', selfconj=True),
A.combiner('DstDkpi', 'D*+ -> D0 pi+:lowpt', 'deltaM < 0.10'),
],
tuples=[
A.ntuple('tup', 'D0', 'tuple.root', [
A.vars('root', ['momentumVars', 'E', 'M']),
A.vars('D0 -> ^pi+ ^K-', ['momentumVars', 'pidVars', 'matchVars']),
]
),
A.ntuple('tup', 'D0', 'tuple.root', [
A.vars('root', ['M', 'deltaM']),
A.vars('D*+ -> D0 ^pi+:lowpt', ['momentumVars', 'pidVars', 'charge']),
]
),
]
)
am.add_analysis(cfg=anaysisCfg)
am.run(evtmax=10**4)
