papi_bindings/
events_set.rs

1use crate::counter::Counter;
2use crate::PAPI_destroy_eventset;
3use crate::{check_error, PAPI_create_eventset};
4use crate::{PAPI_accum, PAPI_read, PAPI_start, PAPI_stop, PAPI_NULL};
5use crate::{PAPI_add_event, PapiError};
6use std::os::raw::{c_int, c_longlong};
7use std::sync::Arc;
8
9pub struct EventsSet {
10    counter_defs: Arc<Vec<Counter>>,
11    counter_values: Vec<c_longlong>,
12    evt_set_id: c_int,
13}
14
15impl EventsSet {
16    fn create_event(counters: &[Counter]) -> Result<c_int, PapiError> {
17        let mut evt_set_id = PAPI_NULL;
18
19        unsafe { check_error(PAPI_create_eventset(&mut evt_set_id as *mut c_int))? };
20
21        for evt in counters {
22            unsafe {
23                check_error(PAPI_add_event(evt_set_id, evt.code))?;
24            }
25        }
26
27        Ok(evt_set_id)
28    }
29
30    pub fn new(counters: &[Counter]) -> Result<Self, PapiError> {
31        Ok(Self {
32            counter_defs: Arc::new(counters.to_vec()),
33            counter_values: vec![0; counters.len()],
34            evt_set_id: Self::create_event(counters)?,
35        })
36    }
37
38    pub fn len(&self) -> usize {
39        self.counter_defs.len()
40    }
41
42    pub fn try_clone(&self) -> Result<Self, PapiError> {
43        Ok(Self {
44            counter_defs: self.counter_defs.clone(),
45            counter_values: vec![0; self.counter_defs.len()],
46            evt_set_id: Self::create_event(&self.counter_defs)?,
47        })
48    }
49
50    pub fn start(&mut self) -> Result<(), PapiError> {
51        unsafe { check_error(PAPI_start(self.evt_set_id)) }
52    }
53
54    pub fn read(&mut self) -> Result<&[c_longlong], PapiError> {
55        unsafe {
56            check_error(PAPI_read(self.evt_set_id, self.counter_values.as_mut_ptr()))?;
57        }
58        Ok(&self.counter_values)
59    }
60
61    pub fn read_into<'a>(
62        &self,
63        counters: &'a mut [c_longlong],
64    ) -> Result<&'a [c_longlong], PapiError> {
65        assert_eq!(counters.len(), self.counter_defs.len());
66        unsafe {
67            check_error(PAPI_read(self.evt_set_id, counters.as_mut_ptr()))?;
68        }
69        Ok(counters)
70    }
71
72    pub fn accum(&mut self) -> Result<&[c_longlong], PapiError> {
73        unsafe {
74            check_error(PAPI_accum(
75                self.evt_set_id,
76                self.counter_values.as_mut_ptr(),
77            ))?;
78        }
79        Ok(&self.counter_values)
80    }
81
82    pub fn stop(&mut self) -> Result<&[c_longlong], PapiError> {
83        unsafe {
84            check_error(PAPI_stop(self.evt_set_id, self.counter_values.as_mut_ptr()))?;
85        }
86        Ok(&self.counter_values)
87    }
88}
89
90impl Drop for EventsSet {
91    fn drop(&mut self) {
92        let _ = self.stop();
93        unsafe {
94            PAPI_destroy_eventset(&mut self.evt_set_id as *mut c_int);
95        }
96    }
97}
papi_bindings/events_set.rs

papi_bindings/
events_set.rs
