jinwurs 0.3.0__tar.gz

jinwurs-0.3.0/Cargo.lock

@@ -0,0 +1,269 @@
+# This file is automatically @generated by Cargo.
+# It is not intended for manual editing.
+version = 4
+
+[[package]]
+name = "autocfg"
+version = "1.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f2032f911046de80f0a198e0901378627c33f59ea0ac00e363d481118bd70a53"
+
+[[package]]
+name = "cfg-if"
+version = "1.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9330f8b2ff13f34540b44e946ef35111825727b38d33286ef986142615121801"
+
+[[package]]
+name = "heck"
+version = "0.5.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "2304e00983f87ffb38b55b444b5e3b60a884b5d30c0fca7d82fe33449bbe55ea"
+
+[[package]]
+name = "indoc"
+version = "2.0.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "79cf5c93f93228cf8efb3ba362535fb11199ac548a09ce117c9b1adc3030d706"
+dependencies = [
+ "rustversion",
+]
+
+[[package]]
+name = "jinwurs"
+version = "0.3.0"
+dependencies = [
+ "numpy",
+ "pyo3",
+]
+
+[[package]]
+name = "libc"
+version = "0.2.186"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "68ab91017fe16c622486840e4c83c9a37afeff978bd239b5293d61ece587de66"
+
+[[package]]
+name = "matrixmultiply"
+version = "0.3.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a06de3016e9fae57a36fd14dba131fccf49f74b40b7fbdb472f96e361ec71a08"
+dependencies = [
+ "autocfg",
+ "rawpointer",
+]
+
+[[package]]
+name = "memoffset"
+version = "0.9.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "488016bfae457b036d996092f6cb448677611ce4449e970ceaf42695203f218a"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "ndarray"
+version = "0.16.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "882ed72dce9365842bf196bdeedf5055305f11fc8c03dee7bb0194a6cad34841"
+dependencies = [
+ "matrixmultiply",
+ "num-complex",
+ "num-integer",
+ "num-traits",
+ "portable-atomic",
+ "portable-atomic-util",
+ "rawpointer",
+]
+
+[[package]]
+name = "num-complex"
+version = "0.4.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "73f88a1307638156682bada9d7604135552957b7818057dcef22705b4d509495"
+dependencies = [
+ "num-traits",
+]
+
+[[package]]
+name = "num-integer"
+version = "0.1.46"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7969661fd2958a5cb096e56c8e1ad0444ac2bbcd0061bd28660485a44879858f"
+dependencies = [
+ "num-traits",
+]
+
+[[package]]
+name = "num-traits"
+version = "0.2.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "071dfc062690e90b734c0b2273ce72ad0ffa95f0c74596bc250dcfd960262841"
+dependencies = [
+ "autocfg",
+]
+
+[[package]]
+name = "numpy"
+version = "0.23.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b94caae805f998a07d33af06e6a3891e38556051b8045c615470a71590e13e78"
+dependencies = [
+ "libc",
+ "ndarray",
+ "num-complex",
+ "num-integer",
+ "num-traits",
+ "pyo3",
+ "rustc-hash",
+]
+
+[[package]]
+name = "once_cell"
+version = "1.21.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9f7c3e4beb33f85d45ae3e3a1792185706c8e16d043238c593331cc7cd313b50"
+
+[[package]]
+name = "portable-atomic"
+version = "1.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c33a9471896f1c69cecef8d20cbe2f7accd12527ce60845ff44c153bb2a21b49"
+
+[[package]]
+name = "portable-atomic-util"
+version = "0.2.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "c2a106d1259c23fac8e543272398ae0e3c0b8d33c88ed73d0cc71b0f1d902618"
+dependencies = [
+ "portable-atomic",
+]
+
+[[package]]
+name = "proc-macro2"
+version = "1.0.106"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8fd00f0bb2e90d81d1044c2b32617f68fcb9fa3bb7640c23e9c748e53fb30934"
+dependencies = [
+ "unicode-ident",
+]
+
+[[package]]
+name = "pyo3"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7778bffd85cf38175ac1f545509665d0b9b92a198ca7941f131f85f7a4f9a872"
+dependencies = [
+ "cfg-if",
+ "indoc",
+ "libc",
+ "memoffset",
+ "once_cell",
+ "portable-atomic",
+ "pyo3-build-config",
+ "pyo3-ffi",
+ "pyo3-macros",
+ "unindent",
+]
+
+[[package]]
+name = "pyo3-build-config"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94f6cbe86ef3bf18998d9df6e0f3fc1050a8c5efa409bf712e661a4366e010fb"
+dependencies = [
+ "once_cell",
+ "target-lexicon",
+]
+
+[[package]]
+name = "pyo3-ffi"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e9f1b4c431c0bb1c8fb0a338709859eed0d030ff6daa34368d3b152a63dfdd8d"
+dependencies = [
+ "libc",
+ "pyo3-build-config",
+]
+
+[[package]]
+name = "pyo3-macros"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fbc2201328f63c4710f68abdf653c89d8dbc2858b88c5d88b0ff38a75288a9da"
+dependencies = [
+ "proc-macro2",
+ "pyo3-macros-backend",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "pyo3-macros-backend"
+version = "0.23.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fca6726ad0f3da9c9de093d6f116a93c1a38e417ed73bf138472cf4064f72028"
+dependencies = [
+ "heck",
+ "proc-macro2",
+ "pyo3-build-config",
+ "quote",
+ "syn",
+]
+
+[[package]]
+name = "quote"
+version = "1.0.45"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "41f2619966050689382d2b44f664f4bc593e129785a36d6ee376ddf37259b924"
+dependencies = [
+ "proc-macro2",
+]
+
+[[package]]
+name = "rawpointer"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "60a357793950651c4ed0f3f52338f53b2f809f32d83a07f72909fa13e4c6c1e3"
+
+[[package]]
+name = "rustc-hash"
+version = "2.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94300abf3f1ae2e2b8ffb7b58043de3d399c73fa6f4b73826402a5c457614dbe"
+
+[[package]]
+name = "rustversion"
+version = "1.0.22"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b39cdef0fa800fc44525c84ccb54a029961a8215f9619753635a9c0d2538d46d"
+
+[[package]]
+name = "syn"
+version = "2.0.118"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1b9ae57f904213ebb649ce6895b8a66c66f0203b9319718f69a5612a065b1422"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "target-lexicon"
+version = "0.12.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "61c41af27dd6d1e27b1b16b489db798443478cef1f06a660c96db617ba5de3b1"
+
+[[package]]
+name = "unicode-ident"
+version = "1.0.24"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e6e4313cd5fcd3dad5cafa179702e2b244f760991f45397d14d4ebf38247da75"
+
+[[package]]
+name = "unindent"
+version = "0.2.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "7264e107f553ccae879d21fbea1d6724ac785e8c3bfc762137959b5802826ef3"

jinwurs-0.3.0/Cargo.toml

@@ -0,0 +1,16 @@
+[package]
+name = "jinwurs"
+version = "0.3.0"
+edition = "2021"
+description = "Rust-accelerated kernels for jinwu — fast rebinning and X-ray statistics"
+license = "GPL-3.0-or-later"
+repository = "https://github.com/xinxiangsun/JinWu"
+readme = "README.md"
+
+[lib]
+name = "jinwurs"
+crate-type = ["cdylib"]
+
+[dependencies]
+pyo3 = { version = "0.23", features = ["extension-module", "abi3-py311"] }
+numpy = "0.23"

jinwurs-0.3.0/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.4
+Name: jinwurs
+Version: 0.3.0
+Classifier: License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Rust
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Astronomy
+Requires-Dist: numpy>=1.24
+Summary: Rust-accelerated kernels for jinwu — fast rebinning and X-ray statistics
+Keywords: astrophysics,transient,x-ray,rebinning,statistics
+Author-email: Xinxiang Sun <sunxx@nao.cas.cn>
+License: GPL-3.0-or-later
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: Homepage, https://github.com/xinxiangsun/JinWu
+Project-URL: Repository, https://github.com/xinxiangsun/JinWu
+
+# jinwurs
+
+Rust-accelerated kernels for [JinWu](https://github.com/xinxiangsun/JinWu) — fast rebinning and X-ray statistics.
+
+Provides high-performance Rust implementations of core algorithms used in X-ray and gamma-ray astronomy data analysis, exposed as Python extension modules via PyO3.
+

jinwurs-0.3.0/README.md

@@ -0,0 +1,5 @@
+# jinwurs
+
+Rust-accelerated kernels for [JinWu](https://github.com/xinxiangsun/JinWu) — fast rebinning and X-ray statistics.
+
+Provides high-performance Rust implementations of core algorithms used in X-ray and gamma-ray astronomy data analysis, exposed as Python extension modules via PyO3.

jinwurs-0.3.0/conda-recipe/meta.yaml

@@ -0,0 +1,39 @@
+{% set name = "jinwurs" %}
+{% set version = "0.3.0" %}
+
+package:
+  name: {{ name|lower }}
+  version: {{ version }}
+
+source:
+  url: https://pypi.io/packages/source/{{ name[0] }}/{{ name }}/jinwurs-{{ version }}.tar.gz
+  # No jinwurs 0.3.0 sdist is published; this recipe is not yet submit-ready.
+  sha256: placeholder
+
+build:
+  number: 0
+  # ⚠️ 不能 noarch — Rust .so 是平台相关的
+  script: {{ PYTHON }} -m pip install . -vv
+
+requirements:
+  build:
+    - {{ compiler('rust') }}
+    - maturin >=1.0
+  host:
+    - python
+    - pip
+    - maturin
+  run:
+    - python
+
+test:
+  imports:
+    - jinwurs
+  commands:
+    - pip check
+
+about:
+  home: https://github.com/xinxiangsun/jinwu
+  license: GPL-3.0-or-later
+  license_file: LICENSE
+  summary: Rust-accelerated kernels for jinwu — fast rebinning & statistics

jinwurs-0.3.0/pyproject.toml

@@ -0,0 +1,36 @@
+[build-system]
+requires = ["maturin>=1.5"]
+build-backend = "maturin"
+
+[project]
+name = "jinwurs"
+version = "0.3.0"
+description = "Rust-accelerated kernels for jinwu — fast rebinning and X-ray statistics"
+readme = "README.md"
+license = { text = "GPL-3.0-or-later" }
+authors = [
+    { name = "Xinxiang Sun", email = "sunxx@nao.cas.cn" }
+]
+requires-python = ">=3.11"
+dependencies = ["numpy>=1.24"]
+keywords = ["astrophysics", "transient", "x-ray", "rebinning", "statistics"]
+classifiers = [
+    "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Rust",
+    "Intended Audience :: Science/Research",
+    "Topic :: Scientific/Engineering :: Astronomy",
+]
+
+[project.urls]
+Homepage = "https://github.com/xinxiangsun/JinWu"
+Repository = "https://github.com/xinxiangsun/JinWu"
+
+[tool.maturin]
+manifest-path = "Cargo.toml"
+features = ["pyo3/extension-module", "pyo3/abi3-py311"]
+bindings = "pyo3"

jinwurs-0.3.0/src/lib.rs

@@ -0,0 +1,127 @@
+//! jinwurs — Rust-accelerated kernels for jinwu
+
+use numpy::{PyArray1, PyReadonlyArray1};
+use pyo3::prelude::*;
+
+/// Core rebinning: distribute original-bin counts into new uniform bins.
+#[pyfunction]
+#[pyo3(signature = (orig_counts, orig_errors, orig_left, orig_right, orig_width, orig_exposure, new_edges))]
+fn rebin_counts_core<'py>(
+    py: Python<'py>,
+    orig_counts: PyReadonlyArray1<'py, f64>,
+    orig_errors: Option<PyReadonlyArray1<'py, f64>>,
+    orig_left: PyReadonlyArray1<'py, f64>,
+    orig_right: PyReadonlyArray1<'py, f64>,
+    orig_width: PyReadonlyArray1<'py, f64>,
+    orig_exposure: Option<PyReadonlyArray1<'py, f64>>,
+    new_edges: PyReadonlyArray1<'py, f64>,
+) -> (Py<PyArray1<f64>>, Py<PyArray1<f64>>, Py<PyArray1<f64>>) {
+    let oc = orig_counts.as_slice().unwrap();
+    let ol = orig_left.as_slice().unwrap();
+    let or_ = orig_right.as_slice().unwrap();
+    let ow = orig_width.as_slice().unwrap();
+    let ne = new_edges.as_slice().unwrap();
+
+    let oe: Option<&[f64]> = orig_errors.as_ref().map(|a| a.as_slice().unwrap());
+    let oexpo: Option<&[f64]> = orig_exposure.as_ref().map(|a| a.as_slice().unwrap());
+
+    let n_orig = oc.len();
+    let m = ne.len().saturating_sub(1);
+    let have_exposure = oexpo.is_some();
+
+    let mut counts = vec![0.0_f64; m];
+    let mut var = vec![0.0_f64; m];
+    let mut exposure = vec![0.0_f64; m];
+
+    for i in 0..n_orig {
+        let a = ol[i];
+        let b = or_[i];
+        if b <= ne[0] || a >= ne[m] {
+            continue;
+        }
+
+        // binary search for overlapping new-bin range
+        let j0 = ne.partition_point(|&e| e <= a).saturating_sub(1);
+        let j1 = ne.partition_point(|&e| e < b).saturating_sub(1);
+        let j0 = j0.min(m.saturating_sub(1));
+        let j1 = j1.min(m.saturating_sub(1));
+
+        let owi = ow[i];
+        let ci = oc[i];
+        let ei = oe.map(|e| e[i]).unwrap_or_else(|| ci.max(0.0).sqrt());
+
+        for j in j0..=j1 {
+            let new_l = ne[j];
+            let new_r = ne[j + 1];
+            let overlap = (b.min(new_r) - a.max(new_l)).max(0.0);
+            if overlap <= 0.0 {
+                continue;
+            }
+            let frac = overlap / owi;
+            let contrib = ci * frac;
+            counts[j] += contrib;
+            var[j] += (ei * frac).powi(2);
+
+            if have_exposure {
+                exposure[j] += oexpo.unwrap()[i] * frac;
+            } else {
+                exposure[j] += overlap;
+            }
+        }
+    }
+
+    let c = PyArray1::from_vec(py, counts).unbind();
+    let v = PyArray1::from_vec(py, var).unbind();
+    let e = PyArray1::from_vec(py, exposure).unbind();
+    (c, v, e)
+}
+
+/// Finalize rebinned counts into counts space or rate space.
+#[pyfunction]
+fn rebin_finalize<'py>(
+    py: Python<'py>,
+    counts: PyReadonlyArray1<'py, f64>,
+    var: PyReadonlyArray1<'py, f64>,
+    exposure: PyReadonlyArray1<'py, f64>,
+    method: &str,
+    empty_nan: bool,
+) -> (Py<PyArray1<f64>>, Py<PyArray1<f64>>) {
+    let c = counts.as_slice().unwrap();
+    let v = var.as_slice().unwrap();
+    let e = exposure.as_slice().unwrap();
+    let m = c.len();
+
+    let mut value = vec![0.0_f64; m];
+    let mut error = vec![0.0_f64; m];
+
+    if method == "sum" {
+        for j in 0..m {
+            value[j] = c[j];
+            error[j] = v[j].sqrt();
+        }
+    } else {
+        for j in 0..m {
+            let denom = if e[j] > 0.0 { e[j] } else { f64::NAN };
+            value[j] = c[j] / denom;
+            error[j] = v[j].sqrt() / denom;
+        }
+    }
+
+    if empty_nan {
+        for j in 0..m {
+            if e[j] == 0.0 {
+                value[j] = f64::NAN;
+                error[j] = f64::NAN;
+            }
+        }
+    }
+
+    (PyArray1::from_vec(py, value).unbind(), PyArray1::from_vec(py, error).unbind())
+}
+
+#[pymodule]
+fn jinwurs(m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_function(wrap_pyfunction!(rebin_counts_core, m)?)?;
+    m.add_function(wrap_pyfunction!(rebin_finalize, m)?)?;
+    Ok(())
+}