Skip to content
Snippets Groups Projects

CalibNtupleAnalysisAlg: flake8 fixes

Merged Frank Winklmeier requested to merge fwinkl/athena:flake8_muoncalib into master
3 files
+ 55
46
Compare changes
  • Side-by-side
  • Inline
Files
3
@@ -2,20 +2,28 @@
@@ -2,20 +2,28 @@
from __future__ import print_function
from __future__ import print_function
from CalibNtupleAnalysisAlg.CalibNtupleAnalysisAlgConf import *
from CalibNtupleAnalysisAlg.CalibNtupleAnalysisAlgConf import (MuonCalib__ApplyRawTimes,
from MuonCalibStandAloneBase.MuonCalibStandAloneBaseConf import *
MuonCalib__CalibNtupleAnalysisAlg2,
from MdtCalibIOSvc.MdtCalibIOSvcConf import *
MuonCalib__CalibNtupleLoader,
from MuonCalibStandAloneTools.MuonCalibStandAloneToolsConf import *
MuonCalib__UpdateBField,
from MuonCalibStandAloneTools.MuonCalibStandAloneToolsConfig import *
MuonCalib__GoodRunListFilter,
 
MuonCalib__InitialDqFilter,
 
MuonCalib__BarrelSecondCoordinatePreparationTool,
 
MuonCalib__SegmentRawdataSelector,
 
MuonCalib__SegmentRefitter,
 
MuonCalib__SegmentRecalibration,
 
MuonCalib__SegmentRegionSelector)
 
 
from MuonCalibStandAloneBase.MuonCalibStandAloneBaseConf import MuonCalib__CalibrationTeeIOTool, RegionSelectionSvc
 
from MdtCalibIOSvc.MdtCalibIOSvcConf import MdtCalibOutputDbSvc, MdtCalibInputSvc, MuonCalib__CalibrationFileIOTool
from MuonCalibDbOperations.MuonCalibDbOperationsConf import MuonCalib__CalibrationDbIOTool
from MuonCalibDbOperations.MuonCalibDbOperationsConf import MuonCalib__CalibrationDbIOTool
from AthenaCommon.AppMgr import ServiceMgr, ToolSvc
from AthenaCommon.AppMgr import ServiceMgr, ToolSvc
from AthenaCommon.AlgSequence import AlgSequence
from AthenaCommon.AlgSequence import AlgSequence
from CalibNtupleAnalysisAlg.CalibNtupleMetaData import *
from CalibNtupleAnalysisAlg.CalibNtupleMetaData import CalibNtupleMetaData
 
