-
Notifications
You must be signed in to change notification settings - Fork 0
/
textblock.py
105 lines (85 loc) · 2.29 KB
/
textblock.py
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
# textblock.py == manipulate blocks of text
"""
Functions on 2-d blocks of text
"""
from typeguard import typechecked
#---------------------------------------------------------------------
"""
Example:
textRec("foo\n1\n2", "bar", "aaaa\nbbbbbbbb\ncc\ndddd", "baz\nforth")
would return:
'''\
foo bar aaaa baz
1 bbbbbbbb forth
2 cc
dddd
'''
"""
@typechecked
def joinTextRects(*ss: str) -> str:
""" each string in (s) is a multiline string. They are put together
horizontally next to each other.
"""
if len(ss) < 1: return ""
numRows = 0
numCols = 0
colsForSS = []
for s in ss:
r,c = getExtent(s)
numRows = max(numRows, r)
colsForSS += [c]
#//for s
ssJust: list[list[str]] = []
for s, colForS in zip(ss, colsForSS):
sJust = makeSize(s, numRows, colForS)
ssJust += [sJust]
#//for s, colForS
result = ""
for r in range(numRows):
for sJust in ssJust:
result += sJust[r] + " "
#//for sJust
result = result[:-1] + "\n"
#//for r
return result
@typechecked
def getExtent(s: str) -> tuple[int,int]:
""" return the number of rows and columns in a string """
if s=="": return (0,0)
a = s.split("\n")
r = len(a)
if r==0: return (0,0)
c = max(len(line) for line in a)
return (r,c)
@typechecked
def makeRightSize(s: str, r: int) -> list[str]:
""" Split (s) into a list of strings, where each string is the
same length as the longest line in (s). There most be (r) rows.
"""
_, numCols = getExtent(s)
return makeSize(s, r, numCols)
@typechecked
def makeSize(s: str, r: int, c: int) -> list[str]:
""" Output is like (s), but it has (r) rows each with (c) columns,
adding spaces where necessary.
"""
if r==0: return []
a = s.split("\n") + [""]*r
result = []
doneLines = 0
for line in a:
result += [padTo(line, c)]
doneLines += 1
if doneLines >= r: break
#//for line
return result
@typechecked
def padTo(s: str, x: int) -> str:
""" make the string exactly (x) characters lone """
if x <= 0: return ""
if len(s)>x:
return s[:x]
else:
return s + " "*(x-len(s))
#---------------------------------------------------------------------
#end