Пример #1
0
    def __init__(self, data, radius=15):

        self.data = data

        # Add circular node
        self.node = QGraphicsEllipseItem(0, 0, 1, 1)

        # Set radius
        self.radius = radius

        # Add text label
        self.label = QGraphicsTextItem(data.label)
        font = self.label.font()
        font.setPointSize(10)
        self.label.setFont(font)

        # Add line between label and node
        self.line1 = QGraphicsLineItem(0, 0, 1, 1)
        self.line2 = QGraphicsLineItem(0, 0, 1, 1)

        self.node.setZValue(20)
        self.label.setZValue(10)
        self.line1.setZValue(10)
        self.line2.setZValue(10)

        self.line1.setPen(get_pen('0.5'))
        self.line2.setPen(get_pen('0.5'))

        self.color = '0.8'
Пример #2
0
    def __init__(self, lon0, lat0, lon1, lat1, parent=None):
        QGraphicsLineItem.__init__(self, parent=parent)
        MapItem.__init__(self)

        self._lon0 = lon0
        self._lat0 = lat0
        self._lon1 = lon1
        self._lat1 = lat1
Пример #3
0
    def __init__(self, lon0, lat0, lon1, lat1, parent=None):
        QGraphicsLineItem.__init__(self, parent=parent)
        MapItem.__init__(self)

        self._lon0 = lon0
        self._lat0 = lat0
        self._lon1 = lon1
        self._lat1 = lat1
Пример #4
0
    def __init__(self,
                 parent=None,
                 width=1,
                 color1=QColor(0, 0, 0),
                 color2=QColor(255, 255, 255)):
        super(DualColoredLineSegment, self).__init__(parent)

        # create lines
        self.line_1 = QGraphicsLineItem()
        self.line_2 = QGraphicsLineItem()

        self.width = width
        self._length = 2
        self._spacing = 5

        # set pen
        self.setColor1(color1)
        self.setColor2(color2)

        # add lines to group
        self.addToGroup(self.line_1)
        self.addToGroup(self.line_2)
Пример #5
0
    def __init__(self, longitudes, latitudes, parent=None):
        QGraphicsItem.__init__(self, parent=parent)
        MapItem.__init__(self)

        assert len(longitudes) == len(latitudes)
        assert len(longitudes) >= 2

        self._longitudes = np.array(longitudes, dtype=np.float64)
        self._latitudes = np.array(latitudes, dtype=np.float64)

        # Setup internal lines
        linesGroup = QGraphicsItemGroup(parent=self)
        self._linesGroup = linesGroup
        self._lines = [
            QGraphicsLineItem(parent=linesGroup)
            for i in iterRange(len(longitudes) - 1)
        ]
Пример #6
0
 def addLine(self, text, color, width=1.):
     shape = QGraphicsLineItem(10., 10., 20., 20.)
     pen = makePen(color, width=width)
     shape.setPen(pen)
     self.addEntry(MapLegendEntryItem(shape, text))
