def testSelection(self, data):
"""
Given a
Parameters
----------
data : (N, 2) array
Point coordinates
"""
if len(data) == 0:
return numpy.zeros(0, dtype=bool)
def contained(a, left, top, right, bottom):
assert left <= right and bottom <= top
x, y = a.T
return (x >= left) & (x <= right) & (y <= top) & (y >= bottom)
data = numpy.asarray(data)
selected = numpy.zeros(len(data), dtype=bool)
for p1, p2 in self.selection:
r = QRectF(p1, p2).normalized()
# Note the inverted top/bottom (Qt coordinate system)
selected |= contained(data, r.left(), r.bottom(),
r.right(), r.top())
return selected
def testSelection(self, data):
"""
Given a
Parameters
----------
data : (N, 2) array
Point coordinates
"""
if len(data) == 0:
return numpy.zeros(0, dtype=bool)
def contained(a, left, top, right, bottom):
assert left <= right and bottom <= top
x, y = a.T
return (x >= left) & (x <= right) & (y <= top) & (y >= bottom)
data = numpy.asarray(data)
selected = numpy.zeros(len(data), dtype=bool)
for p1, p2 in self.selection:
r = QRectF(p1, p2).normalized()
# Note the inverted top/bottom (Qt coordinate system)
selected |= contained(data, r.left(), r.bottom(), r.right(),
r.top())
return selected
def qrect_aligned_to(
rect_a: QRectF, rect_b: QRectF, alignment: Qt.Alignment) -> QRectF:
res = QRectF(rect_a)
valign = alignment & Qt.AlignVertical_Mask
halign = alignment & Qt.AlignHorizontal_Mask
if valign == Qt.AlignTop:
res.moveTop(rect_b.top())
if valign == Qt.AlignVCenter:
res.moveCenter(QPointF(res.center().x(), rect_b.center().y()))
if valign == Qt.AlignBottom:
res.moveBottom(rect_b.bottom())
if halign == Qt.AlignLeft:
res.moveLeft(rect_b.left())
if halign == Qt.AlignHCenter:
res.moveCenter(QPointF(rect_b.center().x(), res.center().y()))
if halign == Qt.AlignRight:
res.moveRight(rect_b.right())
return res
