I'm coding the final part of a clustering program, I want to parse a file like
--
COLOR
-
POINT COLOR
...
where
COLOR = (R,G,B)
POINT = (X,Y)
Example:
--
(255,0,4)
-
(0,0) (255,0,4)
(0,1) (255,0,4)
(0,2) (255,0,4)
(0,3) (255,0,4)
--
(32,32,12)
-
(1,0) (156,0,42)
(1,1) (156,0,42)
(1,2) (156,0,42)
(1,3) (156,0,42)
I want to save all this information in different classes to process the data easier.
Here is my code in python:
#!/usr/bin/env python3
import sys
class Point:
def __init__(self, x, y):
self.x = x
self.y = y
class Color:
def __init__(self, r, g, b):
self.r = r
self.g = g
self.b = b
class Pixel:
def __init__(self, point, color):
self.point = point
self.color = color
class Cluster:
def __init__(self, meanColor, pixels):
self.meanColor = meanColor
self.pixels = pixels
def printCluster(cl):
print("---------------")
print("(" str(cl.meanColor.r) "," str(cl.meanColor.g) "," str(cl.meanColor.b) ")")
print("I have " str(len(pixels)) " pixels.")
for px in pixels:
print("(" str(px.point.x) "," str(px.point.y) ") (" str(px.color.r) "," str(px.color.g) "," str(px.color.b) ")")
def printClusters(clusters):
for cl in clusters:
printCluster(cl)
def getColor(line):
r = int(line.split(',')[0][1:])
g = int(line.split(',')[1])
b = int(line.split(',')[2][:-1])
color = Color(r,g,b)
return (color)
file = open(sys.argv[1], 'r')
meanColor = None
clusters = []
cluster = ""
pixels = []
import copy
count = 0
file.readline()
line = file.readline().rstrip('\n')
meanColor = getColor(line)
file.readline()
while True:
count = 1
line = file.readline().rstrip('\n')
if (line == "--"):
pixelsCopy = list(pixels)
cluster = Cluster(meanColor, pixelsCopy)
printClusters(clusters)
printCluster(cluster)
clusters.append(cluster)
pixels = []
printCluster(cluster)
line = file.readline().rstrip('\n')
meanColor = getColor(line)
file.readline()
elif (line and line != "--"):
print("-------------------------------")
printClusters(clusters)
print("line: ~" line "~")
x = int(line.split()[0].split(',')[0][1:])
y = int(line.split()[0].split(',')[1][:-1])
print("x: " str(x))
print("y: " str(y))
point = Point(x,y)
r = int(line.split()[1].split(',')[0][1:])
g = int(line.split()[1].split(',')[1])
b = int(line.split()[1].split(',')[2][:-1])
print("r: " str(r))
print("g: " str(g))
print("b: " str(b))
color = Color(r,g,b)
pixels.append(Pixel(point, color))
if not line:
pixelsCopy = list(pixels)
cluster = Cluster(meanColor, pixelsCopy)
clusters.append(cluster)
#printCluster(cluster)
break
printClusters(clusters)
For some reason, after the while loop, when I print all clusters objects, I see that all contains the same pixels list, I try to do a copy with deepcopy, with [:] and with list() but makes no sense. I don't know where my error is, also I print the cluster after reset to [] and I see that is save like a reference list but I don't know why if I use list() to make a copy.
Thanks!
CodePudding user response:
You need Cluster instance's attribute named pixels:
def printCluster(cl):
print("---------------")
print("(" str(cl.meanColor.r) "," str(cl.meanColor.g) "," str(cl.meanColor.b) ")")
print("I have " str(len(pixels)) " pixels.")
for px in cl.pixels:
print("(" str(px.point.x) "," str(px.point.y) ") (" str(px.color.r) "," str(px.color.g) "," str(px.color.b) ")")