Custom setters for float type coercion

Closes #8

Author: VirxEC

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "rlbot_flatbuffers"
-version = "0.14.1"
+version = "0.14.2"
 edition = "2024"
 description = "A Python module implemented in Rust for serializing and deserializing RLBot's flatbuffers"
 repository = "https://github.com/VirxEC/rlbot_flatbuffers_py"

Cargo.toml

@@ -466,6 +466,33 @@ impl StructBindGenerator {
 
         write_str!(self, "        }");
         write_str!(self, "    }");
+
+        if self.is_frozen || self.is_no_set {
+            return;
+        }
+
+        for variable_info in &self.types {
+            let variable_name = variable_info.name.as_str();
+
+            match &variable_info.rust_type {
+                RustType::Base(inner_type) => match inner_type.as_str() {
+                    "f32" => {
+                        write_str!(self, "\n    #[setter]");
+                        write_fmt!(
+                            self,
+                            "    pub fn {variable_name}(&mut self, py: Python, value: f32) {{",
+                        );
+                        write_fmt!(
+                            self,
+                            "        self.{variable_name} = crate::float_to_py(py, value);"
+                        );
+                        write_str!(self, "    }");
+                    }
+                    _ => continue,
+                },
+                _ => continue,
+            }
+        }
     }
 
     fn generate_str_method(&mut self) {
@@ -919,10 +946,8 @@ impl Generator for StructBindGenerator {
                 "#[pyclass(module = \"rlbot_flatbuffers\", subclass, get_all, frozen)]"
             } else if self.types.is_empty() {
                 "#[pyclass(module = \"rlbot_flatbuffers\", subclass, frozen)]"
-            } else if self.is_no_set {
-                "#[pyclass(module = \"rlbot_flatbuffers\", subclass, get_all)]"
             } else {
-                "#[pyclass(module = \"rlbot_flatbuffers\", subclass, get_all, set_all)]"
+                "#[pyclass(module = \"rlbot_flatbuffers\", subclass, get_all)]"
             }
         );
 
@@ -933,11 +958,14 @@ impl Generator for StructBindGenerator {
             return;
         }
 
+        let gen_set = !(self.is_no_set || self.is_frozen);
+
         write_fmt!(self, "pub struct {} {{", self.struct_name);
 
         for variable_info in &self.types {
             let variable_name = variable_info.name.as_str();
 
+            let mut add_set = true;
             let variable_type = match &variable_info.rust_type {
                 RustType::Vec(InnerVecType::U8) => String::from("Py<PyBytes>"),
                 RustType::Vec(InnerVecType::String) => String::from("Vec<String>"),
@@ -952,7 +980,10 @@ impl Generator for StructBindGenerator {
                 }
                 RustType::Option(_, inner_type) => format!("Option<Py<super::{inner_type}>>"),
                 RustType::Base(inner_type) => match inner_type.as_str() {
-                    "f32" => String::from("Py<PyFloat>"),
+                    "f32" => {
+                        add_set = false;
+                        String::from("Py<PyFloat>")
+                    }
                     _ => inner_type.clone(),
                 },
                 RustType::String => String::from("Py<PyString>"),
@@ -962,6 +993,10 @@ impl Generator for StructBindGenerator {
                 RustType::Other(inner_type) => format!("super::{inner_type}"),
             };
 
+            if gen_set && add_set {
+                write_str!(self, "    #[pyo3(set)]");
+            }
+
             write_fmt!(self, "    pub {variable_name}: {variable_type},");
         }
 

codegen/structs.rs

@@ -1,6 +1,7 @@
 from random import randrange
 from time import time_ns
 
+import numpy as np
 from rlbot_flatbuffers import *
 
 
@@ -35,6 +36,11 @@ def random_script_config():
 
 
 if __name__ == "__main__":
+    controller = ControllerState()
+    controller.throttle = 1
+    controller.steer = 0.5
+    controller.pitch = np.array([0.1], dtype=np.float32)[0]
+
     vec1 = MyVector(1, 2, 3)
     vec2 = Vector3(4, 5, 6)
     print(vec1 + vec2)