fixed background correction problem for analyzing PSF, corrected using LinearModel from lmfit as autobackground correction as part of a composite model with Gaussian Model

Author: SarthakJariwala

PythonGUI_apps/Spectrum_analysis/Spectra_plot_fit.py

@@ -17,7 +17,7 @@
 import matplotlib
 import pickle
 import lmfit
-from lmfit.models import GaussianModel
+from lmfit.models import GaussianModel, LinearModel
 from scipy import interpolate
 import customplotting.mscope as cpm
 
@@ -372,6 +372,16 @@ def fit_and_plot(self):
                 self.file = np.zeros((self.x.shape[0], 2))
                 self.file[:,0] = self.x
                 self.file[:,1] = self.y
+                bck = lmfit.models.LinearModel(prefix='line_')
+                gmodel = GaussianModel(prefix='g1_')
+                pars = bck.make_params(intercept=self.y.min(), slope=0)
+                pars += gmodel.guess(self.y, x=self.x)
+                comp_model = gmodel + bck
+                self.result = comp_model.fit(self.y, pars, x=self.x, nan_policy='propagate')
+                self.ui.plot.plot(self.x, self.y, clear=self.clear_check(), pen='r')
+                self.ui.plot.plot(self.x, self.result.best_fit, clear=False, pen='k')
+                self.ui.result_textBrowser.setText(self.result.fit_report())
+                
 
             if self.ui.plot_without_bck_radioButton.isChecked(): #if plot w/o bck, create dummy bck_file
                 self.bck_file = np.zeros(shape=(self.file.shape[0], 2))
@@ -381,6 +391,8 @@ def fit_and_plot(self):
             #     self.ui.result_textBrowser.setText("You need to check the subtract background option!")
             if self.check_loaded_files is None:
                 pass
+            elif self.opened_from_flim:
+                pass
             else:
 
                 if fit_func == "Single Gaussian": #and self.ui.subtract_bck_radioButton.isChecked() == True: