131 lines
4.2 KiB
Python
131 lines
4.2 KiB
Python
from tpers import TPrs
|
|
import ansi
|
|
# probably reinventing the wheel :/
|
|
class Rgseq(TPrs):
|
|
def InitPersistent(self, *rgseq):
|
|
self.rgseq = rgseq
|
|
def __len__(self):
|
|
return sum([len(seq) for seq in self.rgseq])
|
|
|
|
def __getitem__(self, i):
|
|
c = 0
|
|
for seq in self.rgseq:
|
|
cSeq = len(seq)
|
|
if i < c + cSeq:
|
|
return seq[i - c]
|
|
c = c + cSeq
|
|
raise IndexError
|
|
|
|
def index(self, o):
|
|
c = 0
|
|
for seq in self.rgseq:
|
|
try:
|
|
iSeq = seq.index(o)
|
|
return iSeq + c
|
|
except ValueError:
|
|
c = c + len(seq)
|
|
raise ValueError
|
|
|
|
class Snippet(object):
|
|
def __init__(self, st, ichStart, ichLim):
|
|
self.st = st
|
|
self.ichStart = ichStart
|
|
self.ichLim = ichLim
|
|
|
|
def RgstWrapped(st, w, fForEditing = False, dxStart = 0):
|
|
return [snippet.st for snippet in RgSnippetWrapped(st, w, fForEditing, dxStart)]
|
|
|
|
def RgSnippetWrapped(st, w, fForEditing = False, xStart = 0):
|
|
ichWrap = -1
|
|
ichStart = 0
|
|
for ich, ch in enumerate(st):
|
|
if ch == '\n':
|
|
assert not fForEditing, "no support yet for editing strings with newlines"
|
|
ichWrap = ich
|
|
if not fForEditing and ch == ' ':
|
|
ichWrap = ich
|
|
if (ich == ichStart + w - xStart) or (ch == '\n'):
|
|
ichStartNew = ichWrap + 1
|
|
if ichWrap < 0:
|
|
ichWrap = ich
|
|
ichStartNew = ich
|
|
elif fForEditing: # don't swallow spaces
|
|
ichWrap = ichStartNew
|
|
yield Snippet(st[ichStart:ichWrap], ichStart, ichStartNew)
|
|
ichStart = ichStartNew
|
|
ichWrap = -1
|
|
xStart = 0
|
|
if fForEditing and ch == ' ':
|
|
ichWrap = ich
|
|
|
|
if ichStart < len(st) or ichStart == 0:
|
|
yield Snippet(st[ichStart:], ichStart, len(st))
|
|
|
|
# selection helpers
|
|
def IoInc(rgo, io, fnSelectable, fRebound = True):
|
|
io = io + 1
|
|
while io < len(rgo) and not fnSelectable(rgo[io]):
|
|
io = io + 1
|
|
if io >= len(rgo) or not fnSelectable(rgo[io]):
|
|
if fRebound:
|
|
return IoDec(rgo, io, fnSelectable, False)
|
|
return -1
|
|
return io
|
|
|
|
def IoDec(rgo, io, fnSelectable, fRebound = True):
|
|
io = io - 1
|
|
while io > 0 and not fnSelectable(rgo[io]):
|
|
io = io - 1
|
|
if io < 0 or not fnSelectable(rgo[io]):
|
|
if fRebound:
|
|
return IoInc(rgo, io, fnSelectable, False)
|
|
return -1
|
|
return io
|
|
|
|
class Stedit(TPrs):
|
|
def InitPersistent(self, dgEnter = None, stInitial = ""):
|
|
self.dgEnter = dgEnter
|
|
self.SetValue(stInitial)
|
|
def SetValue(self, st):
|
|
self.st = st
|
|
self.ich = len(st)
|
|
def GetValue(self):
|
|
return self.st
|
|
def StForSize(self, fSel = True):
|
|
if fSel and self.ich == len(self.st):
|
|
return self.st + "_"
|
|
return self.st
|
|
|
|
def HandleKey(self, key):
|
|
if key == ansi.K_LEFT and self.ich > 0:
|
|
self.ich = self.ich - 1
|
|
elif key == ansi.K_RIGHT and self.ich < len(self.st):
|
|
self.ich = self.ich + 1
|
|
elif key == ansi.K_BACKSPACE and self.ich > 0:
|
|
self.st = self.st[:self.ich - 1] + self.st[self.ich:]
|
|
self.ich = self.ich - 1
|
|
elif key == ansi.K_DEL and self.ich < len(self.st):
|
|
self.st = self.st[:self.ich] + self.st[self.ich + 1:]
|
|
elif key == ansi.K_HOME:
|
|
self.ich = 0
|
|
elif key == ansi.K_END:
|
|
self.ich = len(self.st)
|
|
elif ansi.FKeyPrintable(key):
|
|
self.st = self.st[:self.ich] + key + self.st[self.ich:]
|
|
self.ich = self.ich + 1
|
|
elif ansi.FEnter(key) and self.dgEnter != None:
|
|
if not self.dgEnter(self.st):
|
|
self.st = ""
|
|
self.ich = 0
|
|
else:
|
|
return False
|
|
return True
|
|
|
|
def DrawCursor(self, ascr, x, y, colFg, colBg, snippet):
|
|
if self.ich >= snippet.ichStart and (snippet.ichLim < 0 or self.ich < snippet.ichLim):
|
|
if self.ich == len(self.st):
|
|
ch = ' '
|
|
else:
|
|
ch = self.st[self.ich]
|
|
ascr.PutAch(ansi.MkAch(ch, colFg, colBg), x + self.ich - snippet.ichStart, y)
|