Update readme and comments

Author: Ducolnd

README.md

@@ -1,5 +1,5 @@
 # Complex
-Complex number plotter using domain coloring
+Complex function plotter using domain coloring
 
 ## Usage
  1. `git clone https://github.com/Ducolnd/complex`
@@ -12,6 +12,6 @@ Complex number plotter using domain coloring
  8. Run by typing `python complex.py` and close by pressing ECS
 
 ## Sources
-[algorithm-archive.org](https://www.algorithm-archive.org/contents/domain_coloring/domain_coloring.html)
-[dynamicmath.xyz](https://www.dynamicmath.xyz/domain-coloring/#basic)
-[wikipedia.org](https://en.wikipedia.org/wiki/Domain_coloring)
+[algorithm-archive.org](https://www.algorithm-archive.org/contents/domain_coloring/domain_coloring.html) <br>
+[dynamicmath.xyz](https://www.dynamicmath.xyz/domain-coloring/#basic)<br>
+[wikipedia.org](https://en.wikipedia.org/wiki/Domain_coloring)<br>

complex.py

@@ -3,22 +3,26 @@
 import numpy as np
 import math
 
-size = 400
+# Setup
+size = 800
+x_range = 3
+y_range = x_range
+threshold = 0.01 # Black line visibility
 
-res = np.linspace(-2, 2, size * 2 + 1)
-ims = np.linspace(-2, 2, size * 2 + 1)
+res = np.linspace(-(x_range / 2), x_range / 2, size * 2 + 1)
+ims = np.linspace(-(y_range / 2), y_range / 2, size * 2 + 1)
 
 nums = np.zeros((len(res) - 1, len(ims) - 1), dtype=np.complex_)
 
 for i in range(-size, size):
     for j in range(-size , size):
-        nums[i+size, j+size] = complex(res[i+size], ims[j+size])
+        nums[i+size, j+size] = complex(ims[i+size], res[j+size])
 
+def g(z):
+    return (z - 1) / (z ** 13 + z + 1)
 
-# Edit this
-def function(complex_num):
-    return complex_num ** 2
-
+def function(z):
+    return z ** (1 + 10j) * np.cos(g(z))
 
 
 nums = function(nums)
@@ -29,12 +33,14 @@ def function(complex_num):
 
 H = np.interp(phase, (-180, 180), (0,  360))
 S = 0.7 + (1 / 3) * (np.log(magnitude) / np.log(1.6) - np.floor(np.log(magnitude) / np.log(1.6)))   # alternatively S = 0.5 + 0.5 * (magnitude - np.floor(magnitude))
-V = ((np.abs(np.sin(math.pi * nums.real)) ** 0.1) * (np.abs(np.sin(math.pi * nums.imag)) ** 0.1)) * 255 # alternatively V = np.full(phase.shape, 255, dtype="float32")
+V = ((np.abs(np.sin(math.pi * nums.real)) ** threshold) * (np.abs(np.sin(math.pi * nums.imag)) ** threshold)) * 255 # alternatively V = np.full(phase.shape, 255, dtype="float32")
 
 
 HSV = np.dstack((H, S, V)).astype("float32", copy=False)
 
-image = cv2.cvtColor(HSV, cv2.COLOR_HSV2BGR) / 255
+image = np.rot90(cv2.cvtColor(HSV, cv2.COLOR_HSV2BGR) / 255, 3)
+
+# cv2.imwrite("complex.png", image * 255)
 
 cv2.imshow("Complex", image)
 cv2.waitKey(0)