| 509 | return data |
| 510 | |
| 511 | def add_viz(self, element, node_data): |
| 512 | viz = node_data.pop("viz", False) |
| 513 | if viz: |
| 514 | color = viz.get("color") |
| 515 | if color is not None: |
| 516 | if self.VERSION == "1.1": |
| 517 | e = Element( |
| 518 | f"{{{self.NS_VIZ}}}color", |
| 519 | r=str(color.get("r")), |
| 520 | g=str(color.get("g")), |
| 521 | b=str(color.get("b")), |
| 522 | ) |
| 523 | else: |
| 524 | e = Element( |
| 525 | f"{{{self.NS_VIZ}}}color", |
| 526 | r=str(color.get("r")), |
| 527 | g=str(color.get("g")), |
| 528 | b=str(color.get("b")), |
| 529 | a=str(color.get("a")), |
| 530 | ) |
| 531 | element.append(e) |
| 532 | |
| 533 | size = viz.get("size") |
| 534 | if size is not None: |
| 535 | e = Element(f"{{{self.NS_VIZ}}}size", value=str(size)) |
| 536 | element.append(e) |
| 537 | |
| 538 | thickness = viz.get("thickness") |
| 539 | if thickness is not None: |
| 540 | e = Element(f"{{{self.NS_VIZ}}}thickness", value=str(thickness)) |
| 541 | element.append(e) |
| 542 | |
| 543 | shape = viz.get("shape") |
| 544 | if shape is not None: |
| 545 | if shape.startswith("http"): |
| 546 | e = Element( |
| 547 | f"{{{self.NS_VIZ}}}shape", value="image", uri=str(shape) |
| 548 | ) |
| 549 | else: |
| 550 | e = Element(f"{{{self.NS_VIZ}}}shape", value=str(shape)) |
| 551 | element.append(e) |
| 552 | |
| 553 | position = viz.get("position") |
| 554 | if position is not None: |
| 555 | e = Element( |
| 556 | f"{{{self.NS_VIZ}}}position", |
| 557 | x=str(position.get("x")), |
| 558 | y=str(position.get("y")), |
| 559 | z=str(position.get("z")), |
| 560 | ) |
| 561 | element.append(e) |
| 562 | return node_data |
| 563 | |
| 564 | def add_parents(self, node_element, node_data): |
| 565 | parents = node_data.pop("parents", False) |