{
  "file_path": "poster/templates/collage_template.py",
  "file_size": 4095,
  "line_count": 124,
  "functions": [
    {
      "name": "generate",
      "line_start": 24,
      "line_end": 56,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "image_paths",
          "type_hint": "List[str]"
        },
        {
          "name": "style",
          "type_hint": "str"
        }
      ],
      "return_type": "Image.Image",
      "docstring": "生成拼图海报\n\nArgs:\n    image_paths (List[str]): 用于制作拼图的图片路径列表\n    style (str): 拼图样式 (例如 'grid_2x2', 'asymmetric', 'filmstrip')\n\nReturns:\n    Image.Image: 生成的拼图海报图像",
      "is_async": false,
      "decorators": [],
      "code": "    def generate(self,\n                 image_paths: List[str],\n                 style: str = 'grid_2x2',\n                 **kwargs) -> Image.Image:\n        \"\"\"\n        生成拼图海报\n\n        Args:\n            image_paths (List[str]): 用于制作拼图的图片路径列表\n            style (str): 拼图样式 (例如 'grid_2x2', 'asymmetric', 'filmstrip')\n        \n        Returns:\n            Image.Image: 生成的拼图海报图像\n        \"\"\"\n        if not self._validate_inputs(['image_paths'], image_paths=image_paths):\n            return None\n        if not image_paths:\n            logger.error(\"生成拼图失败: 图片列表不能为空。\")\n            return None\n\n        images = [self.image_processor.load_image(p) for p in image_paths if p]\n        images = [img for img in images if img]  # 过滤掉加载失败的图片\n\n        if not images:\n            logger.error(\"所有图片都加载失败。\")\n            return None\n        \n        # 根据样式选择对应的私有方法\n        style_method_name = f\"_create_{style}_collage\"\n        style_method = getattr(self, style_method_name, self._create_grid_2x2_collage)\n        \n        logger.info(f\"使用 '{style}' 风格创建拼图...\")\n        return style_method(images, self.size)",
      "code_hash": "096ed2c4dad1d06302e309eee8b8fb27"
    },
    {
      "name": "_resize_and_crop",
      "line_start": 58,
      "line_end": 60,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "img"
        },
        {
          "name": "target_size"
        }
      ],
      "return_type": null,
      "docstring": "辅助方法：调整并裁剪图片以适应目标尺寸",
      "is_async": false,
      "decorators": [],
      "code": "    def _resize_and_crop(self, img, target_size):\n        \"\"\"辅助方法：调整并裁剪图片以适应目标尺寸\"\"\"\n        return self.image_processor.resize_and_crop(img, target_size)",
      "code_hash": "696270183a17e54e5bcfae13e81b60c9"
    },
    {
      "name": "_create_grid_2x2_collage",
      "line_start": 64,
      "line_end": 82,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "images",
          "type_hint": "List[Image.Image]"
        },
        {
          "name": "target_size",
          "type_hint": "Tuple[int, int]"
        }
      ],
      "return_type": "Image.Image",
      "docstring": "创建 2x2 网格拼图",
      "is_async": false,
      "decorators": [],
      "code": "    def _create_grid_2x2_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建 2x2 网格拼图\"\"\"\n        if len(images) < 4:\n            logger.warning(f\"2x2网格需要4张图，但只提供了{len(images)}张。将使用重复图片。\")\n            images.extend([random.choice(images) for _ in range(4 - len(images))])\n        \n        w, h = target_size\n        img_w, img_h = w // 2, h // 2\n        \n        canvas = self.create_canvas()\n        \n        resized_images = [self._resize_and_crop(img, (img_w, img_h)) for img in images[:4]]\n        \n        canvas.paste(resized_images[0], (0, 0))\n        canvas.paste(resized_images[1], (img_w, 0))\n        canvas.paste(resized_images[2], (0, img_h))\n        canvas.paste(resized_images[3], (img_w, img_h))\n        \n        return canvas",
      "code_hash": "82355b1196b092405dc327e4e7678a09"
    },
    {
      "name": "_create_asymmetric_collage",
      "line_start": 84,
      "line_end": 105,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "images",
          "type_hint": "List[Image.Image]"
        },
        {
          "name": "target_size",
          "type_hint": "Tuple[int, int]"
        }
      ],
      "return_type": "Image.Image",
      "docstring": "创建非对称拼图（左一右二）",
      "is_async": false,
      "decorators": [],
      "code": "    def _create_asymmetric_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建非对称拼图（左一右二）\"\"\"\n        if len(images) < 3:\n            logger.warning(f\"非对称拼图需要3张图，将使用重复图片。\")\n            images.extend([random.choice(images) for _ in range(3 - len(images))])\n            \n        w, h = target_size\n        main_w = int(w * 0.6)\n        side_w = w - main_w\n        side_h = h // 2\n        \n        canvas = self.create_canvas()\n\n        main_img = self._resize_and_crop(images[0], (main_w, h))\n        side_img1 = self._resize_and_crop(images[1], (side_w, side_h))\n        side_img2 = self._resize_and_crop(images[2], (side_w, side_h))\n        \n        canvas.paste(main_img, (0, 0))\n        canvas.paste(side_img1, (main_w, 0))\n        canvas.paste(side_img2, (main_w, side_h))\n\n        return canvas",
      "code_hash": "49917abd95156a55d639ba3a89c51208"
    },
    {
      "name": "_create_filmstrip_collage",
      "line_start": 107,
      "line_end": 122,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "images",
          "type_hint": "List[Image.Image]"
        },
        {
          "name": "target_size",
          "type_hint": "Tuple[int, int]"
        }
      ],
      "return_type": "Image.Image",
      "docstring": "创建电影胶片风格拼图",
      "is_async": false,
      "decorators": [],
      "code": "    def _create_filmstrip_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建电影胶片风格拼图\"\"\"\n        num_images = min(len(images), 4) # 最多4张\n        w, h = target_size\n        padding = 10\n        img_h = (h - (num_images + 1) * padding) // num_images\n\n        canvas = self.create_canvas(background_color=(30, 30, 30, 255))\n        \n        current_y = padding\n        for img in images[:num_images]:\n            resized_img = self._resize_and_crop(img, (w - 2 * padding, img_h))\n            canvas.paste(resized_img, (padding, current_y))\n            current_y += img_h + padding\n            \n        return canvas",
      "code_hash": "9ca7d505aaef9100103696bbfc71e660"
    },
    {
      "name": "_validate_inputs",
      "line_start": 124,
      "line_end": 125,
      "args": [
        {
          "name": "self"
        },
        {
          "name": "required_keys",
          "type_hint": "list"
        }
      ],
      "return_type": "bool",
      "docstring": "",
      "is_async": false,
      "decorators": [],
      "code": "    def _validate_inputs(self, required_keys: list, **kwargs) -> bool:\n        return super()._validate_inputs(required_keys, **kwargs) ",
      "code_hash": "476143d3ed1432102ca95ee17d59b7bb"
    }
  ],
  "classes": [
    {
      "name": "CollageTemplate",
      "line_start": 18,
      "line_end": 125,
      "bases": [
        "BaseTemplate"
      ],
      "methods": [
        {
          "name": "generate",
          "line_start": 24,
          "line_end": 56,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "image_paths",
              "type_hint": "List[str]"
            },
            {
              "name": "style",
              "type_hint": "str"
            }
          ],
          "return_type": "Image.Image",
          "docstring": "生成拼图海报\n\nArgs:\n    image_paths (List[str]): 用于制作拼图的图片路径列表\n    style (str): 拼图样式 (例如 'grid_2x2', 'asymmetric', 'filmstrip')\n\nReturns:\n    Image.Image: 生成的拼图海报图像",
          "is_async": false,
          "decorators": [],
          "code": "    def generate(self,\n                 image_paths: List[str],\n                 style: str = 'grid_2x2',\n                 **kwargs) -> Image.Image:\n        \"\"\"\n        生成拼图海报\n\n        Args:\n            image_paths (List[str]): 用于制作拼图的图片路径列表\n            style (str): 拼图样式 (例如 'grid_2x2', 'asymmetric', 'filmstrip')\n        \n        Returns:\n            Image.Image: 生成的拼图海报图像\n        \"\"\"\n        if not self._validate_inputs(['image_paths'], image_paths=image_paths):\n            return None\n        if not image_paths:\n            logger.error(\"生成拼图失败: 图片列表不能为空。\")\n            return None\n\n        images = [self.image_processor.load_image(p) for p in image_paths if p]\n        images = [img for img in images if img]  # 过滤掉加载失败的图片\n\n        if not images:\n            logger.error(\"所有图片都加载失败。\")\n            return None\n        \n        # 根据样式选择对应的私有方法\n        style_method_name = f\"_create_{style}_collage\"\n        style_method = getattr(self, style_method_name, self._create_grid_2x2_collage)\n        \n        logger.info(f\"使用 '{style}' 风格创建拼图...\")\n        return style_method(images, self.size)",
          "code_hash": "096ed2c4dad1d06302e309eee8b8fb27"
        },
        {
          "name": "_resize_and_crop",
          "line_start": 58,
          "line_end": 60,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "img"
            },
            {
              "name": "target_size"
            }
          ],
          "return_type": null,
          "docstring": "辅助方法：调整并裁剪图片以适应目标尺寸",
          "is_async": false,
          "decorators": [],
          "code": "    def _resize_and_crop(self, img, target_size):\n        \"\"\"辅助方法：调整并裁剪图片以适应目标尺寸\"\"\"\n        return self.image_processor.resize_and_crop(img, target_size)",
          "code_hash": "696270183a17e54e5bcfae13e81b60c9"
        },
        {
          "name": "_create_grid_2x2_collage",
          "line_start": 64,
          "line_end": 82,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "images",
              "type_hint": "List[Image.Image]"
            },
            {
              "name": "target_size",
              "type_hint": "Tuple[int, int]"
            }
          ],
          "return_type": "Image.Image",
          "docstring": "创建 2x2 网格拼图",
          "is_async": false,
          "decorators": [],
          "code": "    def _create_grid_2x2_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建 2x2 网格拼图\"\"\"\n        if len(images) < 4:\n            logger.warning(f\"2x2网格需要4张图，但只提供了{len(images)}张。将使用重复图片。\")\n            images.extend([random.choice(images) for _ in range(4 - len(images))])\n        \n        w, h = target_size\n        img_w, img_h = w // 2, h // 2\n        \n        canvas = self.create_canvas()\n        \n        resized_images = [self._resize_and_crop(img, (img_w, img_h)) for img in images[:4]]\n        \n        canvas.paste(resized_images[0], (0, 0))\n        canvas.paste(resized_images[1], (img_w, 0))\n        canvas.paste(resized_images[2], (0, img_h))\n        canvas.paste(resized_images[3], (img_w, img_h))\n        \n        return canvas",
          "code_hash": "82355b1196b092405dc327e4e7678a09"
        },
        {
          "name": "_create_asymmetric_collage",
          "line_start": 84,
          "line_end": 105,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "images",
              "type_hint": "List[Image.Image]"
            },
            {
              "name": "target_size",
              "type_hint": "Tuple[int, int]"
            }
          ],
          "return_type": "Image.Image",
          "docstring": "创建非对称拼图（左一右二）",
          "is_async": false,
          "decorators": [],
          "code": "    def _create_asymmetric_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建非对称拼图（左一右二）\"\"\"\n        if len(images) < 3:\n            logger.warning(f\"非对称拼图需要3张图，将使用重复图片。\")\n            images.extend([random.choice(images) for _ in range(3 - len(images))])\n            \n        w, h = target_size\n        main_w = int(w * 0.6)\n        side_w = w - main_w\n        side_h = h // 2\n        \n        canvas = self.create_canvas()\n\n        main_img = self._resize_and_crop(images[0], (main_w, h))\n        side_img1 = self._resize_and_crop(images[1], (side_w, side_h))\n        side_img2 = self._resize_and_crop(images[2], (side_w, side_h))\n        \n        canvas.paste(main_img, (0, 0))\n        canvas.paste(side_img1, (main_w, 0))\n        canvas.paste(side_img2, (main_w, side_h))\n\n        return canvas",
          "code_hash": "49917abd95156a55d639ba3a89c51208"
        },
        {
          "name": "_create_filmstrip_collage",
          "line_start": 107,
          "line_end": 122,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "images",
              "type_hint": "List[Image.Image]"
            },
            {
              "name": "target_size",
              "type_hint": "Tuple[int, int]"
            }
          ],
          "return_type": "Image.Image",
          "docstring": "创建电影胶片风格拼图",
          "is_async": false,
          "decorators": [],
          "code": "    def _create_filmstrip_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建电影胶片风格拼图\"\"\"\n        num_images = min(len(images), 4) # 最多4张\n        w, h = target_size\n        padding = 10\n        img_h = (h - (num_images + 1) * padding) // num_images\n\n        canvas = self.create_canvas(background_color=(30, 30, 30, 255))\n        \n        current_y = padding\n        for img in images[:num_images]:\n            resized_img = self._resize_and_crop(img, (w - 2 * padding, img_h))\n            canvas.paste(resized_img, (padding, current_y))\n            current_y += img_h + padding\n            \n        return canvas",
          "code_hash": "9ca7d505aaef9100103696bbfc71e660"
        },
        {
          "name": "_validate_inputs",
          "line_start": 124,
          "line_end": 125,
          "args": [
            {
              "name": "self"
            },
            {
              "name": "required_keys",
              "type_hint": "list"
            }
          ],
          "return_type": "bool",
          "docstring": "",
          "is_async": false,
          "decorators": [],
          "code": "    def _validate_inputs(self, required_keys: list, **kwargs) -> bool:\n        return super()._validate_inputs(required_keys, **kwargs) ",
          "code_hash": "476143d3ed1432102ca95ee17d59b7bb"
        }
      ],
      "docstring": "拼图风格模板，用于将多张图片组合成一张海报。\n支持多种布局样式，如网格、非对称、电影胶片等。",
      "decorators": [],
      "code": "class CollageTemplate(BaseTemplate):\n    \"\"\"\n    拼图风格模板，用于将多张图片组合成一张海报。\n    支持多种布局样式，如网格、非对称、电影胶片等。\n    \"\"\"\n\n    def generate(self,\n                 image_paths: List[str],\n                 style: str = 'grid_2x2',\n                 **kwargs) -> Image.Image:\n        \"\"\"\n        生成拼图海报\n\n        Args:\n            image_paths (List[str]): 用于制作拼图的图片路径列表\n            style (str): 拼图样式 (例如 'grid_2x2', 'asymmetric', 'filmstrip')\n        \n        Returns:\n            Image.Image: 生成的拼图海报图像\n        \"\"\"\n        if not self._validate_inputs(['image_paths'], image_paths=image_paths):\n            return None\n        if not image_paths:\n            logger.error(\"生成拼图失败: 图片列表不能为空。\")\n            return None\n\n        images = [self.image_processor.load_image(p) for p in image_paths if p]\n        images = [img for img in images if img]  # 过滤掉加载失败的图片\n\n        if not images:\n            logger.error(\"所有图片都加载失败。\")\n            return None\n        \n        # 根据样式选择对应的私有方法\n        style_method_name = f\"_create_{style}_collage\"\n        style_method = getattr(self, style_method_name, self._create_grid_2x2_collage)\n        \n        logger.info(f\"使用 '{style}' 风格创建拼图...\")\n        return style_method(images, self.size)\n\n    def _resize_and_crop(self, img, target_size):\n        \"\"\"辅助方法：调整并裁剪图片以适应目标尺寸\"\"\"\n        return self.image_processor.resize_and_crop(img, target_size)\n\n    # --- 各种拼图样式的实现 ---\n\n    def _create_grid_2x2_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建 2x2 网格拼图\"\"\"\n        if len(images) < 4:\n            logger.warning(f\"2x2网格需要4张图，但只提供了{len(images)}张。将使用重复图片。\")\n            images.extend([random.choice(images) for _ in range(4 - len(images))])\n        \n        w, h = target_size\n        img_w, img_h = w // 2, h // 2\n        \n        canvas = self.create_canvas()\n        \n        resized_images = [self._resize_and_crop(img, (img_w, img_h)) for img in images[:4]]\n        \n        canvas.paste(resized_images[0], (0, 0))\n        canvas.paste(resized_images[1], (img_w, 0))\n        canvas.paste(resized_images[2], (0, img_h))\n        canvas.paste(resized_images[3], (img_w, img_h))\n        \n        return canvas\n\n    def _create_asymmetric_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建非对称拼图（左一右二）\"\"\"\n        if len(images) < 3:\n            logger.warning(f\"非对称拼图需要3张图，将使用重复图片。\")\n            images.extend([random.choice(images) for _ in range(3 - len(images))])\n            \n        w, h = target_size\n        main_w = int(w * 0.6)\n        side_w = w - main_w\n        side_h = h // 2\n        \n        canvas = self.create_canvas()\n\n        main_img = self._resize_and_crop(images[0], (main_w, h))\n        side_img1 = self._resize_and_crop(images[1], (side_w, side_h))\n        side_img2 = self._resize_and_crop(images[2], (side_w, side_h))\n        \n        canvas.paste(main_img, (0, 0))\n        canvas.paste(side_img1, (main_w, 0))\n        canvas.paste(side_img2, (main_w, side_h))\n\n        return canvas\n\n    def _create_filmstrip_collage(self, images: List[Image.Image], target_size: Tuple[int, int]) -> Image.Image:\n        \"\"\"创建电影胶片风格拼图\"\"\"\n        num_images = min(len(images), 4) # 最多4张\n        w, h = target_size\n        padding = 10\n        img_h = (h - (num_images + 1) * padding) // num_images\n\n        canvas = self.create_canvas(background_color=(30, 30, 30, 255))\n        \n        current_y = padding\n        for img in images[:num_images]:\n            resized_img = self._resize_and_crop(img, (w - 2 * padding, img_h))\n            canvas.paste(resized_img, (padding, current_y))\n            current_y += img_h + padding\n            \n        return canvas\n        \n    def _validate_inputs(self, required_keys: list, **kwargs) -> bool:\n        return super()._validate_inputs(required_keys, **kwargs) ",
      "code_hash": "2bc0edfc12a8ceeab12e40988b8dcfca"
    }
  ],
  "imports": [
    {
      "type": "import",
      "modules": [
        "logging"
      ],
      "aliases": []
    },
    {
      "type": "import",
      "modules": [
        "random"
      ],
      "aliases": []
    },
    {
      "type": "from_import",
      "module": "typing",
      "names": [
        "List",
        "Tuple",
        "Optional"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "PIL",
      "names": [
        "Image",
        "ImageDraw"
      ],
      "aliases": [],
      "level": 0
    },
    {
      "type": "from_import",
      "module": "base_template",
      "names": [
        "BaseTemplate"
      ],
      "aliases": [],
      "level": 1
    }
  ],
  "constants": [],
  "docstring": "Collage风格（拼图风格）海报模板",
  "content_hash": "b04aa331f67aa2626fee2d7665fbf690"
}