| from PySide.QtWidgets import QWidget, QPlainTextEdit |
| from PySide.QtGui import QPainter, QFont |
| from PySide.QtCore import Qt, QRect, QSize |
|
|
|
|
| class LineNumberArea(QWidget): |
| def __init__(self, editor): |
| super().__init__(editor) |
| self.editor = editor |
|
|
| def sizeHint(self): |
| return QSize(self.editor.fontMetrics().horizontalAdvance("999") + 4, 0) |
|
|
| def paintEvent(self, event): |
| painter = QPainter(self) |
| painter.fillRect(event.rect(), Qt.black) |
| font = QFont() |
| font.setFamily(self.editor.font().family()) |
| font.setFixedPitch(self.editor.font().fixedPitch()) |
| font.setPointSize(self.editor.font().pointSize()) |
| painter.setFont(font) |
|
|
| block = self.editor.firstVisibleBlock() |
| blockNumber = block.blockNumber() |
| top = self.editor.blockBoundingGeometry(block).translated(self.editor.contentOffset()).top() |
| bottom = top + self.editor.blockBoundingRect(block).height() |
|
|
| while block.isValid() and top <= event.rect().bottom(): |
| if block.isVisible() and bottom >= event.rect().top(): |
| number = str(blockNumber + 1) |
| painter.setPen(Qt.gray) |
| painter.drawText( |
| -6, |
| int(top), |
| self.width(), |
| self.editor.fontMetrics().height(), |
| Qt.AlignRight, |
| number, |
| ) |
|
|
| block = block.next() |
| top = bottom |
| bottom = top + self.editor.blockBoundingRect(block).height() |
| blockNumber += 1 |
|
|
|
|
| class CodeEditor(QPlainTextEdit): |
| def __init__(self, parent=None): |
| super().__init__(parent) |
| self.lineNumberArea = LineNumberArea(self) |
| self.blockCountChanged.connect(self.updateLineNumberAreaWidth) |
| self.updateRequest.connect(self.updateLineNumberArea) |
| self.verticalScrollBar().valueChanged.connect(self.lineNumberArea.update) |
|
|
| self.updateLineNumberAreaWidth(0) |
|
|
| def lineNumberAreaWidth(self): |
| digits = 2 |
| max_value = max(1, self.blockCount()) |
| while max_value >= 10: |
| max_value /= 10 |
| digits += 1 |
| space = 3 + self.fontMetrics().horizontalAdvance("9") * digits |
| return space |
|
|
| def updateLineNumberAreaWidth(self, newBlockCount): |
| self.setViewportMargins(self.lineNumberAreaWidth(), 0, 0, 0) |
|
|
| def updateLineNumberArea(self, rect, dy): |
| if dy: |
| self.lineNumberArea.scroll(0, dy) |
| else: |
| self.lineNumberArea.update(0, rect.y(), self.lineNumberArea.width(), rect.height()) |
|
|
| if rect.contains(self.viewport().rect()): |
| self.updateLineNumberAreaWidth(0) |
|
|
| def resizeEvent(self, event): |
| super().resizeEvent(event) |
| cr = self.contentsRect() |
| self.lineNumberArea.setGeometry( |
| QRect(cr.left(), cr.top(), self.lineNumberAreaWidth(), cr.height()) |
| ) |
|
|
| def setText(self, text): |
| """Backward compatibility method for setText() calls.""" |
| self.setPlainText(text) |
|
|
