hsi-preprocessing-toolkit 1.1.3__tar.gz → 1.1.5.dev4__tar.gz

{hsi_preprocessing_toolkit-1.1.3 → hsi_preprocessing_toolkit-1.1.5.dev4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hsi-preprocessing-toolkit
-Version: 1.1.3
+Version: 1.1.5.dev4
 Summary: HSI Preprocessing Toolkit
 Author: songyz2023
 Author-email: songyz2023 <songyz2023dlut@outlook.com>
@@ -39,6 +39,7 @@ A Hyperspectral Preprocessing Toolkit that
 3. Preview HSI, and convert it to RGB `.png` file
 4. Crop and rotate the HSI and preview in realtime
 5. Select spectrals of interest visually and save them into a `.mat` file
+6. Mix multiple HSI image with layers. (coming in v2.0.0, now availabe in git)
 
 
 ## Usage

{hsi_preprocessing_toolkit-1.1.3 → hsi_preprocessing_toolkit-1.1.5.dev4}/README.md

@@ -16,6 +16,7 @@ A Hyperspectral Preprocessing Toolkit that
 3. Preview HSI, and convert it to RGB `.png` file
 4. Crop and rotate the HSI and preview in realtime
 5. Select spectrals of interest visually and save them into a `.mat` file
+6. Mix multiple HSI image with layers. (coming in v2.0.0, now availabe in git)
 
 
 ## Usage

{hsi_preprocessing_toolkit-1.1.3 → hsi_preprocessing_toolkit-1.1.5.dev4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hsi-preprocessing-toolkit"
-version = "1.1.3"
+version = "1.1.5.dev4"
 description = "HSI Preprocessing Toolkit"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -33,7 +33,7 @@ url = "https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple"
 default = true
 
 [build-system]
-requires = ["uv_build >= 0.8.0, <0.9.0"]
+requires = ["uv_build >= 0.8.0, <0.10.0"]
 build-backend = "uv_build"
 
 [tool.hatch.build.targets.wheel]

{hsi_preprocessing_toolkit-1.1.3 → hsi_preprocessing_toolkit-1.1.5.dev4}/src/hsi_preprocessing_toolkit/__main__.py

@@ -8,91 +8,19 @@ from rasterio import open as rasterio_open
 from scipy.io import savemat, loadmat
 import gradio as gr
 import gradio.utils
-from scipy.ndimage import rotate
 from rs_fusion_datasets.util.hsi2rgb import _hsi2rgb, hsi2rgb
 from jaxtyping import Float
 from enum import Enum
-from scanner_calc import scanner_calc_tab
-
-
-plt.rcParams['font.family'] = 'SimHei'
-
-i18n = gr.I18n(**{
-    'en': {
-        "title": "hsi-preprocessing-toolkit",
-        "load": "Load",
-        "upload_instructions": "**Upload one of the following formats:**\n1. One .hdr file + one raw data file without extension\n2. One .mat file",
-        "input_format": "Input Image Shape Format",
-        "data_files": "Data Files",
-        "manual_normalize": "Manual Normalize (affects preview only)",
-        "normalize_min": "Normalize Min",
-        "normalize_max": "Normalize Max",
-        "wavelength_start": "Wavelength Range Start",
-        "wavelength_end": "Wavelength Range End",
-        "processing": "Processing",
-        "crop": "Crop",
-        "top": "Top",
-        "bottom": "Bottom",
-        "left": "Left",
-        "right": "Right",
-        "rotate": "Rotate",
-        "rotate_degree": "Rotate Degree",
-        "preview": "Preview",
-        "apply_processing": "Apply Processing Effects",
-        "mat_data_type": "MAT Data Type",
-        "mat_format": "MAT Image Shape Format",
-        "mat_key": "Key of MAT file",
-        "compress_mat": "Produce Compressed MAT File",
-        "spectral_selection": "Spectral Selection",
-        "spectral_selection_help": "Click on the image to select pixels for spectral data extraction. The selected pixels will be plotted in the spectral plot below.",
-        "spectral_plot": "Spectral Plot",
-        "style": "Style",
-        "clear": "Clear",
-        "download": "Download",
-        "output_results": "Output Results",
-        "mat_file": "MAT File",
-        "info": "Info",
-        "same_as_input": "Same",
-        "auto_detect": "Auto Detect",
-    },
-    'zh-CN':{
-        "title": "高光谱图像预处理工具箱",
-        "load": "加载",
-        "upload_instructions": "**应上传以下两种格式中的一种**\n1. 同时上传一个.hdr文件 + 一个无后缀的数据文件\n2. 一个.mat文件",
-        "input_format": "输入数据形状",
-        "data_files": "数据文件",
-        "manual_normalize": "手动归一化(仅影响预览结果)",
-        "normalize_min": "归一化最小值",
-        "normalize_max": "归一化最大值",
-        "wavelength_start": "波长范围起始",
-        "wavelength_end": "波长范围结束",
-        "processing": "处理",
-        "crop": "裁切",
-        "top": "上",
-        "bottom": "下",
-        "left": "左",
-        "right": "右",
-        "rotate": "旋转",
-        "rotate_degree": "旋转角度",
-        "preview": "预览",
-        "apply_processing": "应用处理效果",
-        "mat_data_type": "mat文件数据类型",
-        "mat_format": "mat文件格式",
-        "mat_key": "mat文件的key",
-        "compress_mat": "启用mat文件压缩",
-        "spectral_selection": "光谱选择",
-        "spectral_selection_help": "点击预览图像图像中的像素进行光谱数据提取。选中的像素将在下方的光谱图中绘制。",
-        "spectral_plot": "光谱图",
-        "style": "样式",
-        "clear": "清空",
-        "download": "下载",
-        "output_results": "输出结果",
-        "mat_file": "MAT文件",
-        "info": "信息",
-        "same_as_input": "与输入相同",
-        "auto_detect": "自动检测",
-    }
-})
+from .page.scanner_calc import scanner_calc_tab
+import logging
+from .i18n import i18n
+from .algorithm import composite_img
+
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+logger.info("started")
+
 
 class AppState(Enum):
     NOT_LOADED = 0
@@ -151,19 +79,7 @@ def load_data(
     return data, data_path
 
 
-# HWC -> HWC
-def process_img(img, crop_top, crop_left, crop_bottom, crop_right, rotate_deg):
-    # Rotate
-    if rotate_deg % 360 != 0:
-        img = rotate(img, angle=rotate_deg, axes=(0, 1), reshape=True)
 
-    # Crop
-    if crop_top > 0 or crop_left > 0 or crop_bottom > 0 or crop_right > 0:
-        crop_bottom = None if crop_bottom == 0 else -crop_bottom
-        crop_right  = None if crop_right == 0  else -crop_right
-        img = img[crop_top:crop_bottom, crop_left:crop_right, :]
-
-    return img
 
 # def update_ui(state_app_state: AppState):
 #     gr.Info(f"App state changed to {state_app_state.name}")
@@ -179,11 +95,15 @@ def process_img(img, crop_top, crop_left, crop_bottom, crop_right, rotate_deg):
 #     )
 
 def gr_load(
+        state_current_layer_index,
+        state_transforms :list[dict], state_original_rgb, state_original_data, state_original_data_path,
         dat_files :List[gradio.utils.NamedString] | None,
         input_format, input_mat_key: str | None,
         manual_normalize: bool, normalize_min: float, normalize_max: float,
         wavelength_from: int, wavelength_to: int,
     ):
+    logger.info(f"gr_load {state_current_layer_index=} {len(state_original_rgb)=}")
+
     data, data_path = load_data(dat_files, input_format, input_mat_key)
     if data is None:
         raise gr.Error("No data file provided or data loading failed.")
@@ -201,26 +121,40 @@ def gr_load(
         "original_reflection_mean": float(data.mean()),
         "original_reflection_std": float(data.std()),
     })
-    return rgb, data, data_path, AppState.LOADED, logging_text
 
-def gr_preview(
-        state_rgb :Float[np.ndarray, 'h w c'] | None,
-        crop_top: int, crop_left: int, crop_bottom: int, crop_right: int, 
-        rotate_deg: int
+    # FIXME: Do not use dirty fix
+    while state_current_layer_index >= len(state_original_rgb):
+        state_original_rgb.append(None)
+        state_original_data.append(None)
+        state_original_data_path.append(None)
+        state_transforms.append(DEFAULT_TRANSFORM)
+
+    state_original_rgb[state_current_layer_index]       = rgb
+    state_original_data[state_current_layer_index]      = data
+    state_original_data_path[state_current_layer_index] = data_path
+    state_transforms[state_current_layer_index]         = DEFAULT_TRANSFORM
+
+    logger.info(f"gr_loaded {state_current_layer_index=} {len(state_original_rgb)=}")
+
+    return state_transforms, state_original_rgb, state_original_data, state_original_data_path, AppState.LOADED, logging_text
+
+def gr_composite(
+        state_original_rgb :Float[np.ndarray, 'h w c'] | None,
+        state_transforms,
     ):
-    if state_rgb is None:
-        raise gr.Error("No data provided.")
-    return process_img(
-        state_rgb, crop_top, crop_left, crop_bottom, crop_right, rotate_deg
-    ), AppState.PREVIEWED
+    logger.info("gr_composite")
+    img = composite_img(state_original_rgb, state_transforms)
+    logger.info(f"gr_composited {img.shape=} {type(img)=}")
+    return img, AppState.PREVIEWED
 
-def gr_download_selected_spectral(data, state_select_location, data_path):
+
+def gr_download_selected_spectral(state_current_layer_index, data, state_select_location, data_path):
     if not state_select_location or len(state_select_location) == 0:
         raise gr.Error("No spectral data selected. Please select at least one pixel to download")
     
     gr.Info("Converting ...")
 
-    dat_path = Path(data_path.name)
+    dat_path = Path(data_path[state_current_layer_index].name)
     result = []
     result_location = []
     for (row, col) in state_select_location:
@@ -238,17 +172,17 @@ def gr_download_selected_spectral(data, state_select_location, data_path):
     return str(mat_path)
 
 def gr_convert(
+        state_current_layer_index,
+        state_transforms,
         data, data_path,
-        crop_top: int, crop_left: int, crop_bottom: int, crop_right: int, 
-        rotate_deg: int,
         mat_dtype, output_format, mat_key, compress_mat: bool,
         logging_text: str
     ):
-    if output_format == 'same':
-        output_format = input_format
+    logger.info("gr_convert")
     
-    data = process_img(
-        data, crop_top, crop_left, crop_bottom, crop_right, rotate_deg
+    data_path = data_path[state_current_layer_index]
+    data = composite_img(
+        data, state_transforms
     )
     
     # Convert to mat
@@ -313,26 +247,80 @@ def gr_on_img_clicked(evt: gr.SelectData, state_figure :tuple, data, state_selec
     state_figure = (fig, ax)
     return fig, state_figure, state_select_location
 
+def gr_update_transforms(state_transforms, state_current_layer_index, crop_top, crop_left, crop_bottom, crop_right, rotate_deg, offset_x, offset_y):
+    state_transforms[state_current_layer_index] = {
+        'rotation': rotate_deg,
+        'crop': [crop_top, crop_left, crop_bottom, crop_right],
+        'location': [offset_x, offset_y]
+    }
+    return state_transforms
 
-if __name__ == "__main__":
+def gr_on_state_current_layer_index_changed(state_current_layer_index, state_transforms, state_original_data):
+    if state_current_layer_index >= len(state_transforms):
+        trans = DEFAULT_TRANSFORM
+        shape = [0,0]
+    else:
+        trans = state_transforms[state_current_layer_index]
+        shape = state_original_data[state_current_layer_index].shape
+
+    crop_top, crop_left, crop_bottom, crop_right = trans['crop']
+    rotate_deg = trans['rotation']
+    offset_x, offset_y = trans['location']
+    
+    max_h, max_w = shape[:2]
+    logger.info(f"gr_on_state_current_layer_index_changed {state_current_layer_index=} {max_h=}, {max_w=}")
+    update_crop_top = gr.update(value=crop_top, maximum=max_h, minimum=0)
+    update_crop_bottom = gr.update(value=crop_bottom, maximum=max_h, minimum=0)
+    update_crop_left = gr.update(value=crop_left, maximum=max_w, minimum=0)
+    update_crop_right = gr.update(value=crop_bottom, maximum=max_w, minimum=0)
+
+    return update_crop_top, update_crop_left, update_crop_bottom, update_crop_right, rotate_deg, offset_x, offset_y
+
+
+DEFAULT_TRANSFORM = {
+    'rotation': 0,
+    'crop': [0,0,0,0],
+    'location': [0,0]
+}
+
+def main():
     theme = gr.themes.Default(primary_hue='cyan').set(
         button_primary_background_fill='#39c5bb',
         button_primary_background_fill_hover="#30A8A0",
     )
 
     with gr.Blocks(title="hsi-preprocessing-toolkit", theme=theme) as demo:
-        state_app_state = gr.State(value=AppState.NOT_LOADED)
-        state_original_data = gr.State(value=None)
-        state_processed_data = gr.State(value=None)
-        state_data_path = gr.State(value=None,)
-        state_selected_location = gr.State(value=[])
-        state_original_rgb = gr.State(value=None)
-        state_spectral_figure = gr.State(value=None)
+        # 应用整体状态
+        state_app_state = gr.State(value=AppState.NOT_LOADED) # 操作阶段
+        # 输入的状态，支持多输出
+        state_current_layer_index = gr.State(value=0)      # 已选中的图层的数组index
+        state_max_layer_index   = gr.State(value=0)      # 已选中的图层的数组index
+        state_original_data = gr.State(value=[])              # 原数据
+        state_data_path = gr.State(value=[])                  # 原数据文件路径
+        state_original_rgb = gr.State(value=[])               # 原数据的RGB代理
+        state_transforms = gr.State(value=[])                 # 对数据的变换
+        # 缓存状态
+        # state_original_alpha = gr.State(value=[])           # 用于合成的Alpha通道
+        # state_original_offset = gr.State(value=[])            # 尽管用alpha更好，但先用offset吧
+        # 中间状态，多个
+        # state_local_rgb = gr.State(value=[])
+        # 输出内容状态，保持单个
+        state_processed_data = gr.State(value=None)           # 处理后的数据
+        state_selected_location = gr.State(value=[])          # 已选择的光谱XY坐标点 
+        state_spectral_figure = gr.State(value=None)          # 已选择的光谱的绘图 
 
 
         with gr.Tab(i18n("title"), id="hsi_preprocessing_toolkit"):
             with gr.Row():
                 with gr.Column():
+                    with gr.Column():
+                        current_layer_index_slider = gr.Slider(
+                            label="当前图层",
+                            minimum=0,
+                            maximum=3,
+                            step=1,
+                            value=0
+                        )
                     with gr.Accordion(i18n("load")) as load_panel:
                         gr.Markdown(i18n("upload_instructions"))
                         with gr.Row():
@@ -391,8 +379,10 @@ if __name__ == "__main__":
                                 value=1000,
                                 precision=1,
                             )
-                        reload_btn = gr.Button(i18n("load"), variant="primary")
-                    
+                        with gr.Row():
+                            reload_btn = gr.Button(i18n("load"), variant="primary")
+                            reset_loaded_btn = gr.Button(i18n("clear"), variant="secondary")
+                    # TODO: set maxium to be shape[i]
                     with gr.Accordion(i18n("processing"), visible=False) as preview_panel:
                         with gr.Column():
                             gr.Markdown(f"### {i18n('crop')}")
@@ -400,26 +390,26 @@ if __name__ == "__main__":
                                 crop_top = gr.Slider(
                                     label=i18n("top"),
                                     minimum=0,
-                                    maximum=0,
+                                    maximum=9999,
                                     step=1,
                                 )
                                 crop_bottom = gr.Slider(
                                     label=i18n("bottom"),
                                     minimum=0,
-                                    maximum=0,
+                                    maximum=9999,
                                     step=1,
                                 )
                             with gr.Row():
                                 crop_left = gr.Slider(
                                     label=i18n("left"),
                                     minimum=0,
-                                    maximum=0,
+                                    maximum=9999,
                                     step=1,
                                 )
                                 crop_right = gr.Slider(
                                     label=i18n("right"),
                                     minimum=0,
-                                    maximum=0,
+                                    maximum=9999,
                                     step=1,
                                 )
 
@@ -432,24 +422,36 @@ if __name__ == "__main__":
                                 step=1,
                                 value=0
                             )
-                        # 这个按钮因为实时预览实际上不需要了，但是暂且隐藏备用
-                        preview_btn  = gr.Button(i18n("preview"), variant="primary", visible=False)
                         
+                        with gr.Column():
+                            gr.Markdown(f"### 平移")
+                            with gr.Row():
+                                offset_x = gr.Slider(
+                                    label="X轴",
+                                    minimum=0,
+                                    maximum=+9999,
+                                    step=1,
+                                )
+                                offset_y = gr.Slider(
+                                    label="Y轴",
+                                    minimum=0,
+                                    maximum=+9999,
+                                    step=1,
+                                )
 
                     with gr.Accordion(i18n("apply_processing"), visible=False) as convert_panel:
                         mat_dtype = gr.Radio(
                             label=i18n("mat_data_type"),
-                            choices=["same", "uint8", "uint16", "float32"],
+                            choices=["auto", "uint8", "uint16", "float32"],
                             value="float32"
                         )
                         output_format = gr.Radio(
                             label=i18n("mat_format"),
                             choices=[
-                                'same', 
                                 'HWC', 
                                 'CHW'
                             ],
-                            value='same'
+                            value='CHW'
                         )
                         mat_key = gr.Text(
                             label=i18n("mat_key"),
@@ -496,44 +498,74 @@ if __name__ == "__main__":
                             label=i18n("info"),
                         )
                     
-
-            # dat_files.upload(
-            #     fn=gr_on_file_upload,
-            #     inputs=[dat_files, input_format, manual_normalize, normalize_min, normalize_max, wavelength_from, wavelength_to],
-            #     outputs=[state_original_rgb, state_original_data, state_data_path, state_app_state]
-            # )
             reload_btn.click(
                 fn=gr_load,
-                inputs=[dat_files, input_format, input_mat_key, manual_normalize, normalize_min, normalize_max, wavelength_from, wavelength_to],
+                inputs=[state_current_layer_index, state_transforms, state_original_rgb, state_original_data, state_data_path, dat_files, input_format, input_mat_key, manual_normalize, normalize_min, normalize_max, wavelength_from, wavelength_to],
+                outputs=[state_transforms, state_original_rgb, state_original_data, state_data_path, state_app_state, logging_text]
+            )
+            reset_loaded_btn.click(
+                fn=lambda logging_text: ([],[],[],AppState.NOT_LOADED, "已清空"),
+                inputs=[logging_text],
                 outputs=[state_original_rgb, state_original_data, state_data_path, state_app_state, logging_text]
             )
             convert_btn.click(
                 fn=gr_convert,
                 inputs=[
+                    state_current_layer_index,
+                    state_transforms,
                     state_original_data, state_data_path,
-                    crop_top, crop_left, crop_bottom, crop_right,
-                    rotate_deg,
                     mat_dtype, output_format, mat_key, compress_mat,
                     logging_text
                 ],
                 outputs=[state_processed_data ,mat_file_output, logging_text, state_app_state]
             )
-            for component in [crop_top, crop_left, crop_bottom, crop_right, rotate_deg]:
+            for component in [crop_top, crop_left, crop_bottom, crop_right, rotate_deg, offset_x, offset_y]:
                 component.change(
-                    fn=gr_preview,
-                    inputs=[state_original_rgb, crop_top, crop_left, crop_bottom, crop_right, rotate_deg],
-                    outputs=[preview_img, state_app_state]
+                    fn = gr_update_transforms,
+                    inputs=[state_transforms, state_current_layer_index, crop_top, crop_left, crop_bottom, crop_right, rotate_deg, offset_x, offset_y],
+                    outputs=[state_transforms]
                 )
-            
-            preview_btn.click(
-                fn=gr_preview,
-                inputs=[
-                    state_original_rgb,
-                    crop_top, crop_left, crop_bottom, crop_right,
-                    rotate_deg,
-                ],
+
+            state_transforms.change(
+                fn=gr_composite,
+                inputs=[state_original_rgb, state_transforms],
                 outputs=[preview_img, state_app_state]
             )
+            # Legacy Code:
+            # preview_btn.click(
+            #     fn=gr_composite,
+            #     inputs=[
+            #         state_original_rgb,
+            #         crop_top, crop_left, crop_bottom, crop_right,
+            #         rotate_deg,
+            #     ],
+            #     outputs=[preview_img, state_app_state]
+            # )
+            current_layer_index_slider.change(
+                fn = lambda x: x,
+                inputs=[current_layer_index_slider],
+                outputs=[state_current_layer_index]
+
+            )
+
+            # ------------
+            state_current_layer_index.change(
+                fn=gr_on_state_current_layer_index_changed,
+                inputs=[state_current_layer_index, state_transforms, state_original_data],
+                outputs=[
+                    crop_top, crop_left, crop_bottom, crop_right, rotate_deg, offset_x, offset_y
+                ]
+            )
+            state_original_data.change(
+                fn=gr_on_state_current_layer_index_changed,
+                inputs=[state_current_layer_index, state_transforms, state_original_data],
+                outputs=[
+                    crop_top, crop_left, crop_bottom, crop_right, rotate_deg, offset_x, offset_y
+                ]
+            )
+            # ------------
+            
+            
             preview_img.select(
                 fn=gr_on_img_clicked,
                 inputs=[state_spectral_figure, state_processed_data, state_selected_location, wavelength_from, wavelength_to, plot_hint],
@@ -545,24 +577,26 @@ if __name__ == "__main__":
             )
             download_select_spectral.click(
                 fn=gr_download_selected_spectral,
-                inputs=[state_processed_data, state_selected_location, state_data_path],
+                inputs=[state_current_layer_index, state_processed_data, state_selected_location, state_data_path],
                 outputs=[download_select_spectral]
             )
-            state_original_data.change(
-                fn=lambda x: (
-                    gr.update(maximum=x.shape[1] if x is not None else 0),
-                    gr.update(maximum=x.shape[1] if x is not None else 0),
-                    gr.update(maximum=x.shape[0] if x is not None else 0),
-                    gr.update(maximum=x.shape[0] if x is not None else 0),
-                ),
-                inputs=[state_original_data],
-                outputs=[crop_left, crop_right, crop_top, crop_bottom]
-            )
-            state_original_rgb.change(
-                fn=lambda x:x,
-                inputs=[state_original_rgb],
-                outputs=[preview_img]
-            )
+            # TODO: Bring it back
+            # state_original_data.change(
+            #     fn=lambda x,i: (
+            #         gr.update(maximum=x[i].shape[1]),
+            #         gr.update(maximum=x[i].shape[1]),
+            #         gr.update(maximum=x[i].shape[0]),
+            #         gr.update(maximum=x[i].shape[0]),
+            #     ),
+            #     inputs=[ state_original_data, state_current_layer_index ],
+            #     outputs=[crop_left, crop_right, crop_top, crop_bottom]
+            # )
+            # TODO: shoud we bring it back?
+            # state_original_rgb.change(
+            #     fn=lambda x:x,
+            #     inputs=[state_original_rgb],
+            #     outputs=[preview_img]
+            # )
             state_app_state.change(
                 fn=lambda x: (
                     gr.update(visible=True, open=(x==AppState.NOT_LOADED)),
@@ -573,8 +607,18 @@ if __name__ == "__main__":
                 inputs=[state_app_state],
                 outputs=[load_panel, preview_panel, convert_panel, plot_panel]
             )
+            # FIXME:
+            # state_max_layer_index.change(
+            #     fn=lambda :gr.update(maximum=state_max_layer_index),
+            #     inputs=[],
+            #     outputs=[current_layer_index_slider]
+            # )
         with gr.Tab("推扫参数计算"):
             scanner_calc_tab()
 
     demo.launch(share=False, inbrowser=True, i18n=i18n)
 
+
+if __name__ == "__main__":
+    plt.rcParams['font.family'] = 'SimHei'
+    main()

hsi_preprocessing_toolkit-1.1.5.dev4/src/hsi_preprocessing_toolkit/algorithm.py

@@ -0,0 +1,42 @@
+from scipy.ndimage import rotate
+import numpy as np
+import logging
+from copy import deepcopy
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+logger.info("started")
+
+# Return A HWC Image, the C should work with both RGB and HSI
+def composite_img(imgs :list[np.ndarray], transforms:list[dict]):
+    imgs = deepcopy(imgs)
+    # Transform
+    logger.info(f"{transforms=}")
+    for i, (img,trans) in enumerate(zip(imgs, transforms)):
+        rotate_deg = trans['rotation']
+        if rotate_deg % 360 != 0:
+            imgs[i] = rotate(img, angle=rotate_deg, axes=(0, 1), reshape=True)
+
+        # Crop
+        crop_top, crop_left, crop_bottom, crop_right = trans['crop']
+        if crop_top > 0 or crop_left > 0 or crop_bottom > 0 or crop_right > 0:
+            crop_bottom = None if crop_bottom == 0 else -crop_bottom
+            crop_right  = None if crop_right == 0  else -crop_right
+            imgs[i] = img[crop_top:crop_bottom, crop_left:crop_right, :]
+        
+    # Caclate canvas size
+    shapes = [x.shape for x in imgs]
+    offsets = [ x['location'] for x in transforms ]
+    canvas_c = imgs[0].shape[-1]
+    sizes = [ ((h+abs(x)),w+abs(y)) for (h,w,_),(x,y) in zip(shapes, offsets)]
+    canvas_h, canvas_w = ( max([x[0] for x in sizes]), max([x[1] for x in sizes]) )
+    canvas_shape = (canvas_h, canvas_w, canvas_c)
+    logger.info(f"{sizes=} {canvas_shape=} {shapes=}")
+    canvas = np.zeros(shape=canvas_shape, dtype=imgs[0].dtype)
+
+    # 合成 Compositing
+    for img,trans in zip(imgs, transforms):
+        x,y = trans['location']
+        h,w,_ = img.shape
+        canvas[x:x+h, y:y+w] = img
+    return canvas

hsi_preprocessing_toolkit-1.1.5.dev4/src/hsi_preprocessing_toolkit/i18n.py

@@ -0,0 +1,78 @@
+import gradio as gr
+
+i18n = gr.I18n(**{
+    'en': {
+        "title": "hsi-preprocessing-toolkit",
+        "load": "Load",
+        "upload_instructions": "**Upload one of the following formats:**\n1. One .hdr file + one raw data file without extension\n2. One .mat file",
+        "input_format": "Input Image Shape Format",
+        "data_files": "Data Files",
+        "manual_normalize": "Manual Normalize (affects preview only)",
+        "normalize_min": "Normalize Min",
+        "normalize_max": "Normalize Max",
+        "wavelength_start": "Wavelength Range Start",
+        "wavelength_end": "Wavelength Range End",
+        "processing": "Processing",
+        "crop": "Crop",
+        "top": "Top",
+        "bottom": "Bottom",
+        "left": "Left",
+        "right": "Right",
+        "rotate": "Rotate",
+        "rotate_degree": "Rotate Degree",
+        "preview": "Preview",
+        "apply_processing": "Apply Processing Effects",
+        "mat_data_type": "MAT Data Type",
+        "mat_format": "MAT Image Shape Format",
+        "mat_key": "Key of MAT file",
+        "compress_mat": "Produce Compressed MAT File",
+        "spectral_selection": "Spectral Selection",
+        "spectral_selection_help": "Click on the image to select pixels for spectral data extraction. The selected pixels will be plotted in the spectral plot below.",
+        "spectral_plot": "Spectral Plot",
+        "style": "Style",
+        "clear": "Clear",
+        "download": "Download",
+        "output_results": "Output Results",
+        "mat_file": "MAT File",
+        "info": "Info",
+        "same_as_input": "Same",
+        "auto_detect": "Auto Detect",
+    },
+    'zh-CN':{
+        "title": "高光谱图像预处理工具箱",
+        "load": "加载",
+        "upload_instructions": "**应上传以下两种格式中的一种**\n1. 同时上传一个.hdr文件 + 一个无后缀的数据文件\n2. 一个.mat文件",
+        "input_format": "输入数据形状",
+        "data_files": "数据文件",
+        "manual_normalize": "手动归一化(仅影响预览结果)",
+        "normalize_min": "归一化最小值",
+        "normalize_max": "归一化最大值",
+        "wavelength_start": "波长范围起始",
+        "wavelength_end": "波长范围结束",
+        "processing": "处理",
+        "crop": "裁切",
+        "top": "上",
+        "bottom": "下",
+        "left": "左",
+        "right": "右",
+        "rotate": "旋转",
+        "rotate_degree": "旋转角度",
+        "preview": "预览",
+        "apply_processing": "应用处理效果",
+        "mat_data_type": "mat文件数据类型",
+        "mat_format": "mat文件格式",
+        "mat_key": "mat文件的key",
+        "compress_mat": "启用mat文件压缩",
+        "spectral_selection": "光谱选择",
+        "spectral_selection_help": "点击预览图像图像中的像素进行光谱数据提取。选中的像素将在下方的光谱图中绘制。",
+        "spectral_plot": "光谱图",
+        "style": "样式",
+        "clear": "清空",
+        "download": "下载",
+        "output_results": "输出结果",
+        "mat_file": "MAT文件",
+        "info": "信息",
+        "same_as_input": "与输入相同",
+        "auto_detect": "自动检测",
+    }
+})