A Python library for computing (optimal) anisotropic power diagrams using GPU acceleration. Current main application concerns geometric modelling of polycrystalline materials with curved boundaries with grains of prescribed volumes and fine control over aspect ratio and location of the grains.
Install with pip:
pip install PyAPD
See /notebooks/tutorials/example_usage.ipynb 
To quickly test the speed of the GPU-acceleration, you can play around with the notebook example_usage.ipynb in Google Colab 
Note that by default the Google Colab runtime is CPU-only. To change to a GPU runtime, go to Runtime > Change runtime type and click on T4 GPU. Note that T4 GPU is considered pretty slow and is provided by Google Colab free of charge. To access the gold standard A100 GPU via Google Colab, a subscription is required.
For local use, if you do not have a GPU, you should expect a warning
[KeOps] Warning : Cuda libraries were not detected on the system ; using cpu only mode
but the library is fully functional.
This library is accompanied by the paper
- M. Buze, J. Feydy, S.M. Roper, K. Sedighiani, D.P. Bourne (2024). Anisotropic power diagrams for polycrystal modeling: efficient generation of curved grains via optimal transport. arXiv e-prints 2403.03571, 2024. LINK
The examples presented in the paper can be found in /notebooks/paper_examples, which includes all the Jupyter notebooks as they were run, the data that was generated and the plots from the paper. For the ease of access, here we list them with links to view them statically in NBViewer and also a link to an interactive version in Google Colab. Note that the notebooks relying the loading of data will not work out of the box in Google Colab -- the relevant data will have to be loaded manually.
- Runtime tests
- APD generation
- Gathering data
- Plotting the data for the 2D case
and for the 3D case
- Gathering data
- Finding optimal APDs
- Gathering data
- Plotting the data for the 3D multi-phase case
- Gathering data
- EBSD data examples
- Fitting to data and artificial sample generation
- Pixel level comparison
- Fitting to data and artificial sample generation
- Additive manufacturing example
- APD generation
If you use PyAPD for academic research, you may cite the paper to which our library is tied as follows (preprint version for now).
@misc{pyapd2024,
title={Anisotropic power diagrams for polycrystal modelling: efficient generation of curved grains via optimal transport},
author={Maciej Buze and Jean Feydy and Steven M. Roper and Karo Sedighiani and David P. Bourne},
year={2024},
eprint={2403.03571},
archivePrefix={arXiv},
primaryClass={cond-mat.mtrl-sci}
}