Пример #7
0
class DataNode:

    def __init__(self, data, radius=15):

        self.data = data

        # Add circular node
        self.node = QGraphicsEllipseItem(0, 0, 1, 1)

        # Set radius
        self.radius = radius

        # Add text label
        self.label = QGraphicsTextItem(data.label)
        font = self.label.font()
        font.setPointSize(10)
        self.label.setFont(font)

        # Add line between label and node
        self.line1 = QGraphicsLineItem(0, 0, 1, 1)
        self.line2 = QGraphicsLineItem(0, 0, 1, 1)

        self.node.setZValue(20)
        self.label.setZValue(10)
        self.line1.setZValue(10)
        self.line2.setZValue(10)

        self.line1.setPen(get_pen('0.5'))
        self.line2.setPen(get_pen('0.5'))

        self.color = '0.8'

    @property
    def radius(self):
        return self._radius

    @radius.setter
    def radius(self, value):
        self._radius = value
        self.node.setRect(-value, -value, 2 * value, 2 * value)

    def contains(self, point):

        # Check label
        if self.label.contains(self.label.mapFromScene(point)):
            return True

        # Check node
        if self.node.contains(self.node.mapFromScene(point)):
            return True

        return False

    def update(self):
        self.node.update()

    def add_to_scene(self, scene):
        scene.addItem(self.node)
        scene.addItem(self.label)
        scene.addItem(self.line1)
        scene.addItem(self.line2)

    def remove_from_scene(self, scene):
        scene.removeItem(self.node)
        scene.removeItem(self.label)
        scene.removeItem(self.line1)
        scene.removeItem(self.line2)

    @property
    def node_position(self):
        pos = self.node.pos()
        return pos.x(), pos.y()

    @node_position.setter
    def node_position(self, value):
        self.node.setPos(value[0], value[1])
        self.update_lines()

    @property
    def label_position(self):
        pos = self.label.pos()
        return pos.x(), pos.y()

    @label_position.setter
    def label_position(self, value):
        self.label.setPos(value[0], value[1])
        self.update_lines()

    def update_lines(self):
        x0, y0 = self.label_position
        x2, y2 = self.node_position
        x1 = 0.5 * (x0 + x2)
        y1 = y0
        self.line1.setLine(x0, y0, x1, y1)
        self.line2.setLine(x1, y1, x2, y2)

    @property
    def color(self):
        return qt_to_mpl_color(self.node.brush().color())

    @color.setter
    def color(self, value):
        self.node.setBrush(mpl_to_qt_color(value))
Пример #8
0
class DualColoredLineSegment(QGraphicsItemGroup):
    """
    One individual line segment.  This is a group because it needs to create two
    lines in order to create a multi colored dashed pattern.

    Attributes:
        spacing (int): how much space in pixels are between each line segment
    """
    def __init__(self,
                 parent=None,
                 width=1,
                 color1=QColor(0, 0, 0),
                 color2=QColor(255, 255, 255)):
        super(DualColoredLineSegment, self).__init__(parent)

        # create lines
        self.line_1 = QGraphicsLineItem()
        self.line_2 = QGraphicsLineItem()

        self.width = width
        self._length = 2
        self._spacing = 5

        # set pen
        self.setColor1(color1)
        self.setColor2(color2)

        # add lines to group
        self.addToGroup(self.line_1)
        self.addToGroup(self.line_2)

    """ DISPLAY"""

    def setColor1(self, color):
        """
        Sets the first line to the specified color

        Args:
            color (QColor): color for the line to be set to
        """
        pen = self.createPen(color)
        self.line_1.setPen(pen)

    def setColor2(self, color):
        """
        Sets the first line to the specified color

        Args:
            color (QColor): color for the line to be set to
        """
        pen = self.createPen(color, offset=True)
        self.line_2.setPen(pen)

    def updatePen(self):
        for line in [self.line_1, self.line_2]:
            pen = line.pen()
            total_line_space = self.length() + (2 * self.spacing())
            pen.setDashPattern([self.length(), total_line_space])
            pen.setWidth(self.width)
            line.setPen(pen)

    def createPen(self, color, offset=None):
        """
        Creates a pen of the color specified

        Args:
            color (QColor): color for the line to be set to
            offset (bool): if color should be offset or not.
                Since this is only two colors, this can be a boolean, rather
                than an index.
        """
        pen = QPen()
        pen.setColor(color)
        total_line_space = self.length() + (2 * self.spacing())
        if offset:
            pen.setDashOffset(self.length() + self.spacing())
        pen.setDashPattern([self.length(), total_line_space])
        pen.setWidth(self.width)

        return pen

    def setLine(self, x, y, width, height):
        self.line_1.setLine(x, y, width, height)
        self.line_2.setLine(x, y, width, height)

    """ PROPERTIES """

    def width(self):
        return self._width

    def setWidth(self, width):
        self._width = width

    def length(self):
        return self._length

    def setLength(self, length):
        self._length = length

    def spacing(self):
        return self._spacing

    def setSpacing(self, spacing):
        self._spacing = spacing
Пример #9
0
 def addLine(self, text, color, width=1.):
     shape = QGraphicsLineItem(10., 10., 20., 20.)
     pen = makePen(color, width=width)
     shape.setPen(pen)
     self.addEntry(MapLegendEntryItem(shape, text))