Enables background image size to be modified at run time

Author: wcshds

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "text-image-generator"
-version = "0.2.5"
+version = "0.2.6"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

src/lib.rs

@@ -211,6 +211,10 @@ impl Generator {
         })
     }
 
+    fn set_bg_size(&mut self, height: usize, width: usize) {
+        self.bg_factory = BgFactory::new(&self.bg_factory.bg_dir, height, width);
+    }
+
     // fn set_latin_ch_dict(&mut self, ch: String, font_list: Vec<String>) {
     //     if let Some(content) = &mut self.latin_ch_dict {
     //         *content.entry(ch).or_insert(vec![]) = font_list;

src/merge_util.rs

@@ -14,8 +14,9 @@ use super::effect_helper::{
 #[pyclass]
 pub struct BgFactory {
     images: Vec<GrayImage>,
-    height: usize,
-    width: usize,
+    pub height: usize,
+    pub width: usize,
+    pub bg_dir: String,
 }
 
 impl BgFactory {
@@ -86,6 +87,7 @@ impl BgFactory {
             images,
             height,
             width,
+            bg_dir: dir.as_ref().to_string_lossy().to_string(),
         }
     }
 

text_image_generator.pyi

@@ -118,6 +118,16 @@ class Generator:
     main_font_list: str
 
     def __init__(self, config_path: str) -> None: ...
+    def set_bg_size(
+        self,
+        height: int,
+        width: int,
+    ):
+        """Set the background image's height and width.
+
+        :param height: specify the height of the background image
+        :param width: specify the width of the background image
+        """
     def get_random_chinese(
         self, min: int, max: int, add_extra_symbol: bool = False
     ) -> list[Tuple[str, list[Tuple[str, int, int, int]]]]:
@@ -143,7 +153,7 @@ class Generator:
         text_with_font_list: list[Tuple[str, list[Tuple[str, int, int, int]]]],
         text_color: Tuple[int, int, int],
         background_color: Tuple[int, int, int],
-        apply_effect: bool = False
+        apply_effect: bool = False,
     ) -> npt.NDArray:
         """
         Generate an image based on a given list of characters and font information.