struct literal defines by condition

mozilla · Aug 10, 2024 · 4d06178 · 4d06178
1 parent df1c558
commit 4d06178
Show file tree

Hide file tree

Showing 11 changed files with 117 additions and 1 deletion.
diff --git a/src/bindgen/language_backend/clike.rs b/src/bindgen/language_backend/clike.rs
@@ -927,6 +927,7 @@ impl LanguageBackend for CLikeLanguageBackend<'_> {
                 let ordered_fields = out.bindings().struct_field_names(path);
                 for (i, ordered_key) in ordered_fields.iter().enumerate() {
                     if let Some(lit) = fields.get(ordered_key) {
+                        let condition = lit.cfg.to_condition(self.config);
                         if is_constexpr {
                             out.new_line();
 
@@ -949,7 +950,13 @@ impl LanguageBackend for CLikeLanguageBackend<'_> {
                             } else {
                                 write!(out, ".{} = ", ordered_key);
                             }
-                            self.write_literal(out, &lit.value);
+                            if condition.is_some() {
+                                write!(out, "__{export_name}_{ordered_key}(");
+                                self.write_literal(out, &lit.value);
+                                write!(out, ")")
+                            } else {
+                                self.write_literal(out, &lit.value);
+                            }
                         }
                     }
                 }
@@ -960,6 +967,22 @@ impl LanguageBackend for CLikeLanguageBackend<'_> {
                     write!(out, " ");
                 }
                 write!(out, "}}");
+
+                if self.config.language == Language::C {
+                    for ordered_key in ordered_fields.iter() {
+                        if let Some(lit) = fields.get(ordered_key) {
+                            if let Some(condition) = lit.cfg.to_condition(self.config) {
+                                out.new_line();
+                                condition.write_before(self.config, out);
+                                let define = format!("#define __{export_name}_{ordered_key}(v)");
+                                write!(out, "{define} (v)");
+                                write!(out, "\n#else\n");
+                                write!(out, "{define}");
+                                condition.write_after(self.config, out);
+                            }
+                        }
+                    }
+                }
             }
         }
     }

diff --git a/tests/expectations/cfg.c b/tests/expectations/cfg.c
@@ -77,6 +77,18 @@ typedef struct {
 #endif
   ;
 } ConditionalField;
+#define ConditionalField_ZERO (ConditionalField){ .field = __ConditionalField_field(0) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
+#define ConditionalField_ONE (ConditionalField){ .field = __ConditionalField_field(1) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
 
 typedef struct {
   int32_t x;

diff --git a/tests/expectations/cfg.compat.c b/tests/expectations/cfg.compat.c
@@ -95,6 +95,18 @@ typedef struct {
 #endif
   ;
 } ConditionalField;
+#define ConditionalField_ZERO (ConditionalField){ .field = __ConditionalField_field(0) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
+#define ConditionalField_ONE (ConditionalField){ .field = __ConditionalField_field(1) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
 
 typedef struct {
   int32_t x;

diff --git a/tests/expectations/cfg.cpp b/tests/expectations/cfg.cpp
@@ -201,6 +201,12 @@ struct ConditionalField {
 #endif
   ;
 };
+constexpr static const ConditionalField ConditionalField_ZERO = ConditionalField{ 
+  /* .field = */ 0
+};
+constexpr static const ConditionalField ConditionalField_ONE = ConditionalField{ 
+  /* .field = */ 1
+};
 
 struct Normal {
   int32_t x;

diff --git a/tests/expectations/cfg.pyx b/tests/expectations/cfg.pyx
@@ -57,6 +57,8 @@ cdef extern from *:
 
   ctypedef struct ConditionalField:
     int32_t field;
+  const ConditionalField ConditionalField_ZERO # = <ConditionalField>{ 0 }
+  const ConditionalField ConditionalField_ONE # = <ConditionalField>{ 1 }
 
   ctypedef struct Normal:
     int32_t x;

diff --git a/tests/expectations/cfg_both.c b/tests/expectations/cfg_both.c
@@ -77,6 +77,18 @@ typedef struct ConditionalField {
 #endif
   ;
 } ConditionalField;
+#define ConditionalField_ZERO (ConditionalField){ .field = __ConditionalField_field(0) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
+#define ConditionalField_ONE (ConditionalField){ .field = __ConditionalField_field(1) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
 
 typedef struct Normal {
   int32_t x;

diff --git a/tests/expectations/cfg_both.compat.c b/tests/expectations/cfg_both.compat.c
@@ -95,6 +95,18 @@ typedef struct ConditionalField {
 #endif
   ;
 } ConditionalField;
+#define ConditionalField_ZERO (ConditionalField){ .field = __ConditionalField_field(0) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
+#define ConditionalField_ONE (ConditionalField){ .field = __ConditionalField_field(1) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
 
 typedef struct Normal {
   int32_t x;

diff --git a/tests/expectations/cfg_tag.c b/tests/expectations/cfg_tag.c
@@ -77,6 +77,18 @@ struct ConditionalField {
 #endif
   ;
 };
+#define ConditionalField_ZERO (ConditionalField){ .field = __ConditionalField_field(0) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
+#define ConditionalField_ONE (ConditionalField){ .field = __ConditionalField_field(1) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
 
 struct Normal {
   int32_t x;

diff --git a/tests/expectations/cfg_tag.compat.c b/tests/expectations/cfg_tag.compat.c
@@ -95,6 +95,18 @@ struct ConditionalField {
 #endif
   ;
 };
+#define ConditionalField_ZERO (ConditionalField){ .field = __ConditionalField_field(0) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
+#define ConditionalField_ONE (ConditionalField){ .field = __ConditionalField_field(1) }
+#if defined(X11)
+#define __ConditionalField_field(v) (v)
+#else
+#define __ConditionalField_field(v)
+#endif
 
 struct Normal {
   int32_t x;

diff --git a/tests/expectations/cfg_tag.pyx b/tests/expectations/cfg_tag.pyx
@@ -57,6 +57,8 @@ cdef extern from *:
 
   cdef struct ConditionalField:
     int32_t field;
+  const ConditionalField ConditionalField_ZERO # = <ConditionalField>{ 0 }
+  const ConditionalField ConditionalField_ONE # = <ConditionalField>{ 1 }
 
   cdef struct Normal:
     int32_t x;

diff --git a/tests/rust/cfg.rs b/tests/rust/cfg.rs
@@ -49,6 +49,17 @@ struct ConditionalField {
     field: i32,
 }
 
+impl ConditionalField {
+    pub const ZERO: Self = Self {
+        #[cfg(x11)]
+        field: 0,
+    };
+    pub const ONE: Self = Self {
+        #[cfg(x11)]
+        field: 1,
+    };
+}
+
 #[cfg(all(unix, x11))]
 #[no_mangle]
 pub extern "C" fn root(a: FooHandle, c: C)