1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use std::collections::HashSet;

use maplit::hashmap;
use once_cell::sync::Lazy;

use crate::{
	CompiledInterpolation,
	Element,
	Enum,
	get_debug,
	StrExt,
};

use super::RustNativeGeneratedElement;

impl RustNativeGeneratedElement for Enum<'_> {
	fn element_safe_id(&self) -> String {
		format!("{}-{}", self.rust_module(), self.rust_localname())
	}

	fn gen_rust(&self, opencv_version: &str) -> String {
		static ENUM_TPL: Lazy<CompiledInterpolation> = Lazy::new(
			|| include_str!("tpl/enum/enum.tpl.rs").compile_interpolation()
		);

		static CONST_TPL: Lazy<CompiledInterpolation> = Lazy::new(
			|| include_str!("tpl/enum/const.tpl.rs").compile_interpolation()
		);

		static CONST_IGNORED_TPL: Lazy<CompiledInterpolation> = Lazy::new(
			|| include_str!("tpl/enum/const_ignored.tpl.rs").compile_interpolation()
		);

		let consts = self.consts();
		let mut generated_values = HashSet::with_capacity(consts.len());
		let consts = consts.into_iter()
			.map(|c| {
				let name = c.rust_leafname();
				let value = c.value().expect("Can't get value of enum variant").to_string();
				let is_ignored = generated_values.contains(&value);
				let tpl = if is_ignored {
					&CONST_IGNORED_TPL
				} else {
					&CONST_TPL
				};
				let doc_comment = if is_ignored {
					c.rendered_doc_comment_with_prefix("//", opencv_version)
				} else {
					c.rendered_doc_comment(opencv_version)
				};
				generated_values.insert(value.clone());
				tpl.interpolate(&hashmap! {
					"doc_comment" => doc_comment,
					"name" => name.into_owned(),
					"value" => value,
				})
			}).collect::<Vec<_>>();
		ENUM_TPL.interpolate(&hashmap! {
			"doc_comment" => self.rendered_doc_comment(opencv_version).into(),
			"debug" => get_debug(self).into(),
			"rust_local" => self.rust_localname(),
			"rust_full" => self.rust_fullname(),
			"consts" => consts.join("").into(),
		})
	}
}