Back to home page

Project CMSSW displayed by LXR

 
 

    


File indexing completed on 2025-07-15 22:06:31

0001 import FWCore.ParameterSet.Config as cms
0002 from PhysicsTools.NanoAOD.common_cff import *
0003 from PhysicsTools.NanoAOD.nano_cff import nanoMetadata
0004 from Validation.HGCalValidation.HLT_TICLIterLabels_cff import hltTiclIterLabels
0005 
0006 hltUpgradeNanoTask = cms.Task(nanoMetadata)
0007 
0008 hltSimTracksterTable = cms.EDProducer(
0009     "TracksterCollectionTableProducer",
0010     skipNonExistingSrc=cms.bool(True),
0011     src=cms.InputTag("hltTiclSimTracksters"),
0012     cut=cms.string(""),
0013     name=cms.string("hltTiclSimTracksters"),
0014     doc=cms.string("hltTiclSimTracksters"),
0015     singleton=cms.bool(False),  # the number of entries is variable
0016     variables=cms.PSet(
0017         raw_energy=Var("raw_energy", "float",
0018                        doc="Raw Energy of the trackster [GeV]"),
0019         raw_em_energy=Var("raw_em_energy", "float",
0020                           doc="EM raw Energy of the trackster [GeV]"),
0021         raw_pt=Var(
0022             "raw_pt", "float", doc="Trackster raw pT, computed from trackster raw energy and direction [GeV]"),
0023         regressed_energy=Var("regressed_energy", "float",
0024                              doc="Regressed Energy of the trackster, for the SimTrackster it corresponds to the GEN-energy"),
0025         barycenter_x=Var("barycenter.x", "float",
0026                          doc="Trackster barycenter x [cm]"),
0027         barycenter_y=Var("barycenter.y", "float",
0028                          doc="Trackster barycenter y [cm]"),
0029         barycenter_z=Var("barycenter.z", "float",
0030                          doc="Trackster barycenter z [cm]"),
0031         barycenter_eta=Var("barycenter.eta", "float",
0032                            doc="Trackster barycenter pseudorapidity"),
0033         barycenter_phi=Var("barycenter.phi", "float",
0034                            doc="Trackster barycenter phi"),
0035         EV1=Var("eigenvalues()[0]", "float",
0036                 doc="Trackster PCA eigenvalues 0"),
0037         EV2=Var("eigenvalues()[1]", "float",
0038                 doc="Trackster PCA eigenvalues 1"),
0039         EV3=Var("eigenvalues()[2]", "float",
0040                 doc="Trackster PCA eigenvalues 2"),
0041         eVector0_x=Var(
0042             "eigenvectors()[0].x", "float", doc="Trackster PCA principal axis, x component"),
0043         eVector0_y=Var(
0044             "eigenvectors()[0].z", "float", doc="Trackster PCA principal axis, y component"),
0045         eVector0_z=Var(
0046             "eigenvectors()[0].y", "float", doc="Trackster PCA principal axis, z component"),
0047         time=Var("time", "float", doc="Trackster HGCAL time"),
0048         timeError=Var("timeError", "float",
0049                       doc="Trackster HGCAL time error")
0050     ),
0051     collectionVariables=cms.PSet(
0052         tracksterVertices=cms.PSet(
0053             name=cms.string(f"hltTiclSimTrackstersvertices"),
0054             doc=cms.string("Vertex properties"),
0055             useCount=cms.bool(True),
0056             useOffset=cms.bool(True),
0057             variables=cms.PSet(
0058                 vertices=Var("vertices", "uint",
0059                              doc="Layer clusters indices."),
0060                 vertex_mult=Var(
0061                     "vertex_multiplicity",
0062                     "float",
0063                     doc="Fraction of Layer cluster energy used by the Trackster.",
0064                 ),
0065             ),
0066         )
0067     ),
0068 )
0069 
0070 hltSimTracksterFromCPsTable = cms.EDProducer(
0071     "TracksterCollectionTableProducer",
0072     skipNonExistingSrc=cms.bool(True),
0073     src=cms.InputTag("hltTiclSimTracksters", "fromCPs"),
0074     cut=cms.string(""),
0075     name=cms.string("hltTiclSimTrackstersFromCPs"),
0076     doc=cms.string("hltTiclSimTrackstersFromCPs"),
0077     singleton=cms.bool(False),  # the number of entries is variable
0078     variables=cms.PSet(
0079         raw_energy=Var("raw_energy", "float",
0080                        doc="Raw Energy of the trackster [GeV]"),
0081         raw_em_energy=Var("raw_em_energy", "float",
0082                           doc="EM raw Energy of the trackster [GeV]"),
0083         raw_pt=Var(
0084             "raw_pt", "float", doc="Trackster raw pT, computed from trackster raw energy and direction [GeV]"),
0085         regressed_energy=Var("regressed_energy", "float",
0086                              doc="Regressed Energy of the trackster, for the SimTrackster it corresponds to the GEN-energy"),
0087         barycenter_x=Var("barycenter.x", "float",
0088                          doc="Trackster barycenter x [cm]"),
0089         barycenter_y=Var("barycenter.y", "float",
0090                          doc="Trackster barycenter y [cm]"),
0091         barycenter_z=Var("barycenter.z", "float",
0092                          doc="Trackster barycenter z [cm]"),
0093         barycenter_eta=Var("barycenter.eta", "float",
0094                            doc="Trackster barycenter pseudorapidity"),
0095         barycenter_phi=Var("barycenter.phi", "float",
0096                            doc="Trackster barycenter phi"),
0097         EV1=Var("eigenvalues()[0]", "float",
0098                 doc="Trackster PCA eigenvalues 0"),
0099         EV2=Var("eigenvalues()[1]", "float",
0100                 doc="Trackster PCA eigenvalues 1"),
0101         EV3=Var("eigenvalues()[2]", "float",
0102                 doc="Trackster PCA eigenvalues 2"),
0103         eVector0_x=Var(
0104             "eigenvectors()[0].x", "float", doc="Trackster PCA principal axis, x component"),
0105         eVector0_y=Var(
0106             "eigenvectors()[0].z", "float", doc="Trackster PCA principal axis, y component"),
0107         eVector0_z=Var(
0108             "eigenvectors()[0].y", "float", doc="Trackster PCA principal axis, z component"),
0109         time=Var("time", "float", doc="Trackster HGCAL time"),
0110         timeError=Var("timeError", "float",
0111                       doc="Trackster HGCAL time error")
0112     ),
0113     collectionVariables=cms.PSet(
0114         tracksterVertices=cms.PSet(
0115             name=cms.string(f"hltTiclSimTrackstersFromCPsvertices"),
0116             doc=cms.string("Vertex properties"),
0117             useCount=cms.bool(True),
0118             useOffset=cms.bool(True),
0119             variables=cms.PSet(
0120                 vertices=Var("vertices", "uint",
0121                              doc="Layer clusters indices."),
0122                 vertex_mult=Var(
0123                     "vertex_multiplicity",
0124                     "float",
0125                     doc="Fraction of Layer cluster energy used by the Trackster.",
0126                 ),
0127             ),
0128         )
0129     ),
0130 )
0131 
0132 # Extra tables for SimTracksters using information from SimClusters and CaloParticles 
0133 # Might be replaced in case we save CaloParticles and SimClusters, in that case we just need adding the corresponding SimObject indices in the SimTrackster tables
0134 hltTiclSimTrackstersExtraTable = cms.EDProducer("SimTracksterTableProducer",
0135                                 tableName = cms.string("hltTiclSimTracksters"),
0136                                 skipNonExistingSrc = cms.bool(True),
0137                                 simTracksters = cms.InputTag( "hltTiclSimTracksters" ),
0138                                 caloParticles = cms.InputTag( "mix", "MergedCaloTruth" ),
0139                                 simClusters = cms.InputTag( "mix", "MergedCaloTruth" ),
0140                                 caloParticleToSimClustersMap = cms.InputTag("hltTiclSimTracksters"),
0141                                 precision = cms.int32(7),
0142                                 )
0143 
0144 hltTiclSimTrackstersFromCPsExtraTable = cms.EDProducer("SimTracksterTableProducer",
0145                                 tableName = cms.string("hltTiclSimTrackstersFromCPs"),
0146                                 skipNonExistingSrc = cms.bool(True),
0147                                 simTracksters = cms.InputTag( "hltTiclSimTracksters", "fromCPs"),
0148                                 caloParticles = cms.InputTag( "mix", "MergedCaloTruth" ),
0149                                 simClusters = cms.InputTag( "mix", "MergedCaloTruth" ),
0150                                 caloParticleToSimClustersMap = cms.InputTag("hltTiclSimTracksters"),
0151                                 precision = cms.int32(7),
0152                                 )