CMS 3D CMS Logo

Functions
tools Namespace Reference

Functions

def addCrabInputFile (crabCfg, inputFile)
 
def copyFilesFromCastor (castor_dir, output_dir, type='root')
 
def copyFilesLocal (dir, output_dir, type='root')
 
def create_single_iov_db (inputs, run_number, output_db)
 
def dqmWorkflowName (datasetpath, type, rev=1)
 
def get_iovs (db, tag)
 
def get_process_object (cfg)
 
def get_tags (global_tag, records)
 
def getDatasetStr (datasetpath)
 
def haddInCastor (castor_dir, result_file, type='root', prefix='rfio:', suffix=None)
 
def haddLocal (dir, result_file, type='root')
 
def listFilesInCastor (castor_dir, type='root', prefix='rfio:')
 
def listFilesLocal (dir, type='root')
 
def loadCmsProcess (psetPath)
 
def loadCmsProcessFile (psetName)
 
def loadCrabCfg (cfgName=None)
 
def loadCrabDefault (crabCfg, config)
 
def make_unique_runranges (ali_producer)
 
def parseInput (inputFields, requiredFields=())
 
def prependPaths (process, seqname)
 
def replaceTemplate (template, opts)
 
def run_checked (cmd)
 
def setGridEnv (cmssw_dir)
 
def writeCfg (process, dir, psetName)
 
def writeCfgPkl (process, dir, psetName)
 

Function Documentation

def tools.addCrabInputFile (   crabCfg,
  inputFile 
)

Definition at line 183 of file tools.py.

Referenced by DTDQMValidation.DTDQMValidation.initCrab(), DTAnalysisResiduals.DTAnalysisResiduals.initCrab(), DTTTrigValid.DTTTrigValid.initCrab(), DTResidualCalibration.DTResidualCalibration.initCrab(), DTVDriftSegmentCalibration.DTVDriftSegmentCalibration.initCrab(), and DTVDriftMeanTimerCalibration.DTVDriftMeanTimerCalibration.initCrab().

183 def addCrabInputFile(crabCfg,inputFile):
184  additionalInputFiles = ''
185  if crabCfg.has_option('USER','additional_input_files'):
186  additionalInputFiles = crabCfg.get('USER','additional_input_files')
187 
188  if additionalInputFiles: additionalInputFiles += ',%s' % inputFile
189  else: additionalInputFiles = inputFile
190 
191  crabCfg.set('USER','additional_input_files',additionalInputFiles)
192 
193  return crabCfg
194 
def addCrabInputFile(crabCfg, inputFile)
Definition: tools.py:183
def tools.copyFilesFromCastor (   castor_dir,
  output_dir,
  type = 'root' 
)

Definition at line 55 of file tools.py.

References listFilesInCastor().

Referenced by DTCalibrationWorker.DTCalibrationWorker.runNoiseWorkflow().

55 def copyFilesFromCastor(castor_dir,output_dir,type='root'):
56  from subprocess import call
57  files = listFilesInCastor(castor_dir,type,'')
58 
59  print "Copying files from %s to %s" % (castor_dir,output_dir)
60  for item in files:
61  cmd = ['rfcp',item,output_dir]
62  print "..." + item
63  retcode = call(cmd)
64  if retcode != 0: raise RuntimeError('Error in copying file %s to directory %s' % (item,output_dir))
65 
66  return 0
67 
def listFilesInCastor(castor_dir, type='root', prefix='rfio:')
Definition: tools.py:30
def copyFilesFromCastor(castor_dir, output_dir, type='root')
Definition: tools.py:55
def tools.copyFilesLocal (   dir,
  output_dir,
  type = 'root' 
)

Definition at line 68 of file tools.py.

References listFilesLocal().

Referenced by DTCalibrationWorker.DTCalibrationWorker.runNoiseWorkflow().

68 def copyFilesLocal(dir,output_dir,type='root'):
69  if not dir: raise ValueError('Please specify valid dir')
70  if not output_dir: raise ValueError('Please specify valid output dir')
71 
72  from subprocess import call
73  files = listFilesLocal(dir,type)
74  cmd = ['cp']
75  cmd.extend(files)
76  cmd.append(output_dir)
77  print cmd
78  retcode = call(cmd)
79  return retcode
80 
def copyFilesLocal(dir, output_dir, type='root')
Definition: tools.py:68
def listFilesLocal(dir, type='root')
Definition: tools.py:42
def tools.create_single_iov_db (   inputs,
  run_number,
  output_db 
)
Create an sqlite file with single-IOV tags for alignment payloads.

