Add files via upload

Author: JingShing

script/bayer_dithering.py

@@ -54,7 +54,7 @@ def process_dither_color(img, matrix):
             else:
                 v = 0
 
-            img[row][col] = v    
+            img[row][col] = v
 
 def dither_color_image(image, matrix):
     blue=image[:,:,0]  #taking the blue channel
@@ -63,4 +63,70 @@ def dither_color_image(image, matrix):
     process_dither_color(green, matrix)
     red=image[:,:,2]
     process_dither_color(red, matrix)
-    return cv2.merge((blue, green, red))
+    return cv2.merge((blue, green, red))
+
+def detect_board_color(img, y, x, allow_value, mode, select_color_value=None):
+    height = img.shape[0]
+    width = img.shape[1]
+    self_color = img[y][x]
+    if select_color_value:
+        if self_color >= select_color_value[0] and self_color <= select_color_value[1]:
+            pass
+        else:
+            return False
+    if mode == 0:
+        x_dir = [1, 1, 1, 1, -1, -1, -1, -1]
+        y_dir = [1, -1, 1, -1, 1, -1, 1, -1]
+        for i in range(8):
+            new_y = y + y_dir[i]
+            new_x = x + x_dir[i]
+            new_y = clamp(new_y, height-1, 0)
+            new_x = clamp(new_x, width-1, 0)
+            new_color = img[new_y][new_x]
+            if self_color >= new_color+allow_value or self_color <= new_color-allow_value:
+                return True
+        return False
+    elif mode == 1:
+        x_dir = [1, 1, -1, -1]
+        y_dir = [1, -1, 1, -1]
+        for i in range(4):
+            new_y = y + y_dir[i]
+            new_x = x + x_dir[i]
+            new_y = clamp(new_y, height-1, 0)
+            new_x = clamp(new_x, width-1, 0)
+            new_color = img[new_y][new_x]
+            if self_color >= new_color+allow_value or self_color <= new_color-allow_value:
+                return True
+        return False
+        
+def clamp(value, max_num, min_num):
+    value = max(value, min_num)
+    value = min(value, max_num)
+    # value = value % max_num
+    return value
+
+def process_dither_color_fixed_8_dir(img, matrix, allow_value, mode, select_color_value):
+    or_img = img.copy()
+    height = img.shape[0]
+    width = img.shape[1]
+    matrix_len = len(matrix) - 1
+    for row in range(height):
+        for col in range(width):
+            if detect_board_color(or_img, row, col, allow_value, mode, select_color_value):
+                color = img[row][col]
+                
+                if color >= (color>>2) > matrix[row&matrix_len][col&matrix_len]:
+                    v = 255
+                else:
+                    v = 0
+
+                img[row][col] = v
+
+def dither_color_image_fix(image, matrix, allow_value, mode, select_color_value=None):
+    blue=image[:,:,0]  #taking the blue channel
+    process_dither_color_fixed_8_dir(blue, matrix, allow_value, mode, select_color_value)
+    green=image[:,:,1]
+    process_dither_color_fixed_8_dir(green, matrix, allow_value, mode, select_color_value)
+    red=image[:,:,2]
+    process_dither_color_fixed_8_dir(red, matrix, allow_value, mode, select_color_value)
+    return cv2.merge((blue, green, red))

script/contrast_and_saturation.py

@@ -0,0 +1,42 @@
+import cv2
+import numpy
+import math
+
+def contrast_and_brightness(img, brightness=0 , contrast=100):
+    B = brightness / 255.0
+    c = contrast / 255.0 
+    k = math.tan((45 + 44 * c) / 180 * math.pi)
+
+    img = (img - 127.5 * (1 - B)) * k + 127.5 * (1 + B)
+      
+    # values between 0-255
+    img = numpy.clip(img, 0, 255).astype(numpy.uint8)
+    return img
+    
+def saturation_and_lightness(img, lightness, saturation):
+
+    # origin_img = img
+
+    # astype image and turn to float
+    fImg = img.astype(numpy.float32)
+    fImg = fImg / 255.0
+
+    # color space exchange BGR -> HLS
+    hlsImg = cv2.cvtColor(fImg, cv2.COLOR_BGR2HLS)
+    hlsCopy = numpy.copy(hlsImg)
+
+    # lightness = 0 # lightness edited to "1 +/- n %"
+    # saturation = 300 # saturation edited to "1 +/- n %"
+
+    # brightness
+    hlsCopy[:, :, 1] = (1 + lightness / 100.0) * hlsCopy[:, :, 1]
+    hlsCopy[:, :, 1][hlsCopy[:, :, 1] > 1] = 1  # Between 0-1
+
+    # saturation
+    hlsCopy[:, :, 2] = (1 + saturation / 100.0) * hlsCopy[:, :, 2]
+    hlsCopy[:, :, 2][hlsCopy[:, :, 2] > 1] = 1  # Between 0-1
+
+    # color space exchange back HLS -> BGR 
+    result_img = cv2.cvtColor(hlsCopy, cv2.COLOR_HLS2BGR)
+    result_img = ((result_img * 255).astype(numpy.uint8))
+    return result_img

