Added the scanning code for the slam sensor but there is severe slow of the animation

This commit is contained in:
Alfredo Rwagaju
2016-08-15 12:26:00 -04:00
parent 2085647748
commit 25cb93d805

View File

@@ -48,6 +48,7 @@ TRAJECTORY_COLOR_BGR = (255, 0, 0)
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from math import sin, cos, radians from math import sin, cos, radians
import numpy as np
class SlamShow(object): class SlamShow(object):
@@ -87,14 +88,20 @@ class SlamShow(object):
self._add_vehicle(16000,16000,0) self._add_vehicle(16000,16000,0)
def displayMap(self, mapbytes): def displayMap(self, mapbytes):
mapimg = np.reshape(np.frombuffer(mapbytes, dtype=np.uint8), (self.map_size_pixels, self.map_size_pixels))
plt.imshow(mapimg)
plt.set_cmap('gray')
# Interleave the grayscale map bytes into the color bytes # Interleave the grayscale map bytes into the color bytes
self.bgrbytes[0::3] = mapbytes self.bgrbytes[0::3] = mapbytes
self.bgrbytes[1::3] = mapbytes self.bgrbytes[1::3] = mapbytes
self.bgrbytes[2::3] = mapbytes self.bgrbytes[2::3] = mapbytes
# Put color bytes into image # Put color bytes into image - this line causes error
#plt.set_cmap('BlueRedAlpha')
def displayVelocities(self, dxy_mm, dtheta_deg): def displayVelocities(self, dxy_mm, dtheta_deg):
# Add velocity bars # Add velocity bars