Function parse_poster_xml_for_figures

PosterAgent/tree_split_layout.py:238–273

(xml_path)

def parse_poster_xml_for_figures(xml_path):
    root = parse_xml_with_recovery(xml_path)

    poster_w = float(root.get("Width", "1"))
    poster_h = float(root.get("Height", "1"))
    poster_area = poster_w * poster_h

    records = []

    for panel in root.findall("Panel"):
        px, py = float(panel.get("left", 0)), float(panel.get("right", 0))
        pw, ph = float(panel.get("width", 1)), float(panel.get("height", 1))
        panel_area = pw * ph
        sp = panel_area / poster_area
        rp = pw / ph if ph > 0 else 1.0

        lp = sum(len(t.text.strip()) for t in panel.findall("Text") if t.text)

        for fig in panel.findall("Figure"):
            fx, fy = float(fig.get("left", 0)), float(fig.get("right", 0))
            fw, fh = float(fig.get("width", 1)), float(fig.get("height", 1))

            sg = (fw * fh) / poster_area
            rg = fw / fh if fh > 0 else 1.0
            ug = fw / pw if pw > 0 else 0.1

            panel_center_x = px + pw / 2
            fig_center_x = fx + fw / 2
            delta_x = fig_center_x - panel_center_x

            hg = 0 if delta_x < -pw / 6 else (2 if delta_x > pw / 6 else 1)

            record = {"sp": sp, "rp": rp, "lp": lp, "sg": sg, "rg": rg, "hg": hg, "ug": ug}
            records.append(record)

    return records


def train_figure_model(figure_records):