add python/RunDMCFlags.py - which implements a look-up from Run Number to...

add python/RunDMCFlags.py - which implements a look-up from Run Number to timestamp as discussed in ATCONDDB-21. Tagging RunDependentSimComps-00-00-08 (RunDependentSimComps-00-00-08)

Simulation/RunDependentSim/RunDependentSimComps/python/RunDMCFlags.py

+# Copyright (C) 2002-2017 CERN for the benefit of the ATLAS collaboration
+
+
+from AthenaCommon.Logging import logging
+logRunDMCFlags = logging.getLogger('RunDMCFlags')
+from AthenaCommon.JobProperties import JobProperty, JobPropertyContainer
+from AthenaCommon.JobProperties import jobproperties
+
+class RunToTimestampDict(JobProperty):
+    """This is used to hold a dictionary of the form
+    {152166:1269948352889940910, ...} to allow the
+    timestamp to be determined from the run.
+    """
+    statusOn=True
+    allowedTypes=['dict']
+    StoredValue={
+        152777:1270866809, ##2010 7 TeV p-p for period A (very low mu)
+        154471:1272831478, ##2010 900 GeV p-p
+        155697:1274722685, ##2010 7 TeV p-p for period C (mu~0.2)
+        162623:1282673387, ##2010 7 TeV p-p for period F (36 single bunches, mu~2)
+        167576:1287909620, ##2010 7 TeV p-p (150ns bunch trains)
+        170002:1291063245, ##2010 Heavy Ion run
+        178211:1301070988, ##2011 2.76 TeV
+        180164:1303554669, ##2011 7 TeV p-p (B-D)
+        183003:1306973316, ##2011 7 TeV p-p (E-H)
+        186169:1311379706, ##2011 7 TeV p-p (I-K)
+        189751:1316868299, ##2011 7 TeV p-p (L-M)
+        193662:1322075751, ##2011 Heavy Ion run
+        195847:1328040250, ##Upgrade samples from MC12
+        210184:1347470747, ##2012 Heavy Ion - proton run
+        212272:1349603811, ##2012 8 TeV p-p
+        219114:1360457560, ##2013 Heavy Ion - proton run
+        219305:1360689056, ##2013 2.76 TeV p-p
+        222222:1376009561, ##DC14 (mu=30) 25ns, old R1 geo tag
+        222250:1376075781, ##DC14 (mu=30) 50ns, old R1 geo tag
+        222500:1376646281, ##MC15 50ns, new R2 geo tag (mu=0)
+        222510:1376669101, ##MC15 50ns, new R2 geo tag (nominal mu)
+        222525:1376703331  ##MC15 25ns, new R2 geo tag (nominal mu)
+        }
+    def addEntry( self, run, timestamp, force=False):
+        if not type(run) == int:
+            raise ValueError( 'run is not the expected type (int) for the key of RunToTimestampDict!' )
+        if not type(timestamp) == int:
+            raise ValueError( 'timestamp is not the expected type (int) for the key of RunToTimestampDict!' )
+        run2tslist = self.get_Value()
+        logRunDMCFlags.info("Attempting to add the following pair to RunToTimestampDict: %i:%i.", run, timestamp)
+        if run2tslist.has_key(run):
+            logRunDMCFlags.warning("RunToTimestampDict already has an entry for run %i with timestamp %i", run, run2tslist[run])
+            if force:
+                logRunDMCFlags.warning("Replacing existing entry in RunToTimestampDict!")
+            else:
+                logRunDMCFlags.warning("Not altering RunToTimestampDict!")
+                return
+        run2tslist[run] = timestamp
+        return
+    def getTimestampForRun(self, run):
+        return self.get_Value()[run]
+    def Print(self):
+        """Print the contents of the RunToTimestampList
+        """
+        logRunDMCFlags.info( 'RunToTimestampList:' )
+        run2tslist = self.get_Value()
+        logRunDMCFlags.info('run\t\ttimestamp')
+        for d in run2tslist.keys(): logRunDMCFlags.info('%i\t%i', d, run2tslist[d])
+        return
+
+## Definition and registration of the simulation flag container
+class RunDMCFlags(JobPropertyContainer):
+    """
+    Container for the RunDMCFlags
+
+    This JobPropertyContainer is intended to hold a dictionary to
+    allow the correct timestamp to be inferred from the DataRunNumber
+    specified on the transform command-line. This is intended to be a
+    temporary fix for https://its.cern.ch/jira/browse/ATCONDDB-21
+    A longer term solution will eventually be put in place.
+
+    """
+
+## Register and populate the RunDMCFlags container
+jobproperties.add_Container(RunDMCFlags)
+for jpname in dir():
+    jp = eval(jpname)
+    import inspect
+    if inspect.isclass(jp):
+        if issubclass(jp, JobProperty) and jp is not JobProperty:
+            logRunDMCFlags.debug("Adding RunDMCFlag '%s' to RunDMCFlags container" % jpname)
+            jobproperties.RunDMCFlags.add_JobProperty(jp)
+
+
+## Short-cut alias (standard naming)
+runDMCFlags = jobproperties.RunDMCFlags