filter.py

import numpy as np
from PIL import Image


class GreyImage:
    """ Класс, преобразующий картинку в черно-белый пиксель-арт
            Тест значения градации изображения
            >>> res = GreyImage(np.array(Image.open('picture.jpg')), 10, 100)
            >>> res.grad
            2

            Тест размера блока
            >>> res = GreyImage(np.array(Image.open('picture.jpg')), 75, 10)
            >>> res.size
            75
    """
    def __init__(self, pix_array, pix_size=10, gradation=50):
        self.image = np.array(pix_array)
        self.size = pix_size
        self.grad = 255 // gradation
        self.width = len(self.image)
        self.height = len(self.image[0])

    def get_grey_image(self):
        """ Преобразование картинки в черно-белую, пиксельную
                Тест размера полученной картинки
                >>> GreyImage(np.array(Image.open('picture.jpg')), 10, 50).get_grey_image().size
                (1920, 1080)
                >>> GreyImage(np.array(Image.open('picture_2.jpg')), 10, 50).get_grey_image().size
                (750, 750)

                :return: черно-белая картинка
                """
        for x in range(0, self.width, self.size):
            for y in range(0, self.height, self.size):
                    self.image[x:x + self.size, y:y + self.size] = self.get_middle_color(x, y)
        return Image.fromarray(self.image)

    def get_middle_color(self, x, y):
        """
        Закрашивает блок пикселей одной градацией серого цвета и вычисляет среднее значение яркости блока
                :param get_middle_color: Поиск нужного цвета
                :param x: координата пикселя по оси x
                :param y: координата пикселя по оси y
                :return: Среднее значение яркости

               Тест исходной картинки
               >>> GreyImage(np.array(Image.open('picture.jpg')), 10, 50).get_middle_color(1,1)
               170

               Тест черно-белой картинки
               >>> GreyImage(np.array(Image.open('pics/new_pic.jpg')), 10, 50).get_middle_color(1,1)
               155
        """
        return int(self.image[x:x + self.size, y:y + self.size].sum() / 3 // self.size ** 2 // self.grad * self.grad)


flag = 'no'
while flag == 'no':
    print('Введите название картинки')
    orig_img = Image.open('{}'.format(input()))
    print('Выберите величину ячейки, на которую будут делить изображение\n' +
          'Размер должен быть инициализирован целым положительным числом')
    pic_size = int(input())
    print('Выберите количество оттенков\n' +
          'Количество должно быть инициализировано целым положительным числом')
    grad = int(input())
    arr = GreyImage(orig_img, pix_size=pic_size, gradation=grad).get_grey_image()
    print('Введите название новой картинки')
    name = input()
    print('Введите формат новой картинки(jpg, png и т.д.)')
    formt = input()
    arr.save('{}.{}'.format(name, formt))
    print('Если хотите завершить сессию, наберите \'yes\', если нет-нажмите enter')
    if input() == 'yes':
        flag = 'yes'