Arguments:
- `inputs`: dictionary with input needed for payload extraction
- `run_number`: run for which the IOVs are selected
- `output_db`: name of the output sqlite file

Definition at line 9 of file tools.py.

References join(), run_checked(), and harvestTrackValidationPlots.str.

9 def create_single_iov_db(inputs, run_number, output_db):
10  """Create an sqlite file with single-IOV tags for alignment payloads.
11 
12  Arguments:
13  - `inputs`: dictionary with input needed for payload extraction
14  - `run_number`: run for which the IOVs are selected
15  - `output_db`: name of the output sqlite file
16  """
17 
18  # find the IOV containing `run_number`
19  for record,tag in inputs.iteritems():
20  run_is_covered = False
21  for iov in reversed(tag["iovs"]):
22  if iov <= run_number:
23  tag["since"] = str(iov)
24  run_is_covered = True
25  break
26  if not run_is_covered:
27  msg = ("Run number {0:d} is not covered in '{1:s}' ({2:s}) from"
28  " '{3:s}'.".format(run_number, tag["tag"], record,
29  global_tag))
30  print msg
31  print "Aborting..."
32  sys.exit(1)
33 
34  result = {}
35  if os.path.exists(output_db): os.remove(output_db)
36  for record,tag in inputs.iteritems():
37  result[record] = {"connect": "sqlite_file:"+output_db,
38  "tag": "_".join([tag["tag"], tag["since"]])}
39  cmd = ("conddb_import",
40  "-f", "frontier://PromptProd/cms_conditions",
41  "-c", result[record]["connect"],
42  "-i", tag["tag"],
43  "-t", result[record]["tag"],
44  "-b", str(run_number),
45  "-e", str(run_number))
46  run_checked(cmd)
47  if len(inputs) > 0:
48  run_checked(["sqlite3", output_db, "update iov set since=1"])
49 
50  return result
51 
52 
def create_single_iov_db(inputs, run_number, output_db)
Definition: tools.py:9
static std::string join(char **cmd)
Definition: RemoteFile.cc:18
def run_checked(cmd)
Definition: tools.py:53
def tools.dqmWorkflowName (   datasetpath,
  type,
  rev = 1 
)

Definition at line 23 of file tools.py.

Referenced by DTDqm.DTDqm.initProcess(), and DTDQMHarvesting.DTDQMHarvesting.initProcess().

23 def dqmWorkflowName(datasetpath,type,rev=1):
24  workflowName = datasetpath
25  sections = workflowName.split('/')[1:]
26  workflowName = '/%s/%s-%s-rev%d/%s' % (sections[0],sections[1],type,rev,sections[2])
27 
28  return workflowName
29 
def dqmWorkflowName(datasetpath, type, rev=1)
Definition: tools.py:23
def tools.get_iovs (   db,
  tag 
)
Retrieve the list of IOVs from `db` for `tag`.

Arguments:
- `db`: database connection string
- `tag`: tag of database record

Definition at line 136 of file tools.py.

References Vispa.Plugins.EdmBrowser.EdmDataAccessor.all(), ALCARECOTkAlBeamHalo_cff.filter, createfilelist.int, and python.rootplot.root2matplotlib.replace().

136 def get_iovs(db, tag):
137  """Retrieve the list of IOVs from `db` for `tag`.
138 
139  Arguments:
140  - `db`: database connection string
141  - `tag`: tag of database record
142  """
143 
144  db = db.replace("sqlite_file:", "").replace("sqlite:", "")
145 
146  con = conddb.connect(url = conddb.make_url(db))
147  session = con.session()
148  IOV = session.get_dbtype(conddb.IOV)
149 
150  iovs = set(session.query(IOV.since).filter(IOV.tag_name == tag).all())
151  if len(iovs) == 0:
152  print "No IOVs found for tag '"+tag+"' in database '"+db+"'."
153  sys.exit(1)
154 
155  session.close()
156 
157  return sorted([int(item[0]) for item in iovs])
def tools.get_process_object (   cfg)
Returns cms.Process object defined in `cfg`.

Arguments:
- `cfg`: path to CMSSW config file

Definition at line 69 of file tools.py.

