YangBot/visualizer3d.py at master · GodGamer029/YangBot · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph as pg
import pyqtgraph.opengl as gl
import numpy as np

np.set_printoptions(suppress=True, precision=2, linewidth=200)


## Create a GL View widget to display data
app = QtGui.QApplication([])
w = gl.GLViewWidget()
w.show()
w.setWindowTitle('pyqtgraph example: GLSurfacePlot')
w.setCameraPosition(distance=50)

## Add a grid to the view
g = gl.GLGridItem()
g.scale(2,2,1)
g.setDepthValue(10)  # draw grid after surfaces since they may be translucent
w.addItem(g)

names = ["ang", "speed", "accel", "time", "rise","maxcurv", "steer"]

def loadData(str, rows):
    data = np.genfromtxt(str, skip_header=1)

    driftyBoi = np.reshape(data[:, rows[0]], (data.shape[0], 1))
    driftyBoi = np.insert(driftyBoi, 1, data[:, rows[1]], axis=1)
    driftyBoi = np.insert(driftyBoi, 2, data[:, rows[2]], axis=1)

    #normalize data
    driftyBoi[:, 0] /= np.max(np.abs(driftyBoi[:, 0]))
    driftyBoi[:, 1] /= np.max(np.abs(driftyBoi[:, 1]))
    driftyBoi[:, 2] /= np.max(np.abs(driftyBoi[:, 2]))


    driftColor = np.reshape(driftyBoi[:, 2], (driftyBoi.shape[0], 1))

    driftColor = np.insert(driftColor, [1], [0.5, 1, 1], axis=1)

    return (driftyBoi, driftColor)

#oldData, oldColor = loadData('turn.drift.txt')
#newData, newColor = loadData('turn.drift2.txt')
#newColor[:, 2] = 0
#newColor[:, 1] = 1 - newColor[:, 0]

#driftyBoi = np.append(oldData, newData, axis=0)
#driftColor = np.append(oldColor, newColor, axis=0)

data, color = loadData('data/turntime.csv', [1, 6, 2])


#speedData, _ = loadData('data/turntime.csv', [3, 4, 1])
#data2, color2 = loadData('data/turntime.csv', [3, 4, 0])

#for i in range(data.shape[0]):
#    data[i][2] /= speedData[i][2] # accel /= speed

#color2[:, 2] = 0
#color2[:, 1] = 1 - color2[:, 0]

#(0.5, 0.5, 1, 1)
p1 = gl.GLScatterPlotItem(pos=data, color=color, size=5)
#p1 = gl.GLScatterPlotItem(pos=np.append(data, data2, axis=0), color=np.append(color, color2, axis=0), size=5)
p1.scale(10, 10, 10)
w.addItem(p1)

axis = gl.GLAxisItem()
axis.scale(20, 10, 5)
w.addItem(axis)

## Start Qt event loop unless running in interactive mode.
if __name__ == '__main__':
    import sys
    if (sys.flags.interactive != 1) or not hasattr(QtCore, 'PYQT_VERSION'):
        QtGui.QApplication.instance().exec_()