dinglehopper: Give segment ids from the extracted textequiv_level

pull/38/head
Gerber, Mike 4 years ago
parent b23e4ce30e
commit b23b75b601

@ -157,11 +157,17 @@ class ExtractedText:
def segment_id_for_pos(self, pos): def segment_id_for_pos(self, pos):
# Calculate segment ids once, on the first call # Calculate segment ids once, on the first call
if not self._segment_id_for_pos: if not self._segment_id_for_pos:
if self._text is not None:
segment_id_for_pos = list(repeat(self.segment_id, len(self._text)))
else:
# Recurse
segment_id_for_pos = [] segment_id_for_pos = []
for s in self.segments: for s in self.segments:
segment_id_for_pos.extend(repeat(s.segment_id, len(s.text))) seg_ids = [s.segment_id_for_pos(i) for i in range(len(s.text))]
segment_id_for_pos.extend(seg_ids)
segment_id_for_pos.extend(repeat(None, len(self.joiner))) segment_id_for_pos.extend(repeat(None, len(self.joiner)))
segment_id_for_pos = segment_id_for_pos[:-len(self.joiner)] segment_id_for_pos = segment_id_for_pos[:-len(self.joiner)]
# This is frozen, so we have to jump through the hoop: # This is frozen, so we have to jump through the hoop:
object.__setattr__(self, '_segment_id_for_pos', segment_id_for_pos) object.__setattr__(self, '_segment_id_for_pos', segment_id_for_pos)
assert self._segment_id_for_pos assert self._segment_id_for_pos

Loading…
Cancel
Save