Skip to content

Commit

Permalink
Make iter_lines return the first two lines
Browse files Browse the repository at this point in the history 
There's no need to get more.
  • Loading branch information
@liZe
liZe committed Nov 16, 2018
1 parent 5f0723a commit 32d2359
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 40 deletions.
6 changes: 3 additions & 3 deletions weasyprint/css/computed_values.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -304,7 +304,7 @@ def length(computer, name, value, font_size=None, pixels_only=False):
context=None, font_size=font_size,
style=computer.computed)
layout.set_text('0')
line, = layout.iter_lines()
line, _ = layout.iter_lines()
logical_width, _ = text.get_size(line, computer.computed)
result = value.value * logical_width
elif unit == 'em':
Expand Down Expand Up @@ -682,7 +682,7 @@ def strut_layout(style, context=None):

layout = text.Layout(
context=context, font_size=style['font_size'], style=style)
line, = layout.iter_lines()
line, _ = layout.iter_lines()
_, _, _, _, text_height, baseline = text.first_line_metrics(
line, '', layout, resume_at=None, space_collapse=False, style=style)
if style['line_height'] == 'normal':
Expand All @@ -705,7 +705,7 @@ def ex_ratio(style):
# TODO: use context to use @font-face fonts
layout = text.Layout(context=None, font_size=font_size, style=style)
layout.set_text('x')
line, = layout.iter_lines()
line, _ = layout.iter_lines()
_, ink_height_above_baseline = text.get_ink_position(line)
# Zero means some kind of failure, fallback is 0.5.
# We round to try keeping exact values that were altered by Pango.
Expand Down
58 changes: 21 additions & 37 deletions weasyprint/text.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -593,7 +593,7 @@ def first_line_metrics(first_line, text, layout, resume_at, space_collapse,
first_line_text = first_line_text.rstrip(u' ')
# Remove soft hyphens
layout.set_text(first_line_text.replace(u'\u00ad', u''))
first_line = next(layout.iter_lines(), None)
first_line, _ = layout.iter_lines()
length = first_line.length if first_line is not None else 0
soft_hyphens = 0
if u'\u00ad' in first_line_text:
Expand Down Expand Up @@ -663,13 +663,15 @@ def __init__(self, context, font_size, style):
pango.pango_layout_set_font_description(self.layout, self.font)

def iter_lines(self):
layout_iter = ffi.gc(
pango.pango_layout_get_iter(self.layout),
pango.pango_layout_iter_free)
while 1:
yield pango.pango_layout_iter_get_line_readonly(layout_iter)
if not pango.pango_layout_iter_next_line(layout_iter):
return
layout_iter = pango.pango_layout_get_iter(self.layout)
first_line = pango.pango_layout_iter_get_line_readonly(layout_iter)
if pango.pango_layout_iter_next_line(layout_iter):
second_line = pango.pango_layout_iter_get_line_readonly(
layout_iter)
else:
second_line = None
pango.pango_layout_iter_free(layout_iter)
return first_line, second_line

def set_text(self, text):
# Keep only the first two lines, we don't need the other ones
Expand All @@ -692,7 +694,7 @@ def set_tabs(self, style):
context=self.context, font_size=style['font_size'],
style=style)
layout.set_text(' ' * style['tab_size'])
line, = layout.iter_lines()
line, _ = layout.iter_lines()
width, _ = get_size(line, style)
width = int(round(width))
else:
Expand Down Expand Up @@ -952,19 +954,15 @@ def split_first_line(text, style, context, max_width, justification_spacing,
layout = create_layout(
text[:expected_length], style, context, max_width,
justification_spacing)
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
if second_line is None:
# The small amount of text fits in one line, give up and use
# the whole text
layout = None
if layout is None:
layout = create_layout(
text, style, context, max_width, justification_spacing)
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
resume_at = None if second_line is None else second_line.start_index

# Step #2: Don't split lines when it's not needed
Expand Down Expand Up @@ -998,9 +996,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
# only try when space collapsing is allowed
new_first_line_text = first_line_text + next_word
layout.set_text(new_first_line_text)
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
first_line_width, _ = get_size(first_line, style)
if second_line is None and first_line_text:
# The next word fits in the first line, keep the layout
Expand Down Expand Up @@ -1046,9 +1042,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
first_line_text.rsplit(u' ', 1))
next_word = u' ' + next_word
layout.set_text(first_line_text)
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
resume_at = len(
(first_line_text + u' ').encode('utf8'))
else:
Expand Down Expand Up @@ -1080,9 +1074,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
new_layout = create_layout(
hyphenated_first_line_text, style, context, max_width,
justification_spacing)
new_lines = new_layout.iter_lines()
new_first_line = next(new_lines, None)
new_second_line = next(new_lines, None)
new_first_line, new_second_line = new_layout.iter_lines()
new_first_line_width, _ = get_size(new_first_line, style)
new_space = max_width - new_first_line_width
if new_second_line is None and (
Expand All @@ -1104,9 +1096,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
layout.set_text(hyphenated_first_line_text)
pango.pango_layout_set_width(
layout.layout, units_from_double(-1))
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
resume_at = len(new_first_line_text.encode('utf8'))
if text[len(first_line_text)] == soft_hyphen:
resume_at += len(soft_hyphen.encode('utf8'))
Expand All @@ -1120,9 +1110,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
layout.set_text(hyphenated_first_line_text)
pango.pango_layout_set_width(
layout.layout, units_from_double(-1))
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
resume_at = len(first_line_text.encode('utf8'))

# Step 5: Try to break word if it's too long for the line
Expand All @@ -1143,9 +1131,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
pango.pango_layout_set_width(
layout.layout, units_from_double(max_width))
layout.set_wrap(PANGO_WRAP_MODE['WRAP_CHAR'])
temp_lines = layout.iter_lines()
next(temp_lines, None)
temp_second_line = next(temp_lines, None)
_, temp_second_line = layout.iter_lines()
temp_second_line_index = (
len(text.encode('utf-8')) if temp_second_line is None
else temp_second_line.start_index)
Expand All @@ -1157,9 +1143,7 @@ def split_first_line(text, style, context, max_width, justification_spacing,
# link given in step #3.
first_line_text = utf8_slice(text, slice(temp_second_line_index))
layout.set_text(first_line_text)
lines = layout.iter_lines()
first_line = next(lines, None)
second_line = next(lines, None)
first_line, second_line = layout.iter_lines()
resume_at = (
first_line.length if second_line is None
else second_line.start_index)
Expand All @@ -1181,7 +1165,7 @@ def show_first_line(context, pango_layout, hinting):
# of hinting.
pango.pango_layout_set_width(pango_layout.layout, -1)
pangocairo.pango_cairo_show_layout_line(
context, next(pango_layout.iter_lines()))
context, pango_layout.iter_lines()[0])


def can_break_text(text, lang):
Expand Down

0 comments on commit 32d2359

Please sign in to comment.