107 lines
4.1 KiB
Python
107 lines
4.1 KiB
Python
import numpy as np
|
|
import matplotlib.pyplot as plt
|
|
import matplotlib
|
|
|
|
|
|
if __name__ == "__main__":
|
|
xc = []
|
|
yc = []
|
|
xm = []
|
|
ym = []
|
|
xf = []
|
|
yf = []
|
|
sxc = []
|
|
syc = []
|
|
sxm = []
|
|
sym = []
|
|
sxf = []
|
|
syf = []
|
|
|
|
sccx = []
|
|
sccy = []
|
|
hccx = []
|
|
hccy = []
|
|
|
|
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/SquareCoveredCells.txt") as c:
|
|
for line in c:
|
|
x, y = line.split()
|
|
sccx.append(int(x))
|
|
sccy.append(int(y))
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/SquareDetectedMines.txt") as c:
|
|
for line in c:
|
|
x, y = line.split()
|
|
sxf.append(int(x))
|
|
syf.append(int(y))
|
|
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/HexCoveredCells.txt") as c:
|
|
for line in c:
|
|
x, y = line.split()
|
|
hccx.append(int(x))
|
|
hccy.append(int(y))
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/HexPath.txt") as c:
|
|
for line in c:
|
|
x, y = line.split()
|
|
xc.append(int(x))
|
|
yc.append(int(y))
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/SquarePath.txt") as c:
|
|
for line in c:
|
|
x, y = line.split()
|
|
sxc.append(int(x))
|
|
syc.append(int(y))
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/Mines.txt") as m:
|
|
for line in m:
|
|
x, y = line.split()
|
|
xm.append(int(x))
|
|
ym.append(int(y))
|
|
|
|
with open("/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/HexDetectedMines.txt") as f:
|
|
for line in f:
|
|
x, y = line.split()
|
|
xf.append(int(x))
|
|
yf.append(int(y))
|
|
|
|
fig = plt.figure(1)
|
|
plt.plot(xc, yc, 'h', label='vehicle', color='blue', linestyle='None')
|
|
plt.plot(xm, ym, 'h', label='all mines', color='red', linestyle='None')
|
|
plt.plot(xf, yf, 'h', label='detected mines', color='orange', linestyle='None')
|
|
plt.title('Hex Simulation')
|
|
plt.legend(loc='lower left', fontsize='xx-small')
|
|
matplotlib.pyplot.savefig('/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/HexTest.png')
|
|
|
|
plt.plot()
|
|
plt.close(fig)
|
|
|
|
fig = plt.figure(2)
|
|
plt.plot(sxc, syc, 's', label='vehicle', color='blue', linestyle='None')
|
|
plt.plot(sxm, sym, 's', label='all mines', color='red', linestyle='None')
|
|
plt.plot(sxf, syf, 's', label='detected mines', color='orange', linestyle='None')
|
|
plt.title('Square Simulation')
|
|
plt.legend(loc='lower left', fontsize='xx-small')
|
|
matplotlib.pyplot.savefig('/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/SquareTest.png')
|
|
plt.plot()
|
|
plt.close(fig)
|
|
|
|
fig = plt.figure(3)
|
|
plt.plot(hccx, hccy, 'h', label='covered cells', color='red', linestyle='None')
|
|
plt.plot(xf, yf, 'h', label='detected mines', color='blue', linestyle='None')
|
|
plt.title('Hex Coverage')
|
|
plt.legend(loc='lower left', fontsize='xx-small')
|
|
matplotlib.pyplot.savefig('/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/HexCoverage.png')
|
|
plt.plot()
|
|
plt.close(fig)
|
|
|
|
fig = plt.figure(4)
|
|
plt.plot(sccx, sccy, 's', label='covered cells', color='red', linestyle='None')
|
|
plt.plot(sxf, syf, 's', label='detected mines', color='blue', linestyle='None')
|
|
plt.title('Square Coverage')
|
|
plt.legend(loc='lower left', fontsize='xx-small')
|
|
matplotlib.pyplot.savefig('/Users/brady.bodily/Documents/Repositories/CS5890_Robot_Intelligence/RobotIntelFinal/ConsoleApp/Output/SquareCoverage.png')
|
|
plt.plot()
|
|
plt.close(fig) |