libgunshotmatch 0.9.0__tar.gz → 0.10.0__tar.gz

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: libgunshotmatch
-Version: 0.9.0
+Version: 0.10.0
 Summary: Base library for GunShotMatch.
 Author-email: Dominic Davis-Foster <dominic@davis-foster.co.uk>
 License: MIT
@@ -135,7 +135,7 @@ libgunshotmatch
 .. |language| image:: https://img.shields.io/github/languages/top/GunShotMatch/libgunshotmatch
 	:alt: GitHub top language
 
-.. |commits-since| image:: https://img.shields.io/github/commits-since/GunShotMatch/libgunshotmatch/v0.9.0
+.. |commits-since| image:: https://img.shields.io/github/commits-since/GunShotMatch/libgunshotmatch/v0.10.0
 	:target: https://github.com/GunShotMatch/libgunshotmatch/pulse
 	:alt: GitHub commits since tagged version
 

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/README.rst

@@ -91,7 +91,7 @@ libgunshotmatch
 .. |language| image:: https://img.shields.io/github/languages/top/GunShotMatch/libgunshotmatch
 	:alt: GitHub top language
 
-.. |commits-since| image:: https://img.shields.io/github/commits-since/GunShotMatch/libgunshotmatch/v0.9.0
+.. |commits-since| image:: https://img.shields.io/github/commits-since/GunShotMatch/libgunshotmatch/v0.10.0
 	:target: https://github.com/GunShotMatch/libgunshotmatch/pulse
 	:alt: GitHub commits since tagged version
 

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/libgunshotmatch/__init__.py

@@ -29,5 +29,5 @@ Base library for GunShotMatch.
 __author__: str = "Dominic Davis-Foster"
 __copyright__: str = "2020-2023 Dominic Davis-Foster"
 __license__: str = "MIT License"
-__version__: str = "0.9.0"
+__version__: str = "0.10.0"
 __email__: str = "dominic@davis-foster.co.uk"

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/libgunshotmatch/consolidate/__init__.py

@@ -59,7 +59,8 @@ __all__ = (
 		"ConsolidatedSearchResult",
 		"match_counter",
 		"pairwise_ms_comparisons",
-		"ConsolidatedPeakFilter"
+		"ConsolidatedPeakFilter",
+		"InvertedFilter",
 		)
 
 
@@ -205,12 +206,16 @@ class ConsolidatedSearchResult:
 		:param d:
 		"""
 
+		hit_numbers = [float("nan") if hn == -65535 else hn for hn in d["hit_numbers"]]
+		mf_list = [float("nan") if mf == -65535 else mf for mf in d["mf_list"]]
+		rmf_list = [float("nan") if mf == -65535 else mf for mf in d["rmf_list"]]
+
 		return cls(
 				name=d["name"],
 				cas=d["cas"],
-				mf_list=d["mf_list"],
-				rmf_list=d["rmf_list"],
-				hit_numbers=d["hit_numbers"],
+				mf_list=mf_list,
+				rmf_list=rmf_list,
+				hit_numbers=hit_numbers,
 				reference_data=d["reference_data"],
 				)
 
@@ -768,3 +773,34 @@ class ConsolidatedPeakFilter:
 		"""
 
 		return [cp for cp in consolidated_peaks if not self.should_filter_peak(cp)]
+
+
+@_fix_init_annotations
+@attr.define
+class InvertedFilter(ConsolidatedPeakFilter):
+	"""
+	Inverted version of :class:`~.ConsolidatedPeakFilter`.
+
+	Returns peaks which would be excluded by a :class:`~.ConsolidatedPeakFilter`.
+
+	.. versionadded:: 0.10.0
+	"""
+
+	def print_skip_reason(self, peak: ConsolidatedPeak, reason: str) -> None:  # noqa: D102
+		if self.verbose:
+			print(f"Would reject peak at {peak.rt / 60:0.3f} mins:", reason)
+
+	def filter(self, consolidated_peaks: List[ConsolidatedPeak]) -> List[ConsolidatedPeak]:  # noqa: A003  # pylint: disable=redefined-builtin
+		"""
+		Filter a list of consolidated peaks.
+
+		:param consolidated_peaks:
+		"""
+
+		filtered_consolidated_peaks = []
+
+		for cp in consolidated_peaks:
+			if self.should_filter_peak(cp):
+				filtered_consolidated_peaks.append(cp)
+
+		return filtered_consolidated_peaks

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/libgunshotmatch/gzip_util.py

