Use rust enums as function parameters (#385)

Author: trobanga

rust-package/brainflow/build.rs

@@ -111,6 +111,7 @@ fn generate_constants_binding() {
 
     let bindings = bindings.replace("const First", "const FIRST");
     let bindings = bindings.replace("const Last", "const LAST");
+    let bindings = bindings.replace("#[derive(", "#[derive(FromPrimitive, ToPrimitive, ");
 
     let binding_target_path = PathBuf::new().join("src").join("ffi").join("constants.rs");
     let mut file = File::create(&binding_target_path).unwrap();

rust-package/brainflow/examples/band_power.rs

@@ -8,13 +8,12 @@ use ndarray::s;
 
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
-    let board_id = BoardIds::SyntheticBoard as i32;
+    let board_id = BoardIds::SyntheticBoard;
     let eeg_channels = board_shim::get_eeg_channels(board_id).unwrap();
     let sampling_rate = board_shim::get_sampling_rate(board_id).unwrap();
     let nfft = data_filter::get_nearest_power_of_two(sampling_rate).unwrap();
 
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
     let board = board_shim::BoardShim::new(board_id, params).unwrap();
 
     board.prepare_session().unwrap();
@@ -28,7 +27,7 @@ fn main() {
         data.slice_mut(s![eeg_channels[1], ..])
             .as_slice_mut()
             .unwrap(),
-        DetrendOperations::Linear as i32,
+        DetrendOperations::Linear,
     )
     .unwrap();
     let mut psd = data_filter::get_psd_welch(
@@ -38,7 +37,7 @@ fn main() {
         nfft,
         nfft / 2,
         sampling_rate,
-        WindowFunctions::BlackmanHarris as i32,
+        WindowFunctions::BlackmanHarris,
     )
     .unwrap();
     let band_power_alpha = data_filter::get_band_power(&mut psd, 7.0, 13.0).unwrap();

rust-package/brainflow/examples/denoising.rs

@@ -8,11 +8,10 @@ use ndarray::s;
 
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
-    let board_id = BoardIds::SyntheticBoard as i32;
+    let board_id = BoardIds::SyntheticBoard;
     let eeg_channels = board_shim::get_eeg_channels(board_id).unwrap();
 
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
     let board = board_shim::BoardShim::new(board_id, params).unwrap();
 
     board.prepare_session().unwrap();
@@ -28,7 +27,7 @@ fn main() {
             .as_slice_mut()
             .unwrap(),
         3,
-        AggOperations::Mean as i32,
+        AggOperations::Mean,
     )
     .unwrap();
     println!("{:?}", data.slice(s![eeg_channels[0], ..]));

rust-package/brainflow/examples/downsampling.rs

@@ -8,11 +8,10 @@ use ndarray::s;
 
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
-    let board_id = BoardIds::SyntheticBoard as i32;
+    let board_id = BoardIds::SyntheticBoard;
     let eeg_channels = board_shim::get_eeg_channels(board_id).unwrap();
 
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
     let board = board_shim::BoardShim::new(board_id, params).unwrap();
 
     board.prepare_session().unwrap();
@@ -28,7 +27,7 @@ fn main() {
             .as_slice_mut()
             .unwrap(),
         3,
-        AggOperations::Each as i32,
+        AggOperations::Each,
     )
     .unwrap();
     println!("{:?}", &downsampled);

rust-package/brainflow/examples/eeg_metrics.rs

@@ -9,7 +9,7 @@ use brainflow::{
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
+    let board_id = BoardIds::SyntheticBoard;
     let board = board_shim::BoardShim::new(board_id, params).unwrap();
 
     board.prepare_session().unwrap();

rust-package/brainflow/examples/get_data.rs

@@ -5,8 +5,7 @@ use brainflow::{board_shim, brainflow_input_params::BrainFlowInputParamsBuilder,
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
-    let board = board_shim::BoardShim::new(board_id, params).unwrap();
+    let board = board_shim::BoardShim::new(BoardIds::SyntheticBoard, params).unwrap();
 
     board.prepare_session().unwrap();
     board.start_stream(45000, "").unwrap();

rust-package/brainflow/examples/markers.rs

@@ -8,7 +8,7 @@ use brainflow::BoardIds;
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board = BoardShim::new(BoardIds::SyntheticBoard as i32, params).unwrap();
+    let board = BoardShim::new(BoardIds::SyntheticBoard, params).unwrap();
 
     board.prepare_session().unwrap();
     board.start_stream(45000, "file://data.csv:w").unwrap();

rust-package/brainflow/examples/read_write_file.rs

@@ -7,8 +7,7 @@ use brainflow::{
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
-    let board = board_shim::BoardShim::new(board_id, params).unwrap();
+    let board = board_shim::BoardShim::new(BoardIds::SyntheticBoard, params).unwrap();
 
     board.prepare_session().unwrap();
     board.start_stream(45000, "").unwrap();

rust-package/brainflow/examples/signal_filtering.rs

@@ -8,12 +8,11 @@ use ndarray::s;
 
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
-    let board_id = BoardIds::SyntheticBoard as i32;
+    let board_id = BoardIds::SyntheticBoard;
     let sampling_rate = board_shim::get_sampling_rate(board_id).unwrap();
     let eeg_channels = board_shim::get_eeg_channels(board_id).unwrap();
 
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
     let board = board_shim::BoardShim::new(board_id, params).unwrap();
 
     board.prepare_session().unwrap();
@@ -33,7 +32,7 @@ fn main() {
         sampling_rate,
         13.0,
         3,
-        FilterTypes::Bessel as i32,
+        FilterTypes::Bessel,
         0.0,
     )
     .unwrap();

rust-package/brainflow/examples/transforms.rs

@@ -9,12 +9,11 @@ use ndarray::s;
 fn main() {
     brainflow::board_shim::enable_dev_board_logger().unwrap();
 
-    let board_id = BoardIds::SyntheticBoard as i32;
+    let board_id = BoardIds::SyntheticBoard;
     let eeg_channels = board_shim::get_eeg_channels(board_id).unwrap();
     println!("{:?}", eeg_channels);
 
     let params = BrainFlowInputParamsBuilder::default().build();
-    let board_id = BoardIds::SyntheticBoard as i32;
     let board = board_shim::BoardShim::new(board_id, params).unwrap();
 
     board.prepare_session().unwrap();
@@ -30,7 +29,7 @@ fn main() {
         data.slice_mut(s![eeg_channels[0], ..])
             .as_slice_mut()
             .unwrap(),
-        WindowFunctions::BlackmanHarris as i32,
+        WindowFunctions::BlackmanHarris,
     )
     .unwrap();
     let restored_fft = data_filter::perform_ifft(&fft_data, data_len).unwrap();