anshitsu 0.0.0__py3-none-any.whl

anshitsu/__version__.py

@@ -0,0 +1 @@
+version: str = "v1.5.0"

anshitsu/cli.py

@@ -0,0 +1,14 @@
+import fire
+
+from anshitsu.process import process
+
+
+def main():
+    """
+    main [summary]
+    """
+    fire.Fire(process)
+
+
+if __name__ == "__main__":
+    main()

anshitsu/process.py

@@ -0,0 +1,139 @@
+import datetime
+import glob
+import os
+import os.path
+import re
+from typing import Optional
+
+import fire
+import fire.core
+from PIL import Image, UnidentifiedImageError
+
+from anshitsu.retouch import Retouch
+from anshitsu.__version__ import version as __version__
+
+
+def process(
+        path: Optional[str] = None,
+        colorautoadjust: bool = False,
+        colorstretch: bool = False,
+        grayscale: bool = False,
+        invert: bool = False,
+        tosaka: Optional[float] = None,
+        outputrgb: bool = False,
+        noise: Optional[float] = None,
+        overwrite: bool = False,
+        version: bool = False,
+) -> str:
+    """
+    Process Runnner for Command Line Interface
+
+    This utility converts the colors of images such as photos.
+
+    If you specify a directory path, it will convert
+    the image files in the specified directory.
+    If you specify a file path, it will convert the specified file.
+    If you specify an option, the specified conversion will be performed.
+
+    Tosaka mode is a mode that expresses the preference of
+    Tosaka-senpai, a character in "Kyūkyoku Chōjin R",
+    for "photos taken with Tri-X that look like they were
+    burned onto No. 4 or No. 5 photographic paper".
+    Only use floating-point numbers when using this mode;
+    numbers around 2.4 will make it look right.
+
+    Args:
+        path (Optional[str], optional): Directory or File Path. Defaults to None.
+        overwrite (bool, optional): Overwrite original files. Defaults to False.
+        colorautoadjust (bool, optional): Use colorautoadjust algorithm. Defaults to False.
+        colorstretch (bool, optional): Use colorstretch algorithm. Defaults to False.
+        grayscale (bool, optional): Convert to grayscale. Defaults to False.
+        invert (bool, optional): Invert color. Defaults to False.
+        tosaka (Optional[float], optional): Use Tosaka mode. Defaults to None.
+        outputrgb (bool, optional): Outputs a monochrome image in RGB. Defaults to False.
+        noise (Optional[float], optional): Add Gaussian noise. Defaults to None.
+        version (bool, optional): Show version. Defaults to False.
+
+    Raises:
+        fire.core.FireError: Error that occurs when the specified string is not a path.
+
+    Returns:
+        str: Message.
+    """
+    if version:
+        return "Anshitsu version {}".format(__version__)
+    if path is None:
+        raise fire.core.FireError("No path specified!")
+    types = ("*.jpg", "*.JPG", "*.jpeg", "*.JPEG", "*.png", "*.PNG")
+    files_glob = []
+    return_path = ""
+    now_s = datetime.datetime.now()
+    output_dir = "anshitsu_out"
+    original_dir = "anshitsu_orig"
+    if os.path.isdir(path):
+        for type in types:
+            files_glob.extend(glob.glob(os.path.join(path, '**', type), recursive=True))
+        files_glob = [file for file in files_glob if not file.__contains__(output_dir)]
+        return_path = path
+
+        if len(files_glob) == 0:
+            raise fire.core.FireError(
+                "There are no JPEG or PNG files in this directory."
+            )
+    elif os.path.isfile(path):
+        files_glob.extend(glob.glob(path))
+        return_path = os.path.abspath(os.path.join(path, os.pardir))
+    else:
+        raise fire.core.FireError("A non-path string was passed.")
+    if overwrite is True:
+        os.makedirs(os.path.join(return_path, original_dir))
+    for i, file in enumerate(files_glob):
+        try:
+            image = Image.open(file)
+        except UnidentifiedImageError as e:
+            raise fire.core.FireError(e)
+        exif = image.getexif()
+        original_filename: str = os.path.split(file)[1]
+        extension = original_filename.split(".")[-1]
+        timestamp = now_s.strftime("%Y-%m-%d_%H-%M-%S")
+        if overwrite is True:
+            backup_filename = original_filename
+            image.save(os.path.join(
+                return_path,
+                original_dir,
+                backup_filename
+            ))
+            filename = os.path.join(return_path, re.sub(r"\.[^.]+$", "", original_filename) + ".png")
+            remove_file_list = [".jpg", ".JPG", ".jpeg", ".JPEG", ".PNG"]
+            for remove_file in remove_file_list:
+                remove_file_name = re.sub(r"\.[^.]+$", "", original_filename) + remove_file
+                remove_file_path = os.path.join(return_path, remove_file_name)
+                if os.path.isfile(remove_file_path):
+                    os.remove(remove_file_path)
+        else:
+            filename = os.path.join(
+                return_path,
+                output_dir,
+                re.sub(r"\.[^.]+$", "_", original_filename)
+                + "_{0}_converted_at_{1}.png".format(extension, timestamp))
+        retouch = Retouch(
+            image=image,
+            colorautoadjust=colorautoadjust,
+            colorstretch=colorstretch,
+            grayscale=grayscale,
+            invert=invert,
+            tosaka=tosaka,
+            outputrgb=outputrgb,
+            noise=noise,
+        )
+        saved_image = retouch.process()
+        os.makedirs(os.path.join(return_path, output_dir), exist_ok=True)
+        saved_image.save(
+            filename,
+            quality=100,  # Specify 100 as the highest image quality
+            subsampling=0,
+            exif=exif,
+        )
+        print("{0}/{1} done!".format((i + 1), str(len(files_glob))))
+
+    return "The process was completed successfully."

