Add new analysis scripts and update MyZPeak_cfg.py for improved functionality

Author: alefisico

episodes/files/ZPeakAnalysis.zip

@@ -1,26 +1,42 @@
+ 
 import FWCore.ParameterSet.Config as cms
 from FWCore.ParameterSet.VarParsing import VarParsing
 
 options = VarParsing ('analysis')
 
+# Adding a custom command line option for Pt
+options.inputFiles = "root://cmseos.fnal.gov//store/user/cmsdas/2026/short_exercises/cat/datasets/MINIAODSIM/RunIISummer20UL17MiniAODv2-106X_mc2017_realistic_v9-v2/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/2C5565D7-ADE5-2C40-A0E5-BDFCCF40640E.root"  #### for cmslpc
+#options.inputFiles = "root://xrootd-cms.infn.it//store/group/cat/datasets/MINIAODSIM/RunIISummer20UL17MiniAODv2-106X_mc2017_realistic_v9-v2/DYJetsToLL_M-50_TuneCP5_13TeV-amcatnloFXFX-pythia8/2C5565D7-ADE5-2C40-A0E5-BDFCCF40640E.root"  #### for cern
+options.register('minPt',
+                 20.0, # default value
+                 VarParsing.multiplicity.singleton,
+                 VarParsing.varType.float,
+                 "Minimum Muon Pt threshold")
+
+options.parseArguments()
+
 process = cms.Process("Test")
 
+process.load("FWCore.MessageService.MessageLogger_cfi")
+process.MessageLogger.cerr.FwkReport.reportEvery = 1000
+
 process.source = cms.Source("PoolSource",
-  fileNames = cms.untracked.vstring (options.inputFiles)
+    fileNames = cms.untracked.vstring(options.inputFiles)
 )
 
-process.MessageLogger = cms.Service("MessageLogger")
 process.maxEvents = cms.untracked.PSet(
-  input = cms.untracked.int32(10000)
+    input = cms.untracked.int32(options.maxEvents)
 )
 
 process.analyzeBasicPat = cms.EDAnalyzer("MyZPeakAnalyzer",
-  muonSrc = cms.untracked.InputTag("slimmedMuons"),
-  elecSrc = cms.untracked.InputTag("slimmedElectrons"),
+    muonSrc = cms.untracked.InputTag("slimmedMuons"),
+    elecSrc = cms.untracked.InputTag("slimmedElectrons"),
+    # New parameter added here
+    minMuonPt = cms.double(options.minPt)
 )
 
 process.TFileService = cms.Service("TFileService",
-                                   fileName = cms.string('myZPeak.root')
-                                   )
+    fileName = cms.string('myZPeak.root')
+)
 
-process.p = cms.Path(process.analyzeBasicPat)
+process.p = cms.Path(process.analyzeBasicPat)

episodes/files/ZPeakAnalysis/test/MyZPeak_cfg.py

@@ -0,0 +1,62 @@
+#!/usr/bin/env python
+import sys
+import argparse
+
+def parse_integral_file(filename):
+    """Parses the text file and returns a dictionary of {variable: value}."""
+    data = {}
+    try:
+        with open(filename, 'r') as f:
+            for line in f:
+                if ':' in line:
+                    parts = line.split(':')
+                    key = parts[0].replace('Integral of ', '').strip()
+                    value = float(parts[1].strip())
+                    data[key] = value
+        return data
+    except IOError:
+        print "Error: File '{0}' not found.".format(filename)
+        sys.exit(1)
+    except Exception as e:
+        print "Error parsing '{0}': {1}".format(filename, e)
+        sys.exit(1)
+
+def compare(file1, file2, tolerance=1e-6):
+    data1 = parse_integral_file(file1)
+    data2 = parse_integral_file(file2)
+
+    all_keys = set(data1.keys()).union(set(data2.keys()))
+    match = True
+
+    print "{0:<20} | {1:<15} | {2:<15} | {3}".format('Variable', 'File 1', 'File 2', 'Status')
+    print "-" * 70
+
+    for key in sorted(all_keys):
+        val1 = data1.get(key, None)
+        val2 = data2.get(key, None)
+
+        if val1 is None or val2 is None:
+            status = "MISSING"
+            match = False
+        elif abs(val1 - val2) <= tolerance:
+            status = "PASS"
+        else:
+            status = "FAIL"
+            match = False
+
+        print "{0:<20} | {1:<15} | {2:<15} | {3}".format(key, str(val1), str(val2), status)
+
+    return match
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Compare CMSSW integral outputs.")
+    parser.add_argument("file1", help="First txt file (reference)")
+    parser.add_argument("file2", help="Second txt file (new run)")
+    args = parser.parse_args()
+
+    if compare(args.file1, args.file2):
+        print "\nSUCCESS: The files match."
+        sys.exit(0)
+    else:
+        print "\nFAILURE: Differences detected between the runs."
+        sys.exit(1)

episodes/files/ZPeakAnalysis/test/check_cutflows_cmssw10.py

@@ -0,0 +1,46 @@
+#!/usr/bin/env python
+import ROOT
+import sys
+
+# Define the input and output filenames
+input_file = 'myZPeak.root'
+output_txt = 'number_of_events.txt'
+
+# Open the ROOT file
+f = ROOT.TFile.Open(input_file)
+if not f or f.IsZombie():
+    print "Error: Could not open {0}".format(input_file)
+    sys.exit(1)
+
+# In TFileService, histograms are usually inside a directory
+# named after the module label in your python config.
+# Based on your config, the label is 'analyzeBasicPat'
+dir_name = "analyzeBasicPat"
+hist_dir = f.Get(dir_name)
+
+if not hist_dir:
+    print "Error: Directory {0} not found in {1}".format(dir_name, input_file)
+    f.ls() # List contents to help debugging
+    sys.exit(1)
+
+# Retrieve the histograms
+h_muonMult = hist_dir.Get("muonMult")
+h_eleMult  = hist_dir.Get("eleMult")
+h_mumuMass = hist_dir.Get("mumuMass")
+
+# Calculate integrals
+# Integral() returns the sum of bin contents (excluding under/overflow by default)
+muon_integral = h_muonMult.Integral()
+ele_integral  = h_eleMult.Integral()
+mumuMass_integral = h_mumuMass.Integral()
+
+# Write to a text file
+with open(output_txt, 'w') as f_out:
+    f_out.write("muonMult: {0}\n".format(muon_integral))
+    f_out.write("eleMult:  {0}\n".format(ele_integral))
+    f_out.write("mumuMass: {0}\n".format(mumuMass_integral))
+
+print "Successfully wrote integrals to {0}".format(output_txt)
+
+# Close ROOT file
+f.Close()