Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

Fix (sort of) words crossing right text box edge

Browse Source
This commit is contained in:
Matthias Bisping 2023-02-13 18:44:10 +01:00
parent 4ec7cb8d7b
commit be0c643f75
1 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
synthesis/text/text_block_generator/paragraph.py
View File

@ -2,30 +2,47 @@ import textwrap
from typing import List from typing import List
from faker import Faker from faker import Faker
from funcy import identity, iterate, take, last from funcy import iterate, take, last
from cv_analysis.logging import debug_log
from cv_analysis.utils import star from cv_analysis.utils import star
from cv_analysis.utils.rectangle import Rectangle from cv_analysis.utils.rectangle import Rectangle
from synthesis.random import rnd from synthesis.random import rnd
from synthesis.text.text_block_generator.text_block_generator import TextBlockGenerator from synthesis.text.line_formatter.identity import IdentityLineFormatter
from synthesis.text.line_formatter.line_formatter import LineFormatter
from synthesis.text.line_formatter.paragraph import ParagraphLineFormatter from synthesis.text.line_formatter.paragraph import ParagraphLineFormatter
from synthesis.text.text_block_generator.text_block_generator import TextBlockGenerator
class ParagraphGenerator(TextBlockGenerator): class ParagraphGenerator(TextBlockGenerator):
def __init__(self): def __init__(self):
self.line_formatter = ParagraphLineFormatter(blank_line_percentage=rnd.uniform(0, 0.5)) self.line_formatter = ParagraphLineFormatter(blank_line_percentage=rnd.uniform(0, 0.5))
def __call__(self, rectangle, n_sentences): @debug_log(exits=False)
def __call__(self, rectangle, n_sentences=None):
return self.generate_paragraph(rectangle, n_sentences) return self.generate_paragraph(rectangle, n_sentences)
def generate_paragraph(self, rectangle, n_sentences): @debug_log(exits=False)
def generate_paragraph(self, rectangle, n_sentences=None):
lines = generate_random_text_lines(rectangle, self.line_formatter, n_sentences) lines = generate_random_text_lines(rectangle, self.line_formatter, n_sentences)
return lines return lines
def generate_random_text_lines(rectangle: Rectangle, line_formatter=identity, n_sentences=3000) -> List[str]: @debug_log(exits=False)
def generate_random_text_lines(
rectangle: Rectangle,
line_formatter: LineFormatter = None,
n_sentences=None,
) -> List[str]:
n_sentences = n_sentences or 3000 # TODO: De-hardcode.
line_formatter = line_formatter or IdentityLineFormatter()
text = Faker().paragraph(nb_sentences=n_sentences, variable_nb_sentences=False, ext_word_list=None) text = Faker().paragraph(nb_sentences=n_sentences, variable_nb_sentences=False, ext_word_list=None)
unformatted_lines = textwrap.wrap(text, width=rectangle.width, break_long_words=False) unformatted_lines = textwrap.wrap(
text,
width=rectangle.width // rnd.uniform(4, 5), # TODO: De-hardcode.
break_long_words=True,
)
# each iteration of the line formatter function formats one more line and adds it to the back of the list # each iteration of the line formatter function formats one more line and adds it to the back of the list
formatted_lines_generator = iterate(star(line_formatter), (unformatted_lines, True)) formatted_lines_generator = iterate(star(line_formatter), (unformatted_lines, True))
# hence do as many iterations as there are lines in the rectangle # hence do as many iterations as there are lines in the rectangle