def paintEvent(self, event):
#mtx = self.mtx
mtx = QTransform()
mtx.rotate(self.mrotation)
mtx.scale(self.mscale.x(), self.mscale.y())
mtx.translate(self.mtranslate.x(), self.mtranslate.y())
eyepos = QPointF(self.epx, self.dof)
ppoi = QPointF(self.ppx, self.ppy)
point = QRectF(0.0,0.0,0.05,0.05);
tpoi = mtx.map(ppoi)
teyepos = mtx.map(eyepos)
evec = QVector2D(tpoi - teyepos).normalized()
pts = find_points(float2(tpoi.x(),tpoi.y()), float2(evec.x(), evec.y()))
print pts
qp = QtGui.QPainter()
qp.begin(self)
qp.scale(self.width()/5.0,self.height()/5.0)
qp.translate(2.5,2.5)
#draw voxel bounds
qp.drawRect(QRectF(0.0,0.0,1.0,1.0))
#qp.transform(mtx)
#draw eyepos
point.moveTo(mtx.map(eyepos))
qp.fillRect(point, QColor("black"))
point.moveTo(mtx.map(ppoi))
qp.fillRect(point, QColor("grey"))
qp.setPen(QColor("cyan"))
qp.drawLine(mtx.map(QLineF(-0.5,0.0,1.5,0.0)))
qp.setPen(QColor("blue"))
qp.drawLine(mtx.map(QLineF(-0.0,0.0,1.0,0.0)))
qp.setPen(QColor("lime"))
qp.drawLine(QLineF(eyepos,ppoi))
qp.setPen(QColor("green"))
qp.drawLine(QLineF(teyepos,tpoi))
qp.setPen(QColor("orange"))
qp.drawLine(QLineF(pts['x'],pts['y'],pts['z'], pts['w']))
point.moveTo(QPointF(pts['x'],pts['y']))
qp.fillRect(point, QColor("red"))
point.moveTo(QPointF(pts['z'],pts['w']))
qp.fillRect(point, QColor("pink"))
qp.end()
def getObjectInteraction(self, persons, objects, interaction, d):
# print("getObjectInteration")
plt.close('all')
polylines_object = []
polylines_interacting = []
for o in objects:
obj = Object(o.x, o.z, o.angle, o.space)
# print("OBJETO")
##para dibujarlo
if d:
plt.figure('ObjectSpace')
rect = plt.Rectangle((obj.x - 0.25, obj.y - 0.25),
0.5,
0.5,
fill=False)
plt.gca().add_patch(rect)
x_aux = obj.x + 0.25 * cos(pi / 2 - obj.th)
y_aux = obj.y + 0.25 * sin(pi / 2 - obj.th)
heading = plt.Line2D((obj.x, x_aux), (obj.y, y_aux),
lw=1,
color='k')
plt.gca().add_line(heading)
w = 1.0
#print (obj.x,obj.y)
##para calcular el rectangulo
s = QRectF(QPointF(0, 0), QSizeF(w, obj.sp))
# if (d):
# plt.plot (s.bottomLeft().x(),s.bottomLeft().y(),"go")
# plt.plot(s.bottomRight().x(), s.bottomRight().y(), "ro")
# plt.plot(s.topRight().x(), s.topRight().y(), "yo")
# plt.plot(s.topLeft().x(), s.topLeft().y(), "bo")
space = QPolygonF()
space.append(s.topLeft())
space.append(s.topRight())
space.append(
QPointF(s.bottomRight().x() + obj.sp / 4,
s.bottomRight().y()))
space.append(
QPointF(s.bottomLeft().x() - obj.sp / 4,
s.bottomLeft().y()))
t = QTransform()
t.translate(-w / 2, 0)
space = t.map(space)
t = QTransform()
t.rotateRadians(-obj.th)
space = t.map(space)
t = QTransform()
t.translate(obj.x, obj.y)
space = t.map(space)
# points = []
# for x in xrange(space.count()-1):
# point = space.value(x)
# print ("valor", point)
# points.append([point.x(),point.y()])
# plt.plot(point.x(),point.y(),"go")
polyline = []
for x in xrange(space.count()):
point = space.value(x)
if (d):
plt.plot(point.x(), point.y(), "go")
p = SNGPoint2D()
p.x = point.x()
p.z = point.y()
polyline.append(p)
polylines_object.append(polyline)
for p in persons:
pn = Person(p.x, p.z, p.angle)
# print("PERSONA", persons.index(p)+1)
if d:
body = plt.Circle((pn.x, pn.y), radius=0.3, fill=False)
plt.gca().add_patch(body)
x_aux = pn.x + 0.30 * cos(pi / 2 - pn.th)
y_aux = pn.y + 0.30 * sin(pi / 2 - pn.th)
heading = plt.Line2D((pn.x, x_aux), (pn.y, y_aux),
lw=1,
color='k')
plt.gca().add_line(heading)
plt.axis('equal')
##CHECKING THE ORIENTATION
a = abs(obj.th - abs(pn.th - math.pi))
if a < math.radians(45):
checkangle = True
else:
checkangle = False
##CHECKING IF THE PERSON IS INSIDE THE POLYGON
if space.containsPoint(QPointF(pn.x, pn.y),
Qt.OddEvenFill) and checkangle:
# print("DENTROOOOO Y MIRANDO")
if not polyline in polylines_interacting:
polylines_interacting.append(polyline)
if d:
for ps in polylines_interacting:
# plt.figure()
for p in ps:
plt.plot(p.x, p.z, "ro")
plt.axis('equal')
plt.xlabel('X')
plt.ylabel('Y')
plt.show()
plt.show()
if (interaction):
return polylines_interacting
else:
return polylines_object
def _paint_crop(self, painter, option, index):
"""Paints the crop
"""
source_rect = index.data(RectRole)
crop_rect = self.crop_rect.translated(option.rect.topLeft())
angle = index.data(RotationRole)
# Target rect with same aspect ratio as source
source_aspect = float(source_rect.width()) / source_rect.height()
crop_aspect = float(crop_rect.width()) / crop_rect.height()
# True if the item has been rotated by a multiple of 90 degrees
perpendicular = 1 == (angle / 90) % 2
# Some nasty logic to compute the target rect
if perpendicular:
crop_aspect = 1.0 / crop_aspect
if source_aspect > 1.0:
# Crop is wider than is is tall
if crop_aspect > source_aspect:
fit_to = 'height'
f = 1.0 / source_aspect
else:
fit_to = 'width'
f = source_aspect
else:
# Crop is taller than is is wide
if crop_aspect < source_aspect:
fit_to = 'width'
f = source_aspect
else:
fit_to = 'height'
f = 1.0 / source_aspect
if perpendicular:
if 'width' == fit_to:
size = QSize(crop_rect.height(),
crop_rect.height() / f)
else:
size = QSize(crop_rect.width() / f,
crop_rect.width())
else:
if 'width' == fit_to:
size = QSize(crop_rect.width(),
crop_rect.width() / f)
else:
size = QSize(crop_rect.height() / f,
crop_rect.height())
target_rect = QRect(crop_rect.topLeft(), size)
target_rect.moveCenter(option.rect.center())
# Draw rotated
if angle:
t = QTransform()
t.translate(option.rect.width() / 2+option.rect.left(),
option.rect.height() / 2+option.rect.top())
t.rotate(angle)
t.translate(-option.rect.width() / 2-option.rect.left(),
-option.rect.height() / 2-option.rect.top())
with painter_state(painter):
if angle:
painter.setTransform(t)
painter.drawPixmap(target_rect, index.data(PixmapRole), source_rect)
if QStyle.State_Selected & option.state:
painter.setPen(QPen(Qt.white, 1, Qt.SolidLine))
painter.drawRect(target_rect)
def _paint_crop(self, painter, option, index):
"""The cropped image
"""
source_rect = index.data(RectRole)
crop_rect = self.CROP_RECT.translated(option.rect.topLeft())
angle = index.data(RotationRole)
# Target rect with same aspect ratio as source
source_aspect = float(source_rect.width()) / source_rect.height()
crop_aspect = float(crop_rect.width()) / crop_rect.height()
# True is the item has been rotated by a multiple of 90 degrees
perpendicular = 1 == (angle / 90) % 2
# Some nasty logic to compute the target rect
if perpendicular:
crop_aspect = 1.0 / crop_aspect
if source_aspect > 1.0:
# Crop is wider than is is tall
if crop_aspect > source_aspect:
fit_to = 'height'
f = 1.0 / source_aspect
else:
fit_to = 'width'
f = source_aspect
else:
# Crop is taller than is is wide
if crop_aspect < source_aspect:
fit_to = 'width'
f = source_aspect
else:
fit_to = 'height'
f = 1.0 / source_aspect
if perpendicular:
if 'width' == fit_to:
size = QSize(crop_rect.height(), crop_rect.height() / f)
else:
size = QSize(crop_rect.width() / f, crop_rect.width())
else:
if 'width' == fit_to:
size = QSize(crop_rect.width(), crop_rect.width() / f)
else:
size = QSize(crop_rect.height() / f, crop_rect.height())
target_rect = QRect(crop_rect.topLeft(), size)
target_rect.moveCenter(option.rect.center())
# Draw rotated
if angle:
t = QTransform()
t.translate(option.rect.width() / 2 + option.rect.left(),
option.rect.height() / 2 + option.rect.top())
t.rotate(angle)
t.translate(-option.rect.width() / 2 - option.rect.left(),
-option.rect.height() / 2 - option.rect.top())
with painter_state(painter):
if angle:
painter.setTransform(t)
painter.drawPixmap(target_rect, index.data(PixmapRole),
source_rect)
painter.setPen(QPen(Qt.white, 1, Qt.SolidLine))
painter.drawRect(target_rect)