anshitsu/retouch.py

@@ -0,0 +1,236 @@
+from collections import namedtuple
+from typing import Optional, Tuple
+
+import colorcorrect.algorithm as cca
+import numpy as np
+from colorcorrect.util import from_pil, to_pil
+from PIL import Image, ImageChops, ImageEnhance, ImageOps
+
+
+class Retouch:
+    """
+    Perform retouching.
+
+    Passing an image and options to the constructor will convert the specified image.
+    """
+
+    RGB_RED_VALUE: int = 255
+    RGB_GREEN_VALUE: int = 255
+    RGB_BLUE_VALUE: int = 255
+    RGB_WHITE_COLOR: Tuple[int, int, int] = (
+        RGB_RED_VALUE,
+        RGB_GREEN_VALUE,
+        RGB_BLUE_VALUE,
+    )
+
+    def __init__(
+        self,
+        image: Image,
+        colorautoadjust: bool = False,
+        colorstretch: bool = False,
+        grayscale: bool = False,
+        invert: bool = False,
+        tosaka: Optional[float] = None,
+        outputrgb: bool = False,
+        noise: Optional[float] = None,
+    ) -> None:
+        """
+        __init__ constructor.
+
+        Args:
+            image (Image): Image file.
+            colorautoadjust (bool, optional): Use colorautoadjust algorithm. Defaults to False.
+            colorstretch (bool, optional): Use colorstretch algorithm. Defaults to False.
+            grayscale (bool, optional): Convert to grayscale. Defaults to False.
+            invert (bool, optional): Invert color. Defaults to False.
+            tosaka (Optional[float], optional): Use Tosaka mode. Defaults to None.
+            outputrgb (bool, optional): Outputs a monochrome image in RGB. Defaults to False.
+            noise (Optional[float], optional): Add Gaussian noise. Defaults to None.
+        """
+        self.image = image
+        self.colorautoadjust = colorautoadjust
+        self.colorstretch = colorstretch
+        self.grayscale = grayscale
+        self.invert = invert
+        self.tosaka = tosaka
+        self.output_rgb = outputrgb
+        self.noise = noise
+
+    def process(self) -> Image:
+        self.image = self.__rgba_convert()
+
+        if self.invert:
+            self.image = self.__invert()
+
+        if self.colorautoadjust:
+            self.image = self.__colorautoadjust()
+
+        if self.colorstretch:
+            self.image = self.__colorstretch()
+
+        if self.grayscale:
+            self.image = self.__grayscale()
+
+        if self.noise is not None:
+            self.image = self.__noise()
+
+        if self.tosaka is not None:
+            self.image = self.__tosaka()
+
+        if self.output_rgb:
+            self.image = self.__output_rgb()
+
+        return self.image
+
+    def __colorautoadjust(self) -> Image:
+        """
+        __colorautoadjust
+
+        Use colorautoadjust algorithm.
+
+        Returns:
+            Image: processed image.
+        """
+        if self.image.mode == "L":
+            return self.image
+        return to_pil(cca.automatic_color_equalization(from_pil(self.image)))
+
+    def __colorstretch(self) -> Image:
+        """
+        __colorstretch
+
+        Use colorstretch algorithm.
+
+        Returns:
+            Image: processed image.
+        """
+        if self.image.mode == "L":
+            return self.image
+        return to_pil(cca.stretch(cca.grey_world(from_pil(self.image))))
+
+    def __noise(self) -> Image:
+        """
+        __noise
+
+        Add Gaussian noise.
+        To add noise, you need to specify floating-point numbers;
+        a value of about 10.0 will be just right.
+
+        Returns:
+            Image: processed image.
+        """
+        table = [x * 2 for x in range(256)] * len(self.image.getbands())
+        if self.image.mode == "RGB":
+            noise_image = Image.effect_noise(
+                (self.image.width, self.image.height), self.noise
+            ).convert("RGB")
+            self.image = ImageChops.multiply(self.image, noise_image).point(table)
+        if self.image.mode == "L":
+            noise_image = Image.effect_noise(
+                (self.image.width, self.image.height), self.noise
+            )
+            self.image = ImageChops.multiply(self.image, noise_image).point(table)
+        return self.image
+
+    def __grayscale(self) -> Image:
+        """
+        __grayscale
+
+        Converts to grayscale based on the luminance in the CIE XYZ color space.
+
+        Returns:
+            Image: processed image.
+        """
+        if self.image.mode == "L":
+            return self.image
+
+        rgb = np.array(self.image, dtype="float32")
+
+        GAMMA = 2.2
+        ColorMatrix = namedtuple("ColorMatrix", ("red", "green", "blue"))
+
+        # Coefficients for luminance in CIE XYZ color space
+        CIE_XYZ = ColorMatrix(0.2126, 0.7152, 0.0722)
+
+        # Inverse Gamma Correction
+        rgbL = pow(rgb / 255.0, GAMMA)
+
+        # Extract RGB values
+        r, g, b = rgbL[:, :, 0], rgbL[:, :, 1], rgbL[:, :, 2]
+
+        # Convert to grayscale
+        grayL = CIE_XYZ.red * r + CIE_XYZ.green * g + CIE_XYZ.blue * b
+
+        # gamma correction
+        gray = pow(grayL, 1.0 / GAMMA) * 255
+
+        self.image = Image.fromarray(gray.astype("uint8"))
+
+        return self.image
+
+    def __invert(self) -> Image:
+        """
+        __invert
+
+        Invert color.
+
+        Returns:
+            Image: processed image.
+        """
+        return ImageOps.invert(self.image)
+
+    def __tosaka(self) -> Image:
+        """
+        __tosaka
+
+        Use Tosaka mode.
+
+        Tosaka mode is a mode that expresses the preference of
+        Tosaka-senpai, a character in "Kyūkyoku Chōjin R",
+        for "photos taken with Tri-X that look like they were
+        burned onto No. 4 or No. 5 photographic paper".
+        Only use floating-point numbers when using this mode;
+        numbers around 2.4 will make it look right.
+
+        Returns:
+            Image: processed image.
+        """
+        if self.image.mode != "L":
+            self.image = self.__grayscale()
+        imageC = ImageEnhance.Contrast(self.image)
+        self.image = imageC.enhance(self.tosaka)
+        return self.image
+
+    def __rgba_convert(self) -> Image:
+        """
+        __rgba_convert
+
+        Converts image data that contains transparency to image data that does not contain transparency.
+
+        Returns:
+            Image: processed image.
+        """
+        if self.image.mode == "RGBA":
+            self.image.load()
+            background = Image.new("RGB", self.image.size, self.RGB_WHITE_COLOR)
+            background.paste(self.image, mask=self.image.split()[3])
+            self.image = background
+        if self.image.mode == "LA":
+            self.image.load()
+            background = Image.new("L", self.image.size, 255)
+            background.paste(self.image, mask=self.image.split()[1])
+            self.image = background
+        return self.image
+
+    def __output_rgb(self) -> Image:
+        """
+        __output_rgb
+
+        Outputs a monochrome image in RGB.
+
+        Returns:
+            Image: processed image.
+        """
+        if self.image.mode == "L":
+            self.image = self.image.convert("RGB")
+        return self.image

