def on_init(self):
pygame.init()
infoObject = pygame.display.Info()
self.width_screen = infoObject.current_w
self.height_screen = infoObject.current_h
self.clock = pygame.time.Clock()
self.size = self.width, self.height
self.screen = pygame.display.set_mode(self.size, pygame.HWSURFACE | pygame.DOUBLEBUF)
self.background = pygame.Surface(self.screen.get_size()).convert()
self._running = True
self.sample = GPG_Pos_Dist();
io = GPG_Pos_Dist_IO(self.sample.get_version())
self.sample.release_element()
info = None
if self.filename is not None:
io.FromFile(self.sample,self.filename,info)
self.scale = self.get_scale(self.sample)
self.map = GPG_Map(20,20)
self.map.AddAllToMap(self.sample)
self.offsetx = self.width/2
self.offsety = self.height/2
class App:
filename = None
def __init__(self, width=640, height=400, fps=30):
self._running = True
self._display_surf = None
self.width = width
self.height = height
self.height_screen = 0
self.width_screen = 0
self.fps = fps
self.playtime = 0.0
def on_init(self):
pygame.init()
infoObject = pygame.display.Info()
self.width_screen = infoObject.current_w
self.height_screen = infoObject.current_h
self.clock = pygame.time.Clock()
self.size = self.width, self.height
self.screen = pygame.display.set_mode(self.size, pygame.HWSURFACE | pygame.DOUBLEBUF)
self.background = pygame.Surface(self.screen.get_size()).convert()
self._running = True
self.sample = GPG_Pos_Dist();
io = GPG_Pos_Dist_IO(self.sample.get_version())
self.sample.release_element()
info = None
if self.filename is not None:
io.FromFile(self.sample,self.filename,info)
self.scale = self.get_scale(self.sample)
self.map = GPG_Map(20,20)
self.map.AddAllToMap(self.sample)
self.offsetx = self.width/2
self.offsety = self.height/2
def get_scale(self,sample):
scale = 1
bbox = sample.get_bbox(scale)
scale = 1
scalex = self.width/bbox.w
scaley = self.height/bbox.h
if scalex<scale:
scale=scalex
if scaley<scale:
scale=scaley
scalex = self.width/2/bbox.h
scaley = self.height/2/bbox.w
if scalex<scale:
scale=scalex
if scaley<scale:
scale=scaley
return scale
def on_event_key(self,event):
if event.key == pygame.K_ESCAPE:
self._running = False
def on_event_mouse(self,down,event):
pass
def on_event(self, event):
if event.type == pygame.QUIT:
self._running = False
elif event.type == pygame.KEYDOWN:
self.on_event_key(event)
elif event.type == pygame.MOUSEBUTTONDOWN:
self.on_event_mouse(True,event)
elif event.type == pygame.MOUSEBUTTONUP:
self.on_event_mouse(False,event)
def on_loop(self):
self.miliseconds = self.clock.tick(self.fps)
self.playtime+=self.miliseconds / 1000.0
def on_render(self):
self.screen.blit(self.background, (0,0))
num_element = self.sample.get_num_element()
num = 0
while num < num_element:
points = self.sample.get_frame(self.scale,self.offsetx,self.offsety,num)
if points:
pygame.draw.lines(self.screen,(255,255,255),False,points,1)
num = num +1
num_element = self.map.get_num_element()
num = 0
while num < num_element:
color = self.map.get_level(num)
points = self.map.get_frame(self.scale,self.offsetx,self.offsety,num)
if points:
if color == 1:
rgb = (255,0,0)
elif color == 0:
rgb = (0,255,0)
else:
rgb = (0,0,255)
pygame.draw.lines(self.screen,rgb,False,points,1)
num = num +1
pygame.display.flip()
def on_cleanup(self):
pygame.quit()
def on_execute(self):
if self.on_init() == False:
self._running = False
while( self._running ):
for event in pygame.event.get():
self.on_event(event)
self.on_loop()
self.on_render()
self.on_cleanup()
print 'usage: GenerateSampleGpG.py <path_to_file_to_save> \n You must specify the path to the file you want to save as the first arg'
sys.exit(1)
filename = sys.argv[1]
sensor_dof = Sensor_Dof()
sensor_dof.on_init()
sensor_motor = Sensor_Motor()
sensor_motor.on_init()
sensor_motor.switchdualmotors()
sensor_motor.setMotorSpeed(90)
sensor_motor.on_loop()
gopigo.enable_servo()
print "enable_servo"
sample_acquire = main_usd_sample()
sample = GPG_Pos_Dist()
sample.release_element()
sample_acquire.rotate_usd(0)
print "Current angle :",sensor_dof.angle
one_sample = GPG_Pos_Dist_Element()
istep_main = 66
iangle_main = 0
istep_usd = 10
icone_usd = 60
max_sample_usd = 2
angle_main = sensor_dof.angle
sample_acquire.acquire_cone_usd(angle_main,one_sample,istep_usd,icone_usd,max_sample_usd)
while iangle_main<360-istep_main:
angle_main = sample_acquire.rotate_main(sensor_dof,sensor_motor,istep_main);
sample_acquire.rotate_usd(0)
istepusd = 00
while icurrenta<91:
icurrentb = -15
while icurrentb<16:
usd = []
for add_usd in range(num_usd):
usd.append(base_usd)
one_sample.set_all(0,0,icurrenta,icurrentb)
one_sample.set_all_usd(usd,num_usd)
sample.add_element(one_sample)
base_usd = base_usd+istepusd
icurrentb = icurrentb+ istepb
icurrenta = icurrenta + istepa
if __name__ == "__main__":
if len(sys.argv) !=2:
print 'usage: GenerateTestGpG.py <path_to_file_to_save> \n You must specify the path to the file you want to save as the first arg'
sys.exit(1)
filename = sys.argv[1]
test_acquire = pseudo_usd_sample()
test = GPG_Pos_Dist()
test.release_element()
test_acquire.generate_sample(test,100,5)
io = GPG_Pos_Dist_IO(test.get_version())
info = 'Generated by GenerateTestGpG.py version :'+str(test_acquire.num_version)
io.ToFile(test,filename,info)