@@ -51,7 +51,13 @@ def read_gzip_json(path: PathLike) -> JSONOutput:
 	"""
 
 	with gzip.open(PathPlus(path), 'r') as f:
-		return sdjson.load(f)
+		try:
+			# 3rd party
+			import orjson
+			data = f.read().decode().replace("NaN", "-65535")
+			return orjson.loads(data)
+		except ImportError:
+			return sdjson.load(f)
 
 
 def write_gzip_json(path: PathLike, data: JSONInput, indent: Optional[int] = 2) -> None:

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/libgunshotmatch/project.py

@@ -50,7 +50,7 @@ from libgunshotmatch.datafile import Repeat
 from libgunshotmatch.peak import PeakList, QualifiedPeak, peak_from_dict
 from libgunshotmatch.utils import create_alignment
 
-__all__ = ["Project"]
+__all__ = ("Project", "consolidate")
 
 
 @attr.define
@@ -176,31 +176,7 @@ class Project:
 			between the repeats for each aligned peak.
 		"""
 
-		ms_comparison_df = pairwise_ms_comparisons(self.alignment)
-
-		peak_numbers: List[int] = []
-		peak: Optional[QualifiedPeak]
-
-		qualified_peak_array = []
-
-		# for experiment in project.alignment.expr_code:
-		for experiment in self.datafile_data:
-			qualified_peaks = self.datafile_data[experiment].qualified_peaks
-			assert qualified_peaks is not None
-			for peak in qualified_peaks:
-				assert peak.peak_number is not None
-				peak_numbers.append(peak.peak_number)
-			qualified_peak_array.append(qualified_peaks)
-
-		# Convert peak_numbers to a set and sort smallest to largest
-		peak_numbers = sorted(set(peak_numbers))
-
-		consolidated_peaks = match_counter(
-				engine=engine,
-				peak_numbers=peak_numbers,
-				qualified_peaks=qualified_peak_array,
-				ms_comp_data=ms_comparison_df,
-				)
+		consolidated_peaks, ms_comparison_df = consolidate(self, engine)
 
 		if peak_filter is None:
 			self.consolidated_peaks = consolidated_peaks
@@ -210,3 +186,48 @@ class Project:
 		return ms_comparison_df
 
 		# chart_data = make_chart_data(self)
+
+
+def consolidate(
+		project: Project,
+		engine: pyms_nist_search.Engine,
+		) -> pandas.DataFrame:
+	"""
+	Consolidate the compound identification from the experiments into a single dataset.
+
+	:param project:
+	:param engine:
+
+	:returns: List of consolidated peaks and :class:`pandas.DataFrame`
+		giving the results of pairwise mass spectral comparisons between the repeats for each aligned peak.
+
+	.. versionadded:: 0.10.0
+	"""
+
+	ms_comparison_df = pairwise_ms_comparisons(project.alignment)
+
+	peak_numbers: List[int] = []
+	peak: Optional[QualifiedPeak]
+
+	qualified_peak_array = []
+
+	# for experiment in project.alignment.expr_code:
+	for experiment in project.datafile_data:
+		qualified_peaks = project.datafile_data[experiment].qualified_peaks
+		assert qualified_peaks is not None
+		for peak in qualified_peaks:
+			assert peak.peak_number is not None
+			peak_numbers.append(peak.peak_number)
+		qualified_peak_array.append(qualified_peaks)
+
+	# Convert peak_numbers to a set and sort smallest to largest
+	peak_numbers = sorted(set(peak_numbers))
+
+	consolidated_peaks = match_counter(
+			engine=engine,
+			peak_numbers=peak_numbers,
+			qualified_peaks=qualified_peak_array,
+			ms_comp_data=ms_comparison_df,
+			)
+
+	return consolidated_peaks, ms_comparison_df

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/libgunshotmatch/utils.py

@@ -32,6 +32,7 @@ from typing import TYPE_CHECKING, Any, Iterable, List, Optional, Sequence, Tuple
 
 # 3rd party
 import numpy
+from attr import AttrsInstance
 from chemistry_tools.spectrum_similarity import SpectrumSimilarity
 from mathematical.utils import rounders
 from pyms.DPA.Alignment import Alignment
@@ -114,10 +115,10 @@ def ms_comparison(top_ms: MassSpectrum, bottom_ms: MassSpectrum) -> Optional[flo
 	return match * 1000
 
 
-_O = TypeVar("_O", bound=object)
+_AI = TypeVar("_AI", bound=AttrsInstance)
 
 
-def _fix_init_annotations(method: Type[_O]) -> Type[_O]:
+def _fix_init_annotations(method: Type[_AI]) -> Type[_AI]:
 	init_annotations = method.__init__.__annotations__
 	cls_annotations = method.__annotations__
 

{libgunshotmatch-0.9.0 → libgunshotmatch-0.10.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "whey"
 
 [project]
 name = "libgunshotmatch"
-version = "0.9.0"
+version = "0.10.0"
 description = "Base library for GunShotMatch."
 readme = "README.rst"
 keywords = []