def containsNewCity(self):
w, h = self.width(), self.height()
bounding_rect = QRectF(5, 5, w - 10, h - 10)
r = self.new_city.get_radius()
city_rect = QRectF(self.new_city.pos - QPointF(r, r),
QtCore.QSize(2 * r, 2 * r))
return bounding_rect.contains(city_rect)
def get_random_room():
# --- Se elige en que lado del pasillo se va a insertar la habitación ---
random_side = random.choice(list(dict_orientation_sides[orientation]))
if dict_rooms_per_side[random_side] > max_rooms_per_side:
print(f'side {random_side} has already the maximum of rooms')
random_side = dict_opposite_side[random_side]
print(f'side chosed = {random_side}')
# --- Se escoge un punto aleatorio de ese lado ---
random_point = random.choice(list(corridor_points[random_side]))
print(f'point = {random_point}')
# --. Se escogen un ancho y un alto para la habitacion teninedo en cuenta el lado del pasillo en el que estará la
# habitacion si esta en la parte de abajo del pasillo la altura no puede ser positiva porque se comería el
# pasillo---
signed_width = random.choice(dict_side_sign[random_side][0])
signed_height = random.choice(dict_side_sign[random_side][1])
random_width = signed_width * (random.uniform(2.5, 5))
random_height = signed_height * (random.uniform(2.5, 5))
print(
f'Creating room with width = {random_width} and height = {random_height}'
)
# --- Se crea un rectangulo con el ancho y alto seleccionado
random_room = QRectF(random_point.x(), random_point.y(), random_width,
random_height)
# Se comprueba que la habitación no interseccione con ninguna otra o no contenga a alguan otra
valid_room = True
for prev_room in random_qrect_list:
intersected = random_room.intersects(prev_room)
room_contained_in_prev = prev_room.contains(random_room)
prev_contained_in_room = random_room.contains(prev_room)
if intersected or room_contained_in_prev or prev_contained_in_room:
print('room intersects with existing room')
valid_room = False
break
if valid_room:
random_qrect_list.append(random_room)
dict_rooms_per_side[random_side] += 1
return valid_room
class Callout(QGraphicsItem):
"""
This class code was taken from \
https://code.qt.io/cgit/qt/qtcharts.git/tree/examples/charts/callout/callout.cpp?h=5.13
"""
def __init__(self, chart):
QGraphicsItem.__init__(self, chart)
self.chart = chart
self.rect = QRectF()
self.anchor = QPointF()
self.text_rect = QRectF()
self.text = ""
self.font = QFont()
def boundingRect(self):
anchor = self.mapFromParent(self.chart.mapToPosition(self.anchor))
rect = QRectF()
rect.setLeft(min(self.rect.left(), anchor.x()))
rect.setRight(max(self.rect.right(), anchor.x()))
rect.setTop(min(self.rect.top(), anchor.y()))
rect.setBottom(max(self.rect.bottom(), anchor.y()))
return rect
def paint(self, painter, option, widget):
path = QPainterPath()
path.addRoundedRect(self.rect, 5, 5)
anchor = self.mapFromParent(self.chart.mapToPosition(self.anchor))
if not self.rect.contains(anchor):
point1 = QPointF()
point2 = QPointF()
# establish the position of the anchor point in relation to m_rect
above = anchor.y() <= self.rect.top()
aboveCenter = anchor.y() > self.rect.top() and anchor.y(
) <= self.rect.center().y()
belowCenter = anchor.y() > self.rect.center().y() and anchor.y(
) <= self.rect.bottom()
below = anchor.y() > self.rect.bottom()
onLeft = anchor.x() <= self.rect.left()
leftOfCenter = anchor.x() > self.rect.left() and anchor.x(
) <= self.rect.center().x()
rightOfCenter = anchor.x() > self.rect.center().x() and anchor.x(
) <= self.rect.right()
onRight = anchor.x() > self.rect.right()
# get the nearest m_rect corner
x = (onRight + rightOfCenter) * self.rect.width()
y = (below + belowCenter) * self.rect.height()
cornerCase = (above and onLeft) or (above and onRight) or (
below and onLeft) or (below and onRight)
vertical = qAbs(anchor.x() - x) > qAbs(anchor.y() - y)
x1 = x + leftOfCenter * 10 - rightOfCenter * 20 + cornerCase * (
not vertical) * (onLeft * 10 - onRight * 20)
y1 = y + aboveCenter * 10 - belowCenter * 20 + cornerCase * vertical * (
above * 10 - below * 20)
x2 = x + leftOfCenter * 20 - rightOfCenter * 10 + cornerCase * (
not vertical) * (onLeft * 20 - onRight * 10)
y2 = y + aboveCenter * 20 - belowCenter * 10 + cornerCase * vertical * (
above * 20 - below * 10)
point1.setX(x1)
point1.setY(y1)
point2.setX(x2)
point2.setY(y2)
path.moveTo(point1)
path.lineTo(anchor)
path.lineTo(point2)
path = path.simplified()
painter.setBrush(QColor(255, 255, 255))
painter.drawPath(path)
painter.drawText(self.text_rect, self.text)
def set_anchor(self, point):
self.anchor = point
def updateGeometry(self):
self.prepareGeometryChange()
self.setPos(self.chart.mapToPosition(self.anchor) + QPoint(10, -50))
def set_text(self, text):
self.text = text
metrics = QFontMetrics(self.font)
self.text_rect = metrics.boundingRect(QRect(0, 0, 150, 150),
Qt.AlignLeft, self.text)
self.text_rect.translate(5, 5)
self.prepareGeometryChange()
self.rect = QRectF(self.text_rect.adjusted(-5.0, -5.0, 5.0, 5.0))
class Callout(QGraphicsItem):
def __init__(self, chart):
QGraphicsItem.__init__(self, chart)
self._chart = chart
self._text = ""
self._textRect = QRectF()
self._anchor = QPointF()
self._font = QFont()
self._rect = QRectF()
def boundingRect(self):
anchor = self.mapFromParent(self._chart.mapToPosition(self._anchor))
rect = QRectF()
rect.setLeft(min(self._rect.left(), anchor.x()))
rect.setRight(max(self._rect.right(), anchor.x()))
rect.setTop(min(self._rect.top(), anchor.y()))
rect.setBottom(max(self._rect.bottom(), anchor.y()))
return rect
def paint(self, painter, option, widget):
path = QPainterPath()
path.addRoundedRect(self._rect, 5, 5)
anchor = self.mapFromParent(self._chart.mapToPosition(self._anchor))
if not self._rect.contains(anchor) and not self._anchor.isNull():
point1 = QPointF()
point2 = QPointF()
# establish the position of the anchor point in relation to _rect
above = anchor.y() <= self._rect.top()
aboveCenter = (anchor.y() > self._rect.top()
and anchor.y() <= self._rect.center().y())
belowCenter = (anchor.y() > self._rect.center().y()
and anchor.y() <= self._rect.bottom())
below = anchor.y() > self._rect.bottom()
onLeft = anchor.x() <= self._rect.left()
leftOfCenter = (anchor.x() > self._rect.left()
and anchor.x() <= self._rect.center().x())
rightOfCenter = (anchor.x() > self._rect.center().x()
and anchor.x() <= self._rect.right())
onRight = anchor.x() > self._rect.right()
# get the nearest _rect corner.
x = (onRight + rightOfCenter) * self._rect.width()
y = (below + belowCenter) * self._rect.height()
cornerCase = ((above and onLeft) or (above and onRight)
or (below and onLeft) or (below and onRight))
vertical = abs(anchor.x() - x) > abs(anchor.y() - y)
x1 = (x + leftOfCenter * 10 - rightOfCenter * 20 +
cornerCase * int(not vertical) *
(onLeft * 10 - onRight * 20))
y1 = (y + aboveCenter * 10 - belowCenter * 20 +
cornerCase * vertical * (above * 10 - below * 20))
point1.setX(x1)
point1.setY(y1)
x2 = (x + leftOfCenter * 20 - rightOfCenter * 10 +
cornerCase * int(not vertical) *
(onLeft * 20 - onRight * 10))
y2 = (y + aboveCenter * 20 - belowCenter * 10 +
cornerCase * vertical * (above * 20 - below * 10))
point2.setX(x2)
point2.setY(y2)
path.moveTo(point1)
path.lineTo(anchor)
path.lineTo(point2)
path = path.simplified()
painter.setBrush(QColor(255, 255, 255))
painter.drawPath(path)
painter.drawText(self._textRect, self._text)
def mousePressEvent(self, event):
event.setAccepted(True)
def mouseMoveEvent(self, event):
if event.buttons() & Qt.LeftButton:
self.setPos(
mapToParent(event.pos() - event.buttonDownPos(Qt.LeftButton)))
event.setAccepted(True)
else:
event.setAccepted(False)
def setText(self, text):
self._text = text
metrics = QFontMetrics(self._font)
self._textRect = QRectF(
metrics.boundingRect(QRect(0.0, 0.0, 150.0, 150.0), Qt.AlignLeft,
self._text))
self._textRect.translate(5, 5)
self.prepareGeometryChange()
self._rect = self._textRect.adjusted(-5, -5, 5, 5)
def setAnchor(self, point):
self._anchor = QPointF(point)
def updateGeometry(self):
self.prepareGeometryChange()
self.setPos(self._chart.mapToPosition(self._anchor) + QPointF(10, -50))
class Callout(QGraphicsItem):
def __init__(self, chart):
QGraphicsItem.__init__(self, chart)
self._chart = chart
self._text = ""
self._textRect = QRectF()
self._anchor = QPointF()
self._font = QFont()
self._rect = QRectF()
def boundingRect(self):
anchor = self.mapFromParent(self._chart.mapToPosition(self._anchor))
rect = QRectF()
rect.setLeft(min(self._rect.left(), anchor.x()))
rect.setRight(max(self._rect.right(), anchor.x()))
rect.setTop(min(self._rect.top(), anchor.y()))
rect.setBottom(max(self._rect.bottom(), anchor.y()))
return rect
def paint(self, painter, option, widget):
path = QPainterPath()
path.addRoundedRect(self._rect, 5, 5)
anchor = self.mapFromParent(self._chart.mapToPosition(self._anchor))
if not self._rect.contains(anchor) and not self._anchor.isNull():
point1 = QPointF()
point2 = QPointF()
# establish the position of the anchor point in relation to _rect
above = anchor.y() <= self._rect.top()
aboveCenter = (anchor.y() > self._rect.top() and
anchor.y() <= self._rect.center().y())
belowCenter = (anchor.y() > self._rect.center().y() and
anchor.y() <= self._rect.bottom())
below = anchor.y() > self._rect.bottom()
onLeft = anchor.x() <= self._rect.left()
leftOfCenter = (anchor.x() > self._rect.left() and
anchor.x() <= self._rect.center().x())
rightOfCenter = (anchor.x() > self._rect.center().x() and
anchor.x() <= self._rect.right())
onRight = anchor.x() > self._rect.right()
# get the nearest _rect corner.
x = (onRight + rightOfCenter) * self._rect.width()
y = (below + belowCenter) * self._rect.height()
cornerCase = ((above and onLeft) or (above and onRight) or
(below and onLeft) or (below and onRight))
vertical = abs(anchor.x() - x) > abs(anchor.y() - y)
x1 = (x + leftOfCenter * 10 - rightOfCenter * 20 + cornerCase *
int(not vertical) * (onLeft * 10 - onRight * 20))
y1 = (y + aboveCenter * 10 - belowCenter * 20 + cornerCase *
vertical * (above * 10 - below * 20))
point1.setX(x1)
point1.setY(y1)
x2 = (x + leftOfCenter * 20 - rightOfCenter * 10 + cornerCase *
int(not vertical) * (onLeft * 20 - onRight * 10))
y2 = (y + aboveCenter * 20 - belowCenter * 10 + cornerCase *
vertical * (above * 20 - below * 10))
point2.setX(x2)
point2.setY(y2)
path.moveTo(point1)
path.lineTo(anchor)
path.lineTo(point2)
path = path.simplified()
painter.setBrush(QColor(255, 255, 255))
painter.drawPath(path)
painter.drawText(self._textRect, self._text)
def mousePressEvent(self, event):
event.setAccepted(True)
def mouseMoveEvent(self, event):
if event.buttons() & Qt.LeftButton:
self.setPos(mapToParent(
event.pos() - event.buttonDownPos(Qt.LeftButton)))
event.setAccepted(True)
else:
event.setAccepted(False)
def setText(self, text):
self._text = text
metrics = QFontMetrics(self._font)
self._textRect = QRectF(metrics.boundingRect(
QRect(0.0, 0.0, 150.0, 150.0),Qt.AlignLeft, self._text))
self._textRect.translate(5, 5)
self.prepareGeometryChange()
self._rect = self._textRect.adjusted(-5, -5, 5, 5)
def setAnchor(self, point):
self._anchor = QPointF(point)
def updateGeometry(self):
self.prepareGeometryChange()
self.setPos(self._chart.mapToPosition(
self._anchor) + QPointF(10, -50))
def get_random_room(): # corridor_height, initial_corridor
random_side = random.choice(['top', 'bottom'])
if len(dict_rooms_per_side[random_side]) >= max_rooms_per_side:
print(f'side {random_side} has already the maximum of rooms')
random_side = dict_opposite_side[random_side]
print(f'side chosed = {random_side}')
corridor_locations = dict_corridors_per_side[random_side]
new_corridor_width = initial_corridor.height()
add_corridor = False
# if corridor_location == len(dict_rooms_per_side[random_side]):
if len(dict_rooms_per_side[random_side]) in corridor_locations:
add_corridor = True
if len(dict_rooms_per_side[random_side]) == 0:
if add_corridor:
if random_side == 'bottom':
random_point = initial_corridor.topLeft()
random_point = QPointF(random_point.x() + new_corridor_width,
random_point.y())
else:
random_point = initial_corridor.bottomLeft()
random_point = QPointF(random_point.x() + new_corridor_width,
random_point.y())
else:
if random_side == 'bottom':
random_point = initial_corridor.topLeft()
else:
random_point = initial_corridor.bottomLeft()
# random_point = random.choice(list(dict_corridor_points[random_side]))
else:
if add_corridor:
prev = dict_rooms_per_side[random_side][-1].topRight()
random_point = QPointF(prev.x() + new_corridor_width, prev.y())
else:
random_point = dict_rooms_per_side[random_side][-1].topRight()
print(f'random point = {random_point}')
# --. Se escogen un ancho y un alto para la habitacion teninedo en cuenta el lado del pasillo en el que estará la
# habitacion si esta en la parte de abajo del pasillo la altura no puede ser positiva porque se comería el
# pasillo---
signed_width = dict_side_sign[random_side][0]
signed_height = dict_side_sign[random_side][1]
random_width = signed_width * (random.uniform(3, 6))
random_height = signed_height * fixed_w_h_room
print(
f'Creating room with width = {random_width} and height = {random_height}'
)
# --- Se crea un rectangulo con el ancho y alto seleccionado
random_room = QRectF(random_point.x(), random_point.y(), random_width,
random_height)
# Se comprueba que la habitación no interseccione con ninguna otra o no contenga a alguan otra
valid_room = True
for prev_room in random_qrect_list:
intersected = random_room.intersects(prev_room)
room_contained_in_prev = prev_room.contains(random_room)
prev_contained_in_room = random_room.contains(prev_room)
if intersected or room_contained_in_prev or prev_contained_in_room:
print('room intersects with existing room')
valid_room = False
break
if valid_room:
random_qrect_list.append(random_room)
dict_rooms_per_side[random_side].append(random_room)
return valid_room
def eventFilter(self, obj, event):
if event.type() == QEvent.GraphicsSceneMousePress:
if event.button() == Qt.MouseButton.LeftButton:
self.mouse_pressed = True
self.mouse_pressed_x, self.mouse_pressed_y = event.pos().x(
), event.pos().y()
if self.draw_ellipse:
ellipsis = QGraphicsEllipseItem(self.chart)
ellipsis.setZValue(12)
ellipsis.setBrush(QBrush(QColor(244, 67, 54, 50)))
ellipsis.setPen(QPen(Qt.transparent))
self.ellipses.append(ellipsis)
elif self.write_text:
for t in self.texts:
r = QRectF()
r.setTopLeft(t.pos())
r.setWidth(t.boundingRect().width())
r.setHeight(t.boundingRect().height())
if r.contains(self.mouse_pressed_x,
self.mouse_pressed_y):
return True
"""
The user clicked over an area where there is no text. So we create one.
"""
text = QGraphicsTextItem(self.chart)
text.setZValue(12)
text.setPos(
QPointF(self.mouse_pressed_x, self.mouse_pressed_y))
text.setPlainText("label")
text.setAcceptHoverEvents(True)
text.setTabChangesFocus(True)
text.setFlags(QGraphicsTextItem.ItemIsMovable)
text.installEventFilter(self.text_event_filter)
self.texts.append(text)
return True
elif event.button() == Qt.MouseButton.RightButton:
x, y = event.pos().x(), event.pos().y()
for e in self.ellipses:
if e.rect().contains(x, y):
e.hide()
self.ellipses.remove(e)
for t in self.texts:
r = QRectF()
r.setTopLeft(t.pos())
r.setWidth(t.boundingRect().width())
r.setHeight(t.boundingRect().height())
if r.contains(x, y):
t.hide()
self.texts.remove(t)
return True
return QObject.eventFilter(self, obj, event)
elif event.type() == QEvent.GraphicsSceneMouseRelease:
self.mouse_pressed = False
return True
elif event.type() == QEvent.GraphicsSceneMouseMove:
if self.mouse_pressed:
if self.draw_ellipse:
x, y = event.pos().x(), event.pos().y()
width = x - self.mouse_pressed_x
height = y - self.mouse_pressed_y
self.ellipses[-1].setRect(self.mouse_pressed_x,
self.mouse_pressed_y, width,
height)
return True
return QObject.eventFilter(self, obj, event)
return QObject.eventFilter(self, obj, event)