script/display_image.py

@@ -21,7 +21,6 @@ def display_screen_resize(self, width=10, height=10, mode=0):
         elif mode == 1:
             self.display_screen = pygame.display.set_mode(self.image.get_size())
 
-
     def cv_to_pygame(self, cv_img):
         frame = cv2.cvtColor(cv_img, cv2.COLOR_BGR2RGB)
         frame = cv2.transpose(frame)

script/gui_helper.py

@@ -14,74 +14,99 @@ def __init__(self, display):
         self.image_path = resource_path(sample_image_path)
         self.cv_image = cv2.imread(self.image_path)
 
-        self.root.title('pixel art styler')
-        self.root.geometry('300x300')
+        self.root.title('pixel art styler像素風格濾鏡工具')
+        self.root.geometry('500x300')
 
         self.load_file_name = StringVar()
-        self.load_file_name.set('no file')
+        self.load_file_name.set('no file無檔案')
         self.now_file_name = Label(self.root,textvariable=self.load_file_name)
         self.now_file_name.grid(column=0, row=0, sticky=N+W)
 
-        self.import_img_btn = Button(self.root, text='import img', command=self.import_img)
+        self.import_img_btn = Button(self.root, text='import img導入圖片', command=self.import_img)
         self.import_img_btn.grid(column=4, row=0, sticky=E+N)
 
         # color num
-        self.color_num_name = Label(self.root,text='Color nums')
+        self.color_num_name = Label(self.root,text='Color nums色數')
         self.color_num_name.grid(column=0, row=1, sticky=N+W)
         self.color_num_select_box = Combobox(self.root, width=5)
         self.color_num_select_box.grid(column=0, row=2, sticky=N+W)
         self.color_num_select_box['values'] = ['2', '4', '8', '16']
         self.color_num_select_box.current(0)
         # pixel size
-        self.pixel_size_name = Label(self.root,text='Pixel size')
-        self.pixel_size_name.grid(column=4, row=1, sticky=N+W)
+        self.pixel_size_name = Label(self.root,text='Pixel size像素尺寸')
+        self.pixel_size_name.grid(column=4, row=1, sticky=N+E)
         self.pixel_size_select_box = Combobox(self.root, width=5)
-        self.pixel_size_select_box.grid(column=4, row=2, sticky=N+W)
+        self.pixel_size_select_box.grid(column=4, row=2, sticky=N+E)
         self.pixel_size_select_box['values'] = ['1', '2', '3', '4']
         self.pixel_size_select_box.current(0)
         # smoothing
-        self.smoothing_name = Label(self.root,text='Smoothing')
+        self.smoothing_name = Label(self.root,text='Smoothing光滑')
         self.smoothing_name.grid(column=0, row=3, sticky=N+W)
         self.smoothing_select_box = Combobox(self.root, width=5)
         self.smoothing_select_box.grid(column=0, row=4, sticky=N+W)
         self.smoothing_select_box['values'] = ['None', 'Less', 'Great']
         self.smoothing_select_box.current(0)
         # outlines
-        self.outline_name = Label(self.root,text='Outline inflate')
-        self.outline_name.grid(column=4, row=3, sticky=N+W)
+        self.outline_name = Label(self.root,text='Outline輪廓')
+        self.outline_name.grid(column=4, row=3, sticky=N+E)
         self.outline_select_box = Combobox(self.root, width=5)
-        self.outline_select_box.grid(column=4, row=4, sticky=N+W)
+        self.outline_select_box.grid(column=4, row=4, sticky=N+E)
         self.outline_select_box['values'] = ['None', 'Less', 'Great']
         self.outline_select_box.current(0)
