def plot_data(self, *args):
#ipdb.set_trace()
color_pts = self.colormap.map(args[0][1])
brush = [QtGui.QBrush(QtGui.QColor(*color_pts))]
self.scatter.addPoints(x=[args[0][0][0]],
y=[args[0][0][1]],
brush=brush)
def plot_data(self, *args):
#ipdb.set_trace()
color_pts = self.colormap.map(args[0][1][0])
brush = QtGui.QBrush(QtGui.QColor(*color_pts))
self.brushes[args[0][1][1] % self.n_points] = brush
self.pos[args[0][1][1] % self.n_points, 0] = args[0][0][0]
self.pos[args[0][1][1] % self.n_points, 1] = args[0][0][1]
#self.scatter.addPoints(x=self.x_pos, y=self.y_pos, brush=self.brushes)
self.scatter.setData(pos=self.pos, brush=self.brushes)
def __init__(self, parent, myView=[]):
QtGui.QGraphicsScene.__init__(self, parent)
self.myView = myView
# Some items in scene
self.txt = self.addSimpleText("///////")
self.txt.setPos(2, -20)
self.txt.setScale(2)
self.txt.setBrush(QtGui.QBrush(QtCore.Qt.green))
self.addRect(0, 16, 20, 20, pen=QtGui.QPen(QtCore.Qt.blue))
self.addRect(10, 60, 32, 8, pen=QtGui.QPen(QtCore.Qt.red))
self.addRect(30, 16, 20, 20, pen=QtGui.QPen(QtCore.Qt.blue))
self.N = 0
def __init__(self, *args, **kwargs):
super(map_sim, self).__init__()
self.n_points = 100 #with 1500 could see the whole orbit
self.counter = 0
self.pos = np.random.random([self.n_points, 2]) + 20
values = np.zeros(self.n_points)
##This is simulation orbit
undersamp = 5
self.x_pos = np.loadtxt('x_val')[::undersamp]
self.y_pos = np.loadtxt('y_val')[::undersamp]
##
self.win = pg.GraphicsWindow()
self.win.setWindowTitle('scatt anim')
self.win.show()
p = self.win.addPlot()
p.disableAutoRange()
p.setRange(xRange=[-10, 10], yRange=[-10, 10])
p.showGrid(x=True, y=True)
self.scatter = pg.ScatterPlotItem()
p.addItem(self.scatter)
self.colormap = self.mapping_init(0, 100)
aux = self.colormap.map(values)
self.brushes = [
QtGui.QBrush(QtGui.QColor(*aux[i, :].tolist()))
for i in range(aux.shape[0])
]
#ipdb.set_trace()
#self.values = [QtGui.QBrush(QtGui.QColor(*aux))]
self.threadpool = QtCore.QThreadPool(parent=self.win)
self.threadpool.setMaxThreadCount(1)
self.update()
def get_data(self):
[x,y] = get_pos() ###simular!!
[A, B, re, im] = get_vals() #ver como importarlo..
pow_diff = 10*(np.log10(A+1)-np.log10(B+1))
ang_diff = np.rad2deg(np.arctan2(im, re))
return [[x,y], [pow_diff, ang_diff]]
def process_data(self, *args)
color_ang = self.color_ang.map(args[1][1])
brush_ang = [QtGui.QBrush(QtGui.QColor(*color_ang))]
self.ang.addPoints(x=[args[0][0]], y=[args[0][1]], brush=brush_ang)
color_pow = self.color_pow.map(args[1][0])
brush_pow = [QtGui.QBrush(QtGui.QColor(*color_pow))]
self.pow.addPoints(x=[args[0][0]], y=[args[0][1]], brush=brush_pow)
##################################
"""
Para correrlo falta hacer
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
#
#
# Code:
"""Read data distributed in a directory structure"""
import os
from pyqtgraph import (QtCore, QtGui)
from pyqtgraph import parametertree as ptree
bgBrush = QtGui.QBrush(QtGui.QColor('lightsteelblue'))
class PathParams(ptree.parameterTypes.GroupParameter):
def __init__(self, **opts):
opts['type'] = 'group'
opts['addText'] = "Add rule"
opts['addList'] = ['timestamp', 'regex', 'string']
super(PathParams, self).__init__(**opts)
def addNew(self, typ):
val = {
'timestamp': 'YYYY-mm-dd HH:MM:SS',
'regex': '.*',
'string': '_'
}[typ]
