# import mantid algorithms, numpy and matplotlib
from mantid.simpleapi import *
import matplotlib.pyplot as plt
import numpy as np


import matplotlib.pyplot as plt
from mantid.plots.utility import MantidAxType
# Fit definition, see https://docs.mantidproject.org/algorithms/Fit-v1.html for more details
Function="name=BackToBackExponential,I=63.0628,A=382.979,B=100.266,X0=4.77635,S=0.00549865,ties=(A=382.979,B=100.266)"
InputWorkspace="WISH00056237_H00_foc"
Output="WISH00056237_H00_foc"
StartX=4.68455
EndX=4.89934
MaxIterations=4997
Normalise=True
OutputCompositeMembers=True
Fit(EndX=EndX,Function=Function,InputWorkspace=InputWorkspace,MaxIterations=MaxIterations,Normalise=Normalise,Output=Output,OutputCompositeMembers=OutputCompositeMembers,StartX=StartX)

from mantid.api import AnalysisDataService as ADS

WISH00056237_H00_foc = ADS.retrieve('WISH00056237_H00_foc')
WISH00056237_H00_foc_Workspace = ADS.retrieve('WISH00056237_H00_foc_Workspace')

fig, axes = plt.subplots(figsize=[8.7927, 8.6829], num='WISH00056237_H00_foc-6', subplot_kw={'projection': 'mantid'})
axes.plot(WISH00056237_H00_foc, color='#1f77b4', label='WISH00056237_H00_foc: spec 1', wkspIndex=0)
axes.plot(WISH00056237_H00_foc_Workspace, color='#ff7f0e', label='WISH00056237_H00_foc_Workspace: Calc', wkspIndex=1)
axes.plot(WISH00056237_H00_foc_Workspace, color='#2ca02c', label='WISH00056237_H00_foc_Workspace: Diff', wkspIndex=2)
axes.tick_params(axis='x', which='major', **{'gridOn': False, 'tick1On': True, 'tick2On': False, 'label1On': True, 'label2On': False, 'size': 6, 'tickdir': 'out', 'width': 1})
axes.tick_params(axis='y', which='major', **{'gridOn': False, 'tick1On': True, 'tick2On': False, 'label1On': True, 'label2On': False, 'size': 6, 'tickdir': 'out', 'width': 1})
axes.set_title('WISH00056237_H00_foc')
axes.set_xlabel('d-Spacing ($\\AA$)')
axes.set_ylabel('Counts (microAmp.hour $\\AA$)$^{-1}$')
axes.set_xlim([4.6726, 4.9113])
axes.set_ylim([-454.03, 5809.4])
legend = axes.legend(fontsize=8.0).draggable().legend

plt.show()
# Scripting Plots in Mantid:
# https://docs.mantidproject.org/tutorials/python_in_mantid/plotting/02_scripting_plots.html