Function wrap_text_precise

agentocr/utils.py:305–345

Precise text wrapping using actual font measurements for optimal packing. Returns a list of tuples (line_text, is_paragraph_end) to track paragraph boundaries.

(text: str, max_width: int, font, font_size: int)

def wrap_text_precise(text: str, max_width: int, font, font_size: int) -> List[Tuple[str, bool]]:
    """
    Precise text wrapping using actual font measurements for optimal packing.
    Returns a list of tuples (line_text, is_paragraph_end) to track paragraph boundaries.
    """
    lines = []
    paragraphs = text.split('\n')
    
    for para_idx, paragraph in enumerate(paragraphs):
        paragraph = paragraph.strip()
        if not paragraph:
            lines.append(("", True))
            continue
        
        words = paragraph.split()
        current_line = ""
        
        for word in words:
            test_line = current_line + (" " if current_line else "") + word
            
            # Use actual font measurement
            try:
                bbox = font.getbbox(test_line)
                text_width = bbox[2] - bbox[0]
            except:
                # Fallback to character-based estimation
                text_width = len(test_line) * font_size * 0.6
            
            if text_width <= max_width:
                current_line = test_line
            else:
                if current_line:
                    lines.append((current_line, False))
                current_line = word
        
        if current_line:
            # Mark the last line of each paragraph
            is_last_para = para_idx == len(paragraphs) - 1
            lines.append((current_line, not is_last_para))
    
    return lines


# Cache for font metrics to avoid repeated calculations