Merge pull request #269 from wmvanvliet/fix-data-ptidx

fix saving overlays with text elements other than data-ptidx
gallantlab · Jun 5, 2018 · 539c4d4 · 539c4d4
2 parents 8046bc5 + 199723a
commit 539c4d4
Showing 1 changed file with 8 additions and 7 deletions.
diff --git 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