def VisualizeSimData(icList,allplans=False,showpaths=True,showtrace=True,showtraffic=True,xmin=-100,ymin=-100,xmax=100,ymax=100,playbkspeed=1,interval=30,record=False,filename="",network=[]):
'''
ic: icarous object
allplans: True - plot all computed plans, False - plot only the mission plan
xmin,ymin : plot axis min values
xmax,ymax : plot axis max values
interval : Interval between frames
'''
if record:
import matplotlib; matplotlib.use('Agg')
from Animation import AgentAnimation
anim= AgentAnimation(xmin,ymin, xmax,ymax,showtrace,playbkspeed,interval,record,filename)
vehicleSize1 = np.abs(xmax - xmin)/100
vehicleSize2 = np.abs(ymax - ymin)/100
vehicleSize = np.max([vehicleSize1,vehicleSize2])
homePos = icList[0].home_pos
getLocPos = lambda pos: np.array(ConvertToLocalCoordinates(homePos,pos))
for j,ic in enumerate(icList):
anim.AddAgent(ic.callsign,vehicleSize,'r',ic.ownshipLog,show_circle=True,circle_rad=ic.daa_radius)
for i,pln in enumerate(ic.plans):
planTime = pln[0][0]
planPositions = np.array(GetPlanPositions(ic.plans[i],0.1))
points = np.array(list(map(getLocPos,planPositions)))
if i == 0:
planWPs = np.array(pln)[:,1:]
labels = [[TcpType.getString(val[3]),TcpType.getString(val[4]),TcpType.getString(val[5])]\
for val in planWPs]
if showpaths:
anim.AddPath(np.array(list(map(getLocPos,planWPs))),'k--',points,labels,time=planTime)
if i > 0 and allplans:
planWPs = np.array(pln)[:,1:]
if showpaths:
anim.AddPath(np.array(list(map(getLocPos,planWPs))),'k--',points,time=planTime)
tfids = ic.trafficLog.keys()
for key in tfids:
if showtraffic or key[0:2] == 'tf':
anim.AddAgent('traffic_'+str(key),vehicleSize,'b',ic.trafficLog[key])
for fence in ic.localFences:
fence.append(fence[0])
anim.AddFence(np.array(fence),'c-.')
for fix in icList[0].localMergeFixes:
anim.AddZone(fix[::-1][1:3],icList[0].params['COORD_ZONE'],'r')
anim.AddZone(fix[::-1][1:3],icList[0].params['SCHEDULE_ZONE'],'b')
anim.AddZone(fix[::-1][1:3],icList[0].params['ENTRY_RADIUS'],'g')
for plan in network:
planWPs = np.array(plan)[:,1:]
anim.AddPath(np.array(list(map(getLocPos,planWPs))),'k--',color2='k')
anim.run()
def ConvertToLocalCoordinates(self, pos):
return ConvertToLocalCoordinates(self.home_pos, pos)