mimical 0.0.7__py3-none-any.whl → 0.0.9__py3-none-any.whl

mimical/fitting/fitter.py

@@ -39,7 +39,7 @@ class mimical(object):
         A 3D array of image data with slices for each filter. Each image
         must be the same shape.
 
-    filt_list : list - str
+    filt_list : str or list
         A list of path strings to the filter transmission curve files, relative
         to the current working directory. Must be in ascending order with effective wavelength.
 
@@ -60,6 +60,14 @@ class mimical(object):
 
     def __init__(self, id, images, filt_list, psfs, user_prior, astropy_model=models.Sersic2D(), pool=None, timeout=3600):
         
+        # Helper when only one image is passed
+        if len(images.shape)==2:
+            images = np.array(([images]))
+        if (type(filt_list).__name__=='str') | (type(filt_list).__name__=='str_'):
+            filt_list = [filt_list]
+        if len(psfs.shape)==2:
+            psfs = np.array(([psfs]))
+
         self.id = id
 
         print(f"Fitting object {id}")
@@ -88,6 +96,7 @@ class mimical(object):
 
 
 
+
     def lnlike(self, param_dict):
         """ Returns the log-likelihood for a given parameter vector. """
 
@@ -165,7 +174,7 @@ class mimical(object):
 
         if self.success != True:
 
-            print("Sampling failed (timout).")
+            print("Sampling failed (timeout).")
             print(" ")
             return {}
     
@@ -195,7 +204,7 @@ class mimical(object):
 
     def plot_model(self):
         if self.success != True:
-            print(f'Failed sampling - cannot plot model for {self.id}.')
+            print(f'Failed sampling, cannot plot model for {self.id}.')
         else:
             # Plot and save the median-parameter fit
             Plotter().plot_median(self.images, self.wavs, self.convolved_models, self.samples, list(self.fitter_prior.keys), self.prior_handler)

mimical/plotting/plotting.py

@@ -57,16 +57,31 @@ class Plotter(object):
 
         median_models = np.median(models, axis=0)
 
-        for i in range(len(wavs)):
 
-            v = np.percentile(images[-1], 99.9)
+        if len(wavs)==1:
+            for i in range(len(wavs)):
+                v = np.percentile(images[-1], 99.9)
 
-            axes[0,i].imshow(images[i], vmax=v, vmin=-v)
-            axes[0,i].set_axis_off()
+                axes[0].imshow(images[i], vmax=v, vmin=-v)
+                axes[0].set_axis_off()
 
-            axes[1,i].imshow(median_models[i], vmax=v, vmin=-v)
-            axes[1,i].set_axis_off()
+                axes[1].imshow(median_models[i], vmax=v, vmin=-v)
+                axes[1].set_axis_off()
 
-            axes[2,i].imshow(images[i]-median_models[i], vmax=v, vmin=-v)
-            axes[2,i].set_axis_off()
+                axes[2].imshow(images[i]-median_models[i], vmax=v, vmin=-v)
+                axes[2].set_axis_off()
+
+        else:
+            for i in range(len(wavs)):
+
+                v = np.percentile(images[-1], 99.9)
+
+                axes[0,i].imshow(images[i], vmax=v, vmin=-v)
+                axes[0,i].set_axis_off()
+
+                axes[1,i].imshow(median_models[i], vmax=v, vmin=-v)
+                axes[1,i].set_axis_off()
+
+                axes[2,i].imshow(images[i]-median_models[i], vmax=v, vmin=-v)
+                axes[2,i].set_axis_off()
 

{mimical-0.0.7.dist-info → mimical-0.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mimical
-Version: 0.0.7
+Version: 0.0.9
 Summary: Intesity modelling of multiply-imaged objects
 Author: Struan Stevenson
 Author-email: struan.stevenson@ed.ac.uk

{mimical-0.0.7.dist-info → mimical-0.0.9.dist-info}/RECORD

@@ -1,12 +1,12 @@
 mimical/__init__.py,sha256=cBuyDtPwn6KcIkjlT7oct36jigvYDJmbZNmnt1I1QJU,75
 mimical/fitting/__init__.py,sha256=xcnLoeBDtt4Le5NhD6GhJe2ZRPNSsEySRHFYNuJyNEQ,69
-mimical/fitting/fitter.py,sha256=snF9OS6enzjBvcmFPyS5NmVljRRL54So4JN9QfE-SQY,8309
+mimical/fitting/fitter.py,sha256=zYbhc5iQsj_-b5Q6FbkCbh8fex2MFxGsBN2w4PB-1EY,8622
 mimical/fitting/prior_handler.py,sha256=yXvxbohS6f94CMRyeIViFc83hRYAj82IZNA5mzhMSRw,4193
 mimical/plotting/__init__.py,sha256=I1ZpQA48g-YPkqwrDGgTrqaTxDUgR8n0mbBX0MdbZtU,30
-mimical/plotting/plotting.py,sha256=OWTR-uOT4UpBw0XcKpl_uFSODuEdcfw7nIk3Sf1cOGY,2646
+mimical/plotting/plotting.py,sha256=RW6PgrX3UbafXmUnthV0mttKRjeUnS03eeX1c6fgZHU,3131
 mimical/utils/__init__.py,sha256=yolMgYFpvXoHuyV-ijGo26SQ0_niw0gz__iAFgdiQiY,35
 mimical/utils/filter_set.py,sha256=EITLa2c3FG3n1-v7KatKRwHCRyfZ4TTjsi_WM7Bz64k,6819
-mimical-0.0.7.dist-info/METADATA,sha256=KMd6HC02m-F7FQ64pHZi_aHBjyOG__rL-Lpc-mUwUvk,831
-mimical-0.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mimical-0.0.7.dist-info/top_level.txt,sha256=z6HTYpsoNjLUFayXjn8WyjX8C1mIbZw8Arb334cUbwc,8
-mimical-0.0.7.dist-info/RECORD,,
+mimical-0.0.9.dist-info/METADATA,sha256=WzyzgjIu15SVhcoTyLTIBYFWhiLEVrY759LSjRz46vU,831
+mimical-0.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mimical-0.0.9.dist-info/top_level.txt,sha256=z6HTYpsoNjLUFayXjn8WyjX8C1mIbZw8Arb334cUbwc,8
+mimical-0.0.9.dist-info/RECORD,,