Skip to content

Commit

Permalink
Checkpointing progress on fixing pylint warnings (2). Refs #11289.
Browse files Browse the repository at this point in the history
  • Loading branch information
wdzhou committed Apr 29, 2015
1 parent f0e4348 commit 2a69ede
Showing 1 changed file with 28 additions and 22 deletions.
50 changes: 28 additions & 22 deletions Code/Mantid/scripts/HFIRPowderReduction/HfirPDReductionControl.py
@@ -1,3 +1,4 @@
#pylint: disable=too-many-lines,relative-import,invalid-name,too-many-instance-attributes,too-many-arguments
############################################################################
#
# HFIR powder reduction control class
Expand All @@ -18,11 +19,10 @@
if os.path.exists(libpath) is False:
libpath = os.path.join(curdir.split('Code')[0], 'Code/release/bin')
sys.path.append(libpath)
import mantid
import mantid.simpleapi as api
import mantid.kernel
from mantid.simpleapi import AnalysisDataService
from mantid.kernel import ConfigService
#from mantid.kernel import ConfigService


VanadiumPeakPositions = [0.5044,0.5191,0.5350,0.5526,0.5936,0.6178,0.6453,0.6768,
Expand Down Expand Up @@ -54,6 +54,7 @@ def __init__(self, exp, scan):
self._processedVanWSTemp = None
self._processVanNote = ""
self._applySmoothVan = False
self._vanadiumPeakPosList = []

self._wavelength = None

Expand All @@ -72,7 +73,7 @@ def applySmoothVanadium(self, smoothaccept):
def getProcessedVanadiumWS(self):
"""
"""
return self._processedVanWS
return self._processedVanWS

def getProcessedVanadiumWSTemp(self):
"""
Expand Down Expand Up @@ -147,7 +148,7 @@ def setupMDWrokspaces(self, datamdws, monitormdws):

def setProcessedVanadiumDataTemp(self, vanws, note):
""" Set tempory processed vanadium data
Arguments:
Arguments:
- vanws :: workspace
- note :: string as note
"""
Expand Down Expand Up @@ -185,7 +186,7 @@ def setWavelength(self, wavelength):

return


#pylint: disable=too-many-public-methods
class HFIRPDRedControl(object):
""" Class for controlling HFIR powder reduction
"""
Expand Down Expand Up @@ -237,7 +238,7 @@ def getIndividualDetCounts(self, exp, scan, detid, xlabel):
# END-IF-ELSE

# Get raw counts
# FIXME : use **args
# FUTURE: use **args
if xlabel is None:
tempoutws = \
api.GetSpiceDataRawCountsFromMD(InputWorkspace=datamdws,
Expand Down Expand Up @@ -343,7 +344,7 @@ def getSampleLogValue(self, expno, scanno, samplelogname, xlabel):
# END-IF-ELSE

# get the complete list of Pt. number
ptnolist = self._getRunNumberList(datamdws=rmanager.datamdws)
# ptnolist = self._getRunNumberList(datamdws=rmanager.datamdws)

# get data
tempoutws = api.GetSpiceDataRawCountsFromMD(InputWorkspace=datamdws,
Expand Down Expand Up @@ -672,10 +673,14 @@ def parseSpiceData(self, expno, scanno, detefftablews=None):
return True


def reduceSpicePDData(self, exp, scan, unit, xmin, xmax, binsize, wavelength=None, excludeddetlist=[],scalefactor=None):
def reduceSpicePDData(self, exp, scan, unit, xmin, xmax, binsize, wavelength=None, excludeddetlist=None,scalefactor=None):
""" Reduce SPICE powder diffraction data.
Return - Boolean as reduction is successful or not
"""
# Default
if excludeddetlist is None:
excludeddetlist = None

# Get reduction manager
try:
wsmanager = self._myWorkspaceDict[(int(exp), int(scan))]
Expand Down Expand Up @@ -800,8 +805,8 @@ def saveMergedScan(self, sfilename, mergeindex):
else:
raise NotImplementedError('Unable to locate the merged scan workspace.')

api.SaveFocusedXYE(InputWorkspace=wksp,
StartAtBankNumber=1,
api.SaveFocusedXYE(InputWorkspace=wksp,
StartAtBankNumber=1,
Filename=sfilename)

return
Expand All @@ -824,29 +829,29 @@ def savePDFile(self, exp, scan, filetype, sfilename):
if sfilename.endswith('.dat') is True:
sfilename.replace('.dat', '.gsa')

api.SaveGSS(InputWorkspace=wksp,
Filename=sfilename,
api.SaveGSS(InputWorkspace=wksp,
Filename=sfilename,
SplitFiles=False, Append=False,
MultiplyByBinWidth=False,
Bank=1,
Format="SLOG",
Bank=1,
Format="SLOG",
ExtendedHeader=True)
# ENDIF

if "fullprof" in filetype:
if sfilename.endswith('.gsa') is True:
sfilename.replace('.gsa', '.dat')

api.SaveFocusedXYE(InputWorkspace=wksp,
StartAtBankNumber=1,
api.SaveFocusedXYE(InputWorkspace=wksp,
StartAtBankNumber=1,
Filename=sfilename)
# ENDIF

if "topas" in filetype:
sfilename = sfilename[:-4]+".xye"
api.SaveFocusedXYE(InputWorkspace=wksp,
api.SaveFocusedXYE(InputWorkspace=wksp,
StartAtBankNumber=info["bank"],
Filename=sfilename,
Filename=sfilename,
Format="TOPAS")
# ENDIF

Expand Down Expand Up @@ -900,9 +905,9 @@ def smoothVanadiumSpectrum(self, expno, scanno, smoothparams_str):
IgnoreXBins=True,
AllSpectra=True)

if outws is not None:
if outws is not None:
wsmanager.setProcessedVanadiumDataTemp(outws, "FFT smooth")

return True


Expand Down Expand Up @@ -936,7 +941,6 @@ def stripVanadiumPeaks(self, exp, scan, binparams, vanpeakposlist=None):
raise NotImplementedError('No vanadium peaks has been set up.')
# ENDIF


outwsname = wksp.name()+"_rmVan"
wksp = api.StripPeaks(InputWorkspace=wksp,
OutputWorkspace=outwsname,
Expand Down Expand Up @@ -991,7 +995,9 @@ def _getValueFromTable(self, tablews, colname, rowindex=0):

return rvalue

""" External Methods """
#-------------------------------------------------------------------------------
# External Methods
#-------------------------------------------------------------------------------
def downloadFile(url, localfilepath):
"""
Test: 'http://neutron.ornl.gov/user_data/hb2a/exp400/Datafiles/HB2A_exp0400_scan0001.dat'
Expand Down

0 comments on commit 2a69ede

Please sign in to comment.