darn-dmap 0.2.0__tar.gz → 0.2.1__tar.gz

{darn_dmap-0.2.0 → darn_dmap-0.2.1}/Cargo.lock

@@ -210,7 +210,7 @@ checksum = "460fbee9c2c2f33933d720630a6a0bac33ba7053db5344fac858d4b8952d77d5"
 
 [[package]]
 name = "dmap"
-version = "0.2.0"
+version = "0.2.1"
 dependencies = [
  "bzip2",
  "criterion",

{darn_dmap-0.2.0 → darn_dmap-0.2.1}/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "dmap"
-version = "0.2.0"
+version = "0.2.1"
 edition = "2021"
 rust-version = "1.63.0"
 

{darn_dmap-0.2.0 → darn_dmap-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: darn-dmap
-Version: 0.2.0
+Version: 0.2.1
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Rust
 Requires-Dist: numpy<3

{darn_dmap-0.2.0 → darn_dmap-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "darn-dmap"
-version = "0.2.0"
+version = "0.2.1"
 requires-python = ">=3.8"
 authors = [
     { name = "Remington Rohel" }

{darn_dmap-0.2.0 → darn_dmap-0.2.1}/src/lib.rs

@@ -163,7 +163,7 @@ read_rust!(dmap);
 /// Generates two functions: `read_[type]` and `read_[type]_lax`, for strict and lax
 /// reading, respectively.
 macro_rules! read_py {
-    ($name:ident, $py_name:literal, $lax_name:literal, $bytes_name:literal) => {
+    ($name:ident, $py_name:literal, $lax_name:literal, $bytes_name:literal, $lax_bytes_name:literal) => {
         paste! {
             #[doc = "Reads a `" $name:upper "` file, returning a list of dictionaries containing the fields." ]
             #[pyfunction]
@@ -204,27 +204,44 @@ macro_rules! read_py {
                     .collect()
                 )
             }
+
+            #[doc = "Reads a `" $name:upper "` file, returning a tuple of" ]
+            #[doc = "(list of dictionaries containing the fields, byte where first corrupted record starts). "]
+            #[pyfunction]
+            #[pyo3(name = $lax_bytes_name)]
+            #[pyo3(text_signature = "(buf: bytes, /)")]
+            fn [< read_ $name _bytes_lax_py >](
+                bytes: &[u8],
+            ) -> PyResult<(Vec<IndexMap<String, DmapField>>, Option<usize>)> {
+                let result = [< $name:camel Record >]::read_records_lax(bytes).map_err(PyErr::from)?;
+                Ok((
+                    result.0.into_iter().map(|rec| rec.inner()).collect(),
+                    result.1,
+                ))
+            }
         }
     }
 }
 
-read_py!(iqdat, "read_iqdat", "read_iqdat_lax", "read_iqdat_bytes");
+read_py!(iqdat, "read_iqdat", "read_iqdat_lax", "read_iqdat_bytes", "read_iqdat_bytes_lax");
 read_py!(
     rawacf,
     "read_rawacf",
     "read_rawacf_lax",
-    "read_rawacf_bytes"
+    "read_rawacf_bytes",
+    "read_rawacf_bytes_lax"
 );
 read_py!(
     fitacf,
     "read_fitacf",
     "read_fitacf_lax",
-    "read_fitacf_bytes"
+    "read_fitacf_bytes",
+    "read_fitacf_bytes_lax"
 );
-read_py!(grid, "read_grid", "read_grid_lax", "read_grid_bytes");
-read_py!(map, "read_map", "read_map_lax", "read_map_bytes");
-read_py!(snd, "read_snd", "read_snd_lax", "read_snd_bytes");
-read_py!(dmap, "read_dmap", "read_dmap_lax", "read_dmap_bytes");
+read_py!(grid, "read_grid", "read_grid_lax", "read_grid_bytes", "read_grid_bytes_lax");
+read_py!(map, "read_map", "read_map_lax", "read_map_bytes", "read_map_bytes_lax");
+read_py!(snd, "read_snd", "read_snd_lax", "read_snd_bytes", "read_snd_bytes_lax");
+read_py!(dmap, "read_dmap", "read_dmap_lax", "read_dmap_bytes", "read_dmap_bytes_lax");
 
 /// Checks that a list of dictionaries contains DMAP records, then appends to outfile.
 ///
@@ -315,6 +332,15 @@ fn dmap(m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_function(wrap_pyfunction!(read_grid_bytes_py, m)?)?;
     m.add_function(wrap_pyfunction!(read_map_bytes_py, m)?)?;
 
+    // Lax read functions from byte buffer
+    m.add_function(wrap_pyfunction!(read_dmap_bytes_lax_py, m)?)?;
+    m.add_function(wrap_pyfunction!(read_iqdat_bytes_lax_py, m)?)?;
+    m.add_function(wrap_pyfunction!(read_rawacf_bytes_lax_py, m)?)?;
+    m.add_function(wrap_pyfunction!(read_fitacf_bytes_lax_py, m)?)?;
+    m.add_function(wrap_pyfunction!(read_snd_bytes_lax_py, m)?)?;
+    m.add_function(wrap_pyfunction!(read_grid_bytes_lax_py, m)?)?;
+    m.add_function(wrap_pyfunction!(read_map_bytes_lax_py, m)?)?;
+
     // Write functions
     m.add_function(wrap_pyfunction!(write_dmap_py, m)?)?;
     m.add_function(wrap_pyfunction!(write_iqdat_py, m)?)?;