FFT_presenter.py

from __future__ import (absolute_import, division, print_function)
from six import iteritems
import mantid.simpleapi as mantid


class FFTPresenter(object):

    def __init__(self,view):
        self.view=view
        # set data
        self.getWorkspaceNames()
        #connect
        self.view.tableClickSignal.connect(self.tableClicked)
        self.view.buttonSignal.connect(self.handleButton)

    # only get ws that are groups or pairs
    # ignore raw
    def getWorkspaceNames(self):
        options=mantid.AnalysisDataService.getObjectNames()
        options=[item.replace(" ","") for item in options]
        final_options=[]
        for pick in options:
            if ";" in pick and "Raw" not in pick:
                final_options.append(pick)
        self.view.addItems(final_options)

    #functions
    def tableClicked(self,row,col):
        if row == self.view.getImBoxRow() and col ==1:
            self.view.changedHideUnTick(self.view.getImBox(),self.view.getImBoxRow()+1)
        elif  row == self.view.getShiftBoxRow() and col ==1:
            self.view.changed(self.view.getShiftBox(),self.view.getShiftBoxRow()+1)

    def handleButton(self):
        inputs = self.get_FFT_input()
        alg=mantid.AlgorithmManager.create("FFT")
        alg.initialize()
        alg.setChild(False)
        for name,value in iteritems(inputs):
            alg.setProperty(name,value)
            mantid.logger.warning(name+"  "+str(value))
        alg.execute()
        ws=alg.getPropertyValue("OutputWorkspace")
        if mantid.AnalysisDataService.doesExist("FFTMuon"):
            FFTMuon=mantid.AnalysisDataService.retrieve("FFTMuon")
            FFTMuon.add(ws)
        else:
            FFTMuon=mantid.GroupWorkspaces(InputWorkspaces=ws)

    def get_FFT_input(self):
        inputs=self.view.initFFTInput()
        if self.view.isRaw():
            self.view.addRaw(inputs,"InputWorkspace")
        if  self.view.isAutoShift():
            inputs["AutoShift"]=True
        else:
            self.view.addFFTShift(inputs)
        if self.view.isComplex():
            self.view.addFFTComplex(inputs)
            if self.view.isRaw():
                self.view.addRaw(inputs,"InputImagWorkspace")
        return inputs