Minor adjustments

Author: lucidLuckylee

src/lib.rs

@@ -307,7 +307,7 @@ pub fn define_pushable(_: TokenStream) -> TokenStream {
                 fn bitcoin_script_push(self, builder: Builder) -> Builder {
                     let (optimal, replacement_opcode) =
                         check_optimality(builder.last_opcode(), self.first_opcode());
-                    let mut script_vec = vec![];
+                    let mut script_vec = Vec::with_capacity(builder.0.as_bytes().len() + self.as_bytes().len());
                     if optimal {
                         script_vec.extend_from_slice(builder.as_bytes());
                         script_vec.extend_from_slice(self.as_bytes());

src/parse.rs

@@ -42,7 +42,7 @@ macro_rules! abort {
 
 pub fn parse(tokens: TokenStream) -> Vec<(Syntax, Span)> {
     let mut tokens = tokens.into_iter().peekable();
-    let mut syntax = Vec::with_capacity(64);
+    let mut syntax = Vec::with_capacity(2048);
 
     while let Some(token) = tokens.next() {
         let token_str = token.to_string();
@@ -63,9 +63,10 @@ pub fn parse(tokens: TokenStream) -> Vec<(Syntax, Span)> {
                     // Not a native Bitcoin opcode
                     // Allow functions without arguments to be identified by just their name
                     _ => {
-                        let mut pseudo_stream = TokenStream::from(token.clone());
+                        let span = token.span();
+                        let mut pseudo_stream = TokenStream::from(token);
                         pseudo_stream.extend(TokenStream::from_str("()"));
-                        (Syntax::Escape(pseudo_stream), token.span())
+                        (Syntax::Escape(pseudo_stream), span)
                     }
                 }
             }
@@ -100,7 +101,7 @@ where
 {
     // Use a Vec here to get rid of warnings when the variable is overwritten
     let mut escape = quote! {
-        let mut script_var = vec![];
+        let mut script_var = Vec::with_capacity(256);
     };
     escape.extend(std::iter::once(token.clone()));
 

tests/test.rs

@@ -144,11 +144,12 @@ fn test_performance_loop() {
         OP_ADD
         OP_ADD
         OP_ADD
+        script_from_func
     };
 
     let script = script! {
         for _ in 0..5_000_000 {
-            { loop_script.clone() }
+            {loop_script.clone()}
         }
     };