70  """Returns cms.Process object defined in `cfg`.
71 
72  Arguments:
73  - `cfg`: path to CMSSW config file
74  """
75 
76  sys.path.append(os.path.dirname(cfg)) # add location to python path
77  cache_stdout = sys.stdout
78  sys.stdout = open(os.devnull, "w") # suppress unwanted output
79  __configuration = \
80  importlib.import_module(os.path.splitext(os.path.basename(cfg))[0])
81  sys.stdout = cache_stdout
82  sys.path.pop() # clean up python path again
83  try:
84  os.remove(cfg+"c") # try to remove temporary .pyc file
85  except OSError as e:
86  if e.args == (2, "No such file or directory"): pass
87  else: raise
88 
89  return __configuration.process
90 
91 
def get_process_object(cfg)
Definition: tools.py:69
def tools.get_tags (   global_tag,
  records 
)
Get tags for `records` contained in `global_tag`.

Arguments:
- `global_tag`: global tag of interest
- `records`: database records of interest

Definition at line 110 of file tools.py.

References Vispa.Plugins.EdmBrowser.EdmDataAccessor.all(), and ALCARECOTkAlBeamHalo_cff.filter.

110 def get_tags(global_tag, records):
111  """Get tags for `records` contained in `global_tag`.
112 
113  Arguments:
114  - `global_tag`: global tag of interest
115  - `records`: database records of interest
116  """
117 
118  if len(records) == 0: return {} # avoid useless DB query
119 
120  # setting up the DB session
121  con = conddb.connect(url = conddb.make_url())
122  session = con.session()
123  GlobalTagMap = session.get_dbtype(conddb.GlobalTagMap)
124 
125  # query tag names for records of interest contained in `global_tag`
126  tags = session.query(GlobalTagMap.record, GlobalTagMap.tag_name).\
127  filter(GlobalTagMap.global_tag_name == global_tag,
128  GlobalTagMap.record.in_(records)).all()
129 
130  # closing the DB session
131  session.close()
132 
133  return {item[0]: {"tag": item[1], "connect": "pro"} for item in tags}
134 
135 
def get_tags(global_tag, records)
Definition: tools.py:110
def tools.haddInCastor (   castor_dir,
  result_file,
  type = 'root',
  prefix = 'rfio:',
  suffix = None 
)

Definition at line 81 of file tools.py.

References listFilesInCastor().

Referenced by DTCalibrationWorker.DTCalibrationWorker.runAnalysisResiduals(), DTCalibrationWorker.DTCalibrationWorker.runResidualCalib(), DTCalibrationWorker.DTCalibrationWorker.runTtrigProd(), DTCalibrationWorker.DTCalibrationWorker.runVDriftMeanTimerCalib(), and DTCalibrationWorker.DTCalibrationWorker.runVDriftSegmentCalib().

81 def haddInCastor(castor_dir,result_file,type = 'root',prefix = 'rfio:',suffix = None):
82  if not castor_dir: raise ValueError('Please specify valid castor dir')
83  if not result_file: raise ValueError('Please specify valid output file name')
84 
85  #cmd = 'hadd %s `./listfilesCastor %s | grep %s`'%(result_file,castor_dir,type)
86  #print "Running",cmd
87  #os.system(cmd)
88  from subprocess import call
89  files = listFilesInCastor(castor_dir,type,prefix)
90  if suffix: files = [item + suffix for item in files]
91 
92  cmd = ['hadd',result_file]
93  cmd.extend(files)
94  #print cmd
95  retcode = call(cmd)
96  return retcode
97 
def listFilesInCastor(castor_dir, type='root', prefix='rfio:')
Definition: tools.py:30
def haddInCastor(castor_dir, result_file, type='root', prefix='rfio:', suffix=None)
Definition: tools.py:81
def tools.haddLocal (   dir,
  result_file,
  type = 'root' 
)

Definition at line 98 of file tools.py.

References listFilesLocal().

Referenced by DTCalibrationWorker.DTCalibrationWorker.runAnalysisResiduals(), DTCalibrationWorker.DTCalibrationWorker.runResidualCalib(), DTCalibrationWorker.DTCalibrationWorker.runTtrigProd(), DTCalibrationWorker.DTCalibrationWorker.runVDriftMeanTimerCalib(), and DTCalibrationWorker.DTCalibrationWorker.runVDriftSegmentCalib().

