masster 0.2.4__tar.gz → 0.2.5__tar.gz

{masster-0.2.4 → masster-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: masster
-Version: 0.2.4
+Version: 0.2.5
 Summary: Mass spectrometry data analysis package
 Project-URL: homepage, https://github.com/zamboni-lab/masster
 Project-URL: repository, https://github.com/zamboni-lab/masster

{masster-0.2.4 → masster-0.2.5}/pyproject.toml

@@ -1,7 +1,7 @@
 
 [project]
 name = "masster"
-version = "0.2.4"
+version = "0.2.5"
 description = "Mass spectrometry data analysis package"
 authors = [
     { name = "Zamboni Lab" }

{masster-0.2.4 → masster-0.2.5}/src/masster/_version.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 
-__version__ = "0.2.4"
+__version__ = "0.5.1"
 
 
 def get_version():

{masster-0.2.4 → masster-0.2.5}/src/masster/sample/save.py

@@ -168,7 +168,7 @@ def export_mgf(
     q1_ratio_max=None,
     eic_corr_min=None,
     deisotope=True,
-    precursor_trim=-(-10.0),
+    precursor_trim=10.0,
     centroid_algo=None,
 ):
     """
@@ -354,81 +354,98 @@ def export_mgf(
             elif use_cache:
                 spect = row["ms2_specs"]
                 if spect is None:
-                    print(f"No MS2 spectrum for feature {row['fid']} is cached.")
-                    continue
-                # check if spec is a list of spectra
-                if isinstance(spect, list):
-                    if selection == "best":
-                        s = spect[0]
-                        scan_uid = row["ms2_scans"][0]
-                        s.energy = self.get_spectrum(scan_uid).energy
-                        spect = [s]
-                        scan_uids = [scan_uid]
-                    else:
-                        scan_uids = row["ms2_scans"]
-
-                    for i, s in enumerate(spect):
-                        if s is None:
-                            print(
-                                f"No MS2 spectrum for feature {row['fid']} is cached.",
+                    # No cached spectra, fall through to fetch from scan_uid
+                    use_cache = False
+                else:
+                    # check if spec is a list of spectra
+                    if isinstance(spect, list):
+                        if selection == "best":
+                            s = spect[0]
+                            scan_uid = row["ms2_scans"][0]
+                            s.energy = self.get_spectrum(scan_uid).energy
+                            spect = [s]
+                            scan_uids = [scan_uid]
+                        else:
+                            scan_uids = row["ms2_scans"]
+
+                        for i, s in enumerate(spect):
+                            if s is None:
+                                print(
+                                    f"No MS2 spectrum for feature {feature_uid} is cached.",
+                                )
+                                continue
+                            # check if s is a spectrum
+                            if type(s).__name__ == "Spectrum":
+                                s = filter_peaks(
+                                    s,
+                                    inty_min=inty_min,
+                                    q1_min=q1_ratio_min,
+                                    eic_min=eic_corr_min,
+                                    q1_max=q1_ratio_max,
+                                )
+                                # Get the corresponding scan_uid from the list
+                                current_scan_uid = scan_uids[i] if i < len(scan_uids) else "unknown"
+                                write_ion(
+                                    f,
+                                    f"fid:{feature_uid}, rt:{rt_str}, mz:{mz_str}, scan_uid:{current_scan_uid}",
+                                    feature_uid,
+                                    mz,
+                                    rt,
+                                    charge,
+                                    s,
+                                )
+                                c += 1
+                        continue  # Skip the rest of the processing for this feature
+            
+            # If we reach here, either use_cache=False or no cached spectra were available
+            if split_energy:
+                # get energy of all scans with scan_uid in ms2_scans by fetching them
+                ms2_scan_uids = row["ms2_scans"]
+                if isinstance(ms2_scan_uids, list) and len(ms2_scan_uids) > 0:
+                    # Fetch spectra to get energy information
+                    spectra_with_energy = []
+                    for scan_uid in ms2_scan_uids:
+                        spec = self.get_spectrum(scan_uid)
+                        if spec is not None:
+                            spectra_with_energy.append((scan_uid, spec.energy if hasattr(spec, 'energy') else 0))
+                    
+                    # Group by energy
+                    energy_groups: dict[float, list[int]] = {}
+                    for scan_uid, energy in spectra_with_energy:
+                        if energy not in energy_groups:
+                            energy_groups[energy] = []
+                        energy_groups[energy].append(scan_uid)
+                    
+                    for energy, scan_uids_for_energy in energy_groups.items():
+                        if selection == "best":
+                            # Keep only the first scan for this energy
+                            scan_uids_for_energy = [scan_uids_for_energy[0]]
+                        
+                        for scan_uid in scan_uids_for_energy:
+                            spect = self.get_spectrum(
+                                scan_uid,
+                                centroid=centroid,
+                                deisotope=deisotope,
+                                precursor_trim=precursor_trim,
+                                centroid_algo=centroid_algo,
                             )
-                            continue
-                        # check if s is a spectrum
-                        if type(s).__name__ == "spec":
-                            s = filter_peaks(
-                                s,
+                            spect = filter_peaks(
+                                spect,
                                 inty_min=inty_min,
                                 q1_min=q1_ratio_min,
                                 eic_min=eic_corr_min,
                                 q1_max=q1_ratio_max,
                             )
-                            # Get the corresponding scan_uid from the list
-                            current_scan_uid = scan_uids[i] if i < len(scan_uids) else "unknown"
                             write_ion(
                                 f,
-                                f"fid:{feature_uid}, rt:{rt_str}, mz:{mz_str}, scan_uid:{current_scan_uid}",
+                                f"fid:{feature_uid}, rt:{rt_str}, mz:{mz_str}, scan_uid:{scan_uid}, energy:{energy}",
                                 feature_uid,
                                 mz,
                                 rt,
                                 charge,
-                                s,
+                                spect,
                             )
                             c += 1
-            elif split_energy:
-                # get energy of all scans with scan_uid in ms2_scans
-                energy = [s.energy for s in row["ms2_specs"]]
-                # find unique energies
-                unique_energies = list(set(energy))
-                for e in unique_energies:
-                    ms2_scans = [row["ms2_scans"][i] for i, s in enumerate(row["ms2_specs"]) if s.energy == e]
-                    if selection == "best":
-                        # Keep as list with single element
-                        ms2_scans = [ms2_scans[0]]
-                    for scan_uid in ms2_scans:
-                        spect = self.get_spectrum(
-                            scan_uid,
-                            centroid=centroid,
-                            deisotope=deisotope,
-                            precursor_trim=precursor_trim,
-                            centroid_algo=centroid_algo,
-                        )
-                        spect = filter_peaks(
-                            spect,
-                            inty_min=inty_min,
-                            q1_min=q1_ratio_min,
-                            eic_min=eic_corr_min,
-                            q1_max=q1_ratio_max,
-                        )
-                        write_ion(
-                            f,
-                            f"fid:{feature_uid}, rt:{rt_str}, mz:{mz_str}, scan_uid:{scan_uid}, energy:{e}",
-                            feature_uid,
-                            mz,
-                            rt,
-                            charge,
-                            spect,
-                        )
-                        c += 1
             else:
                 if selection == "best":
                     ms2_scans = row["ms2_scans"][0]

{masster-0.2.4 → masster-0.2.5}/uv.lock

@@ -1585,7 +1585,7 @@ wheels = [
 
 [[package]]
 name = "masster"
-version = "0.2.4"
+version = "0.2.5"
 source = { editable = "." }
 dependencies = [
     { name = "alphabase" },