Content warnings.
[librarian.git] / src / librarian / hyphenator.py
1 """
2
3 This is a Pure Python module to hyphenate text.
4
5 It is inspired by Ruby's Text::Hyphen, but currently reads standard *.dic files,
6 that must be installed separately.
7
8 In the future it's maybe nice if dictionaries could be distributed together with
9 this module, in a slightly prepared form, like in Ruby's Text::Hyphen.
10
11 Wilbert Berendsen, March 2008
12 info@wilbertberendsen.nl
13
14 License: LGPL.
15
16 """
17 from __future__ import print_function, unicode_literals
18
19 import sys
20 import re
21
22 __all__ = ("Hyphenator")
23
24 # cache of per-file Hyph_dict objects
25 hdcache = {}
26
27 # precompile some stuff
28 parse_hex = re.compile(r'\^{2}([0-9a-f]{2})').sub
29 parse = re.compile(r'(\d?)(\D?)').findall
30
31 def hexrepl(matchObj):
32     return unichr(int(matchObj.group(1), 16))
33
34
35 class parse_alt(object):
36     """
37     Parse nonstandard hyphen pattern alternative.
38     The instance returns a special int with data about the current position
39     in the pattern when called with an odd value.
40     """
41     def __init__(self, pat, alt):
42         alt = alt.split(',')
43         self.change = alt[0]
44         if len(alt) > 2:
45             self.index = int(alt[1])
46             self.cut = int(alt[2]) + 1
47         else:
48             self.index = 1
49             self.cut = len(re.sub(r'[\d\.]', '', pat)) + 1
50         if pat.startswith('.'):
51             self.index += 1
52
53     def __call__(self, val):
54         self.index -= 1
55         val = int(val)
56         if val & 1:
57             return dint(val, (self.change, self.index, self.cut))
58         else:
59             return val
60
61
62 class dint(int):
63     """
64     Just an int some other data can be stuck to in a data attribute.
65     Call with ref=other to use the data from the other dint.
66     """
67     def __new__(cls, value, data=None, ref=None):
68         obj = int.__new__(cls, value)
69         if ref and type(ref) == dint:
70             obj.data = ref.data
71         else:
72             obj.data = data
73         return obj
74
75
76 class Hyph_dict(object):
77     """
78     Reads a hyph_*.dic file and stores the hyphenation patterns.
79     Parameters:
80     -filename : filename of hyph_*.dic to read
81     """
82     def __init__(self, filename):
83         self.patterns = {}
84         f = open(filename)
85         charset = f.readline().strip()
86         if charset.startswith('charset '):
87             charset = charset[8:].strip()
88
89         for pat in f:
90             pat = pat.decode(charset).strip()
91             if not pat or pat[0] == '%': continue
92             # replace ^^hh with the real character
93             pat = parse_hex(hexrepl, pat)
94             # read nonstandard hyphen alternatives
95             if '/' in pat:
96                 pat, alt = pat.split('/', 1)
97                 factory = parse_alt(pat, alt)
98             else:
99                 factory = int
100             tag, value = zip(*[(s, factory(i or "0")) for i, s in parse(pat)])
101             # if only zeros, skip this pattern
102             if max(value) == 0: continue
103             # chop zeros from beginning and end, and store start offset.
104             start, end = 0, len(value)
105             while not value[start]: start += 1
106             while not value[end-1]: end -= 1
107             self.patterns[''.join(tag)] = start, value[start:end]
108         f.close()
109         self.cache = {}
110         self.maxlen = max(map(len, self.patterns.keys()))
111
112     def positions(self, word):
113         """
114         Returns a list of positions where the word can be hyphenated.
115         E.g. for the dutch word 'lettergrepen' this method returns
116         the list [3, 6, 9].
117
118         Each position is a 'data int' (dint) with a data attribute.
119         If the data attribute is not None, it contains a tuple with
120         information about nonstandard hyphenation at that point:
121         (change, index, cut)
122
123         change: is a string like 'ff=f', that describes how hyphenation
124             should take place.
125         index: where to substitute the change, counting from the current
126             point
127         cut: how many characters to remove while substituting the nonstandard
128             hyphenation
129         """
130         word = word.lower()
131         points = self.cache.get(word)
132         if points is None:
133             prepWord = '.%s.' % word
134             res = [0] * (len(prepWord) + 1)
135             for i in range(len(prepWord) - 1):
136                 for j in range(i + 1, min(i + self.maxlen, len(prepWord)) + 1):
137                     p = self.patterns.get(prepWord[i:j])
138                     if p:
139                         offset, value = p
140                         s = slice(i + offset, i + offset + len(value))
141                         res[s] = map(max, value, res[s])
142
143             points = [dint(i - 1, ref=r) for i, r in enumerate(res) if r % 2]
144             self.cache[word] = points
145         return points
146
147
148 class Hyphenator(object):
149     """
150     Reads a hyph_*.dic file and stores the hyphenation patterns.
151     Provides methods to hyphenate strings in various ways.
152     Parameters:
153     -filename : filename of hyph_*.dic to read
154     -left: make the first syllabe not shorter than this
155     -right: make the last syllabe not shorter than this
156     -cache: if true (default), use a cached copy of the dic file, if possible
157
158     left and right may also later be changed:
159       h = Hyphenator(file)
160       h.left = 1
161     """
162     def __init__(self, filename, left=2, right=2, cache=True):
163         self.left  = left
164         self.right = right
165         if not cache or filename not in hdcache:
166             hdcache[filename] = Hyph_dict(filename)
167         self.hd = hdcache[filename]
168
169     def positions(self, word):
170         """
171         Returns a list of positions where the word can be hyphenated.
172         See also Hyph_dict.positions. The points that are too far to
173         the left or right are removed.
174         """
175         right = len(word) - self.right
176         return [i for i in self.hd.positions(word) if self.left <= i <= right]
177
178     def iterate(self, word):
179         """
180         Iterate over all hyphenation possibilities, the longest first.
181         """
182         if isinstance(word, str):
183             word = word.decode('latin1')
184         for p in reversed(self.positions(word)):
185             if p.data:
186                 # get the nonstandard hyphenation data
187                 change, index, cut = p.data
188                 if word.isupper():
189                     change = change.upper()
190                 c1, c2 = change.split('=')
191                 yield word[:p+index] + c1, c2 + word[p+index+cut:]
192             else:
193                 yield word[:p], word[p:]
194
195     def wrap(self, word, width, hyphen='-'):
196         """
197         Return the longest possible first part and the last part of the
198         hyphenated word. The first part has the hyphen already attached.
199         Returns None, if there is no hyphenation point before width, or
200         if the word could not be hyphenated.
201         """
202         width -= len(hyphen)
203         for w1, w2 in self.iterate(word):
204             if len(w1) <= width:
205                 return w1 + hyphen, w2
206
207     def inserted(self, word, hyphen='-'):
208         """
209         Returns the word as a string with all the possible hyphens inserted.
210         E.g. for the dutch word 'lettergrepen' this method returns
211         the string 'let-ter-gre-pen'. The hyphen string to use can be
212         given as the second parameter, that defaults to '-'.
213         """
214         if isinstance(word, str):
215             word = word.decode('latin1')
216         l = list(word)
217         for p in reversed(self.positions(word)):
218             if p.data:
219                 # get the nonstandard hyphenation data
220                 change, index, cut = p.data
221                 if word.isupper():
222                     change = change.upper()
223                 l[p + index : p + index + cut] = change.replace('=', hyphen)
224             else:
225                 l.insert(p, hyphen)
226         return ''.join(l)
227
228     __call__ = iterate
229
230
231 if __name__ == "__main__":
232
233     dict_file = sys.argv[1]
234     word = sys.argv[2].decode('latin1')
235
236     h = Hyphenator(dict_file, left=1, right=1)
237
238     for i in h(word):
239         print(i)
240