File size: 3,133 Bytes
985c397 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 | 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) # Simple update
self.updateLineNumberAreaWidth(0) # Initial call
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)
|