lgarrison

# /// script
# requires-python = ">=3.13"
# dependencies = [
#     "click",
#     "tqdm",
#     "zarr",
#     "zarrs",
# ]
# ///

lgarrison

# /// script
# requires-python = ">=3.11"
# dependencies = [
#     "numba",
#     "numpy",
# ]
# ///


# Example output:

lgarrison

# %ECSV 1.0
# ---
# datatype:
# - {name: x, datatype: float32}
# - {name: y, datatype: float32}
# - {name: z, datatype: float32}
# meta: {L: 2000.0}
# schema: astropy-2.0
x y z
1002.2712 1022.61536 251.77208

lgarrison

lgarrison

lgarrison

#!/usr/bin/env python3
'''
Compress a density grid.

Usage: ./compress.py --help
'''

from pathlib import Path

import click

lgarrison

lgarrison

lgarrison

import numpy as np
import threading
import queue

def multithreaded_read(fns, nthreads=2):
    '''
    A simple multithreaded reader that spins up N threads
    to read a list of files, returning the result in a list.
    Reads in Numpy format by default.
    

lgarrison

import numba as nb
import numpy as np

def nb_ecdf(X,Rmax,ngrid=64,nthreads=1):
    '''Compute the empirical CDF of points X (shape (N,D))
    in domain [0,Rmax) using a histogram with `ngrid` cells
    per dimension.
    
    Rmax and ngrid can also be length D arrays.
    '''