anshitsu-0.0.0.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 Iosif Takakura
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

anshitsu-0.0.0.dist-info/METADATA

@@ -0,0 +1,183 @@
+Metadata-Version: 2.1
+Name: anshitsu
+Version: 0.0.0
+Summary: A tiny digital photographic utility.
+Home-page: https://github.com/huideyeren
+License: MIT
+Author: Iosif Takakura
+Author-email: iosif@huideyeren.info
+Requires-Python: >=3.10,<3.13
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Dist: Pillow (>=10.1.0,<11.0.0)
+Requires-Dist: colorcorrect (>=0.9.1,<0.10.0)
+Requires-Dist: fire (>=0.5.0,<0.6.0)
+Requires-Dist: numpy (>=1.26.0,<2.0.0)
+Project-URL: Documentation, https://github.com/huideyeren/anshitsu
+Project-URL: Repository, https://github.com/huideyeren/anshitsu
+Description-Content-Type: text/markdown
+
+# Anshitsu
+
+[![Testing](https://github.com/huideyeren/anshitsu/actions/workflows/testing.yml/badge.svg)](https://github.com/huideyeren/anshitsu/actions/workflows/testing.yml)
+
+[![codecov](https://codecov.io/gh/huideyeren/anshitsu/branch/main/graph/badge.svg?token=ZYRX8NBTLQ)](https://codecov.io/gh/huideyeren/anshitsu)
+
+A tiny digital photographic utility.
+
+"Anshitsu" means a darkroom in Japanese.
+
+## Install
+
+Run this command in an environment where Python 3.10 or higher is installed.
+
+We have tested it on Windows, Mac, and Ubuntu on GitHub Actions, but we have not tested it on Macs with Apple Silicon, so please use it at your own risk on Macs with Apple Silicon.
+
+``` shell
+pip install anshitsu
+```
+
+## Usage
+
+It is as described in the following help.
+
+``` shell
+INFO: Showing help with the command 'anshitsu -- --help'.
+
+NAME
+    anshitsu - Process Runnner for Command Line Interface
+
+SYNOPSIS
+    anshitsu PATH <flags>
+
+DESCRIPTION
+    This utility converts the colors of images such as photos.
+
+    If you specify a directory path, it will convert
+    the image files in the specified directory.
+    If you specify a file path, it will convert the specified file.
+    If you specify an option, the specified conversion will be performed.
+
+    Tosaka mode is a mode that expresses the preference of
+    Tosaka-senpai, a character in "Kyūkyoku Chōjin R",
+    for "photos taken with Tri-X that look like they were
+    burned onto No. 4 or No. 5 photographic paper".
+    Only use floating-point numbers when using this mode;
+    numbers around 2.4 will make it look right.
+
+POSITIONAL ARGUMENTS
+    PATH
+        Type: str
+        Directory or File Path
+
+FLAGS
+    --colorautoadjust=COLORAUTOADJUST
+        Type: bool
+        Default: False
+        Use colorautoadjust algorithm. Defaults to False.
+    --colorstretch=COLORSTRETCH
+        Type: bool
+        Default: False
+        Use colorstretch algorithm. Defaults to False.
+    --grayscale=GRAYSCALE
+        Type: bool
+        Default: False
+        Convert to grayscale. Defaults to False.
+    --invert=INVERT
+        Type: bool
+        Default: False
+        Invert color. Defaults to False.
+    --tosaka=TOSAKA
+        Type: Optional[Optional]
+        Default: None
+        Use Tosaka mode. Defaults to None.
+    --outputrgb=OUTPUTRGB
+        Type: bool
+        Default: False
+        Outputs a monochrome image in RGB. Defaults to False.
+    --noise=NOISE
+        Type: Optional[Optional]
+        Default: None
+        Add Gaussian noise. Defaults to None.
+
+NOTES
+    You can also use flags syntax for POSITIONAL ARGUMENTS
+```
+
+If a directory is specified in the path, an `out` directory will be created in the specified directory, and the converted JPEG and PNG images will be stored in PNG format.
+
+If you specify a JPEG or PNG image file as the path, an `out` directory will be created in the directory where the image is stored, and the converted image will be stored in PNG format.
+
+**Note:** If you specify a file in any other format in the path, be aware there is no error handling. The program will terminate abnormally.
+
+## Algorithms
+
+The following algorithms are available in this tool.
+
+### RGBA to RGB Convert
+
+Converts an image that contains Alpha, such as RGBA, to image data that does not contain Alpha.
+Transparent areas will be filled with white.
+
+This algorithm is performed on any image file.
+
+### invert
+
+Inverts the colors of an image using Pillow's built-in algorithm.
+
+In the case of negative film, color conversion that takes into account the film base color is not performed, but we plan to follow up with a feature to be developed in the future.
+
+### colorautoajust
+
+We will use the "automatic color equalization" algorithm described in the following paper to apply color correction.
+
+This process is more time consuming than the algorithm used in "colorstretch", but it can reproduce more natural colors.
+
+(References)
+
+A. Rizzi, C. Gatta and D. Marini, "A new algorithm for unsupervised global and local color correction.", Pattern Recognition Letters, vol. 24, no. 11, 2003.
+
+### colorstretch
+
+The "gray world" and "stretch" algorithms described in the following paper are combined to apply color correction.
+
+This process is faster than the algorithm used in "colorautoajust".
+
+(References)
+
+D. Nikitenko, M. Wirth and K. Trudel, "Applicability Of White-Balancing Algorithms to Restoring Faded Colour Slides: An Empirical Evaluation.", Journal of Multimedia, vol. 3, no. 5, 2008.
+
+### grayscale
+
+Convert a color image to grayscale using the algorithm described in the following article.
+
+[Python でグレースケール(grayscale)化](https://qiita.com/yoya/items/dba7c40b31f832e9bc2a#pilpillow-%E3%81%A7%E3%82%B0%E3%83%AC%E3%83%BC%E3%82%B9%E3%82%B1%E3%83%BC%E3%83%AB%E5%8C%96-numpy-%E3%81%A7%E4%BD%8E%E8%BC%9D%E5%BA%A6%E5%AF%BE%E5%BF%9C)
+
+Note: This article is written in Japanese.
+
+### Tosaka mode
+
+Tosaka mode is a mode that expresses the preference of Tosaka-senpai, a character in "Kyūkyoku Chōjin R", for "photos taken with Tri-X that look like they were burned onto No. 4 or No. 5 photographic paper".
+
+Only use floating-point numbers when using this mode; numbers around 2.4 will make it look right.
+
+When this mode is specified, color images will also be converted to grayscale.
+
+### outputrgb
+
+Outputs a monochrome image in RGB.
+
+### noise
+
+Add Gaussian noise.
+
+To add noise, you need to specify a floating-point number; a value of about 10.0 will be just right.
+
+## Special Thanks
+
+We are using the following libraries.
+
+[shunsukeaihara/colorcorrect](https://github.com/shunsukeaihara/colorcorrect)
+

anshitsu-0.0.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+anshitsu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+anshitsu/__version__.py,sha256=jO-0XZe2LK4sxY9Jy21OX9LMsd02dSx3yjI9we0PSY4,24
+anshitsu/cli.py,sha256=yY16URQm-ndXBW0QpcivkeLr7J3kNOJIXy4QKY9_PcM,162
+anshitsu/process.py,sha256=QeMRoqz9LF40Rc9lr1P-hdti8oUec2cQrWdwRkA-nIk,5413
+anshitsu/retouch.py,sha256=gbMYITD4RmBoIHClFtIcJF9kGpWLCE4EwducVVIU6GA,6982
+anshitsu-0.0.0.dist-info/LICENSE,sha256=Hfj88p2Eh9As-ZvaGoE02XkqFlmKks71SnjV-dUpE1k,1071
+anshitsu-0.0.0.dist-info/METADATA,sha256=AVheXk7yTzB5WyfSL2DxuRonLHRQbTdz6bEkxnvrXrg,6389
+anshitsu-0.0.0.dist-info/WHEEL,sha256=d2fvjOD7sXsVzChCqf0Ty0JbHKBaLYwDbGQDwQTnJ50,88
+anshitsu-0.0.0.dist-info/entry_points.txt,sha256=vMtbdj3zt7D9SEzY-icd22vCq9WGIm3wIYBbrB5RuSs,46
+anshitsu-0.0.0.dist-info/RECORD,,

anshitsu-0.0.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 1.7.0
+Root-Is-Purelib: true
+Tag: py3-none-any

anshitsu-0.0.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+anshitsu=anshitsu.cli:main
+