98 def haddLocal(dir,result_file,type = 'root'):
99  if not dir: raise ValueError('Please specify valid dir')
100  if not result_file: raise ValueError('Please specify valid output file name')
101 
102  from subprocess import call
103  files = listFilesLocal(dir,type)
104  cmd = ['hadd',result_file]
105  cmd.extend(files)
106  #print cmd
107  retcode = call(cmd)
108  return retcode
109 
def haddLocal(dir, result_file, type='root')
Definition: tools.py:98
def listFilesLocal(dir, type='root')
Definition: tools.py:42
def tools.listFilesInCastor (   castor_dir,
  type = 'root',
  prefix = 'rfio:' 
)

Definition at line 30 of file tools.py.

Referenced by copyFilesFromCastor(), haddInCastor(), DTDqm.runDQM(), DTDQMMerge.runDQM(), DTCalibrationWorker.DTCalibrationWorker.runDQMClient(), DTCalibrationWorker.DTCalibrationWorker.runDQMHarvesting(), and DTCalibrationWorker.DTCalibrationWorker.runDQMMerge().

30 def listFilesInCastor(castor_dir,type = 'root',prefix = 'rfio:'):
31  if not castor_dir: raise ValueError('Please specify valid castor dir')
32 
33  from subprocess import Popen,PIPE
34  p1 = Popen(['nsls',castor_dir],stdout=PIPE)
35  #p2 = Popen(['grep',type],stdin=p1.stdout,stdout=PIPE)
36  #files = [prefix + castor_dir + "/" + item[:-1] for item in p2.stdout]
37  #p2.stdout.close()
38  files = [ "%s%s/%s" % (prefix,castor_dir,item.rstrip()) for item in p1.stdout if item.find(type) != -1 ]
39  p1.stdout.close()
40  return files
41 
def listFilesInCastor(castor_dir, type='root', prefix='rfio:')
Definition: tools.py:30
def tools.listFilesLocal (   dir,
  type = 'root' 
)

Definition at line 42 of file tools.py.

Referenced by copyFilesLocal(), haddLocal(), DTCalibrationWorker.DTCalibrationWorker.runDQMClient(), DTCalibrationWorker.DTCalibrationWorker.runDQMHarvesting(), and DTCalibrationWorker.DTCalibrationWorker.runDQMMerge().

42 def listFilesLocal(dir,type = 'root'):
43  if not dir: raise ValueError('Please specify valid dir')
44 
45  #from subprocess import Popen,PIPE
46  #p1 = Popen(['ls',dir],stdout=PIPE)
47  #p2 = Popen(['grep',type],stdin=p1.stdout,stdout=PIPE)
48  #files = [dir + "/" + item[:-1] for item in p2.stdout]
49  #p2.stdout.close()
50  files = os.listdir(dir)
51  files = [ "%s/%s" % (dir,item) for item in files if item.find(type) != -1 ]
52 
53  return files
54 
def listFilesLocal(dir, type='root')
Definition: tools.py:42
def tools.loadCmsProcessFile (   psetName)

Definition at line 140 of file tools.py.

140 def loadCmsProcessFile(psetName):
141  pset = imp.load_source("psetmodule",psetName)
142  return pset.process
143 
def loadCmsProcessFile(psetName)
Definition: tools.py:140
def tools.loadCrabDefault (   crabCfg,
  config 
)

Definition at line 195 of file tools.py.

Referenced by DTNoiseCalibration.DTNoiseCalibration.initCrab(), DTTTrigProd.DTTTrigProd.initCrab(), DTDQMValidation.DTDQMValidation.initCrab(), DTResidualCalibration.DTResidualCalibration.initCrab(), DTAnalysisResiduals.DTAnalysisResiduals.initCrab(), DTTTrigValid.DTTTrigValid.initCrab(), DTVDriftSegmentCalibration.DTVDriftSegmentCalibration.initCrab(), and DTVDriftMeanTimerCalibration.DTVDriftMeanTimerCalibration.initCrab().

