Merge pull request #43 from syuvivida/eiko_addpacking

add packing so that one could access softdrop subjets via jetptr->sub…
cms-jet · Oct 11, 2016 · 48e7d4f · 48e7d4f
2 parents 65c15d1 + a3557da
commit 48e7d4f
Showing 1 changed file with 35 additions and 1 deletion.
diff --git a/python/jetToolbox_cff.py b/python/jetToolbox_cff.py
@@ -394,7 +394,7 @@ def jetToolbox( proc, jetType, jetSequence, outputFile,
 				jetAlgorithm = algorithm, 
 				useExplicitGhosts=True,
 				R0= cms.double(jetSize),
-				#zcut=zCutSD, 
+				zcut=zCutSD, 
 				beta=betaCut,
 				doAreaFastjet = cms.bool(True),
 				writeCompound = cms.bool(True),
@@ -484,6 +484,24 @@ def jetToolbox( proc, jetType, jetSequence, outputFile,
 			toolsUsed.append( 'selectedPatJets'+jetALGO+'PF'+PUMethod+'SoftDropPacked' )
 			toolsUsed.append( 'selectedPatJets'+jetALGO+'PF'+PUMethod+'SoftDropSubjets' )
 
+                        ## Pack fat jets with subjets
+			setattr( proc, 'packedPatJets'+jetALGO+'PF'+PUMethod+'SoftDrop', 
+				 cms.EDProducer("JetSubstructurePacker",
+						jetSrc=cms.InputTag('selectedPatJets'+jetALGO+'PF'+PUMethod),
+						distMax = cms.double( jetSize ),
+						fixDaughters = cms.bool(False),
+						algoTags = cms.VInputTag(
+						cms.InputTag('selectedPatJets'+jetALGO+'PF'+PUMethod+'SoftDropPacked')
+						), 
+						algoLabels =cms.vstring('SoftDrop')
+						)
+				 )
+			jetSeq += getattr(proc, 'packedPatJets'+jetALGO+'PF'+PUMethod+'SoftDrop')
+			elemToKeep += [ 'keep *_packedPatJets'+jetALGO+'PF'+PUMethod+'SoftDrop_*_*' ]
+			toolsUsed.append( 'packedPatJets'+jetALGO+'PF'+PUMethod+'SoftDrop' )
+
+
+
 	if addPruning or addPrunedSubjets: 
 
 		setattr( proc, jetalgo+'PFJets'+PUMethod+'Pruned', 
@@ -576,6 +594,22 @@ def jetToolbox( proc, jetType, jetSequence, outputFile,
 			toolsUsed.append( 'selectedPatJets'+jetALGO+'PF'+PUMethod+'PrunedPacked' )
 			toolsUsed.append( 'selectedPatJets'+jetALGO+'PF'+PUMethod+'PrunedSubjets' )
 
+                        ## Pack fat jets with subjets
+			setattr( proc, 'packedPatJets'+jetALGO+'PF'+PUMethod+'Pruned', 
+				 cms.EDProducer("JetSubstructurePacker",
+						jetSrc=cms.InputTag('selectedPatJets'+jetALGO+'PF'+PUMethod),
+						distMax = cms.double( jetSize ),
+						fixDaughters = cms.bool(False),
+						algoTags = cms.VInputTag(
+						cms.InputTag('selectedPatJets'+jetALGO+'PF'+PUMethod+'PrunedPacked')
+						), 
+						algoLabels =cms.vstring('Pruned')
+						)
+				 )
+			jetSeq += getattr(proc, 'packedPatJets'+jetALGO+'PF'+PUMethod+'Pruned')
+			elemToKeep += [ 'keep *_packedPatJets'+jetALGO+'PF'+PUMethod+'Pruned_*_*' ]
+			toolsUsed.append( 'packedPatJets'+jetALGO+'PF'+PUMethod+'Pruned' )
+
 
 	if addTrimming: