opencosmo 1.0.0__tar.gz

opencosmo-1.0.0/LICENSE.md

@@ -0,0 +1,11 @@
+BSD 3-Clause License
+
+Copyright (c) 2026, OpenCosmo Project All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
+
+Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
+Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
+Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

opencosmo-1.0.0/PKG-INFO

@@ -0,0 +1,182 @@
+Metadata-Version: 2.3
+Name: opencosmo
+Version: 1.0.0
+Summary: OpenCosmo Python Toolkit
+Author: Patrick Wells, Will Hicks, Patricia Larsen, Michael Buehlmann
+Author-email: Patrick Wells <pwells@anl.gov>, Will Hicks <whicks@anl.gov>, Patricia Larsen <prlarsen@anl.gov>, Michael Buehlmann <mbuehlmann@anl.gov>
+Requires-Dist: h5py>=3.12.1,<4.0.0
+Requires-Dist: astropy>=7.2.0,<8.0.0
+Requires-Dist: pydantic>=2.10.6,<3.0.0
+Requires-Dist: hdf5plugin>=5.0.0,<6.0.0
+Requires-Dist: healpy>=1.18.1,<2.0.0
+Requires-Dist: healsparse==1.11.1
+Requires-Dist: deprecated>=1.2.18,<2.0.0
+Requires-Dist: numpy>=2.0,<2.4
+Requires-Dist: click>=8.2.1,<9.0.0
+Requires-Dist: numba>=0.62.1
+Requires-Dist: rustworkx>=0.17.1
+Requires-Dist: pyarrow>=21.0.0 ; extra == 'io'
+Requires-Python: >=3.11, <3.15
+Provides-Extra: io
+Description-Content-Type: text/markdown
+
+<h1 align="center">
+    <picture>
+        <source srcset="https://raw.githubusercontent.com/ArgonneCPAC/opencosmo/main/branding/opencosmo_dark.png" media="(prefers-color-scheme: dark)">
+        <source srcset="https://raw.githubusercontent.com/ArgonneCPAC/opencosmo/main/branding/opencosmo_light.png" media="(prefers-color-scheme: light)">
+        <img src="https://raw.githubusercontent.com/ArgonneCPAC/opencosmo/main/branding/opencosmo_light.png" alt="OpenCosmo">
+    </picture>
+</h1><br>
+
+[![CI](https://github.com/ArgonneCPAC/OpenCosmo/actions/workflows/merge.yaml/badge.svg)](https://github.com/ArgonneCPAC/OpenCosmo/actions/workflows/merge.yaml)
+[![PyPI - Version](https://img.shields.io/pypi/v/opencosmo)](https://pypi.org/project/opencosmo/)
+[![Conda Version](https://img.shields.io/conda/vn/conda-forge/opencosmo)](https://anaconda.org/conda-forge/opencosmo)
+[![GitHub License](https://img.shields.io/github/license/ArgonneCPAC/opencosmo)](https://github.com/ArgonneCPAC/OpenCosmo/blob/main/LICENSE.md)
+
+
+The OpenCosmo Python Toolkit provides utilities for reading, writing and manipulating data from cosmological simulations produced by the Cosmolgical Physics and Advanced Computing (CPAC) group at Argonne National Laboratory. It can be used to work with smaller quantities data retrieved with the CosmoExplorer, as well as the much larget datasets these queries draw from. The OpenCosmo toolkit integrates with standard tools such as AstroPy, and allows you to manipulate data in a fully-consistent cosmological context.
+
+### Installation
+
+The OpenCosmo library is available for Python 3.11 and up on Linux and MacOS (and Windows via [WSL](https://learn.microsoft.com/en-us/windows/wsl/setup/environment)). It can be installed easily with `pip`:
+
+```bash
+pip install opencosmo
+```
+
+
+There's a good chance the default version of Python on your system is less than 3.11. Whether or not this is the case, we recommend installing `opencosmo` into a virtual environment. If you're using [Conda](https://docs.conda.io/projects/conda/en/stable/user-guide/getting-started.html), you can create a new environment and install `opencosmo` into it automatically:
+
+```bash
+conda create -n opencosmo_env conda-forge::opencosmo
+conda activate opencosmo_env
+```
+
+or if you already have a virtual environment to use:
+
+```bash
+conda install conda-forge::opencosmo
+```
+
+If you plan to use `opencosmo` in a Jupyter notebook, you can install the `ipykernel` package to make the environment available as a kernel:
+
+```bash
+pip install ipykernel # can also be installed with conda
+python -m ipykernel install --user --name=opencosmo
+```
+
+Be sure you have run the "activate" command shown above before running the `ipykernel` command.
+
+## Getting Started
+
+To get started, download the "haloproperites.hdf5" from the [OpenCosmo Google Drive](https://drive.google.com/drive/folders/1CYmZ4sE-RdhRdLhGuYR3rFfgyA3M1mU-?usp=sharing). This file contains properties of dark-matter halos from a small hydrodynamical simulation run with HACC. You can easily open the data with the `open` command:
+
+```python
+import opencosmo as oc
+
+dataset = oc.open("haloproperties.hdf5")
+print(dataset)
+```
+
+```text
+OpenCosmo Dataset (length=237441)
+Cosmology: FlatLambdaCDM(name=None, H0=<Quantity 67.66 km / (Mpc s)>, Om0=0.3096446816186967, Tcmb0=<Quantity 0. K>, Neff=3.04, m_nu=None, Ob0=0.04897468161869667)
+First 10 rows:
+block fof_halo_1D_vel_disp fof_halo_center_x ... sod_halo_sfr unique_tag
+             km / s               Mpc        ... solMass / yr
+int32       float32             float32      ...   float32      int64
+----- -------------------- ----------------- ... ------------ ----------
+    0            32.088795         1.4680439 ...       -101.0      21674
+    0             41.14525        0.19616994 ...       -101.0      44144
+    0             73.82962         1.5071135 ...    3.1447952      48226
+    0             31.17231         0.7526525 ...       -101.0      58472
+    0            23.038841         5.3246417 ...       -101.0      60550
+    0            37.071426         0.5153746 ...       -101.0     537760
+    0            26.203058         2.1734374 ...       -101.0     542858
+    0              78.7636         2.1477687 ...          0.0     548994
+    0             37.12636         6.9660196 ...       -101.0     571540
+    0             58.09235          6.072006 ...    1.5439711     576648
+```
+
+The `open` function returns a `Dataset` object, which holds the raw data as well as information about the simulation. You can easily access the data and cosmology as Astropy objects:
+```python
+dataset.data
+dataset.cosmology
+```
+
+The first will return an astropy table of the data, with all associated units already applied. The second will return the astropy cosmology object that represents the cosmology the simulation was run with. 
+
+### Basic Querying
+
+Although you can access data directly, `opencosmo` provides tools for querying and transforming the data in a fully cosmology-aware context. For example, suppose we wanted to plot the concentration-mass relationship for the halos in our simulation above a certain mass. One way to perform this would be as follows:
+
+```python
+dataset = dataset
+    .filter(oc.col("fof_halo_mass") > 1e13)
+    .take(1000, at="random")
+    .select(("fof_halo_mass", "sod_halo_cdelta"))
+
+print(dataset)
+
+```
+
+```text
+OpenCosmo Dataset (length=1000)
+Cosmology: FlatLambdaCDM(name=None, H0=<Quantity 67.66 km / (Mpc s)>, Om0=0.3096446816186967, Tcmb0=<Quantity 0. K>, Neff=3.04, m_nu=None, Ob0=0.04897468161869667)
+First 10 rows:
+ fof_halo_mass   sod_halo_cdelta
+    solMass
+    float32          float32
+---------------- ---------------
+11220446000000.0       4.5797048
+17266723000000.0       7.4097505
+51242150000000.0       1.8738283
+70097712000000.0       4.2764015
+51028305000000.0        2.678151
+11960567000000.0       3.9594727
+15276915000000.0        5.793542
+16002001000000.0       2.4318497
+47030307000000.0       3.7146702
+15839942000000.0        3.245569
+```
+
+We could then plot the data, or perform further transformations. This is cool on its own, but the real power of `opencosmo` comes from its ability to work with different data types. Go ahead and download the "haloparticles" file from the [OpenCosmo Google Drive](https://drive.google.com/drive/folders/1CYmZ4sE-RdhRdLhGuYR3rFfgyA3M1mU-?usp=sharing) and try the following:
+
+```python
+import opencosmo as oc
+
+data = oc.open("haloproperties.hdf5", "haloparticles.hdf5")
+```
+This will return a data *collection* that will allow you to query and transform the data as before, but will associate the halos with their particles. 
+
+```python
+
+data = data
+    .filter(oc.col("fof_halo_mass") > 1e13)
+    .take(1000, at="random")
+
+for halo in data.halos():
+    halo_properties = halo["halo_properties"]
+    dm_particles = halo["dm_particles"]
+    star_particles = halo["star_particles"]
+```
+
+In each iteration, "halo properties" will be a dictionary containing the properties of the halo (such as its total mass), while "dm_particles" and "star_particles" will be OpenCosmo datasets containing the dark matter and stars associated with the halo, respectively. Because these are just like the dataset object we saw eariler, we can further query and transform the particles as needed for our analysis. For more details on how to use the library, check out the [full documentation](https://opencosmo.readthedocs.io/en/latest/).
+
+### Testing
+
+To run tests, first download the test data [from Google Drive](https://drive.google.com/drive/folders/1CYmZ4sE-RdhRdLhGuYR3rFfgyA3M1mU-?usp=sharing). Set environment variable `OPENCOSMO_DATA_PATH` to the path where the data is stored. Then run the tests with `pytest`:
+
+```bash
+export OPENCOSMO_DATA_PATH=/path/to/data
+# From the repository root
+pytest --ignore test/parallel 
+```
+
+Although opencosmo does support multi-core processing via MPI, the default installation does not include the necessary dependencies to work in an MPI environment. If you need these capabilities, check out the guide in our documentation.
+
+### Contributing
+
+We welcome bug reports and feature requests from the community. If you would like to contribute to the project, please check out the [contributing guide](CONTRIBUTING.md) for more information.
+
+```

opencosmo-1.0.0/README.md

@@ -0,0 +1,160 @@
+<h1 align="center">
+    <picture>
+        <source srcset="https://raw.githubusercontent.com/ArgonneCPAC/opencosmo/main/branding/opencosmo_dark.png" media="(prefers-color-scheme: dark)">
+        <source srcset="https://raw.githubusercontent.com/ArgonneCPAC/opencosmo/main/branding/opencosmo_light.png" media="(prefers-color-scheme: light)">
+        <img src="https://raw.githubusercontent.com/ArgonneCPAC/opencosmo/main/branding/opencosmo_light.png" alt="OpenCosmo">
+    </picture>
+</h1><br>
+
+[![CI](https://github.com/ArgonneCPAC/OpenCosmo/actions/workflows/merge.yaml/badge.svg)](https://github.com/ArgonneCPAC/OpenCosmo/actions/workflows/merge.yaml)
+[![PyPI - Version](https://img.shields.io/pypi/v/opencosmo)](https://pypi.org/project/opencosmo/)
+[![Conda Version](https://img.shields.io/conda/vn/conda-forge/opencosmo)](https://anaconda.org/conda-forge/opencosmo)
+[![GitHub License](https://img.shields.io/github/license/ArgonneCPAC/opencosmo)](https://github.com/ArgonneCPAC/OpenCosmo/blob/main/LICENSE.md)
+
+
+The OpenCosmo Python Toolkit provides utilities for reading, writing and manipulating data from cosmological simulations produced by the Cosmolgical Physics and Advanced Computing (CPAC) group at Argonne National Laboratory. It can be used to work with smaller quantities data retrieved with the CosmoExplorer, as well as the much larget datasets these queries draw from. The OpenCosmo toolkit integrates with standard tools such as AstroPy, and allows you to manipulate data in a fully-consistent cosmological context.
+
+### Installation
+
+The OpenCosmo library is available for Python 3.11 and up on Linux and MacOS (and Windows via [WSL](https://learn.microsoft.com/en-us/windows/wsl/setup/environment)). It can be installed easily with `pip`:
+
+```bash
+pip install opencosmo
+```
+
+
+There's a good chance the default version of Python on your system is less than 3.11. Whether or not this is the case, we recommend installing `opencosmo` into a virtual environment. If you're using [Conda](https://docs.conda.io/projects/conda/en/stable/user-guide/getting-started.html), you can create a new environment and install `opencosmo` into it automatically:
+
+```bash
+conda create -n opencosmo_env conda-forge::opencosmo
+conda activate opencosmo_env
+```
+
+or if you already have a virtual environment to use:
+
+```bash
+conda install conda-forge::opencosmo
+```
+
+If you plan to use `opencosmo` in a Jupyter notebook, you can install the `ipykernel` package to make the environment available as a kernel:
+
+```bash
+pip install ipykernel # can also be installed with conda
+python -m ipykernel install --user --name=opencosmo
+```
+
+Be sure you have run the "activate" command shown above before running the `ipykernel` command.
+
+## Getting Started
+
+To get started, download the "haloproperites.hdf5" from the [OpenCosmo Google Drive](https://drive.google.com/drive/folders/1CYmZ4sE-RdhRdLhGuYR3rFfgyA3M1mU-?usp=sharing). This file contains properties of dark-matter halos from a small hydrodynamical simulation run with HACC. You can easily open the data with the `open` command:
+
+```python
+import opencosmo as oc
+
+dataset = oc.open("haloproperties.hdf5")
+print(dataset)
+```
+
+```text
+OpenCosmo Dataset (length=237441)
+Cosmology: FlatLambdaCDM(name=None, H0=<Quantity 67.66 km / (Mpc s)>, Om0=0.3096446816186967, Tcmb0=<Quantity 0. K>, Neff=3.04, m_nu=None, Ob0=0.04897468161869667)
+First 10 rows:
+block fof_halo_1D_vel_disp fof_halo_center_x ... sod_halo_sfr unique_tag
+             km / s               Mpc        ... solMass / yr
+int32       float32             float32      ...   float32      int64
+----- -------------------- ----------------- ... ------------ ----------
+    0            32.088795         1.4680439 ...       -101.0      21674
+    0             41.14525        0.19616994 ...       -101.0      44144
+    0             73.82962         1.5071135 ...    3.1447952      48226
+    0             31.17231         0.7526525 ...       -101.0      58472
+    0            23.038841         5.3246417 ...       -101.0      60550
+    0            37.071426         0.5153746 ...       -101.0     537760
+    0            26.203058         2.1734374 ...       -101.0     542858
+    0              78.7636         2.1477687 ...          0.0     548994
+    0             37.12636         6.9660196 ...       -101.0     571540
+    0             58.09235          6.072006 ...    1.5439711     576648
+```
+
+The `open` function returns a `Dataset` object, which holds the raw data as well as information about the simulation. You can easily access the data and cosmology as Astropy objects:
+```python
+dataset.data
+dataset.cosmology
+```
+
+The first will return an astropy table of the data, with all associated units already applied. The second will return the astropy cosmology object that represents the cosmology the simulation was run with. 
+
+### Basic Querying
+
+Although you can access data directly, `opencosmo` provides tools for querying and transforming the data in a fully cosmology-aware context. For example, suppose we wanted to plot the concentration-mass relationship for the halos in our simulation above a certain mass. One way to perform this would be as follows:
+
+```python
+dataset = dataset
+    .filter(oc.col("fof_halo_mass") > 1e13)
+    .take(1000, at="random")
+    .select(("fof_halo_mass", "sod_halo_cdelta"))
+
+print(dataset)
+
+```
+
+```text
+OpenCosmo Dataset (length=1000)
+Cosmology: FlatLambdaCDM(name=None, H0=<Quantity 67.66 km / (Mpc s)>, Om0=0.3096446816186967, Tcmb0=<Quantity 0. K>, Neff=3.04, m_nu=None, Ob0=0.04897468161869667)
+First 10 rows:
+ fof_halo_mass   sod_halo_cdelta
+    solMass
+    float32          float32
+---------------- ---------------
+11220446000000.0       4.5797048
+17266723000000.0       7.4097505
+51242150000000.0       1.8738283
+70097712000000.0       4.2764015
+51028305000000.0        2.678151
+11960567000000.0       3.9594727
+15276915000000.0        5.793542
+16002001000000.0       2.4318497
+47030307000000.0       3.7146702
+15839942000000.0        3.245569
+```
+
+We could then plot the data, or perform further transformations. This is cool on its own, but the real power of `opencosmo` comes from its ability to work with different data types. Go ahead and download the "haloparticles" file from the [OpenCosmo Google Drive](https://drive.google.com/drive/folders/1CYmZ4sE-RdhRdLhGuYR3rFfgyA3M1mU-?usp=sharing) and try the following:
+
+```python
+import opencosmo as oc
+
+data = oc.open("haloproperties.hdf5", "haloparticles.hdf5")
+```
+This will return a data *collection* that will allow you to query and transform the data as before, but will associate the halos with their particles. 
+
+```python
+
+data = data
+    .filter(oc.col("fof_halo_mass") > 1e13)
+    .take(1000, at="random")
+
+for halo in data.halos():
+    halo_properties = halo["halo_properties"]
+    dm_particles = halo["dm_particles"]
+    star_particles = halo["star_particles"]
+```
+
+In each iteration, "halo properties" will be a dictionary containing the properties of the halo (such as its total mass), while "dm_particles" and "star_particles" will be OpenCosmo datasets containing the dark matter and stars associated with the halo, respectively. Because these are just like the dataset object we saw eariler, we can further query and transform the particles as needed for our analysis. For more details on how to use the library, check out the [full documentation](https://opencosmo.readthedocs.io/en/latest/).
+
+### Testing
+
+To run tests, first download the test data [from Google Drive](https://drive.google.com/drive/folders/1CYmZ4sE-RdhRdLhGuYR3rFfgyA3M1mU-?usp=sharing). Set environment variable `OPENCOSMO_DATA_PATH` to the path where the data is stored. Then run the tests with `pytest`:
+
+```bash
+export OPENCOSMO_DATA_PATH=/path/to/data
+# From the repository root
+pytest --ignore test/parallel 
+```
+
+Although opencosmo does support multi-core processing via MPI, the default installation does not include the necessary dependencies to work in an MPI environment. If you need these capabilities, check out the guide in our documentation.
+
+### Contributing
+
+We welcome bug reports and feature requests from the community. If you would like to contribute to the project, please check out the [contributing guide](CONTRIBUTING.md) for more information.
+
+```

opencosmo-1.0.0/pyproject.toml

@@ -0,0 +1,121 @@
+[project]
+name = "opencosmo"
+version = "1.0.0"
+description = "OpenCosmo Python Toolkit"
+authors = [
+  { name = "Patrick Wells", email = "pwells@anl.gov" },
+  { name = "Will Hicks", email = "whicks@anl.gov"},
+  { name = "Patricia Larsen", email = "prlarsen@anl.gov" },
+  { name = "Michael Buehlmann", email = "mbuehlmann@anl.gov" }
+
+]
+readme = "README.md"
+requires-python = ">=3.11,<3.15"
+
+dependencies = [
+  "h5py>=3.12.1,<4.0.0",
+  "astropy>=7.2.0,<8.0.0",
+  "pydantic>=2.10.6,<3.0.0",
+  "hdf5plugin>=5.0.0,<6.0.0",
+  "healpy>=1.18.1,<2.0.0",
+  "healsparse==1.11.1",
+  "deprecated>=1.2.18,<2.0.0",
+  "numpy>=2.0,<2.4",
+  "click (>=8.2.1,<9.0.0)",
+  "numba>=0.62.1",
+  "rustworkx>=0.17.1",
+]
+
+[dependency-groups]
+dev = [
+  "ruff>=0.9.4,<1.0.0",
+  "mypy>=1.15,<2.0.0",
+  "types-deprecated>=1.2.15.20250304",
+  "pre-commit>=4.2.0,<5.0.0",
+  "towncrier (>=24.8.0,<25.0.0)",
+  "pip>=25.1.1",
+  "pytest>=8.3.4,<9.0.0",
+  "pytest-timeout>=2.4.0",
+]
+docs = [
+  "sphinx>=9.0.0",
+  "sphinx-rtd-theme>=3.0.2,<4.0.0",
+  "autodoc-pydantic>=2.2.0,<3.0.0",
+]
+mpi = [
+  "mpi4py>=3.1.0,<5.0.0"
+]
+test = [
+    "pandas>=2.3.3",
+    "pandas-stubs>=2.3.2.250926",
+    "polars>=1.35.2",
+    "pyarrow>=21.0.0",
+    "pyxsim>=4.4.3",
+    "yt>=4.4.1",
+]
+test-mpi = [
+  "mpi-pytest>=2025.4.0,<2026.0.0"
+]
+[project.scripts]
+opencosmo = "opencosmo.analysis.cli:cli"
+
+[project.optional-dependencies]
+io = [
+    "pyarrow>=21.0.0",
+]
+
+[build-system]
+requires = ["uv_build>=0.8.2,<0.9.0", "pip"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+source-include = ["LICENSE.md"]
+
+
+[[tool.mypy.overrides]]
+module = ["h5py", "astropy.*", "healpy", "healsparse", "numba"]
+ignore_missing_imports = true
+
+[tool.towncrier]
+directory = "changes"
+package = "opencosmo"
+filename = "docs/source/changelog.rst"
+
+[tool.pytest.ini_options]
+testpaths = ["test"]
+log_cli = true
+log_cli_level = "INFO"
+
+[tool.pyrefly]
+project-includes = [
+    "**/*.py*",
+]
+replace-imports-with-any = ["astropy.units", "astropy.constants"]
+
+[lint.pycodestyle]
+max-doc-length = 90
+
+[tool.ruff.lint]
+future-annotations = true
+preview = true
+extend-select = ["TC"]
+
+[tool.towncrier.fragment.improvement]
+name = "Improvements"
+showcontent = true
+
+[tool.towncrier.fragment.feature]
+name = "New Features"
+
+[tool.towncrier.fragment.bugfix]
+name = "Bugfixes"
+
+[tool.towncrier.fragment.doc]
+name = "Documentation"
+
+[tool.towncrier.fragment.removal]
+name = "Deprecations and Removals"
+
+[tool.towncrier.fragment.misc]
+name = "Miscellaneous"
+

opencosmo-1.0.0/src/opencosmo/__init__.py

@@ -0,0 +1,26 @@
+from .collection import (
+    HealpixMap,
+    Lightcone,
+    SimulationCollection,
+    StructureCollection,
+)
+from .column import col
+from .dataset import Dataset
+from .io import open, write
+from .spatial import make_box, make_cone
+
+__version__ = "1.0.0"
+
+__all__ = [
+    "write",
+    "col",
+    "open",
+    "Dataset",
+    "StructureCollection",
+    "SimulationCollection",
+    "Lightcone",
+    "HealpixMap",
+    "make_box",
+    "make_cone",
+    "__version__",
+]

opencosmo-1.0.0/src/opencosmo/analysis/__init__.py

@@ -0,0 +1,42 @@
+# ruff: noqa
+from importlib import import_module
+
+known_yt_tools = [
+    "create_yt_dataset",
+    "ProjectionPlot",
+    "SlicePlot",
+    "ParticleProjectionPlot",
+    "ProfilePlot",
+    "PhasePlot",
+    "visualize_halo",
+    "halo_projection_array",
+]
+
+
+"""
+Right now, we have only have two analysis modules so we can handle them directly. In the 
+future we will need to implement a more robust system that handles things automatically.
+"""
+
+
+def __getattr__(name):
+    if name in known_yt_tools:
+        yt_viz = import_module(".yt_viz", package="opencosmo.analysis")
+        yt_utils = import_module(".yt_utils", package="opencosmo.analysis")
+        try:
+            return getattr(yt_viz, name)
+        except AttributeError:
+            pass
+        try:
+            return getattr(yt_utils, name)
+        except AttributError:
+            pass
+
+        # except ImportError as ie:
+        #    raise ImportError(
+        #        "You tried to import one of the OpenCosmo YT tools, but your python "
+        #        "environment does not have the necessary dependencies. You can do install "
+        #        "them with `pip install opencosmo[analysis]`\n"
+        #        f"{ie}"
+        #    )
+    raise ImportError(f"Cannot import name '{name}' from opencosmo.analysis")

opencosmo-1.0.0/src/opencosmo/analysis/cli.py

@@ -0,0 +1,32 @@
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Optional
+
+import click
+
+from opencosmo.analysis.install import get_file_versions, install_spec
+
+if TYPE_CHECKING:
+    from pathlib import Path
+
+
+@click.group()
+def cli():
+    pass
+
+
+@cli.command(name="install")
+@click.argument("spec_name", required=True)
+@click.option("--file", type=click.Path(exists=True), required=False)
+@click.option("--dev", is_flag=True)
+def install(spec_name: str, file: Optional[Path] = None, dev: bool = False):
+    if file is not None:
+        versions = get_file_versions(spec_name, file)
+    else:
+        versions = {}
+
+    install_spec(spec_name, versions, dev=dev)
+
+
+if __name__ == "__main__":
+    cli()

opencosmo-1.0.0/src/opencosmo/analysis/diffsky.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from collections import namedtuple
+from typing import TYPE_CHECKING, NamedTuple, Type, TypeVar
+
+if TYPE_CHECKING:
+    from opencosmo import Dataset
+
+DIFFMAH_INPUT = namedtuple(
+    "DIFFMAH_INPUT", ["logm0", "logtc", "early_index", "late_index", "t_peak"]
+)
+
+T = TypeVar("T", bound=NamedTuple)
+
+
+def make_named_tuple(dataset: Dataset, input_tuple: Type[T]) -> T:
+    required_columns = input_tuple._fields
+    data = dataset.select(required_columns).data
+    output = {c: data[c].value for c in required_columns}
+    return input_tuple(**output)  # type: ignore

opencosmo-1.0.0/src/opencosmo/analysis/install/__init__.py

@@ -0,0 +1,5 @@
+from .install import install_spec
+from .specs import get_specs
+from .versions import get_file_versions
+
+__all__ = ["get_specs", "install_spec", "get_file_versions"]