Skip to content

Commit

Permalink
more scripts!
Browse files Browse the repository at this point in the history
  • Loading branch information
@hearues-zueke-github
hearues-zueke-github committed Nov 26, 2018
1 parent b6d1899 commit fc51cc9
Show file tree
Hide file tree
Showing 6 changed files with 281 additions and 145 deletions.
1 change: 1 addition & 0 deletions .gitignore
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
*.dot
*.class

complex_numbers/images
labyrinth/images
picture_diff/images
picture_manipulation/images
Expand Down
Binary file added complex_numbers/Monospace.ttf
View file
Binary file not shown.
Binary file added complex_numbers/monofonto.ttf
View file
Binary file not shown.
285 changes: 181 additions & 104 deletions complex_numbers/show_2d_complex_function.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -47,23 +47,24 @@ def calculate_angle_x_y(x, y):

if __name__ == "__main__":
# all_symbols_64 = np.array(list(string.ascii_letters+string.digits+"-_"))
all_symbols_16 = np.array(list("0123456789ABCDEF"))
# all_symbols_16 = np.array(list("0123456789ABCDEF"))

def get_random_string(n):
# l = np.random.randint(0, 64, (n, ))
l = np.random.randint(0, 16, (n, ))
return "".join(all_symbols_16[l])
# def get_random_string(n):
# # l = np.random.randint(0, 64, (n, ))
# l = np.random.randint(0, 16, (n, ))
# return "".join(all_symbols_16[l])

path_image = "images/random_names/"

# rand_str = get_random_string(16)
# print("rand_str: {}".format(rand_str))
# path_image += rand_str+"/"
path_image = "images/"

if not os.path.exists(path_image):
os.makedirs(path_image)

func_str = "z**2-4*z+0.1*z*complex(z.imag, z.real*3)"
# func_str = "z**2-4*z+0.1*z*complex(z.imag, z.real*3)"
# func_str = "complex(np.sin(z.imag), z.real)*z"
func_str = "z+np.sin(z.real+z.imag)+complex(z.imag*np.sin(0.1*z.real), z*np.sin(z.imag+1))"
# func_str = "2*z**2+4*z*complex(np.sin(z.real), np.cos(z.imag)**2)"
# func_str = "z**3-2*z**2+4*z"
# func_str = "complex(np.sin(z.imag), z.real)*z+z*complex(z.real+z.imag*0.5, 2*np.cos(z.real))"
func_str_complete = "lambda z: "+func_str
def get_f(modulo, func_str_complete):
f_str = func_str_complete
Expand All @@ -82,9 +83,10 @@ def f_temp(z):

plt.close("all")

n1 = 500
scale = 40
x_offset = scale/2
n1 = 300
scale = 30
x_offset = 15
# x_offset = scale/2
scale_y = 1.
y_offset = scale/2*scale_y
n1_y = int(n1*scale_y)
Expand All @@ -97,8 +99,21 @@ def f_temp(z):
arr_complex = np.empty((ys1.shape[0], xs1.shape[0]), dtype=complex)
arr_complex_mod = np.empty((ys1.shape[0], xs1.shape[0]), dtype=complex)

# TODO: create the other magtrices too!

pix = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.uint8)
pix_hvs = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.float)
pix_mod = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.uint8)
pix_hvs_mod = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.float)

pix_angle = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.uint8)
pix_hvs_angle = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.float)
pix_abs = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.uint8)
pix_hvs_abs = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.float)
pix_x = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.uint8)
pix_hvs_x = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.float)
pix_y = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.uint8)
pix_hvs_y = np.zeros((ys1.shape[0], xs1.shape[0], 3), dtype=np.float)

print("xs1: {}".format(xs1))
print("ys1: {}".format(ys1))
Expand All @@ -109,9 +124,11 @@ def f_temp(z):
z, scale_z = f(complex(x, y))
# z, scale_z = f(complex(x, y), modulo)
arr_complex[yi, xi] = z
arr_complex_mod[yi, xi] = z*scale_z
z_scale = z*scale_z
arr_complex_mod[yi, xi] = z_scale

zx, zy = z.real, z.imag
zx_s, zy_s = z_scale.real, z_scale.imag
if zx == 0. and zy == 0.:
continue

Expand All @@ -120,120 +137,180 @@ def f_temp(z):
h = alpha/(np.pi*2)
h = 1. if h > 1. else h

