asteroid_spinprops 0.2.15__tar.gz → 0.2.16__tar.gz

{asteroid_spinprops-0.2.15 → asteroid_spinprops-0.2.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: asteroid_spinprops
-Version: 0.2.15
+Version: 0.2.16
 Summary: Collection of tools used for fitting sHG1G2 and ssHG1G2 photometric models to sparse asteroid photometry
 License: MIT
 Author: Odysseas

{asteroid_spinprops-0.2.15 → asteroid_spinprops-0.2.16}/asteroid_spinprops/ssolib/modelfit.py

@@ -26,6 +26,7 @@ def get_fit_params(
     alt_spin=False,
     period_in=None,
     terminator=False,
+    pole_metadata=False,
 ):
     if survey_filter is None:
         filter_mask = np.array(data["cfid"].values[0]) >= 0
@@ -91,6 +92,14 @@ def get_fit_params(
                 ra_init, dec_init = utils.generate_initial_points(
                     ra0, dec0, dec_shift=45
                 )
+                if pole_metadata:
+                    pole_md = pd.DataFrame()
+                    pole_md["ra0"] = [ra0]
+                    pole_md["dec0"] = [dec0]
+                    pole_md["ra_init"] = [ra_init]
+                    pole_md["dec_init"] = [dec_init]
+
+                    ra_fin, dec_fin = [], []
                 for ra, dec in zip(ra_init, dec_init):
                     for period_in in period_scan:
                         p_in = [
@@ -115,12 +124,24 @@ def get_fit_params(
                         try:
                             rms.append(sshg1g2["rms"])
                             model.append(sshg1g2)
+                            if pole_metadata:
+                                ra_fin.append(sshg1g2["alpha0"])
+                                dec_fin.append(sshg1g2["dec0"])
                         except Exception:
                             continue
                 rms = np.array(rms)
                 sshg1g2_opt = model[rms.argmin()]
 
-                return sshg1g2_opt
+                if pole_metadata:
+                    ra_fin = np.array(ra_fin)
+                    dec_fin = np.array(dec_fin)
+
+                    pole_md["ra_fin"] = [ra_fin]
+                    pole_md["dec_fin"] = [dec_fin]
+
+                    return sshg1g2_opt, pole_md
+                else:
+                    return sshg1g2_opt
             else:
                 period_si_t, alt_period_si_t, _ = utils.estimate_sidereal_period(
                     data=data, model_parameters=shg1g2_params, synodic_period=period_sy

{asteroid_spinprops-0.2.15 → asteroid_spinprops-0.2.16}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "asteroid_spinprops"
-version = "0.2.15"
+version = "0.2.16"
 description = "Collection of tools used for fitting sHG1G2 and ssHG1G2 photometric models to sparse asteroid photometry"
 authors = [
     {name = "Odysseas",email = "odysseas.xenos@proton.me"}