Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Run3gridpatch #3788

Closed
wants to merge 8 commits into from
Closed

Run3gridpatch #3788

Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
6dc1c83
run3 gridpack runcmsgrid patch
efeyazgan Oct 28, 2024
b1afa88
run3 gridpack runcmsgrid patch
efeyazgan Oct 28, 2024
5b8c2de
run3 gridpack runcmsgrid patch
efeyazgan Oct 28, 2024
23fa821
run3 gridpack runcmsgrid patch
efeyazgan Oct 28, 2024
7863f3a
run3 runcmsgrid patch
efeyazgan Oct 28, 2024
2da6264
run3 runcmsgrid patch
efeyazgan Oct 28, 2024
5e9fdf3
run3 runcmsgrid patch
efeyazgan Oct 28, 2024
2dab351
debug run3 patch
efeyazgan Oct 28, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
110 changes: 78 additions & 32 deletions bin/utils/request_fragment_check.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
import glob
import json
import ast
import subprocess
from datetime import datetime
from json import dumps

Expand Down Expand Up @@ -313,34 +314,74 @@ def ul_consistency(dn,pi,jhu_gp):
if not error_ul: print("UL consistency check is OK.")
return warning_ul,error_ul

def gridpack_copy(gridpack_eos_path,pi):
error_gp_copy = []
targz_flag = 0
if "Run3" in pi:
copy_name = "_original_Run3_wo_runcmsgrid_sys_patch"
else:
copy_name = "_original"
if ".tar.gz" in gridpack_eos_path:
targz_flag = 1
gridpack_eos_path_backup = gridpack_eos_path.replace('.tar.gz',copy_name+'.tar.gz')
if ".tgz" in gridpack_eos_path: gridpack_eos_path_backup = gridpack_eos_path.replace('.tgz',copy_name+'.tgz')
if ".tar.xz" in gridpack_eos_path:
gridpack_eos_path_backup = gridpack_eos_path.replace('.tar.xz',copy_name+'.tar.xz')
targz_flag = 2
if not os.path.exists(gridpack_eos_path_backup):
print("Backup gridpack does not exist.")
print("Copying "+gridpack_eos_path+" to "+gridpack_eos_path_backup+" before patching runcms.grid")
os.system('cp -n -p '+gridpack_eos_path+' '+gridpack_eos_path_backup)
md5_1 = os.popen('md5sum'+' '+gridpack_eos_path).read().split(' ')[0]
md5_2 = os.popen('md5sum'+' '+gridpack_eos_path_backup).read().split(' ')[0]
if md5_1 == md5_2:
print("Backup and original file checksums are equal.")
else:
error_gp_copy.append("backup gridpack has a problem.")
print("Backup gridpack: "+gridpack_eos_path_backup)
return error_gp_copy

def gridpack_repack_and_copy(gridpack_eos_path,my_path,pi):
error_gridpack_repack = []
gp_extension = ".tar.xz"
if ".tar.gz" in gridpack_eos_path:
gp_extension = ".tar.gz"
if ".tgz" in gridpack_eos_path:
gp_extension = ".tgz"
gp_name = "gridpack"+gp_extension
print("re-tarring to "+gp_name)
cur_dir = os.getcwd()
os.chdir(my_path+'/'+pi)
print(os.getcwd())
os.environ['XZ_OPT'] = "--lzma2=preset=9,dict=512MiB"
os.system('XZ_OPT="$XZ_OPT" tar -cJpf '+gp_name+' --exclude='+gp_name+' --exclude='+pi+' ./*')
print('cp -p '+gp_name+' '+gridpack_eos_path)
os.system('cp -p '+gp_name+' '+gridpack_eos_path)
md5_1 = os.popen('md5sum '+gp_name).read().split(' ')[0]
md5_2 = os.popen('md5sum'+' '+gridpack_eos_path).read().split(' ')[0]
print(gp_name)
print(gridpack_eos_path)
print("Checksums = [local=",md5_1,"copied=",md5_2,"]")
print(type(md5_1),type(md5_2))
if md5_1 == md5_2:
print("Updated gridpack copied succesfully.")
else:
error_gridpack_repack.append("There was a problem copying in the updated gridpack to eos.")
os.chdir(cur_dir)
print(os.getcwd())
return error_gridpack_repack