pix_hvs[yi, xi] = (h, np.sqrt(zx**2+zy**2), 1.)
a = np.sqrt(zx**2+zy**2)
a_s = np.sqrt(zx_s**2+zy_s**2)

pix_hvs[yi, xi] = (h, a, 1.)
pix_hvs_mod[yi, xi] = (h, a_s, 1.)

pix_hvs_angle[yi, xi] = (1., h, 1.)
pix_hvs_abs[yi, xi] = (1., a, 1.)
pix_hvs_x[yi, xi] = (1., z.real, 1.)
pix_hvs_y[yi, xi] = (1., z.imag, 1.)

# pix_hvs_angle[yi, xi] = (h, 1., 1.)
# pix_hvs_abs[yi, xi] = (a, 1., 1.)
# pix_hvs_x[yi, xi] = (z.real, 1., 1.)
# pix_hvs_y[yi, xi] = (z.imag, 1., 1.)

# pix_hvs[:, :, 1] /= np.max(pix_hvs[:, :, 1])

vals = pix_hvs[:, :, 1].reshape((-1, ))
def normalize_v_in_hvs(pix, pix_hvs, col=1):
vals = pix_hvs[:, :, col].reshape((-1, ))

combo = np.empty((0, vals.shape[0]), dtype=np.object)
combo = np.vstack((combo, np.arange(0, vals.shape[0])))
combo = np.vstack((combo, vals)).T

combo = np.empty((0, vals.shape[0]), dtype=np.object)
combo = np.vstack((combo, np.arange(0, vals.shape[0])))
combo = np.vstack((combo, vals)).T
combo = combo[combo[:, 1].argsort()]
combo[:, 1] = combo[:, 1]-combo[0, 1]

combo = combo[combo[:, 1].argsort()]
combo[:, 1] = (lambda x: x/x[-1])(combo[:, 1]**(1/4))/2+0.5
vals[combo[:, 0].astype(np.int)] = combo[:, 1]

combo[:, 1] = (lambda x: x/x[-1])(combo[:, 1]**(1/4))/2+0.5
vals[combo[:, 0].astype(np.int)] = combo[:, 1]
print("combo.shape: {}".format(combo.shape))
print("combo.dtype: {}".format(combo.dtype))

print("combo.shape: {}".format(combo.shape))
print("combo.dtype: {}".format(combo.dtype))
vals_sort = np.sort(vals)
vals_1 = vals_sort
vals_2 = vals_1/vals_1[-1]
x_vals = np.arange(0, vals.shape[0])

vals_sort = np.sort(vals)
vals_1 = vals_sort#**(1/4)
vals_2 = vals_1/vals_1[-1]
x_vals = np.arange(0, vals.shape[0])
for y in range(0, pix.shape[0]):
print("yi2: {}".format(y))
for x in range(0, pix.shape[1]):
args = pix_hvs[y, x]
c = colorsys.hsv_to_rgb(*args)
c_uint8 = (np.array(c)*255.49).astype(np.uint8)
pix[y, x] = c_uint8

for y in range(0, pix.shape[0]):
print("yi2: {}".format(y))
for x in range(0, pix.shape[1]):
args = pix_hvs[y, x]
c = colorsys.hsv_to_rgb(*args)
c_uint8 = (np.array(c)*255.49).astype(np.uint8)
pix[y, x] = c_uint8
normalize_v_in_hvs(pix, pix_hvs, col=1)
normalize_v_in_hvs(pix_mod, pix_hvs_mod, col=1)

print("np.min(pix_hvs[:, :, 0]): {}".format(np.min(pix_hvs[:, :, 0])))
print("np.max(pix_hvs[:, :, 0]): {}".format(np.max(pix_hvs[:, :, 0])))
normalize_v_in_hvs(pix_angle, pix_hvs_angle, col=1)
normalize_v_in_hvs(pix_abs, pix_hvs_abs, col=1)
normalize_v_in_hvs(pix_x, pix_hvs_x, col=1)
normalize_v_in_hvs(pix_y, pix_hvs_y, col=1)

print("\nnp.min(pix_hvs[:, :, 1]): {}".format(np.min(pix_hvs[:, :, 1])))
print("np.max(pix_hvs[:, :, 1]): {}".format(np.max(pix_hvs[:, :, 1])))
# print("np.min(pix_hvs[:, :, 0]): {}".format(np.min(pix_hvs[:, :, 0])))
# print("np.max(pix_hvs[:, :, 0]): {}".format(np.max(pix_hvs[:, :, 0])))

