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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use std::time::Duration;

use bevy::app::{RunMode, ScheduleRunnerPlugin};
use pybevy_core::PyPlugin;
use pyo3::prelude::*;

use crate::prelude::PyApp;

#[pyclass(name = "ScheduleRunnerPlugin", extends = PyPlugin)]
#[derive(Clone)]
pub struct PyScheduleRunnerPlugin {
    run_mode: PyRunMode,
}

#[pyclass(name = "RunMode", frozen, eq)]
#[derive(Debug, Clone, Copy, PartialEq)]
pub enum PyRunMode {
    Loop { wait: Option<u64> },
    Once(),
}

impl From<PyRunMode> for RunMode {
    fn from(py_run_mode: PyRunMode) -> Self {
        match py_run_mode {
            PyRunMode::Loop { wait } => RunMode::Loop {
                wait: wait.map(Duration::from_millis),
            },
            PyRunMode::Once() => RunMode::Once,
        }
    }
}

impl PyScheduleRunnerPlugin {
    fn with_mode(run_mode: PyRunMode) -> Self {
        PyScheduleRunnerPlugin { run_mode }
    }
}

#[pymethods]
impl PyScheduleRunnerPlugin {
    #[new]
    #[pyo3(signature = (run_mode = PyRunMode::Loop { wait: None }))]
    pub fn new(run_mode: PyRunMode) -> (Self, PyPlugin) {
        (PyScheduleRunnerPlugin { run_mode }, PyPlugin)
    }

    #[staticmethod]
    pub fn run_once(py: Python) -> PyResult<Py<Self>> {
        Py::new(py, (Self::with_mode(PyRunMode::Once()), PyPlugin))
    }

    #[staticmethod]
    pub fn run_loop(py: Python, wait_duration: Option<u64>) -> PyResult<Py<Self>> {
        Py::new(
            py,
            (
                Self::with_mode(PyRunMode::Loop {
                    wait: wait_duration,
                }),
                PyPlugin,
            ),
        )
    }

    pub fn build(&self, app: Bound<'_, PyApp>) -> PyResult<()> {
        let run_mode: RunMode = self.run_mode.into();

        app.borrow().with_bevy_app(|bevy_app| {
            bevy_app.add_plugins(ScheduleRunnerPlugin { run_mode });
            Ok(())
        })
    }
}