195 def loadCrabDefault(crabCfg,config):
196  # CRAB section
197  if not crabCfg.has_section('CRAB'): crabCfg.add_section('CRAB')
198  crabCfg.set('CRAB','jobtype','cmssw')
199 
200  if hasattr(config,'scheduler') and config.scheduler: crabCfg.set('CRAB','scheduler',config.scheduler)
201  else: crabCfg.set('CRAB','scheduler','CAF')
202 
203  if hasattr(config,'useserver') and config.useserver: crabCfg.set('CRAB','use_server',1)
204 
205  # CMSSW section
206  if not crabCfg.has_section('CMSSW'): crabCfg.add_section('CMSSW')
207  if hasattr(config,'datasetpath') and config.datasetpath: crabCfg.set('CMSSW','datasetpath',config.datasetpath)
208  else: crabCfg.set('CMSSW','datasetpath','/XXX/YYY/ZZZ')
209  crabCfg.set('CMSSW','pset','pset.py')
210 
211  # Splitting config
212  crabCfg.remove_option('CMSSW','total_number_of_events')
213  crabCfg.remove_option('CMSSW','events_per_job')
214  crabCfg.remove_option('CMSSW','number_of_jobs')
215  crabCfg.remove_option('CMSSW','total_number_of_lumis')
216  crabCfg.remove_option('CMSSW','lumis_per_job')
217  crabCfg.remove_option('CMSSW','lumi_mask')
218  crabCfg.remove_option('CMSSW','split_by_run')
219 
220  """
221  if hasattr(config,'totalnumberevents'): crabCfg.set('CMSSW','total_number_of_events',config.totalnumberevents)
222  if hasattr(config,'eventsperjob'): crabCfg.set('CMSSW','events_per_job',config.eventsperjob)
223  """
224  if hasattr(config,'splitByLumi') and config.splitByLumi:
225  crabCfg.set('CMSSW','total_number_of_lumis',config.totalnumberlumis)
226  crabCfg.set('CMSSW','lumis_per_job',config.lumisperjob)
227  if hasattr(config,'lumimask') and config.lumimask: crabCfg.set('CMSSW','lumi_mask',config.lumimask)
228  elif hasattr(config,'splitByEvent') and config.splitByEvent:
229  crabCfg.set('CMSSW','total_number_of_events',config.totalnumberevents)
230  crabCfg.set('CMSSW','events_per_job',config.eventsperjob)
231  else:
232  crabCfg.set('CMSSW','split_by_run',1)
233 
234  if hasattr(config,'splitByEvent') and config.splitByEvent:
235  crabCfg.remove_option('CMSSW','runselection')
236  else:
237  if hasattr(config,'runselection') and config.runselection:
238  crabCfg.set('CMSSW','runselection',config.runselection)
239 
240  # USER section
241  if not crabCfg.has_section('USER'): crabCfg.add_section('USER')
242 
243  # Stageout config
244  if hasattr(config,'stageOutCAF') and config.stageOutCAF:
245  crabCfg.set('USER','return_data',0)
246  crabCfg.set('USER','copy_data',1)
247  crabCfg.set('USER','storage_element','T2_CH_CAF')
248  crabCfg.set('USER','user_remote_dir',config.userdircaf)
249  crabCfg.set('USER','check_user_remote_dir',0)
250  elif hasattr(config,'stageOutLocal') and config.stageOutLocal:
251  crabCfg.set('USER','return_data',1)
252  crabCfg.set('USER','copy_data',0)
253  crabCfg.remove_option('USER','storage_element')
254  crabCfg.remove_option('USER','user_remote_dir')
255  crabCfg.remove_option('USER','check_user_remote_dir')
256 
257  if hasattr(config,'email') and config.email: crabCfg.set('USER','eMail',config.email)
258  crabCfg.set('USER','xml_report','crabReport.xml')
259 
260  if hasattr(config,'runOnGrid') and config.runOnGrid:
261  crabCfg.remove_section('CAF')
262  if hasattr(config,'ce_black_list'):
263  if not crabCfg.has_section('GRID'): crabCfg.add_section('GRID')
264  crabCfg.set('GRID','ce_black_list', config.ce_black_list)
265  if hasattr(config,'ce_white_list'):
266  if not crabCfg.has_section('GRID'): crabCfg.add_section('GRID')
267  crabCfg.set('GRID','ce_white_list', config.ce_white_list)
268  else:
269  if not crabCfg.has_section('CAF'): crabCfg.add_section('CAF')
270  crabCfg.set('CAF','queue',config.queueAtCAF)
271 
272  return crabCfg
273 
def loadCrabDefault(crabCfg, config)
Definition: tools.py:195
def tools.make_unique_runranges (   ali_producer)
Derive unique run ranges from AlignmentProducer PSet.

Arguments:
- `ali_producer`: cms.PSet containing AlignmentProducer configuration

Definition at line 92 of file tools.py.

References createfilelist.int.

