Copyright (c) 2014 Lycaon (lycaon.me)

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.

from PIL import Image import random, sys

def quick_sort(pixels): #Quicksort function that sorts pixels based on combined RGB values (R + B + G) if pixels == []: return pixels

else:
	pivot = pixels[0]
	lesser = quick_sort([x for x in pixels[1:] if (x[0] + x[1] + x[2]) < (pivot[0] + pivot[1] + pivot[2])])
	greater = quick_sort([x for x in pixels[1:] if (x[0] + x[1] + x[2]) >= (pivot[0] + pivot[1] + pivot[2])])
	return lesser + [pivot] + greater

def sort_all_pixels(image): #sorts every line of pixels print("Sorting all pixels.")

print("Opening image...")
img = Image.open(image)
img = img.convert('RGBA')
print("Get data...")
data = img.load()

new = Image.new('RGBA', img.size)

pixels = []
sorted_pixels = []
print("Getting pixels...")
#Load all of the pixels into the pixels list
for y in range(img.size[1]):
	pixels.append([])
	for x in range(img.size[0]):
		pixels[y].append(data[x, y])

print("Quicksorting pixels...")
for y in range(img.size[1]):
		sorted_pixels.append(quick_sort(pixels[y]))

print("Placing pixels...")
for y in range(img.size[1]):
	for x in range(img.size[0]):
		new.putpixel((x, y), sorted_pixels[y][x]) #apply the pixels to the new image

print("Saving image...")
new.save('output-sortall.png')

def random_sort_pixels(image, intensity): #sorts pixels in a random fashion

if intensity > 100:
	intensity = 100

print("Sorting " + image + " with intensity " + str(intensity))

#Open the image, convert it to RGBA, get the pixels 
print("Opening image...")
img = Image.open(image)
img = img.convert('RGBA')
print("Get data...")
data = img.load()

new = Image.new('RGBA', img.size)

pixels = []
sorted_pixels = []
print("Getting pixels...")
#Load all of the pixels into the pixels list
for y in range(img.size[1]):
	pixels.append([])
	for x in range(img.size[0]):
		pixels[y].append(data[x, y])

print("Quicksorting pixels...")
for y in range(img.size[1]):
	#pick different starting points for each line
	if(random.randint(0, 100) > intensity):
		sorted_pixels.append(pixels[y]) #Don't sort this line of pixels
	else:
		minsort = random.randint(3, len(pixels[y]) - 3) #pick the start of the sorted area on this pixel line
		maxsort = random.randint(minsort, len(pixels[y]) - 1)# pick the end of the sorted area on this pixel line
		sort = []
		for x in range(minsort, maxsort):
			sort.append(pixels[y][x]) 

		sort = quick_sort(sort) #sort the pixels by brightness

		i = 0
		for x in range(minsort, maxsort):
			pixels[y][x] = sort[i]
			i = i + 1

		sorted_pixels.append(pixels[y])


print("Placing pixels...")
for y in range(img.size[1]):
	for x in range(img.size[0]):
		new.putpixel((x, y), sorted_pixels[y][x]) #apply the pixels to the new image

print("Saving image...")
new.save('output-sortrandom.png')

def sort_pixels_pivot(image): print ("Sorting pixels on pivot.")

#Open the image, convert it to RGBA, get the pixels 
print("Opening image...")
img = Image.open(image)
img = img.convert('RGBA')
print("Get data...")
data = img.load()

new = Image.new('RGBA', img.size)

pixels = []
sorted_pixels = []
print("Getting pixels...")
#Load all of the pixels into the pixels list
for y in range(img.size[1]):
	pixels.append([])
	for x in range(img.size[0]):
		pixels[y].append(data[x, y])

print("Quicksorting pixels...")

minsort = random.randint(3, img.size[0] - 3) #get sorting pivot
for y in range(img.size[1]):
	maxsort = random.randint(minsort, len(pixels[y]) - 1) #pick the end of the sorted area on this pixel line
	sort = []
	for x in range(minsort, maxsort):
		sort.append(pixels[y][x])
	sort = quick_sort(sort) #sort the pixels by brightness

	i = 0
	for x in range(minsort, maxsort):
		pixels[y][x] = sort[i]
		i = i + 1

	sorted_pixels.append(pixels[y])


print("Placing pixels...")
for y in range(img.size[1]):
	for x in range(img.size[0]):
		new.putpixel((x, y), sorted_pixels[y][x]) #apply the pixels to the new image

print("Saving image...")
new.save('output-sortpivot.png')

sys.setrecursionlimit(10000) #Increase the recursion depth limit. Without this, the script fails on larger images because quicksort recurses too much. sort_all_pixels("image.jpg") random_sort_pixels("image.jpg", random.randint(1, 100)) sort_pixels_pivot("image.jpg")