+        # saturation
+        self.saturation_name = Label(self.root,text='Saturation飽和')
+        self.saturation_name.grid(column=0, row=5, sticky=N+W)
+        self.saturation_select_box = Combobox(self.root, width=5)
+        self.saturation_select_box.grid(column=0, row=6, sticky=N+W)
+        self.saturation_select_box['values'] = ['-200', '-100', '0', '100', '200']
+        self.saturation_select_box.current(2)
+        # contrast
+        self.contrast_name = Label(self.root,text='Contrast對比')
+        self.contrast_name.grid(column=4, row=5, sticky=N+E)
+        self.contrast_select_box = Combobox(self.root, width=5)
+        self.contrast_select_box.grid(column=4, row=6, sticky=N+E)
+        self.contrast_select_box['values'] = ['-200', '-100', '0', '100', '200']
+        self.contrast_select_box.current(2)
 
         # dithering
         self.dithering_bool = BooleanVar()
-        self.dithering_check = Checkbutton(self.root, text='Dithering', variable=self.dithering_bool)
+        self.dithering_check = Checkbutton(self.root, text='Dithering顆粒抖動', variable=self.dithering_bool)
         self.dithering_check.grid(column=0, row=7, sticky=N+W)
 
         # transform button
-        self.transform_img_btn = Button(self.root, text='transform', command=self.transform_img)
+        self.transform_img_btn = Button(self.root, text='transform變換', command=self.transform_img)
         self.transform_img_btn.grid(column=0, row=8, sticky=W+N)
         # save_img button
-        self.save_img_btn = Button(self.root, text='save', command=self.save_img)
+        self.save_img_btn = Button(self.root, text='save保存', command=self.save_img)
         self.save_img_btn.grid(column=0, row=9, sticky=W+N)
 
     def import_img(self):
         image_path = filedialog.askopenfilename()
         if image_path:
             file_name = re.split('/|\.', image_path)[-2]
-            print(image_path)
+            # print(image_path)
             self.load_file_name.set(file_name)
             self.display.image_load(image_path)
             self.image_path = image_path
+            self.cv_image = cv2.imread(self.image_path)
 
     def transform_img(self):
         k = int(self.color_num_select_box.get())
         scale = int(self.pixel_size_select_box.get())
         blur = self.get_textbox_value(self.smoothing_select_box.get())
         erode = self.get_textbox_value(self.outline_select_box.get())
         dither = self.dithering_bool.get()
-        self.cv_image = convert(self.image_path, k=k, scale=scale, blur=blur, erode=erode, dither=dither)
+        saturation = int(self.saturation_select_box.get())
+        contrast = int(self.contrast_select_box.get())
+
+        self.cv_image = convert(self.image_path, 
+                                k=k, 
+                                scale=scale, 
+                                blur=blur, 
+                                erode=erode, 
+                                dither=dither, 
+                                saturation=saturation, 
+                                contrast=contrast)
         self.display.cv_to_pygame(self.cv_image)
 
     def get_textbox_value(self, text):

script/pixel_converter.py

@@ -4,16 +4,16 @@
 from pixel_transform import transform
 from settings import resource_path
 
-def convert(img_path, k=16, scale=1, blur=0, erode=0,alpha = True, to_tw = False, dither=False):
+def convert(img_path, k=16, scale=1, blur=0, erode=0,alpha = True, to_tw = False, dither=False, saturation=0, contrast=0):
     # k is color num
     # scale is pixel size
     # to_tw means to twitter
     img_path.replace("\\", "/")
     img_file_name = re.split("/", img_path)[-1]
-    img_res = transform(img_path, k=k, scale=scale, blur=blur, erode=erode, alpha=alpha, to_tw=to_tw, dither=dither)
+    img_res = transform(img_path, k=k, scale=scale, blur=blur, erode=erode, alpha=alpha, to_tw=to_tw, dither=dither, saturation=saturation, contrast=contrast)
     img_path = img_path.split(img_file_name)[0]
     return img_res
 
 def save_cv_img(img, img_path, img_name):
     result_path = os.path.join(img_path + img_name + '.png')
-    cv2.imwrite(result_path, img, [int(cv2.IMWRITE_PNG_COMPRESSION)])
+    cv2.imwrite(result_path, img, [cv2.IMWRITE_PNG_COMPRESSION, 9])