pybevy 0.2.1

PyBevy: A Python Real-Time Engine Built on Bevy
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
39
40
41
42
43
44
45
46
47
48
49
50
51

use crate::ecs::{
    component_type::PyComponentType, filter::QueryFilter,
    query::query_helpers::construct_query_class_item,
};

pub mod query_helpers;
pub mod query_param;
pub mod query_runtime;
pub mod single;
pub mod single_runtime;

use pyo3::{
    Bound,
    prelude::*,
    types::{PyAny, PyType},
};

pub(crate) enum ParamType {
    Entity,
    Component {
        ty: PyComponentType,
        mutable: bool,
        optional: bool,
    },
    Filter(QueryFilter),
}

impl From<QueryFilter> for ParamType {
    fn from(filter: QueryFilter) -> Self {
        ParamType::Filter(filter)
    }
}

/// An ECS Query that can be used to retrieve entities and their components.
#[pyclass(name = "Query", frozen)]
pub struct PyQuery;

#[pymethods]
impl PyQuery {
    /// Creates a new `QueryParam` from the given type parameters.
    ///
    /// This enables syntax like `Query[Transform]` or `Query[tuple[Transform, Velocity]]`
    #[classmethod]
    #[pyo3(signature = (key, /))]
    pub fn __class_getitem__(
        cls: &Bound<'_, PyType>,
        key: &Bound<'_, PyAny>,
    ) -> PyResult<Py<PyAny>> {
        construct_query_class_item(cls, key)
    }
}