diff options
-rw-r--r-- | src/lib.rs | 5 | ||||
-rw-r--r-- | src/repeat.rs | 30 |
2 files changed, 35 insertions, 0 deletions
@@ -15,6 +15,7 @@ mod str; mod token_stream; const OPENGL_CMD_NAME_REPLACE: &str = "gl_command_name"; +const OPENGL_CMD_NAME_NOPREFIX_REPLACE: &str = "gl_command_name_noprefix"; const OPENGL_CMD_RET_TYPE_REPLACE: &str = "gl_command_ret_type"; const OPENGL_CMD_ARGS_REPLACE: &str = "gl_command_args"; const OPENGL_CMD_ARG_NAMES_REPLACE: &str = "gl_command_arg_names"; @@ -32,6 +33,10 @@ static OPENGL_REGISTRY: Lazy<Registry> = Lazy::new(|| Registry::retrieve().unwra /// **`gl_command_name`**<br> /// Will be replaced by the full command name. For example, `glViewport`. /// +/// **`gl_command_name_noprefix`**<br> +/// Will be replaced by the command name without the "gl" prefix. For example, +/// `Viewport`. +/// /// **`gl_command_ret_type`**<br> /// Will be replaced by the command return type. For example, `GLuint`. /// diff --git a/src/repeat.rs b/src/repeat.rs index 668fefb..594d23e 100644 --- a/src/repeat.rs +++ b/src/repeat.rs @@ -10,6 +10,7 @@ use crate::{ OPENGL_CMD_ARGS_REPLACE, OPENGL_CMD_ARG_NAMES_REPLACE, OPENGL_CMD_ARG_TYPES_REPLACE, + OPENGL_CMD_NAME_NOPREFIX_REPLACE, OPENGL_CMD_NAME_REPLACE, OPENGL_CMD_RET_TYPE_REPLACE, }; @@ -31,6 +32,19 @@ pub fn for_each_opengl_command_impl( )), ); + let cmd_prototype = gl_command.prototype(); + + let stream = stream.replace_ident( + &Ident::new(OPENGL_CMD_NAME_NOPREFIX_REPLACE, Span::call_site()), + &TokenTree::Ident(Ident::new( + cmd_prototype + .name() + .strip_prefix("gl") + .unwrap_or_else(|| cmd_prototype.name()), + Span::call_site(), + )), + ); + let return_type = gl_command.prototype().return_type(); let stream = stream.replace_ident( @@ -197,5 +211,21 @@ mod tests } .to_string(), ); + + assert_eq!( + for_each_opengl_command_impl( + "e! { + fn #ht gl_command_name_noprefix(#ht gl_command_args) -> #ht gl_command_ret_type {} + }, + ®istry, + ) + .to_string(), + quote! { + fn BindBuffer() -> std::ffi::c_void {} + fn ActiveTexture(abc: GLuint, xyz: GLshort) -> std::ffi::c_void {} + fn DrawArrays(foo: GLubyte) -> GLint {} + } + .to_string(), + ); } } |