def xml_check_and_patch(f,cont,gridpack_eos_path,my_path,pi):
xml = str(re.findall('xmllint.*',cont))
cur_dir = os.getcwd()
warning_xml = []
error_xml = []
if "stream" not in xml or len(xml) < 3:
targz_flag = 0
if "stream" not in xml and len(xml) > 3:
warning_xml.append(" --stream option is missing in XMLLINT, will update runcmsgrid.")
if len(xml) < 3:
warning_xml.append("[WARNING] XMLLINT does not exist in runcmsgrid, will update it.")
if ".tar.gz" in gridpack_eos_path:
targz_flag = 1
gridpack_eos_path_backup = gridpack_eos_path.replace('.tar.gz','_original.tar.gz')
if ".tgz" in gridpack_eos_path: gridpack_eos_path_backup = gridpack_eos_path.replace('.tgz','_original.tgz')
if ".tar.xz" in gridpack_eos_path:
gridpack_eos_path_backup = gridpack_eos_path.replace('.tar.xz','_original.tar.xz')
targz_flag = 2
if not os.path.exists(gridpack_eos_path_backup):
print("Backup gridpack does not exist.")
print("Copying "+gridpack_eos_path+" to "+gridpack_eos_path_backup+" before patching runcms.grid")
os.system('cp -n -p '+gridpack_eos_path+' '+gridpack_eos_path_backup)
md5_1 = os.popen('md5sum'+' '+gridpack_eos_path).read().split(' ')[0]
md5_2 = os.popen('md5sum'+' '+gridpack_eos_path_backup).read().split(' ')[0]
if md5_1 == md5_2:
print("Backup and original file checksums are equal.")
else:
error_xml.append("backup gridpack has a problem.")
error_xml.append(gridpack_copy(gridpack_eos_path,pi))
print("Updating XMLLINT line in runcmsgrid.")
os.chdir(my_path+'/'+pi)
if "stream" not in xml and len(xml) > 3: cont = re.sub("xmllint","xmllint --stream",cont)
Expand All @@ -351,19 +392,7 @@ def xml_check_and_patch(f,cont,gridpack_eos_path,my_path,pi):
f.seek(0)
f.write(cont)
f.truncate()
if targz_flag == 0: gridpackname = "gridpack.tgz"
if targz_flag == 1: gridpackname = "gridpack.tar.gz"
if targz_flag == 2: gridpackname = "gridpack.tar.xz"
os.chdir(my_path+'/'+pi)
os.system('tar cfJ '+gridpackname+' ./* --exclude='+gridpackname+' --exclude='+pi)
os.system('cp '+gridpackname+' '+gridpack_eos_path)
md5_1 = os.popen('md5sum '+gridpackname).read().split(' ')[0]
md5_2 = os.popen('md5sum'+' '+gridpack_eos_path).read().split(' ')[0]
if md5_1 == md5_2:
print("Updated gridpack copied succesfully.")
else:
error_xml.append("There was a problem copying in the updated gridpack to eos.")
os.chdir(cur_dir)
error_xml = gridpack_repack_and_copy(gridpack_eos_path,my_path,pi)
return warning_xml,error_xml

def evtgen_check(fragment):
Expand Down Expand Up @@ -1150,7 +1179,7 @@ def root_requests_from_ticket(ticket_prepid, include_docs=False):
warnings.append(pf[1]+" "+pf[2]+" You may try to request more events per phase-space region in the gridpack.")
if os.path.isfile(gp_log_loc) and ('madgraph' in dn.lower() or 'amcatnlo' in dn.lower()):
print("------------------------------------------------------------------------------------")
print("Summary for madgraph for experts fron gridpack log (cross section BEFORE matching (if there is matching/merging)):")
print("Summary for madgraph for experts from gridpack log (cross section BEFORE matching (if there is matching/merging)):")
print(os.popen('grep Summary '+gp_log_loc+' -A 5 -B 1').read())
print("------------------------------------------------------------------------------------")
if mg_gp:
Expand Down Expand Up @@ -1514,6 +1543,23 @@ def root_requests_from_ticket(ticket_prepid, include_docs=False):
if mg5_aMC_version >= 260:
mg_lo = int(os.popen('grep "systematics" '+str(runcmsgrid_file)+' | grep -c madevent').read())
mg_nlo = int(os.popen('grep "systematics" '+str(runcmsgrid_file)+' | grep -c aMCatNLO').read())
if mg_lo: print("LO gridpack")
if mg_nlo: print("NLO gridpack")
if "Run3" or "RunII" in pi:
if int(os.popen('grep -c "systematics $runlabel" '+str(runcmsgrid_file)).read()):
if int(os.popen('grep -c "Encounter Error in Running Systematics Module" '+str(runcmsgrid_file)).read()) < 1:
print("-----------------------------------------")
print("runcmsgrid script patch for Run3 missing!")
print("-----------------------------------------")
print("I will patch the runcmsgrid script.")
err_gpr = gridpack_copy(gridpack_eos_path,pi)
errors.extend(err_gpr)
if mg_nlo:
os.system("patch "+runcmsgrid_file+" < /eos/cms/store/group/phys_generator/cvmfs/gridpacks/mg_amg_patch/runcmsgrid_systematics_NLO.patch")
if mg_lo:
os.system("patch "+runcmsgrid_file+" < /eos/cms/store/group/phys_generator/cvmfs/gridpacks/mg_amg_patch/runcmsgrid_systematics_LO.patch")
err_gpr = gridpack_repack_and_copy(gridpack_eos_path,my_path,pi)
errors.extend(err_gpr)
if mg5_aMC_version < 260:
mg_lo = int(os.popen('grep -c syscalc '+str(runcmsgrid_file)).read())
if mg_nlo > 0:
Expand Down
Loading