From 199723a04830ca3709e79c06c349c94dbc8cf9d0 Mon Sep 17 00:00:00 2001 From: Marijn van Vliet Date: Fri, 18 May 2018 09:51:24 +0300 Subject: [PATCH] fix saving overlays with text elements other than data-ptidx --- cortex/brainctm.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/cortex/brainctm.py b/cortex/brainctm.py index 41a38d0c2..03a68248f 100644 --- a/cortex/brainctm.py +++ b/cortex/brainctm.py @@ -170,13 +170,14 @@ def save(self, path, method='mg2', **kwargs): # assign coordinates in left hemisphere negative values with open(svgname, "wb") as fp: for element in svg.svg.findall(".//{http://www.w3.org/2000/svg}text"): - idx = int(element.attrib["data-ptidx"]) - if idx < len(inverse[0]): - idx = inverse[0][idx] - else: - idx -= len(inverse[0]) - idx = inverse[1][idx] + len(inverse[0]) - element.attrib["data-ptidx"] = str(idx) + if 'data-ptidx' in element.attrib: + idx = int(element.attrib["data-ptidx"]) + if idx < len(inverse[0]): + idx = inverse[0][idx] + else: + idx -= len(inverse[0]) + idx = inverse[1][idx] + len(inverse[0]) + element.attrib["data-ptidx"] = str(idx) fp.write(svg.toxml()) return ptmap