| #------------------------------------------------------------------------------- | |
| # Name: tree_utils.py | |
| # Purpose: classes of node (joint) and tree-node (joint with its parent and children) | |
| # RigNet Copyright 2020 University of Massachusetts | |
| # RigNet is made available under General Public License Version 3 (GPLv3), or under a Commercial License. | |
| # Please see the LICENSE README.txt file in the main directory for more information and instruction on using and licensing RigNet. | |
| #------------------------------------------------------------------------------- | |
| class Node(object): | |
| def __init__(self, name, pos): | |
| self.name = name | |
| self.pos = pos | |
| class TreeNode(Node): | |
| def __init__(self, name, pos): | |
| super(TreeNode, self).__init__(name, pos) | |
| self.children = [] | |
| self.parent = None | |