comparison visualclient/visclient.py @ 22:858fc44a17c1

added s option to stop blobs from fading

 ecol   = (0,0,0,255)
 
 # X,Y=1140,900
 X,Y = 600,400
 NBLOBS = 15
-BLOBSIZE = 20
+BLOBSIZE = 25
 G=110
-FADE = 40
+FADE = 15
 
 class Blob(object):
 	
-	def __init__(self,bg,x,y,color=(255,255,255),mood=None):
+	def __init__(self,bg,x,y,color=(255,255,255),mood=None,fade=FADE):
 		self.x = x
 		self.y = y
 		self.xs = x * X
 		self.ys = Y - (y * Y)
 		self.bg = bg

 		self.time = time.time()
 		self.alpha = 255
 		self.c = color
 		self.count = 1
 		self.visible = True
+		self.FADE = fade
 		if mood and mood.color :
 			self.c = mood.color
 				
 	def __cmp__(self,other):
 		d = math.sqrt( math.pow((self.x-other.x),2) + math.pow((self.y-other.y),2) )

 			return -1
 		
 	def draw(self):
 		d=int(self.size)
 		self.bg.blit(gradients.radial(self.size, (self.c[0],self.c[1],self.c[2],self.alpha), (0,0,0,self.alpha)), (self.xs-d,self.ys-d))
-		self.alpha = 255 - int(self.age()*FADE) 
+		self.alpha = 255 - int(self.age()*self.FADE) 
 		if self.alpha < 5 : 
 			self.alpha = 1
 			self.visible = False
 				
 	def age(self):

 		self.count = count
 		self.size = int(BLOBSIZE * int(self.count/1.5))
 		
 	def get_distance(self,x,y):
 		return math.sqrt( math.pow((self.x-x),2) + math.pow((self.y-y),2) )
+		
+	def fade(self,fade):
+		if not fade : self.alpha = 255
+		self.FADE = fade
+		
+	def reset_time(self):
+		self.time = time.time()
 
 
 
 
 class Mood():

 		self.moods = []
 		self.read_mood_data()
 		# pg.quit()
 		# sys.exit(-1)
 		
+		self.FADE = FADE
+		
 		pass
 		
 	def read_mood_data(self):
 		
 		with open('moods.csv') as mf:
 			data = mf.readlines()[1:]
 			for line in data :
 				l = line.split(',')
 				mood = Mood(l[0],l[1],l[2])
 				self.moods.append(mood)
-		with open('feelings.txt') as ff:
-			data = ff.readlines()
-			data = map(lambda x: x.split('\t'),data)
+		# with open('feelings.txt') as ff:
+		# 	data = ff.readlines()
+		# 	data = map(lambda x: x.split('\t'),data)
+		# 	for mood in self.moods :
+		# 		for colors in data :
+		# 			if mood.word == colors[0] :
+		# 				mood.color = colors[2]
+		with open('colors.txt') as ff:
+			data = ff.readlines()[1:]
+			data = map(lambda x: x.split(','),data)
 			for mood in self.moods :
+				d = cd = sys.float_info.max
 				for colors in data :
-					if mood.word == colors[0] :
-						mood.color = colors[2]
-		pass
-		for mood in self.moods:
-			if mood.color :
-				mood.color = map(lambda x: '0x'+str(x).strip(),[mood.color[0:2],mood.color[2:4],mood.color[4:]])
-				mood.color = tuple(map(lambda x: int(eval(x)),mood.color))
-				# print mood.color
+					d = mood.get_distance(float(colors[0]),float(colors[1]))
+					if d < cd :
+						cd = d
+						mood.color = tuple(map(lambda x: int(pow(math.atan((float(x)/7.0)),12.5)),(colors[2],colors[3],colors[4])))
+		
+		# for mood in self.moods:
+		# 	if mood.color :
+		# 		# mood.color = map(lambda x: '0x'+str(x).strip(),[mood.color[0:2],mood.color[2:4],mood.color[4:]])
+		# 		mood.color = tuple(map(lambda x: int(x),mood.color))
+		# 		print mood.color
 		
 		
 	def update(self):
 		
 		# delete invisibles

 				d = mood.get_distance(x,y)
 				if d < cd :
 					cd = d
 					cmood = mood
 
-			new = Blob(self.bg,x,y,mood=cmood)
+			new = Blob(self.bg,x,y,mood=cmood,fade=self.FADE)
 			if not new in self.blobs :
 				self.blobs.insert(0,new)
 			elif c > self.blobs[self.blobs.index(new)].count:
 				self.blobs[self.blobs.index(new)].increment(c)
 

 					pg.quit()
 					sys.exit()
 				elif event.type == KEYDOWN :
 					if event.key == K_ESCAPE :
 						pg.event.post(pg.event.Event(QUIT))
+					elif event.key == K_SPACE :
+						self.blobs = []
+					elif event.key == K_s :
+						if self.FADE :
+							print "fade off"
+							self.FADE = 0
+							for blob in self.blobs :
+								blob.fade(0)
+						else:
+							print "fade on"
+							self.FADE = 15
+							for blob in self.blobs :
+								blob.fade(15)
+								blob.reset_time()
+							
+								
 
 
 			# Draw 
 			self.draw()