Fixing heatmap, as per esdalmaijer#9 and esdalmaijer#10

jfrey-xx · Sep 6, 2018 · 0080d50 · 0080d50
1 parent efdb608
commit 0080d50
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygazeanalyser/gazeplotter.py b/pygazeanalyser/gazeplotter.py
@@ -224,7 +224,7 @@ def draw_heatmap(fixations, dispsize, imagefile=None, durationweight=True, alpha
 				pass
 		else:				
 			# add Gaussian to the current heatmap
-			heatmap[y:y+gwh,x:x+gwh] += gaus * fix['dur'][i]
+			heatmap[int(round(y)):int(round(y+gwh)),int(round(x)):int(round(x+gwh))] += gaus * fix['dur'][i]
 	# resize heatmap
 	heatmap = heatmap[strt:dispsize[1]+strt,strt:dispsize[0]+strt]
 	# remove zeros