Add manual default overrides

Closes #2

Author: VirxEC

Cargo.lock

@@ -60,6 +60,9 @@ impl PythonBindType {
         "RenderType",
     ];
 
+    pub const DEFAULT_OVERRIDES: [(&'static str, &'static str, &'static str); 1] =
+        [("Color", "a", "255")];
+
     fn new(path: &Path) -> Option<Self> {
         // get the filename without the extension
         let filename = path.file_stem().unwrap().to_str().unwrap();

codegen/main.rs

@@ -274,6 +274,16 @@ pub fn generator(type_data: &[PythonBindType]) -> io::Result<()> {
                         for (variable_info, python_type) in bind.types.iter().zip(&python_types) {
                             let variable_name = variable_info.name.as_str();
 
+                            if let Some((field, value)) = bind.default_override {
+                                if field == variable_name {
+                                    write_fmt!(
+                                        file,
+                                        "        {variable_name}: {python_type} = {value},"
+                                    );
+                                    continue;
+                                }
+                            }
+
                             let default_value = match variable_info.raw_type.as_str() {
                                 "bool" => Cow::Borrowed("False"),
                                 "i32" | "u32" | "f32" | "u8" => Cow::Borrowed("0"),

codegen/pyi.rs

@@ -54,6 +54,7 @@ pub struct StructBindGenerator {
     has_complex_pack: bool,
     pub is_frozen: bool,
     is_no_set: bool,
+    pub default_override: Option<(&'static str, &'static str)>,
 }
 
 macro_rules! write_str {
@@ -103,6 +104,17 @@ impl StructBindGenerator {
         file_contents.push(Cow::Borrowed("use pyo3::{prelude::*, types::*};"));
         file_contents.push(Cow::Borrowed(""));
 
+        let default_override =
+            PythonBindType::DEFAULT_OVERRIDES
+                .iter()
+                .find_map(|&(name, field, value)| {
+                    if name == struct_name.as_str() {
+                        Some((field, value))
+                    } else {
+                        None
+                    }
+                });
+
         Some(Self {
             filename,
             struct_name,
@@ -113,6 +125,7 @@ impl StructBindGenerator {
             has_complex_pack,
             is_frozen,
             is_no_set,
+            default_override,
         })
     }
 
@@ -297,6 +310,13 @@ impl StructBindGenerator {
         for variable_info in &self.types {
             let variable_name = variable_info.name.as_str();
 
+            if let Some((field, value)) = self.default_override {
+                if field == variable_name {
+                    signature_parts.push(format!("{variable_name}={value}"));
+                    continue;
+                }
+            }
+
             let sig_part = match &variable_info.rust_type {
                 RustType::Option(_, _) => {
                     if variable_info.is_special_base.is_some() {
@@ -955,6 +975,13 @@ impl Generator for StructBindGenerator {
         for variable_info in &self.types {
             let variable_name = variable_info.name.as_str();
 
+            if let Some((field, value)) = self.default_override {
+                if field == variable_name {
+                    write_fmt!(self, "            {variable_name}: {value},");
+                    continue;
+                }
+            }
+
             let end = match &variable_info.rust_type {
                 RustType::Vec(InnerVecType::U8) => Cow::Borrowed("PyBytes::new(py, &[]).unbind()"),
                 RustType::Vec(InnerVecType::Custom(_)) => {

codegen/structs.rs

@@ -82,8 +82,12 @@ def random_script_config():
             Color(255),
         )
     )
+
     if isinstance(render_type.variety.item, Line3D):
+        assert render_type.variety.item.color.r == 255
+        assert render_type.variety.item.color.a == 255
         render_type.variety.item.color.a = 150
+        assert render_type.variety.item.color.a == 150
     else:
         raise ValueError("Expected Line3D")