Merge pull request #12685 from bendavid/lheheaderscript_53x

Add script to dump LHE header to file

Author: cmsbuild

GeneratorInterface/LHEInterface/plugins/LHEWriter.cc

@@ -26,12 +26,15 @@ class LHEWriter : public edm::EDAnalyzer {
 	virtual void beginRun(const edm::Run &run, const edm::EventSetup &es);
 	virtual void endRun(const edm::Run &run, const edm::EventSetup &es);
 	virtual void analyze(const edm::Event &event, const edm::EventSetup &es);
+        
+        std::string output;
 
     private:
 	std::ofstream	file;
 };
 
-LHEWriter::LHEWriter(const edm::ParameterSet &params)
+LHEWriter::LHEWriter(const edm::ParameterSet &params) :
+  output(params.getUntrackedParameter<std::string>("output"))
 {
 }
 
@@ -43,8 +46,12 @@ void LHEWriter::beginRun(const edm::Run &run, const edm::EventSetup &es)
 {
 	edm::Handle<LHERunInfoProduct> product;
 	run.getByLabel("source", product);
+        
+        if (!product.isValid()) {
+          run.getByLabel("externalLHEProducer", product);
+        }
 
-	file.open("writer.lhe", std::fstream::out | std::fstream::trunc);
+	file.open(output.c_str(), std::fstream::out | std::fstream::trunc);
 	std::copy(product->begin(), product->end(),
 	          std::ostream_iterator<std::string>(file));
 }
@@ -59,6 +66,10 @@ void LHEWriter::analyze(const edm::Event &event, const edm::EventSetup &es)
 {
 	edm::Handle<LHEEventProduct> product;
 	event.getByLabel("source", product);
+        
+        if (!product.isValid()) {
+          event.getByLabel("externalLHEProducer", product);
+        }        
 
 	std::copy(product->begin(), product->end(),
 	          std::ostream_iterator<std::string>(file));

GeneratorInterface/LHEInterface/scripts/dumpLHEHeader.py

@@ -0,0 +1,43 @@
+#!/usr/bin/env cmsRun
+import FWCore.ParameterSet.Config as cms
+
+from FWCore.ParameterSet.VarParsing import VarParsing
+
+options = VarParsing ('analysis')
+
+# add a list of strings for events to process
+options.register ('input',
+				  '',
+				  VarParsing.multiplicity.list,
+				  VarParsing.varType.string,
+				  "Input File")
+
+options.register ('output',
+				  'writer.lhe',
+				  VarParsing.multiplicity.singleton,
+				  VarParsing.varType.string,
+				  "Output File")
+
+options.parseArguments()
+
+inFileName = cms.untracked.vstring (options.input)
+outFileName = cms.untracked.string (options.output)
+
+process = cms.Process("Writer")
+
+process.maxEvents = cms.untracked.PSet(input = cms.untracked.int32(-1))
+
+process.source = cms.Source("PoolSource",
+        processingMode = cms.untracked.string('Runs'),
+	fileNames = cms.untracked.vstring(inFileName)
+)
+
+process.load("FWCore.MessageService.MessageLogger_cfi")
+process.MessageLogger.cerr.threshold = 'INFO'
+
+process.writer = cms.EDAnalyzer("LHEWriter",
+                                output=outFileName)
+
+process.path = cms.Path(process.writer)
+
+process.schedule = cms.Schedule(process.path)

GeneratorInterface/LHEInterface/test/testWriter_cfg.py

@@ -12,6 +12,7 @@
 process.load("FWCore.MessageService.MessageLogger_cfi")
 process.MessageLogger.cerr.threshold = 'INFO'
 
-process.writer = cms.EDAnalyzer("LHEWriter")
+process.writer = cms.EDAnalyzer("LHEWriter",
+                                outputFile='writer.lhe')
 
 process.outpath = cms.EndPath(process.writer)