def edit(self, cursor):
"""Edit the block at the specified QTextCursor."""
if self._document:
self._document.closed.disconnect(self.reject)
self._document = cursor.document()
self._document.closed.connect(self.reject)
# don't change the cursor
c = self._range = QTextCursor(cursor)
cursorpos = c.position() - c.block().position()
cursortools.strip_indent(c)
indentpos = c.position() - c.block().position()
c.movePosition(QTextCursor.EndOfBlock, QTextCursor.KeepAnchor)
self.view.setPlainText(c.selection().toPlainText())
self.highlighter.setInitialState(tokeniter.state(cursortools.block(cursor)))
self.highlighter.setHighlighting(metainfo.info(cursor.document()).highlighting)
self.highlighter.rehighlight()
# let autocomplete query the real document as if we're at the start
# of the current block
self.completer.document_cursor = QTextCursor(cursor.block())
self.completer.autoComplete = QSettings().value("autocomplete", True, bool)
cursor = self.view.textCursor()
cursor.setPosition(max(0, cursorpos-indentpos))
self.view.setTextCursor(cursor)
self.updateMessage()
def edit(self, cursor):
"""Edit the block at the specified QTextCursor."""
if self._document:
self._document.closed.disconnect(self.reject)
self._document = cursor.document()
self._document.closed.connect(self.reject)
# don't change the cursor
c = self._range = QTextCursor(cursor)
cursorpos = c.position() - c.block().position()
cursortools.strip_indent(c)
indentpos = c.position() - c.block().position()
c.movePosition(QTextCursor.EndOfBlock, QTextCursor.KeepAnchor)
self.view.setPlainText(c.selection().toPlainText())
self.highlighter.setInitialState(
tokeniter.state(cursortools.block(cursor)))
self.highlighter.setHighlighting(
metainfo.info(cursor.document()).highlighting)
self.highlighter.rehighlight()
# let autocomplete query the real document as if we're at the start
# of the current block
self.completer.document_cursor = QTextCursor(cursor.block())
self.completer.autoComplete = QSettings().value("autocomplete",
True) not in ('false',
False)
cursor = self.view.textCursor()
cursor.setPosition(max(0, cursorpos - indentpos))
self.view.setTextCursor(cursor)
self.updateMessage()
def cut_assign(cursor):
"""Cuts selected text and assigns it to a LilyPond variable."""
# ask the variable name
name = inputdialog.getText(None, _("Cut and Assign"), _(
"Please enter the name for the variable to assign the selected "
"text to:"), regexp="[A-Za-z]+")
if not name:
return
cursortools.strip_selection(cursor)
# determine state at cursor
block = cursortools.block(cursor)
state = tokeniter.state(block)
for t in tokeniter.partition(cursor).left:
state.follow(t)
mode = ""
for p in state.parsers():
if isinstance(p, ly.lex.lilypond.ParseInputMode):
if isinstance(p, ly.lex.lilypond.ParseLyricMode):
mode = " \\lyricmode"
elif isinstance(p, ly.lex.lilypond.ParseChordMode):
mode = " \\chordmode"
elif isinstance(p, ly.lex.lilypond.ParseFigureMode):
mode = " \\figuremode"
elif isinstance(p, ly.lex.lilypond.ParseDrumMode):
mode = " \\drummode"
break
# find insertion place:
found = False
while block.previous().isValid():
block = block.previous()
state = tokeniter.state(block)
if isinstance(state.parser(), ly.lex.lilypond.ParseGlobal):
found = True
break
tokens = tokeniter.tokens(block)
for t in tokens:
if isinstance(t, ly.lex.lilypond.Name):
found = True
break
elif not isinstance(t, (ly.lex.Space, ly.lex.Comment)):
break
if found:
break
insert = QTextCursor(block)
text = cursor.selection().toPlainText()
space = '\n' if '\n' in text else ' '
text = ''.join((name, ' =', mode, ' {', space, text, space, '}\n\n'))
with cursortools.compress_undo(cursor):
cursor.insertText('\\' + name)
pos = insert.selectionStart()
insert.insertText(text)
if metainfo.info(cursor.document()).auto_indent:
insert.setPosition(pos, QTextCursor.KeepAnchor)
with cursortools.compress_undo(insert, True):
indent.re_indent(insert)
def state(blockOrCursor):
"""Returns a thawn ly.lex.State() object at the beginning of the given QTextBlock.
If the argument is a QTextCursor, uses the current block or the first block of its selection.
"""
if isinstance(blockOrCursor, QTextCursor):
block = cursortools.block(blockOrCursor)
else:
block = blockOrCursor
if block.userState() == -1:
highlighter.highlighter(block.document()).rehighlight()
return highlighter.highlighter(block.document()).state(block)
def partition(cursor):
"""Returns a named three-tuple (left, middle, right).
left is a tuple of tokens left to the cursor.
middle is the token that overlaps the cursor at both sides (or None).
right is a tuple of tokens right to the cursor.
"""
block = cursortools.block(cursor)
t = tokens(block)
i = index(cursor)
if t:
if i < len(t) and t[i].pos < cursor.selectionStart() - block.position():
return Partition(t[:i], t[i], t[i+1:])
return Partition(t[:i], None, t[i:])
def partition(cursor):
"""Returns a named three-tuple (left, middle, right).
left is a tuple of tokens left to the cursor.
middle is the token that overlaps the cursor at both sides (or None).
right is a tuple of tokens right to the cursor.
"""
block = cursortools.block(cursor)
t = tokens(block)
i = index(cursor)
if t:
if i < len(
t) and t[i].pos < cursor.selectionStart() - block.position():
return Partition(t[:i], t[i], t[i + 1:])
return Partition(t[:i], None, t[i:])
def index(cursor):
"""Returns the index of the token at the cursor (right or overlapping).
The index can range from 0 (if there are no tokens or the cursor is in the
first token) to the total count of tokens in the cursor's block (if the
cursor is at the very end of the block).
"""
block = cursortools.block(cursor)
tokens_ = tokens(block)
if cursor.atBlockEnd():
return len(tokens_)
pos = cursor.selectionStart() - block.position()
lo, hi = 0, len(tokens_)
while lo < hi:
mid = (lo + hi) // 2
if pos < tokens_[mid].pos:
hi = mid
else:
lo = mid + 1
return lo - 1
def index(cursor):
"""Returns the index of the token at the cursor (right or overlapping).
The index can range from 0 (if there are no tokens or the cursor is in the
first token) to the total count of tokens in the cursor's block (if the
cursor is at the very end of the block).
"""
block = cursortools.block(cursor)
tokens_ = tokens(block)
if cursor.atBlockEnd():
return len(tokens_)
pos = cursor.selectionStart() - block.position()
lo, hi = 0, len(tokens_)
while lo < hi:
mid = (lo + hi) // 2
if pos < tokens_[mid].pos:
hi = mid
else:
lo = mid + 1
return lo - 1
def from_cursor(cursor, state=None, first=1):
"""Yields block, tokens starting at the cursor position.
If state is given, it should be the state at the start of the block
the selection begins. (Use state(cursor) to get that.)
If first is -1: starts with the token that touches the cursor at the right
If first is 0: starts with the token that overlaps the cursor
If first is 1 (default): starts with the first token to the right.
"""
block = cursortools.block(cursor)
pos = cursor.selectionStart() - block.position()
if state:
def token_source(block):
for t in tokens(block):
state.follow(t)
yield t
else:
def token_source(block):
return iter(tokens(block))
if first == -1:
pred = lambda t: t.end < pos
elif first == 0:
pred = lambda t: t.end <= pos
else:
pred = lambda t: t.pos < pos
def source_start(block):
source = token_source(block)
for t in source:
if not pred(t):
yield t
for t in source:
yield t
source = source_start
while block.isValid():
yield block, source(block)
block = block.next()
source = token_source
def cut_assign(cursor):
"""Cuts selected text and assigns it to a LilyPond variable."""
# ask the variable name
name = inputdialog.getText(
None,
_("Cut and Assign"),
_("Please enter the name for the variable to assign the selected "
"text to:"),
regexp="[A-Za-z]+")
if not name:
return
cursortools.strip_selection(cursor)
# determine state at cursor
block = cursortools.block(cursor)
state = tokeniter.state(block)
for t in tokeniter.partition(cursor).left:
state.follow(t)
mode = ""
for p in state.parsers():
if isinstance(p, ly.lex.lilypond.ParseInputMode):
if isinstance(p, ly.lex.lilypond.ParseLyricMode):
mode = " \\lyricmode"
elif isinstance(p, ly.lex.lilypond.ParseChordMode):
mode = " \\chordmode"
elif isinstance(p, ly.lex.lilypond.ParseFigureMode):
mode = " \\figuremode"
elif isinstance(p, ly.lex.lilypond.ParseDrumMode):
mode = " \\drummode"
break
# find insertion place:
found = False
while block.previous().isValid():
block = block.previous()
state = tokeniter.state(block)
if isinstance(state.parser(), ly.lex.lilypond.ParseGlobal):
found = True
break
tokens = tokeniter.tokens(block)
for t in tokens:
if isinstance(t, ly.lex.lilypond.Name):
found = True
break
elif not isinstance(t, (ly.lex.Space, ly.lex.Comment)):
break
if found:
break
insert = QTextCursor(block)
text = cursor.selection().toPlainText()
space = '\n' if '\n' in text else ' '
text = ''.join((name, ' =', mode, ' {', space, text, space, '}\n\n'))
with cursortools.compress_undo(cursor):
cursor.insertText('\\' + name)
pos = insert.selectionStart()
insert.insertText(text)
if metainfo.info(cursor.document()).auto_indent:
insert.setPosition(pos, QTextCursor.KeepAnchor)
with cursortools.compress_undo(insert, True):
indent.re_indent(insert)
