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
|
from qtpy.QtCore import Qt, QRect
from qtpy.QtWidgets import QWidget
from qtpy.QtGui import QPainter
class PromptArea(QWidget):
"""Widget that displays the prompts on the left of the input area."""
def __init__(self, edit, get_text, highlighter):
super(PromptArea, self).__init__(edit)
self.setFixedWidth(0)
self.edit = edit
self.get_text = get_text
self.highlighter = highlighter
edit.updateRequest.connect(self.updateContents)
def paintEvent(self, event):
edit = self.edit
height = edit.fontMetrics().height()
block = edit.firstVisibleBlock()
count = block.blockNumber()
painter = QPainter(self)
painter.fillRect(event.rect(), edit.palette().base())
first = True
while block.isValid():
count += 1
block_top = edit.blockBoundingGeometry(block).translated(
edit.contentOffset()).top()
if not block.isVisible() or block_top > event.rect().bottom():
break
rect = QRect(0, int(block_top), self.width(), height)
self.draw_block(painter, rect, block, first)
first = False
block = block.next()
painter.end()
super(PromptArea, self).paintEvent(event)
def updateContents(self, rect, scroll):
if scroll:
self.scroll(0, scroll)
else:
self.update()
def adjust_width(self, new_text):
width = calc_text_width(self.edit, new_text)
if width > self.width():
self.setFixedWidth(width)
def draw_block(self, painter, rect, block, first):
"""Draw the info corresponding to a given block (text line) of the text
document."""
pen = painter.pen()
text = self.get_text(block.blockNumber())
default = self.edit.currentCharFormat()
formats = [default] * len(text)
painter.setFont(self.edit.font())
for index, length, format in self.highlighter.highlight(text):
formats[index:index+length] = [format] * length
for idx, (char, format) in enumerate(zip(text, formats)):
rpos = len(text) - idx - 1
pen.setColor(format.foreground().color())
painter.setPen(pen)
painter.drawText(rect, Qt.AlignRight, text[idx] + ' ' * rpos)
def calc_text_width(widget, text):
"""Estimate the width that the given text would take within the widget."""
return (widget.fontMetrics().width(text) +
widget.fontMetrics().width('M') +
widget.contentsMargins().left() +
widget.contentsMargins().right())
|
