script/dom/webxr/
xrinputsourcearray.rs

1/* This Source Code Form is subject to the terms of the Mozilla Public
2 * License, v. 2.0. If a copy of the MPL was not distributed with this
3 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */
4
5use dom_struct::dom_struct;
6use webxr_api::{InputId, InputSource};
7
8use crate::dom::bindings::cell::DomRefCell;
9use crate::dom::bindings::codegen::Bindings::XRInputSourceArrayBinding::XRInputSourceArrayMethods;
10use crate::dom::bindings::inheritance::Castable;
11use crate::dom::bindings::reflector::{DomGlobal, Reflector, reflect_dom_object};
12use crate::dom::bindings::root::{Dom, DomRoot};
13use crate::dom::event::Event;
14use crate::dom::window::Window;
15use crate::dom::xrinputsource::XRInputSource;
16use crate::dom::xrinputsourceschangeevent::XRInputSourcesChangeEvent;
17use crate::dom::xrsession::XRSession;
18use crate::script_runtime::CanGc;
19
20#[dom_struct]
21pub(crate) struct XRInputSourceArray {
22    reflector_: Reflector,
23    input_sources: DomRefCell<Vec<Dom<XRInputSource>>>,
24}
25
26impl XRInputSourceArray {
27    fn new_inherited() -> XRInputSourceArray {
28        XRInputSourceArray {
29            reflector_: Reflector::new(),
30            input_sources: DomRefCell::new(vec![]),
31        }
32    }
33
34    pub(crate) fn new(window: &Window, can_gc: CanGc) -> DomRoot<XRInputSourceArray> {
35        reflect_dom_object(
36            Box::new(XRInputSourceArray::new_inherited()),
37            window,
38            can_gc,
39        )
40    }
41
42    pub(crate) fn add_input_sources(
43        &self,
44        session: &XRSession,
45        inputs: &[InputSource],
46        can_gc: CanGc,
47    ) {
48        let global = self.global();
49        let window = global.as_window();
50
51        let mut added = vec![];
52        for info in inputs {
53            // This is quadratic, but won't be a problem for the only case
54            // where we add multiple input sources (the initial input sources case)
55            debug_assert!(
56                !self
57                    .input_sources
58                    .borrow()
59                    .iter()
60                    .any(|i| i.id() == info.id),
61                "Should never add a duplicate input id!"
62            );
63            let input = XRInputSource::new(window, session, info.clone(), can_gc);
64            self.input_sources.borrow_mut().push(Dom::from_ref(&input));
65            added.push(input);
66        }
67
68        let event = XRInputSourcesChangeEvent::new(
69            window,
70            atom!("inputsourceschange"),
71            false,
72            true,
73            session,
74            &added,
75            &[],
76            can_gc,
77        );
78        event.upcast::<Event>().fire(session.upcast(), can_gc);
79    }
80
81    pub(crate) fn remove_input_source(&self, session: &XRSession, id: InputId, can_gc: CanGc) {
82        let global = self.global();
83        let window = global.as_window();
84        let removed = if let Some(i) = self.input_sources.borrow().iter().find(|i| i.id() == id) {
85            i.gamepad().update_connected(false, false, can_gc);
86            [DomRoot::from_ref(&**i)]
87        } else {
88            return;
89        };
90
91        let event = XRInputSourcesChangeEvent::new(
92            window,
93            atom!("inputsourceschange"),
94            false,
95            true,
96            session,
97            &[],
98            &removed,
99            can_gc,
100        );
101        self.input_sources.borrow_mut().retain(|i| i.id() != id);
102        event.upcast::<Event>().fire(session.upcast(), can_gc);
103    }
104
105    pub(crate) fn add_remove_input_source(
106        &self,
107        session: &XRSession,
108        id: InputId,
109        info: InputSource,
110        can_gc: CanGc,
111    ) {
112        let global = self.global();
113        let window = global.as_window();
114        let root;
115        let removed = if let Some(i) = self.input_sources.borrow().iter().find(|i| i.id() == id) {
116            i.gamepad().update_connected(false, false, can_gc);
117            root = [DomRoot::from_ref(&**i)];
118            &root as &[_]
119        } else {
120            warn!("Could not find removed input source with id {:?}", id);
121            &[]
122        };
123        self.input_sources.borrow_mut().retain(|i| i.id() != id);
124        let input = XRInputSource::new(window, session, info, can_gc);
125        self.input_sources.borrow_mut().push(Dom::from_ref(&input));
126
127        let added = [input];
128
129        let event = XRInputSourcesChangeEvent::new(
130            window,
131            atom!("inputsourceschange"),
132            false,
133            true,
134            session,
135            &added,
136            removed,
137            can_gc,
138        );
139        event.upcast::<Event>().fire(session.upcast(), can_gc);
140    }
141
142    pub(crate) fn find(&self, id: InputId) -> Option<DomRoot<XRInputSource>> {
143        self.input_sources
144            .borrow()
145            .iter()
146            .find(|x| x.id() == id)
147            .map(|x| DomRoot::from_ref(&**x))
148    }
149}
150
151impl XRInputSourceArrayMethods<crate::DomTypeHolder> for XRInputSourceArray {
152    /// <https://immersive-web.github.io/webxr/#dom-xrinputsourcearray-length>
153    fn Length(&self) -> u32 {
154        self.input_sources.borrow().len() as u32
155    }
156
157    /// <https://immersive-web.github.io/webxr/#xrinputsourcearray>
158    fn IndexedGetter(&self, n: u32) -> Option<DomRoot<XRInputSource>> {
159        self.input_sources
160            .borrow()
161            .get(n as usize)
162            .map(|x| DomRoot::from_ref(&**x))
163    }
164}