norbitAnalyzer 0.0.1__py3-none-any.whl

norbitAnalyzer/Sequence_frame/generate_hex_indexed_data.py

@@ -0,0 +1,57 @@
+import numpy as np
+
+
+def generate_hex_indices(start, stop, step):
+    """
+    Generate a list of hexadecimal frame indices.
+
+    Parameters
+    ----------
+    start : str
+        Starting hex index (e.g. "8000")
+    stop : str
+        Final hex index, inclusive (e.g. "10000")
+    step : str or int
+        Hex increment (e.g. "40" or 0x40)
+
+    Returns
+    -------
+    indices : list[str]
+        List of hex indices as lowercase strings without '0x'
+    """
+    # Convert inputs to integers
+    start_i = int(start, 16)
+    stop_i  = int(stop, 16)
+    step_i  = int(step, 16) if isinstance(step, str) else int(step)
+
+    if step_i <= 0:
+        raise ValueError("step must be positive")
+    if stop_i < start_i:
+        raise ValueError("stop must be >= start")
+
+    indices = []
+    i = start_i
+    while i <= stop_i:
+        indices.append(f"{i:x}")  # hex string without 0x
+        i += step_i
+
+    return indices
+
+
+# config
+N_ROWS = 2 ** 20            # rows per file
+VAL_RANGE = (-100, 100)     # min, max for all 6 columns
+PREFIX = "r10A"             # filename prefix
+OUT_DIR = "."                # output directory
+
+file_paths = []
+indices = generate_hex_indices("7000", "8000", "40")
+
+for idx in indices:
+    out_path = f"{OUT_DIR}/{PREFIX}_{idx}.txt"   # r10A_<index>.txt convention
+    data = np.random.uniform(VAL_RANGE[0], VAL_RANGE[1], size=(N_ROWS, 6))
+    np.savetxt(out_path, data, fmt="%.6f", header="x y z vx vy vz", comments="")
+    file_paths.append(out_path)
+    print(f"wrote {N_ROWS} rows to {out_path}")
+
+print(f"\ndone: {len(file_paths)} files written")

norbitAnalyzer/__init__.py

@@ -0,0 +1,24 @@
+import os
+
+__version__ = "0.0.1"
+
+# Fixed: Changed os.path_dirname to os.path.dirname
+norbit_dir = os.path.dirname(__file__)
+DATADIR = os.path.join(norbit_dir, "norbit_data_ex/")
+
+# Check for PyCUDA availability
+try: 
+    import pycuda.driver as cuda 
+    import pycuda.autoinit
+    from pycuda.compiler import SourceModule
+    cuda_ext = True
+except ImportError: 
+    cuda_ext = False
+
+
+try: 
+    from . import _kepler
+    cext = True 
+except ImportError: 
+    cext = False  
+