print("\nnp.min(pix_hvs[:, :, 2]): {}".format(np.min(pix_hvs[:, :, 2])))
print("np.max(pix_hvs[:, :, 2]): {}".format(np.max(pix_hvs[:, :, 2])))
# print("\nnp.min(pix_hvs[:, :, 1]): {}".format(np.min(pix_hvs[:, :, 1])))
# print("np.max(pix_hvs[:, :, 1]): {}".format(np.max(pix_hvs[:, :, 1])))

# print("\nnp.min(pix_hvs[:, :, 2]): {}".format(np.min(pix_hvs[:, :, 2])))
# print("np.max(pix_hvs[:, :, 2]): {}".format(np.max(pix_hvs[:, :, 2])))

pix2 = pix.copy()[:, :300]
pix2[:] = 0

# img = Image.fromarray(pix)
img2 = Image.fromarray(pix2)

try:
# get a font
fnt = ImageFont.truetype('monofonto.ttf', 16)
d = ImageDraw.Draw(img2)
d.text((8, 8), "Used function:", font=fnt, fill=(255, 255, 255))
func_str_split = [func_str_complete[30*i:30*(i+1)] for i in range(0, len(func_str_complete)//30+1)]

for i, func_str_part in enumerate(func_str_split, 1):
d.text((8, 8+24*i), func_str_part, font=fnt, fill=(255, 255, 255))
font_y_next = 8+24*(i+2)
d.text((8, font_y_next), "n1: {:4}".format(n1), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "scale: {:3.02f}".format(scale), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "x_offset: {:3.02f}".format(x_offset), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "scale_y: {:3.02f}".format(scale_y), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "y_offset: {:3.02f}".format(y_offset), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "n1_y: {:3.02f}".format(n1_y), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "delta: {:3.02f}".format(delta), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "modulo: {:3.02f}".format(modulo), font=fnt, fill=(255, 255, 255))

font_y_next += 24*2
d.text((8, font_y_next), "x_min: {:3.02f}, x_max: {:3.02f}".format(-x_offset, scale-x_offset), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "y_min: {:3.02f}, y_max: {:3.02f}".format(-y_offset, scale-y_offset), font=fnt, fill=(255, 255, 255))
except:
print("Something is wrong!!!")

traceback.print_exc()

pix2 = np.array(img2)
# get a font
fnt = ImageFont.truetype('monofonto.ttf', 16)
d = ImageDraw.Draw(img2)
d.text((8, 8), "Used function:", font=fnt, fill=(255, 255, 255))
func_str_split = [func_str_complete[30*i:30*(i+1)] for i in range(0, len(func_str_complete)//30+1)]

for i, func_str_part in enumerate(func_str_split, 1):
d.text((8, 8+24*i), func_str_part, font=fnt, fill=(255, 255, 255))

font_y_next = 8+24*(i+2)
d.text((8, font_y_next), "x_min: {:3.02f}, x_max: {:3.02f}".format(-x_offset, scale-x_offset), font=fnt, fill=(255, 255, 255))

font_y_next += 24
d.text((8, font_y_next), "y_min: {:3.02f}, y_max: {:3.02f}".format(-y_offset, scale-y_offset), font=fnt, fill=(255, 255, 255))

pix2 = np.array(img2).copy()

font_y_next += 24

img2_mod = img2.copy()
d = ImageDraw.Draw(img2_mod)
d.text((8, font_y_next), "modulo: {:3.02f}".format(modulo), font=fnt, fill=(255, 255, 255))
pix2_mod = np.array(img2_mod)

img2_angle = img2.copy()
d = ImageDraw.Draw(img2_angle)
d.text((8, font_y_next), "only with angle", font=fnt, fill=(255, 255, 255))
pix2_angle = np.array(img2_angle)

img2_abs = img2.copy()
d = ImageDraw.Draw(img2_abs)
d.text((8, font_y_next), "only with abs", font=fnt, fill=(255, 255, 255))
pix2_abs = np.array(img2_abs)

img2_x = img2.copy()
d = ImageDraw.Draw(img2_x)
d.text((8, font_y_next), "only with x", font=fnt, fill=(255, 255, 255))
pix2_x = np.array(img2_x)

img2_y = img2.copy()
d = ImageDraw.Draw(img2_y)
d.text((8, font_y_next), "only with y", font=fnt, fill=(255, 255, 255))
pix2_y = np.array(img2_y)

find_x = 0.
find_y = 0.
rows, cols = pix.shape[:2]
line_col = np.argmin((xs1-find_x)**2)
line_row = np.argmin((ys1-find_y)**2)

if not(line_row < 0 or line_row >= rows):
for y in range(0, rows):
pix[y, line_col] = pix[y, line_col]^(0xFF, )*3
pix_mod[y, line_col] = pix_mod[y, line_col]^(0xFF, )*3

pix_angle[y, line_col] = pix_angle[y, line_col]^(0xFF, )*3
pix_abs[y, line_col] = pix_abs[y, line_col]^(0xFF, )*3
pix_x[y, line_col] = pix_x[y, line_col]^(0xFF, )*3
pix_y[y, line_col] = pix_y[y, line_col]^(0xFF, )*3

if not(line_col < 0 or line_col >= cols):
for x in range(0, cols):
pix[line_row, x] = pix[line_row, x]^(0xFF, )*3
pix_mod[line_row, x] = pix_mod[line_row, x]^(0xFF, )*3

pix_angle[line_row, x] = pix_angle[line_row, x]^(0xFF, )*3
pix_abs[line_row, x] = pix_abs[line_row, x]^(0xFF, )*3
pix_x[line_row, x] = pix_x[line_row, x]^(0xFF, )*3
pix_y[line_row, x] = pix_y[line_row, x]^(0xFF, )*3

pix = np.hstack((pix2, pix))
pix_mod = np.hstack((pix2_mod, pix_mod))
pix_angle = np.hstack((pix2_mod, pix_angle))
pix_abs = np.hstack((pix2_mod, pix_abs))
pix_x = np.hstack((pix2_mod, pix_x))
pix_y = np.hstack((pix2_mod, pix_y))

# # find the pixels of the font
# font_pix_pos = np.where(np.sum(pix2!=(0, 0, 0), axis=2)>0)
img = Image.fromarray(pix)
img_mod = Image.fromarray(pix_mod)
img_angle = Image.fromarray(pix_angle)
img_abs = Image.fromarray(pix_abs)
img_x = Image.fromarray(pix_x)
img_y = Image.fromarray(pix_y)

# for y, x in zip(*font_pix_pos):
# pix[y, x] = (((pix[y, x]^0xFF).astype(np.float)+(pix2[y, x]).astype(np.float))/2).astype(np.uint8)
file_name = "{:%Y_%m_%d_%H_%M_%S_%f}.png".format(datetime.datetime.now())
img.save(path_image+file_name)

# # and then invert the pixel at the position!
file_name = "{:%Y_%m_%d_%H_%M_%S_%f}.png".format(datetime.datetime.now())
img_mod.save(path_image+file_name)

img = Image.fromarray(pix)

file_name = "{:%Y_%m_%d}_{}.png".format(datetime.datetime.now(), get_random_string(16))
img.save(path_image+file_name)
# img.save(path_image+"complex_scale_{}_x_off_{}_y_off_{}.png"
# .format(
# "{:02.02f}".format(scale).replace(".", "_"),
# "{:02.02f}".format(x_offset).replace(".", "_"),
# "{:02.02f}".format(y_offset).replace(".", "_")
# ))

# img2.show()
img.show()
file_name = "{:%Y_%m_%d_%H_%M_%S_%f}.png".format(datetime.datetime.now())
img_angle.save(path_image+file_name)

file_name = "{:%Y_%m_%d_%H_%M_%S_%f}.png".format(datetime.datetime.now())
img_abs.save(path_image+file_name)

file_name = "{:%Y_%m_%d_%H_%M_%S_%f}.png".format(datetime.datetime.now())
img_x.save(path_image+file_name)

file_name = "{:%Y_%m_%d_%H_%M_%S_%f}.png".format(datetime.datetime.now())
img_y.save(path_image+file_name)

# img.show()
# img_mod.show()

# TODO: Need to save this in a data format (e.g. pkl.gz with a dotmap dict)
# TODO: saving with all important parameters, like n1, scale, x_offset, ...
Expand Down
Loading

0 comments on commit fc51cc9

Please sign in to comment.