Skip to content

Commit

Permalink
Remove code duplication for function writing, and other minor clean-ups.
Browse files Browse the repository at this point in the history
  • Loading branch information
@emilio
emilio committed Apr 14, 2024
1 parent 3354606 commit 9fff78c
Show file tree
Hide file tree
Showing 5 changed files with 105 additions and 293 deletions.
2 changes: 1 addition & 1 deletion src/bindgen/cdecl.rs
View file
Expand Up @@ -184,7 +184,7 @@ impl CDecl {
});
self.declarators.push(CDeclarator::Func {
args,
layout: config.function.args.clone(),
layout: config.function.args,
never_return: *never_return,
});
self.build_type(ret, false, config);
Expand Down
2 changes: 1 addition & 1 deletion src/bindgen/config.rs
View file
Expand Up @@ -134,7 +134,7 @@ impl FromStr for Braces {
deserialize_enum_str!(Braces);

/// A type of layout to use when generating long lines of code.
#[derive(Debug, Clone, PartialEq, Eq)]
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
pub enum Layout {
Horizontal,
Vertical,
Expand Down
138 changes: 5 additions & 133 deletions src/bindgen/language_backend/clike.rs
View file
@@ -1,12 +1,12 @@
use crate::bindgen::ir::{
to_known_assoc_constant, ConditionWrite, DeprecatedNoteKind, Documentation, Enum, EnumVariant,
Field, Function, GenericParams, Item, Literal, OpaqueItem, ReprAlign, Static, Struct,
ToCondition, Type, Typedef, Union,
Field, GenericParams, Item, Literal, OpaqueItem, ReprAlign, Static, Struct, ToCondition, Type,
Typedef, Union,
};
use crate::bindgen::language_backend::LanguageBackend;
use crate::bindgen::rename::IdentifierType;
use crate::bindgen::writer::{ListType, SourceWriter};
use crate::bindgen::{cdecl, Bindings, Config, Language, Layout};
use crate::bindgen::{cdecl, Bindings, Config, Language};
use crate::bindgen::{DocumentationLength, DocumentationStyle};
use std::io::Write;

Expand Down Expand Up @@ -780,6 +780,7 @@ impl LanguageBackend for CLikeLanguageBackend<'_> {
}

fn write_static<W: Write>(&mut self, out: &mut SourceWriter<W>, s: &Static) {
self.write_documentation(out, &s.documentation);
out.write("extern ");
if let Type::Ptr { is_const: true, .. } = s.ty {
} else if !s.mutable {
Expand All @@ -789,135 +790,6 @@ impl LanguageBackend for CLikeLanguageBackend<'_> {
out.write(";");
}

fn write_function<W: Write>(&mut self, out: &mut SourceWriter<W>, f: &Function) {
fn write_1<W: Write>(
func: &Function,
language_backend: &mut CLikeLanguageBackend,
out: &mut SourceWriter<W>,
) {
let prefix = language_backend.config.function.prefix(&func.annotations);
let postfix = language_backend.config.function.postfix(&func.annotations);

let condition = func.cfg.to_condition(language_backend.config);
condition.write_before(language_backend.config, out);

language_backend.write_documentation(out, &func.documentation);

if func.extern_decl {
out.write("extern ");
} else {
if let Some(ref prefix) = prefix {
write!(out, "{} ", prefix);
}
if func.annotations.must_use(language_backend.config) {
if let Some(ref anno) = language_backend.config.function.must_use {
write!(out, "{} ", anno);
}
}
if let Some(note) = func
.annotations
.deprecated_note(language_backend.config, DeprecatedNoteKind::Function)
{
write!(out, "{} ", note);
}
}
cdecl::write_func(
language_backend,
out,
func,
Layout::Horizontal,
language_backend.config,
);

if !func.extern_decl {
if let Some(ref postfix) = postfix {
write!(out, " {}", postfix);
}
}

if let Some(ref swift_name_macro) = language_backend.config.function.swift_name_macro {
if let Some(swift_name) = func.swift_name(language_backend.config) {
write!(out, " {}({})", swift_name_macro, swift_name);
}
}

out.write(";");

condition.write_after(language_backend.config, out);
}

fn write_2<W: Write>(
func: &Function,
language_backend: &mut CLikeLanguageBackend,
out: &mut SourceWriter<W>,
) {
let prefix = language_backend.config.function.prefix(&func.annotations);
let postfix = language_backend.config.function.postfix(&func.annotations);

let condition = func.cfg.to_condition(language_backend.config);

condition.write_before(language_backend.config, out);

language_backend.write_documentation(out, &func.documentation);

if func.extern_decl {
out.write("extern ");
} else {
if let Some(ref prefix) = prefix {
write!(out, "{}", prefix);
out.new_line();
}
if func.annotations.must_use(language_backend.config) {
if let Some(ref anno) = language_backend.config.function.must_use {
write!(out, "{}", anno);
out.new_line();
}
}
if let Some(note) = func
.annotations
.deprecated_note(language_backend.config, DeprecatedNoteKind::Function)
{
write!(out, "{}", note);
out.new_line();
}
}
cdecl::write_func(
language_backend,
out,
func,
Layout::Vertical,
language_backend.config,
);
if !func.extern_decl {
if let Some(ref postfix) = postfix {
out.new_line();
write!(out, "{}", postfix);
}
}

if let Some(ref swift_name_macro) = language_backend.config.function.swift_name_macro {
if let Some(swift_name) = func.swift_name(language_backend.config) {
write!(out, " {}({})", swift_name_macro, swift_name);
}
}

out.write(";");

condition.write_after(language_backend.config, out);
}

match self.config.function.args {
Layout::Horizontal => write_1(f, self, out),
Layout::Vertical => write_2(f, self, out),
Layout::Auto => {
let max_line_length = self.config.line_length;
if !out.try_write(|out| write_1(f, self, out), max_line_length) {
write_2(f, self, out)
}
}
}
}

fn write_type<W: Write>(&mut self, out: &mut SourceWriter<W>, t: &Type) {
cdecl::write_type(self, out, t, self.config);
}
Expand Down Expand Up @@ -1116,7 +988,7 @@ impl LanguageBackend for CLikeLanguageBackend<'_> {
}

fn write_functions<W: Write>(&mut self, out: &mut SourceWriter<W>, b: &Bindings) {
// Override default method to close various blocs containing both globals and functions
// Override default method to close various blocks containing both globals and functions
// these blocks are opened in [`write_globals`] that is also overridden
if !b.functions.is_empty() || !b.globals.is_empty() {
self.write_functions_default(out, b);
Expand Down
133 changes: 2 additions & 131 deletions src/bindgen/language_backend/cython.rs
View file
@@ -1,12 +1,10 @@
use crate::bindgen::ir::{
to_known_assoc_constant, ConditionWrite, DeprecatedNoteKind, Documentation, Enum, EnumVariant,
Field, Function, Item, Literal, OpaqueItem, ReprAlign, Static, Struct, ToCondition, Type,
Typedef, Union,
Field, Item, Literal, OpaqueItem, ReprAlign, Static, Struct, ToCondition, Type, Typedef, Union,
};
use crate::bindgen::language_backend::LanguageBackend;
use crate::bindgen::writer::{ListType, SourceWriter};
use crate::bindgen::DocumentationLength;
use crate::bindgen::Layout;
use crate::bindgen::{cdecl, Bindings, Config};
use std::io::Write;

Expand Down Expand Up @@ -323,6 +321,7 @@ impl LanguageBackend for CythonLanguageBackend<'_> {
}

fn write_static<W: Write>(&mut self, out: &mut SourceWriter<W>, s: &Static) {
self.write_documentation(out, &s.documentation);
out.write("extern ");
if let Type::Ptr { is_const: true, .. } = s.ty {
} else if !s.mutable {
Expand All @@ -332,134 +331,6 @@ impl LanguageBackend for CythonLanguageBackend<'_> {
out.write(";");
}

fn write_function<W: Write>(&mut self, out: &mut SourceWriter<W>, f: &Function) {
fn write_1<W: Write>(
func: &Function,
language_backend: &mut CythonLanguageBackend,
out: &mut SourceWriter<W>,
) {
let prefix = language_backend.config.function.prefix(&func.annotations);
let postfix = language_backend.config.function.postfix(&func.annotations);

let condition = func.cfg.to_condition(language_backend.config);
condition.write_before(language_backend.config, out);

language_backend.write_documentation(out, &func.documentation);

if func.extern_decl {
out.write("extern ");
} else {
if let Some(ref prefix) = prefix {
write!(out, "{} ", prefix);
}
if func.annotations.must_use(language_backend.config) {
if let Some(ref anno) = language_backend.config.function.must_use {
write!(out, "{} ", anno);
}
}
if let Some(note) = func
.annotations
.deprecated_note(language_backend.config, DeprecatedNoteKind::Function)
{
write!(out, "{} ", note);
}
}
cdecl::write_func(
language_backend,
out,
func,
Layout::Horizontal,
language_backend.config,
);

if !func.extern_decl {
if let Some(ref postfix) = postfix {
write!(out, " {}", postfix);
}
}

if let Some(ref swift_name_macro) = language_backend.config.function.swift_name_macro {
if let Some(swift_name) = func.swift_name(language_backend.config) {
write!(out, " {}({})", swift_name_macro, swift_name);
}
}

out.write(";");

condition.write_after(language_backend.config, out);
}

fn write_2<W: Write>(
func: &Function,
language_backend: &mut CythonLanguageBackend,
out: &mut SourceWriter<W>,
) {
let prefix = language_backend.config.function.prefix(&func.annotations);
let postfix = language_backend.config.function.postfix(&func.annotations);

let condition = func.cfg.to_condition(language_backend.config);

condition.write_before(language_backend.config, out);

language_backend.write_documentation(out, &func.documentation);

if func.extern_decl {
out.write("extern ");
} else {
if let Some(ref prefix) = prefix {
write!(out, "{}", prefix);
out.new_line();
}
if func.annotations.must_use(language_backend.config) {
if let Some(ref anno) = language_backend.config.function.must_use {
write!(out, "{}", anno);
out.new_line();
}
}
if let Some(note) = func
.annotations
.deprecated_note(language_backend.config, DeprecatedNoteKind::Function)
{
write!(out, "{} ", note);
}
}
cdecl::write_func(
language_backend,
out,
func,
Layout::Vertical,
language_backend.config,
);
if !func.extern_decl {
if let Some(ref postfix) = postfix {
out.new_line();
write!(out, "{}", postfix);
}
}

if let Some(ref swift_name_macro) = language_backend.config.function.swift_name_macro {
if let Some(swift_name) = func.swift_name(language_backend.config) {
write!(out, " {}({})", swift_name_macro, swift_name);
}
}

out.write(";");

condition.write_after(language_backend.config, out);
}

match &self.config.function.args {
Layout::Horizontal => write_1(f, self, out),
Layout::Vertical => write_2(f, self, out),
Layout::Auto => {
let max_line_length = self.config.line_length;
if !out.try_write(|out| write_1(f, self, out), max_line_length) {
write_2(f, self, out)
}
}
}
}

fn write_type<W: Write>(&mut self, out: &mut SourceWriter<W>, t: &Type) {
cdecl::write_type(self, out, t, self.config);
}
Expand Down

0 comments on commit 9fff78c

Please sign in to comment.