92 def make_unique_runranges(ali_producer):
93  """Derive unique run ranges from AlignmentProducer PSet.
94 
95  Arguments:
96  - `ali_producer`: cms.PSet containing AlignmentProducer configuration
97  """
98 
99  if (hasattr(ali_producer, "RunRangeSelection") and
100  len(ali_producer.RunRangeSelection) > 0):
101  iovs = set([int(iov)
102  for sel in ali_producer.RunRangeSelection
103  for iov in sel.RunRanges])
104  if len(iovs) == 0: return [1] # single IOV starting from run 1
105  return sorted(iovs)
106  else:
107  return [1] # single IOV starting from run 1
108 
109 
def make_unique_runranges(ali_producer)
Definition: tools.py:92
def tools.parseInput (   inputFields,
  requiredFields = () 
)

Definition at line 122 of file tools.py.

Referenced by edmStreamStallGrapher.readLogFile().

122 def parseInput(inputFields,requiredFields = ()):
123 
124  class options: pass
125  for item in sys.argv:
126  option = item.split('=')[0]
127  if option in inputFields:
128  value = item.split('=')[1]
129  if value in ('true','True','yes','Yes'): value = True
130  elif value in ('false','False','no','No'): value = False
131 
132  setattr(options,option,value)
133 
134  for item in requiredFields:
135  if not hasattr(options,item):
136  raise RuntimeError('Need to set "%s"' % item)
137 
138  return options
139 
def parseInput(inputFields, requiredFields=())
Definition: tools.py:122
def tools.replaceTemplate (   template,
  opts 
)

Definition at line 5 of file tools.py.

References harvestTrackValidationPlots.str.

5 def replaceTemplate(template,**opts):
6  result = open(template).read()
7  for item in opts:
8  old = '@@%s@@'%item
9  new = str(opts[item])
10  print "Replacing",old,"to",new
11  result = result.replace(old,new)
12 
13  return result
14 
def replaceTemplate(template, opts)
Definition: tools.py:5
def tools.run_checked (   cmd)
Run `cmd` and exit in case of failures.

Arguments:
- `cmd`: list containing the strings of the command

Definition at line 53 of file tools.py.

References join().

Referenced by create_single_iov_db().

53 def run_checked(cmd):
54  """Run `cmd` and exit in case of failures.
55 
56  Arguments:
57  - `cmd`: list containing the strings of the command
58  """
59 
60  try:
61  with open(os.devnull, "w") as devnull:
62  subprocess.check_call(cmd, stdout = devnull)
63  except subprocess.CalledProcessError as e:
64  print "Problem in running the following command:"
65  print " ".join(e.cmd)
66  sys.exit(1)
67 
68 
static std::string join(char **cmd)
Definition: RemoteFile.cc:18
def run_checked(cmd)
Definition: tools.py:53
def tools.setGridEnv (   cmssw_dir)

Definition at line 110 of file tools.py.

110 def setGridEnv(cmssw_dir):
111  cwd = os.getcwd()
112  os.chdir(cmssw_dir)
113 
114  os.system('source /afs/cern.ch/cms/LCG/LCG-2/UI/cms_ui_env.sh')
115  os.system('cmsenv')
116  os.system('source /afs/cern.ch/cms/ccs/wm/scripts/Crab/crab.sh')
117 
118  os.chdir(cwd)
119 
120  return
121 
def setGridEnv(cmssw_dir)
Definition: tools.py:110
def tools.writeCfg (   process,
  dir,
  psetName 
)

Definition at line 158 of file tools.py.

References TriggerAnalyzer.write().

158 def writeCfg(process,dir,psetName):
159  if not os.path.exists(dir): os.makedirs(dir)
160  open(dir + '/' + psetName,'w').write(process.dumpPython())
161 
def writeCfg(process, dir, psetName)
Definition: tools.py:158
def write(self, setup)
def tools.writeCfgPkl (   process,
  dir,
  psetName 
)

Definition at line 162 of file tools.py.

162 def writeCfgPkl(process,dir,psetName):
163  if not os.path.exists(dir): os.makedirs(dir)
164 
165  pklFileName = psetName.split('.')[0] + '.pkl'
166  pklFile = open(dir + '/' + pklFileName,"wb")
167  myPickle = pickle.Pickler(pklFile)
168  myPickle.dump(process)
169  pklFile.close()
170 
171  outFile = open(dir + '/' + psetName,"w")
172  outFile.write("import FWCore.ParameterSet.Config as cms\n")
173  outFile.write("import pickle\n")
174  outFile.write("process = pickle.load(open('%s', 'rb'))\n" % pklFileName)
175  outFile.close()
176 
177 
def writeCfgPkl(process, dir, psetName)
Definition: tools.py:162