mirror of https://github.com/ivanch/tcc.git
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
|
from wand.image import Image
|
||
|
|
|
||
|
|
|
||
|
|
def box_blur_image_separable(image, blurred_image, radius_x, radius_y):
|
||
|
|
pixels = image.get_pixels()
|
||
|
|
blurred_pixels = blurred_image.get_pixels()
|
||
|
|
|
||
|
|
for pixel in pixels:
|
||
|
|
x, y = pixel[0], pixel[1]
|
||
|
|
|
||
|
|
r_total, g_total, b_total = 0, 0, 0
|
||
|
|
pixel_count = 0
|
||
|
|
for offset_y in range(-radius_y, radius_y + 1):
|
||
|
|
for offset_x in range(-radius_x, radius_x + 1):
|
||
|
|
new_x = x + offset_x
|
||
|
|
new_y = y + offset_y
|
||
|
|
|
||
|
|
if 0 <= new_x < image.width and 0 <= new_y < image.height:
|
||
|
|
pixel_color = pixels[new_y * image.width + new_x]
|
||
|
|
r_total += pixel_color.red
|
||
|
|
g_total += pixel_color.green
|
||
|
|
b_total += pixel_color.blue
|
||
|
|
pixel_count += 1
|
||
|
|
|
||
|
|
blurred_pixel = blurred_pixels[y * image.width + x]
|
||
|
|
blurred_pixel.red = int(r_total / pixel_count)
|
||
|
|
blurred_pixel.green = int(g_total / pixel_count)
|
||
|
|
blurred_pixel.blue = int(b_total / pixel_count)
|
||
|
|
|
||
|
|
return blurred_image
|
||
|
|
|
||
|
|
|
||
|
|
class ImageService:
|
||
|
|
def __init__(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def box_blur_image(self, image, radius):
|
||
|
|
temp_path = 'temp_image.png'
|
||
|
|
image.save(temp_path)
|
||
|
|
|
||
|
|
with Image(filename=temp_path) as img:
|
||
|
|
img.blur(radius, 2) # Apply blur directly to the image object
|
||
|
|
blurred_temp_path = 'blurred_temp_image.png'
|
||
|
|
img.save(filename=blurred_temp_path) # Save the modified image
|
||
|
|
return blurred_temp_path
|
||
|
|
|
||
|
|
|
||
|
|
def save_image(self, file_stream):
|
||
|
|
with open("image.png", "wb") as file:
|
||
|
|
file.write(file_stream.read())
|
||
|
|
file.close()
|
||
|
|
|
||
|
|
def get_simple_image(self):
|
||
|
|
return Image(filename='path_to_simple_image.png')
|
||
|
|
|
||
|
|
def get_big_image(self):
|
||
|
|
return Image(filename='path_to_big_image.png')
|