import MuonCalibStandAloneTools.MuonCalibStandAloneToolsConfig as calibtools
import sys
import sys
#ToolSvc = Service( "ToolSvc" )
class CalibNtupleAnalysisConfig:
class CalibNtupleAnalysisConfig:
#--------------------------------------user settings----------------------------
#--------------------------------------user settings----------------------------
@@ -31,9 +39,9 @@ class CalibNtupleAnalysisConfig:
@@ -31,9 +39,9 @@ class CalibNtupleAnalysisConfig:
# event display Display
# event display Display
# rpc timing RpcTiming
# rpc timing RpcTiming
# control histograms ControlHistograms
# control histograms ControlHistograms
CalibrationAlgorithm = "RunScan"
CalibrationAlgorithm = "RunScan"
#file list
#file list
FileList="fl.txt"
FileList="fl.txt"
@@ -49,7 +57,7 @@ class CalibNtupleAnalysisConfig:
@@ -49,7 +57,7 @@ class CalibNtupleAnalysisConfig:
#select segment author - 3=MuonBoy 4=Moore - Unused if SegmentsFromRawdata=True
#select segment author - 3=MuonBoy 4=Moore - Unused if SegmentsFromRawdata=True
SegmentAuthor=4
SegmentAuthor=4
#first and last event to analyse
#first and last event to analyse
FirstEvent=0
FirstEvent=0
LastEvent=-1
LastEvent=-1
@@ -71,7 +79,7 @@ class CalibNtupleAnalysisConfig:
@@ -71,7 +79,7 @@ class CalibNtupleAnalysisConfig:
MaxSegmentHits = -1
MaxSegmentHits = -1
#perform a segment refit - not for algorithms likr MTT0, Integration, RunScan..
#perform a segment refit - not for algorithms likr MTT0, Integration, RunScan..
SegmentRefit=False;
SegmentRefit=False
#segment refit parameters
#segment refit parameters
RefineT0=False
RefineT0=False
@@ -80,7 +88,7 @@ class CalibNtupleAnalysisConfig:
@@ -80,7 +88,7 @@ class CalibNtupleAnalysisConfig:
RefitTimeOut=2
RefitTimeOut=2
RefitMinSegmentHits=4
RefitMinSegmentHits=4
RefitMaxSegmentHits=-1
RefitMaxSegmentHits=-1
#calibrate with b-field
#calibrate with b-field
BFieldCorrection=False
BFieldCorrection=False
@@ -103,25 +111,25 @@ class CalibNtupleAnalysisConfig:
@@ -103,25 +111,25 @@ class CalibNtupleAnalysisConfig:
#output to database - MP/RM/MI/NONE
#output to database - MP/RM/MI/NONE
CalibDB=None
CalibDB=None
#initial data quality list - suppress Tubes
#initial data quality list - suppress Tubes
InitialDQList = None
InitialDQList = None
#if set to true no segment recalibraiton will be done, independent on the tool
#if set to true no segment recalibraiton will be done, independent on the tool
SuppressRecalibration=False
SuppressRecalibration=False
#use good run list to filter events
#use good run list to filter events
GoodRunList=None
GoodRunList=None
#apply time slewing corrections
#apply time slewing corrections
ApplyTimeSlewingCorrections=False
ApplyTimeSlewingCorrections=False
#apply multilayer rt-scaling
#apply multilayer rt-scaling
ApplyRtScaling=True
ApplyRtScaling=True
sToolSvc= ToolSvc
sToolSvc= ToolSvc
sServiceMgr = ServiceMgr
sServiceMgr = ServiceMgr
SegmentRecalibration={}
SegmentRecalibration={}
SegmentRefitter={}
SegmentRefitter={}
@@ -151,6 +159,7 @@ class CalibNtupleAnalysisConfig:
@@ -151,6 +159,7 @@ class CalibNtupleAnalysisConfig:
self._initial_DQ_list()
self._initial_DQ_list()
#apply RPC Timing Correction
#apply RPC Timing Correction
if self.RPCTimingCorr:
if self.RPCTimingCorr:
 
