1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

//! Python bindings for QSym² symmetry analysis via representation and corepresentation theories.
//!
//! See [`crate::drivers::representation_analysis`] for more information.

use num_complex::Complex;
use numpy::{PyArray1, PyArray2, PyArray4};
use pyo3::prelude::*;

pub mod density;
pub mod multideterminant;
pub mod slater_determinant;
pub mod vibrational_coordinate;

type C128 = Complex<f64>;

/// Python-exposed enumerated type to handle the union type of numpy float 1d-arrays and numpy
/// complex 1d-arrays in Python.
#[derive(FromPyObject)]
pub enum PyArray1RC<'a> {
    Real(Bound<'a, PyArray1<f64>>),
    Complex(Bound<'a, PyArray1<C128>>),
}

/// Python-exposed enumerated type to handle the union type of numpy float 2d-arrays and numpy
/// complex 2d-arrays in Python.
#[derive(FromPyObject)]
pub enum PyArray2RC<'a> {
    Real(Bound<'a, PyArray2<f64>>),
    Complex(Bound<'a, PyArray2<C128>>),
}

/// Python-exposed enumerated type to handle the union type of numpy float 4d-arrays and numpy
/// complex 4d-arrays in Python.
#[derive(FromPyObject)]
pub enum PyArray4RC<'a> {
    Real(Bound<'a, PyArray4<f64>>),
    Complex(Bound<'a, PyArray4<C128>>),
}