In [1]:
import os
import sys
root_folder = os.path.dirname(os.getcwd())
sys.path.append(root_folder)
from ResoFit._pulse_shape import NeutronPulse
from ResoFit._pulse_shape import ProtonPulse
from lmfit import Parameters
import numpy as np
import matplotlib.pyplot as plt
from lmfit import Model
import pprint
from lmfit.models import LinearModel
In [2]:
%matplotlib notebook
In [3]:
# path1 = '/Users/Shawn/Dropbox (ORNL)/Postdoc_Research/neutron_beam_shape/SNS/neutron_pulse/source_section_1.dat'
# path2 = '/Users/Shawn/Dropbox (ORNL)/Postdoc_Research/neutron_beam_shape/SNS/neutron_pulse/source_section_2.dat'
path1 = '/Users/y9z/Dropbox (ORNL)/Postdoc_Research/neutron_beam_shape/SNS/neutron_pulse/source_section_1.dat'
path2 = '/Users/y9z/Dropbox (ORNL)/Postdoc_Research/neutron_beam_shape/SNS/neutron_pulse/source_section_2.dat'
In [4]:
neutron_pulse = NeutronPulse(path1, model_index=1)
neutron_pulse.load_shape_each(path2)
In [5]:
neutron_pulse.plot_shape_total(x)
In [5]:
neutron_pulse.fit_shape(e_min=1, e_max=500,
drop=False, norm=True, check_each=True,
save_fig=False, overwrite_csv=True)