Clusterization sketch

other/Clusters/Clusters.pyde

@@ -0,0 +1,120 @@
+import random
+
+class Point:
+    def __init__(self, x, y):
+        self.x, self.y = x, y
+
+def ddist(x1, y1, x2, y2):
+    return sqrt((abs(y2 - y1) ** 2) + (abs(x2 - x1) ** 2))
+
+def clusterize(cpoints, points):
+    clusters = [[] for _ in range(len(cpoints))]
+    for point in points:
+        distances = []
+        # Get distances from current point to all the cluster points
+        for cpoint in cpoints:
+            distances.append(ddist(point.x, point.y, cpoint.x, cpoint.y))
+            
+        curPos, curDist = 0, distances[0]
+        # Get the index of the lowest distance
+        for pos, dist in enumerate(distances):
+            if curDist > dist:
+                curPos = pos
+                curDist = dist
+        clusters[curPos].append(point)
+    return clusters
+
+def avgPoint(points):
+    Xs = [point.x for point in points]
+    Ys = [point.y for point in points]
+    return Point(sum(Xs) / len(points), sum(Ys) / len(points))
+
+def reclusterize(cpoints, points, n=10):    
+    for _ in range(n):
+        clusters = clusterize(cpoints, points)
+        for pos, _ in enumerate(cpoints):
+            if len(clusters[pos]) > 0:
+                cpoints[pos] = avgPoint(clusters[pos])
+    return cpoints, clusters
+
+def rndPoints(n, topX, topY):
+    return [Point(random.randint(0, topX), random.randint(0, topY)) for _ in range(n)]
+    
+def allButIndex(arrs, index):
+    return [arr for pos, arr in enumerate(arrs) if pos != index]
+
+def concat(arrs):
+    while type(arrs[0]) is list:
+        temp = []
+        for arr in arrs:
+            temp += arr
+        arrs = temp
+    return arrs
+def displayClusters(clusters):
+    global colors, colorDiv
+    pColorDiv = 100 / len(clusters)
+    if pColorDiv != colorDiv:
+        print('Colors regenerated.')
+        colorDiv = pColorDiv
+        offset = random.randint(0, 360)
+        colors = [((colorDiv * n) + offset) % 100 for n in range(len(clusters))]
+    for pos, cluster in enumerate(clusters):
+        fill(colors[pos], 100, 100)
+        stroke(0,0,0)
+
+        strokeWeight(0)
+        for p1pos, point1 in enumerate(cluster):            
+            # ellipse(point1.x, point1.y, 5, 5)
+            curdist = 999999999999
+            # line(point1.x, point1.y, cpoints[pos].x, cpoints[pos].y)
+            # for p2pos, point2 in enumerate(cluster):
+            #     if p1pos != p2pos:
+            #         dddist = ddist(point1.x, point1.y, point2.x, point2.y)
+            #         if dddist < 250:
+            #             stroke(colors[pos], 100, 100, 50)
+            #             line(point1.x, point1.y, point2.x, point2.y)
+            for p2pos, point2 in enumerate(concat(allButIndex(clusters, pos))):
+                if p1pos != p2pos:
+                    newdist = ddist(point1.x, point1.y, point2.x, point2.y)
+                    if curdist > newdist:
+                        curdist = newdist
+            fill(colors[pos], 100, 100, 5)
+            noStroke()
+            ellipse(point1.x, point1.y, int(curdist) * 2, int(curdist) * 2)
+        fill(colors[pos], 100, 100)
+        ellipse(cpoints[pos].x, cpoints[pos].y, 10, 10)
+
+def generate():
+    global numClusters, numPoints, numIterations, cpoints, points, clusters
+    cpoints = rndPoints(numClusters, width, height)
+    points = rndPoints(numPoints, width, height)
+    cpoints, clusters = reclusterize(cpoints, points, n=numIterations)
+
+def setup():
+    size(750, 750)
+    colorMode(HSB, 100)
+    blendMode(BLEND)
+    global numClusters, numPoints, numIterations
+    numClusters, numPoints, numIterations = 6, 150, 1
+    generate()
+    global colors, colorDiv
+    colors = []
+    colorDiv = 0
+    noLoop()
+    redraw()
+    
+def draw():
+    background(0, 0, 90)
+    global clusters
+    displayClusters(clusters)
+
+def keyPressed():
+    if key == TAB:
+        global cpoints, points, clusters
+        cpoints, clusters = reclusterize(cpoints, points, n=1)
+        print('Reclustered.')
+        redraw()
+    elif key == ENTER:
+        generate()
+        print('Regenerated.')
+        redraw()

other/Clusters/sketch.properties

@@ -0,0 +1,2 @@
+mode=Python
+mode.id=jycessing.mode.PythonMode

other/README.md

@@ -7,3 +7,5 @@ Other sketches without any specific purpose. If enough fit into a specific categ
 ## Projects
 
 - **Chaser** A test demonstrating easing equations.
+
+- **Clusters** A test of a simple clusterization algorithm.