| |
|
|
| import os |
| import numpy as np |
| import torch |
| import folder_paths |
| |
| from nodes import LoadImage |
| from PIL import Image,ImageOps,ImageSequence,ImageDraw,ImageFont |
| from .imagefunc import pil2tensor, log, generate_text_image, get_resource_dir |
|
|
|
|
| class LoadPSD(LoadImage): |
| @classmethod |
| def INPUT_TYPES(s): |
| input_dir = folder_paths.get_input_directory() |
| files = [f for f in os.listdir(input_dir) if os.path.isfile( |
| os.path.join(input_dir, f)) and f.endswith(".psd")] |
| fine_layer_method = ["layer_index", "layer_name"] |
| return {"required":{ |
| "image": (sorted(files), {"image_upload": True}), |
| "file_path": ("STRING", {"default": ""}), |
| "include_hidden_layer": ("BOOLEAN", {"default": False}), |
| "find_layer_by": (fine_layer_method,), |
| "layer_index": ("INT", {"default": 0, "min": -1, "max": 999, "step": 1}), |
| "layer_name": ("STRING", {"default": ""}), |
| }, |
| "optional": { |
| } |
| } |
|
|
|
|
| RETURN_TYPES = ("IMAGE", "IMAGE", "IMAGE",) |
| RETURN_NAMES = ("flat_image", "layer_image", "all_layers",) |
| FUNCTION = "load_psd" |
| CATEGORY = '😺dzNodes/LayerUtility/SystemIO' |
|
|
| def load_psd(self, image, file_path, include_hidden_layer, layer_index, find_layer_by, layer_name,): |
|
|
| from psd_tools import PSDImage |
| from psd_tools.api.layers import Layer |
|
|
| (LUT_DICT, FONT_DICT) = get_resource_dir() |
|
|
| NODE_NAME = 'LoadPSD' |
| number_of_layers = 1 |
| layer_image = [] |
| all_layers = [] |
| if file_path == "": |
| psd_file_path = folder_paths.get_annotated_filepath(image) |
| else: |
| psd_file_path = folder_paths.get_annotated_filepath(file_path) |
| flat_image = Image.open(psd_file_path).convert("RGB") |
| width, height = flat_image.size |
|
|
| if image.endswith(".psd"): |
| from psd_tools import PSDImage |
| from psd_tools.api.layers import Layer |
| log(f"{NODE_NAME} -> Loading PSD file: {psd_file_path}") |
| psd_image = PSDImage.open(psd_file_path) |
| layers = [] |
| for layer in psd_image: |
| if include_hidden_layer: |
| if not layer.is_visible(): |
| layer.visible = True |
| layers.append(layer) |
| else: |
| if layer.is_visible(): |
| layers.append(layer) |
|
|
| number_of_layers = len(layers) |
| for i in range(number_of_layers): |
| layer_canvas = Image.new("RGBA", (width, height), (0, 0, 0, 0)) |
| layer_canvas.paste(layers[i].composite(), layers[i].bbox) |
| all_layers.append(pil2tensor(layer_canvas)) |
| if find_layer_by == "layer_name": |
| if layers[i].name == layer_name: |
| layer_image.append(pil2tensor(layer_canvas)) |
| log(f"{NODE_NAME} -> Layer {i} : {layer.name} found.") |
| elif find_layer_by == "layer_index": |
| if i == layer_index: |
| layer_image.append(pil2tensor(layer_canvas)) |
| log(f"{NODE_NAME} -> Layer {i} : {layer.name} found.") |
|
|
| text = "Layer Not Found!" |
| font_file = list(FONT_DICT.values())[0] |
| empty_layer_image = generate_text_image(flat_image.width, flat_image.height, text, font_file, font_color="#F01000") |
| if layer_image == []: |
| if layer_index == -1: |
| layer_image.append(all_layers[-1]) |
| elif find_layer_by == "layer_name": |
| log(f'{NODE_NAME} -> Layer "{layer_name}" not found, top layer will be output.', message_type="warning") |
| elif find_layer_by == "layer_index": |
| log(f'{NODE_NAME} -> Layer index {layer_index} not found, top layer will be output.', message_type="warning") |
| layer_image.append(pil2tensor(empty_layer_image)) |
| else: |
| layer_image.append(pil2tensor(flat_image)) |
| all_layers.append(pil2tensor(flat_image)) |
|
|
| return (torch.cat([pil2tensor(flat_image),], dim=0), torch.cat(layer_image, dim=0), torch.cat(all_layers, dim=0),) |
|
|
|
|
| NODE_CLASS_MAPPINGS = { |
| "LayerUtility: LoadPSD": LoadPSD, |
| } |
|
|
| NODE_DISPLAY_NAME_MAPPINGS = { |
| "LayerUtility: LoadPSD": "LayerUtility: Load PSD(Advance)", |
| } |