from CalibNtupleAnalysisAlg.CalibNtupleAnalysisAlgConf import MuonCalib__RpcTimingCorr
self.RpcTiming=MuonCalib__RpcTimingCorr()
self.RpcTiming=MuonCalib__RpcTimingCorr()
self.sToolSvc += self.RpcTiming
self.sToolSvc += self.RpcTiming
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append(self.RpcTiming)
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append(self.RpcTiming)
@@ -172,13 +181,13 @@ class CalibNtupleAnalysisConfig:
@@ -172,13 +181,13 @@ class CalibNtupleAnalysisConfig:
self._config_calibIO()
self._config_calibIO()
#create tool
#create tool
self._create_calib_tool()
self._create_calib_tool()
#-----------------------protected functions------------------------------------
#-----------------------protected functions------------------------------------
def _create_CalibNtupleAnalysisAlg(self):
def _create_CalibNtupleAnalysisAlg(self):
topSequence = AlgSequence()
topSequence = AlgSequence()
self.CalibNtupleAnalysisAlg = MuonCalib__CalibNtupleAnalysisAlg2()
self.CalibNtupleAnalysisAlg = MuonCalib__CalibNtupleAnalysisAlg2()
self.CalibNtupleAnalysisAlg.NumberOfSegments = self.NumberOfSegments
self.CalibNtupleAnalysisAlg.NumberOfSegments = self.NumberOfSegments
if self.CalibrationAlgorithm in LimitSegmentsFor:
if self.CalibrationAlgorithm in calibtools.LimitSegmentsFor:
self.CalibNtupleAnalysisAlg.NumberOfSegments = self.AutoLimitSegments
self.CalibNtupleAnalysisAlg.NumberOfSegments = self.AutoLimitSegments
topSequence += self.CalibNtupleAnalysisAlg
topSequence += self.CalibNtupleAnalysisAlg
@@ -195,7 +204,7 @@ class CalibNtupleAnalysisConfig:
@@ -195,7 +204,7 @@ class CalibNtupleAnalysisConfig:
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.__good_run_list_filter )
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.__good_run_list_filter )
# self.__good_run_list_tool.EventSelectorMode = True
# self.__good_run_list_tool.EventSelectorMode = True
self.__good_run_list_tool.PassThrough = False
self.__good_run_list_tool.PassThrough = False
def _initial_DQ_list(self):
def _initial_DQ_list(self):
if self.InitialDQList:
if self.InitialDQList:
if self.InitialDQList.upper()=="NONE":
if self.InitialDQList.upper()=="NONE":
@@ -204,7 +213,7 @@ class CalibNtupleAnalysisConfig:
@@ -204,7 +213,7 @@ class CalibNtupleAnalysisConfig:
self.InitialDqFilter.InitialDqFile = self.InitialDQList
self.InitialDqFilter.InitialDqFile = self.InitialDQList
self.sToolSvc += self.InitialDqFilter
self.sToolSvc += self.InitialDqFilter
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.InitialDqFilter)
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.InitialDqFilter)
def _region_selection(self):
def _region_selection(self):
if self.SegmentsFromRawdata:
if self.SegmentsFromRawdata:
self.RegionSelection = MuonCalib__SegmentRawdataSelector()
self.RegionSelection = MuonCalib__SegmentRawdataSelector()
@@ -226,10 +235,10 @@ class CalibNtupleAnalysisConfig:
@@ -226,10 +235,10 @@ class CalibNtupleAnalysisConfig:
self.BarrelSecondCoordinatePreparationTool = MuonCalib__BarrelSecondCoordinatePreparationTool()
self.BarrelSecondCoordinatePreparationTool = MuonCalib__BarrelSecondCoordinatePreparationTool()
self.sToolSvc += self.BarrelSecondCoordinatePreparationTool
self.sToolSvc += self.BarrelSecondCoordinatePreparationTool
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.BarrelSecondCoordinatePreparationTool)
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.BarrelSecondCoordinatePreparationTool)
def _configure_segment_recalibration(self, ToolName=""):
def _configure_segment_recalibration(self, ToolName=""):
#tools that do not need segment recalibration
#tools that do not need segment recalibration
if self.CalibrationAlgorithm in NoRecalibration:
if self.CalibrationAlgorithm in calibtools.NoRecalibration:
return
return
if ToolName:
if ToolName:
self.SegmentRecalibration[ToolName] = MuonCalib__SegmentRecalibration(ToolName)
self.SegmentRecalibration[ToolName] = MuonCalib__SegmentRecalibration(ToolName)
@@ -243,39 +252,39 @@ class CalibNtupleAnalysisConfig:
@@ -243,39 +252,39 @@ class CalibNtupleAnalysisConfig:
else:
else:
self.SegmentRecalibration[ToolName].TimeSlewingSwitch='UNAPPLY'
self.SegmentRecalibration[ToolName].TimeSlewingSwitch='UNAPPLY'
#tools that revert all calibrations
#tools that revert all calibrations
if self.CalibrationAlgorithm in RevertT0:
if self.CalibrationAlgorithm in calibtools.RevertT0:
self.SegmentRecalibration[ToolName].T0Switch = "UNAPPLY"
self.SegmentRecalibration[ToolName].T0Switch = "UNAPPLY"
self.SegmentRecalibration[ToolName].BSwitch = "LEAVE"
self.SegmentRecalibration[ToolName].BSwitch = "LEAVE"
self.SegmentRecalibration[ToolName].RecalcR = False
self.SegmentRecalibration[ToolName].RecalcR = False
if self.CalibrationAlgorithm in RecalibT0 :
if self.CalibrationAlgorithm in calibtools.RecalibT0 :
self.SegmentRecalibration[ToolName].T0Switch = "FILE"
self.SegmentRecalibration[ToolName].T0Switch = "FILE"
self.SegmentRecalibration[ToolName].BSwitch = "LEAVE"
self.SegmentRecalibration[ToolName].BSwitch = "LEAVE"
self.SegmentRecalibration[ToolName].RecalcR = False
self.SegmentRecalibration[ToolName].RecalcR = False
if self.CalibrationAlgorithm in RecalibAll:
if self.CalibrationAlgorithm in calibtools.RecalibAll:
self.SegmentRecalibration[ToolName].T0Switch = "FILE"
self.SegmentRecalibration[ToolName].T0Switch = "FILE"
self.SegmentRecalibration[ToolName].BSwitch = "LEAVE"
self.SegmentRecalibration[ToolName].BSwitch = "LEAVE"
self.SegmentRecalibration[ToolName].RecalcR = True
self.SegmentRecalibration[ToolName].RecalcR = True
#id b-field should be corrected, set BSwitch either to "UNAPPLY" or to "FILE"
#id b-field should be corrected, set BSwitch either to "UNAPPLY" or to "FILE"
if self.CalibrationAlgorithm in RevertRtScaling:
if self.CalibrationAlgorithm in calibtools.RevertRtScaling:
self.SegmentRecalibration[ToolName].RtMultilayerScaling = "UNAPPLY"
self.SegmentRecalibration[ToolName].RtMultilayerScaling = "UNAPPLY"
if self.CalibrationAlgorithm in ApplRtScaling and self.ApplyRtScaling:
if self.CalibrationAlgorithm in calibtools.ApplRtScaling and self.ApplyRtScaling:
self.SegmentRecalibration[ToolName].RtMultilayerScaling = "FILE"
self.SegmentRecalibration[ToolName].RtMultilayerScaling = "FILE"
if self.BFieldCorrection:
if self.BFieldCorrection:
if self.SegmentRecalibration[ToolName].RecalcR:
if self.SegmentRecalibration[ToolName].RecalcR:
self.SegmentRecalibration[ToolName].BSwitch = "FILE"
self.SegmentRecalibration[ToolName].BSwitch = "FILE"
# else:
# else:
# self.SegmentRecalibration[ToolName].BSwitch = "UNAPPLY"
# self.SegmentRecalibration[ToolName].BSwitch = "UNAPPLY"
def _config_UpdateBField(self):
def _config_UpdateBField(self):
if self.UpdateBField in NoSegmentRefit:
if self.UpdateBField in calibtools.NoSegmentRefit:
self.UpdateBField = None
self.UpdateBField = None
return
return
self.UpdateBField = MuonCalib__UpdateBField()
self.UpdateBField = MuonCalib__UpdateBField()
self.sToolSvc += self.UpdateBField
self.sToolSvc += self.UpdateBField
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.UpdateBField )
self.CalibNtupleAnalysisAlg.CalibSegmentPreparationTools.append( self.UpdateBField )
def _config_segment_refitter(self, ToolName=""):
def _config_segment_refitter(self, ToolName=""):
if self.CalibrationAlgorithm in NoSegmentRefit:
if self.CalibrationAlgorithm in calibtools.NoSegmentRefit:
return
return
if ToolName:
if ToolName:
self.SegmentRefitter[ToolName] = MuonCalib__SegmentRefitter(ToolName)
self.SegmentRefitter[ToolName] = MuonCalib__SegmentRefitter(ToolName)
@@ -290,7 +299,7 @@ class CalibNtupleAnalysisConfig:
@@ -290,7 +299,7 @@ class CalibNtupleAnalysisConfig:
self.SegmentRefitter[ToolName].TimeOut = self.RefitTimeOut
self.SegmentRefitter[ToolName].TimeOut = self.RefitTimeOut
self.SegmentRefitter[ToolName].MinSegmentHits = self.RefitMinSegmentHits
self.SegmentRefitter[ToolName].MinSegmentHits = self.RefitMinSegmentHits
self.SegmentRefitter[ToolName].MaxSegmentHits = self.RefitMaxSegmentHits
self.SegmentRefitter[ToolName].MaxSegmentHits = self.RefitMaxSegmentHits
def _config_calibIO(self):
def _config_calibIO(self):
if self.CalibDir:
if self.CalibDir:
self.CalibrationFileIOTool = MuonCalib__CalibrationFileIOTool()
self.CalibrationFileIOTool = MuonCalib__CalibrationFileIOTool()
@@ -302,7 +311,7 @@ class CalibNtupleAnalysisConfig:
@@ -302,7 +311,7 @@ class CalibNtupleAnalysisConfig:
if not self.__set_db():
if not self.__set_db():
print ("Unknown database location " + str(self.CalibOutputDB))
print ("Unknown database location " + str(self.CalibOutputDB))
sys.exit(1)
sys.exit(1)
self.MdtCalibOutputDbSvc = MdtCalibOutputDbSvc()
self.MdtCalibOutputDbSvc = MdtCalibOutputDbSvc()
self.MdtCalibOutputDbSvc.PostprocessCalibration = True
self.MdtCalibOutputDbSvc.PostprocessCalibration = True
self.sServiceMgr += self.MdtCalibOutputDbSvc
self.sServiceMgr += self.MdtCalibOutputDbSvc
@@ -332,16 +341,16 @@ class CalibNtupleAnalysisConfig:
@@ -332,16 +341,16 @@ class CalibNtupleAnalysisConfig:
self.MdtCalibInputSvc.CalibrationInputTool = self.CalibrationFileIOTool
self.MdtCalibInputSvc.CalibrationInputTool = self.CalibrationFileIOTool
def _create_calib_tool(self):
def _create_calib_tool(self):
self.CalibrationTool=CreateCalibTool(self.CalibrationAlgorithm)
self.CalibrationTool=calibtools.CreateCalibTool(self.CalibrationAlgorithm)
if self.CalibrationTool:
if self.CalibrationTool:
self.sToolSvc += self.CalibrationTool
self.sToolSvc += self.CalibrationTool
self.CalibNtupleAnalysisAlg.CalibrationTool = self.CalibrationTool
self.CalibNtupleAnalysisAlg.CalibrationTool = self.CalibrationTool
#------------------------------------------------------------
#------------------------------------------------------------
def __set_db(self):
def __set_db(self):
self.DbIoTool.SiteName = self.CalibDB
self.DbIoTool.SiteName = self.CalibDB
self.DbIoTool.UseValidaedT0 = self.CalibrationAlgorithm in ValidatedT0
self.DbIoTool.UseValidaedT0 = self.CalibrationAlgorithm in calibtools.ValidatedT0
self.DbIoTool.UseValidaedRt = self.CalibrationAlgorithm in ValidatedRt
self.DbIoTool.UseValidaedRt = self.CalibrationAlgorithm in calibtools.ValidatedRt
if (self.CalibDB == "MP"):
if (self.CalibDB == "MP"):
self.DbIoTool.ConnectionString = "oracle://oracle01.mppmu.mpg.de/ATLMPI.mppmu.mpg.de/ATLAS_MUONCALIB_MPI"
self.DbIoTool.ConnectionString = "oracle://oracle01.mppmu.mpg.de/ATLMPI.mppmu.mpg.de/ATLAS_MUONCALIB_MPI"
self.DbIoTool.WorkingSchema = "ATLAS_MUONCALIB_MPI"
self.DbIoTool.WorkingSchema = "ATLAS_MUONCALIB_MPI"
Loading