FFTJetProducers/python/fftjettreedump_cfi.py

0001 # FFTJet clustering tree dumper configuration
0002
0003 import os, errno, sys
0004 import FWCore.ParameterSet.Config as cms
0005
0006 from RecoJets.FFTJetProducers.fftjetcommon_cfi import *
0007
0008 def mkdir_p(path):
0009     try:
0010         os.makedirs(path)
0011     except OSError as exc:
0012         if exc.errno == errno.EEXIST:
0013             pass
0014         else: raise
0015
0016 # Output directory for the trees
0017 clustering_trees_outdir = "./ClusteringTrees"
0018 if (sys.argv[0] == "cmsRun"):
0019     mkdir_p(clustering_trees_outdir)
0020
0021 # Base name for the output files
0022 trees_basename = "clustree"
0023
0024 # Configure the FFTJet tree dumper module
0025 fftjetTreeDumper = cms.EDAnalyzer(
0026     "FFTJetTreeDump",
0027     #
0028     # Label for the input clustering tree (either sparse or dense)
0029     treeLabel = cms.InputTag("fftjetpatreco", "FFTJetPatternRecognition"),
0030     #
0031     # Prefix for the output trees
0032     outputPrefix = cms.string(clustering_trees_outdir + '/' + trees_basename),
0033     #
0034     # Eta limit for the plot
0035     etaMax = cms.double(fftjet_standard_eta_range),
0036     #
0037     # Do we have the complete event at the lowest tree scale?
0038     insertCompleteEvent = cms.bool(fftjet_insert_complete_event),
0039     completeEventScale = cms.double(fftjet_complete_event_scale),
0040     #
0041     # The initial set of scales used by the pattern recognition stage.
0042     # This is also the final set unless clustering tree construction
0043     # is adaptive. Needed here for reading back non-adaptive trees.
0044     InitialScales = fftjet_patreco_scales_50,
0045     #
0046     # Clustering tree distance functor
0047     TreeDistanceCalculator = fftjet_fixed_bandwidth_distance,
0048     #
0049     # Which quantity will be mapped into OpenDX glyph size?
0050     GlyphSize = cms.PSet(
0051         Class = cms.string("ScaledMagnitude2")
0052     ),
0053     #
0054     # Which quantity will be mapped into OpenDX glyph color?
0055     GlyphColor = cms.PSet(
0056         Class = cms.string("ScaledHessianDet")
0057     )
0058 )