amethyst_test 0.15.3

Amethyst test utilities crate
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

use std::marker::PhantomData;

use amethyst::{
    core::{bundle::SystemBundle, RunNowDesc},
    ecs::prelude::*,
    error::Error,
};
use derive_new::new;

/// Adds a specified thread local `System` to the dispatcher.
#[derive(Debug, new)]
pub struct ThreadLocalInjectionBundle<'a, 'b, RNDesc, RN>
where
    RNDesc: RunNowDesc<'a, 'b, RN>,
    RN: for<'s> RunNow<'s>,
{
    /// Function to instantiate `System` to add to the dispatcher.
    run_now_desc: RNDesc,
    /// Marker.
    #[new(default)]
    system_marker: PhantomData<(&'a RNDesc, &'b RN)>,
}

impl<'a, 'b, RNDesc, RN> SystemBundle<'a, 'b> for ThreadLocalInjectionBundle<'a, 'b, RNDesc, RN>
where
    RNDesc: RunNowDesc<'a, 'b, RN>,
    RN: for<'s> RunNow<'s> + 'b,
{
    fn build(
        self,
        world: &mut World,
        builder: &mut DispatcherBuilder<'a, 'b>,
    ) -> Result<(), Error> {
        let system = self.run_now_desc.build(world);
        builder.add_thread_local(system);
        Ok(())
    }
}