RecoEgamma/test/electronValidationCheck_Env.py

0001 #! /usr/bin/env python3
0002 # -*-coding: utf-8 -*-
0003
0004 import os, sys
0005
0006
0007 class env:
0008     def checkSample(self):
0009         if ('DD_SAMPLE' not in os.environ) or (os.environ['DD_SAMPLE'] == ''):
0010             print('no existing sample name')
0011             print('nb of element : ', len(sys.argv))
0012             print('0 : ', sys.argv[0])
0013             print('1 : ', sys.argv[1])
0014             if (len(sys.argv) > 1):  # no else part since if sample does not exist, we had quit previously
0015                 sampleName = str(sys.argv[1])
0016                 os.environ['DD_SAMPLE'] = 'RelVal' + sampleName
0017                 print('Sample name:', sampleName, ' - ', os.environ['DD_SAMPLE'])
0018             else:
0019                 print('====================')
0020                 print('no sample name, quit')
0021                 print('====================')
0022                 quit()
0023         else:
0024             print('Existing Sample name:', sampleName, ' - ', os.environ['DD_SAMPLE'])
0025
0026     def beginTag(self):
0027         beginTag = 'Phase2'
0028         #beginTag = 'Run3'
0029         return beginTag
0030
0031     def dd_tier(self):
0032         dd_tier = 'GEN-SIM-RECO'
0033         dd_tier = 'MINIAODSIM'
0034         return dd_tier
0035
0036     def tag_startup(self):
0037         #tag_startup = '125X_mcRun3_2022_realistic_v3'
0038         tag_startup = '140X_mcRun4_realistic_v4_STD_2026D110_noPU'
0039         # tag_startup = '113X_mcRun3_2021_realistic_v7'
0040         return tag_startup
0041
0042     def data_version(self):
0043         data_version = 'v1'
0044         return data_version
0045
0046     def test_global_tag(self):
0047         test_global_tag = self.tag_startup()
0048         return test_global_tag
0049
0050     def dd_cond(self):
0051         # dd_cond = 'PU25ns_' + self.test_global_tag() + '-' + self.data_version() # PU
0052         dd_cond = self.test_global_tag() + '-' + self.data_version()  # noPU
0053         return dd_cond
0054
0055     def checkValues(self):
0056         print('-----')
0057         print(self.dd_tier())
0058         print(self.tag_startup())
0059         print(self.data_version())
0060         print(self.test_global_tag())
0061         print(self.dd_cond())
0062         print('-----')
0063
0064         os.environ['beginTag'] = self.beginTag()
0065
0066         if ('DD_TIER' not in os.environ) or (os.environ['DD_TIER'] == ''):
0067             os.environ['DD_TIER'] = self.dd_tier()  # 'GEN-SIM-RECO'
0068         if 'TAG_STARTUP' not in os.environ:  # TAG_STARTUP from OvalFile
0069             os.environ['TAG_STARTUP'] = self.tag_startup()  # '93X_upgrade2023_realistic_v0_D17PU200'
0070         if 'DATA_VERSION' not in os.environ:  # DATA_VERSION from OvalFile
0071             os.environ['DATA_VERSION'] = self.data_version()  # 'v1'
0072         if 'TEST_GLOBAL_TAG' not in os.environ:  # TEST_GLOBAL_TAG from OvalFile
0073             os.environ['TEST_GLOBAL_TAG'] = self.test_global_tag()  # os.environ['TAG_STARTUP']
0074         if ('DD_COND' not in os.environ) or (os.environ['DD_COND'] == ''):
0075             os.environ[
0076                 'DD_COND'] = self.dd_cond()  # 'PU25ns_' + os.environ['TEST_GLOBAL_TAG'] + '-' + os.environ['DATA_VERSION']
0077
0078         os.environ['DD_RELEASE'] = os.environ['CMSSW_VERSION']
0079         # os.environ['DD_RELEASE'] = "CMSSW_11_3_0_pre3"
0080
0081         print('=====')
0082         if ('DD_SAMPLE_OUT' not in os.environ) or (os.environ['DD_SAMPLE_OUT'] == ''):
0083             os.environ['DD_SAMPLE_OUT'] = os.environ['DD_SAMPLE'].replace("RelVal", "ValFull")
0084         print('=====')
0085
0086         os.environ['DD_SOURCE'] = '/eos/cms/store/relval/' + os.environ['DD_RELEASE'] + '/' + os.environ[
0087             'DD_SAMPLE'] + '/' + os.environ['DD_TIER'] + '/' + os.environ['DD_COND']
0088         os.environ['data'] = '/' + os.environ['DD_SAMPLE'] + '/' + os.environ['DD_RELEASE'] + '-' + os.environ[
0089             'DD_COND'] + '/' + os.environ['DD_TIER']
0090         os.environ['outputFile'] = 'electronHistos.' + os.environ['DD_SAMPLE_OUT'] + '_gedGsfE.root'
0091         if (os.environ['DD_TIER'] == 'MINIAODSIM'):
0092             os.environ['outputFile'] = 'electronHistos.' + os.environ['DD_SAMPLE_OUT'] + '_miniAOD.root'
0093         if ('inputPostFile' not in os.environ) or (os.environ['inputPostFile'] == ''):
0094             print('inputPostFile : %s' % os.environ['outputFile'])
0095             os.environ['inputPostFile'] = os.environ['outputFile']
0096
0097         print('DD_RELEASE', os.environ['DD_RELEASE'])
0098         print('DD_SAMPLE', os.environ['DD_SAMPLE'])
0099         print('DD_SAMPLE_OUT', os.environ['DD_SAMPLE_OUT'])
0100         print('DD_COND', os.environ['DD_COND'])
0101         print('DD_TIER', os.environ['DD_TIER'])
0102         print('DD_SOURCE', os.environ['DD_SOURCE'])
0103         print('data', os.environ['data'])
0104         print('outputFile    :', os.environ['outputFile'])
0105         print('inputPostFile :', os.environ['inputPostFile'])
0106         print('beginTag : ', self.beginTag())