From a61ab5a9781136c021d876c8dbc242958cfcff01 Mon Sep 17 00:00:00 2001
From: Magic Len <len@magiclen.org>
Date: Tue, 6 Nov 2018 19:06:13 +0800
Subject: [PATCH] optimize the allocation of the result vec

---
 rust/src/qr_segment_advanced.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/rust/src/qr_segment_advanced.rs b/rust/src/qr_segment_advanced.rs
index 1e4444f..53bc640 100644
--- a/rust/src/qr_segment_advanced.rs
+++ b/rust/src/qr_segment_advanced.rs
@@ -125,7 +125,8 @@ fn compute_character_modes(code_points: &[char], version: Version) -> Vec<QrSegm
 
     let mut cur_mode = cur_mode.unwrap();
 
-    let mut result = vec![QrSegmentMode::Byte; char_modes.len()];
+    let mut result = Vec::with_capacity(char_modes.len());
+    unsafe { result.set_len(char_modes.len()); }
 
     // Get optimal mode for each code point by tracing backwards
     for i in (0..char_modes.len()).rev() {
@@ -240,4 +241,4 @@ fn is_kanji(_: char) -> bool {
 
 #[cfg(feature = "kanji")]
 // Load the unpacked the computation-friendly Shift JIS table
-static UNICODE_TO_QR_KANJI: [i16; 1 << 16] = include!("unicode_to_qr_kanji.json");
\ No newline at end of file
+static UNICODE_TO_QR_KANJI: [i16; 1 << 16] = include!("unicode_to_qr_kanji.json");