Ete4 create option of rotation for polygon

[dengzq1234]

Allow rotation in polygon, useful for setting direction of triangles

test code

from ete4 import Tree
from ete4.smartview import Layout, RectFace, CircleFace, PolygonFace, TextFace, BASIC_LAYOUT
import random

class TreeLayout:
    def __init__(self, name, shape, rotation, fill_color, stroke_color, column, header, wmax=50, hmax=50, rmax=30):
        self.name = name
        self.shape = shape  # 'square', 'circle', 'triangle'
        self.rotation = rotation
        self.fill_color = fill_color
        self.stroke_color = stroke_color
        self.column = column
        self.header = header
        self.wmax = wmax
        self.hmax = hmax
        self.rmax = rmax
        self.active = True

    def draw_node(self, node, collapsed=False):
        if node.is_leaf:
            binary_value = node.props.get('binary')
            style = {'fill': self.fill_color if binary_value == 1 else 'none',
                     'stroke': self.stroke_color, 'stroke-width': 1}
            
            if self.shape == 'square':
                yield RectFace(wmax=self.wmax, hmax=self.hmax, style=style, column=self.column, position='aligned')
            elif self.shape == 'circle':
                yield CircleFace(rmax=self.rmax, style=style, column=self.column, position='aligned')
            elif self.shape == 'triangle':
               yield PolygonFace(rmax=self.wmax, shape=3, rotation=self.rotation, style=style, column=self.column, position='aligned')
            # hack padding
            yield RectFace(wmax=20, hmax=1, style={'fill': 'none'}, column=self.column + 1, position='aligned')
    
    def draw_tree(self, tree):
        yield TextFace(self.header, rotation=-45, position='header', column=self.column)

# Generate a larger random tree
random.seed(42)
tree = Tree()
tree.populate(50, dist_fn=random.random, support_fn=random.random)

# Assign random binary data to leaves, including empty values
for leaf in tree:
    leaf.add_prop('binary', random.choice([0, 1, None]))  # Adding None for empty values

# Define different layouts using TreeLayout
# square_layout = TreeLayout('square', 'square', 0, 'blue', 'blue', 1, 'Square')
# circle_layout = TreeLayout('circle', 'circle', 0, 'red', 'red', 3, 'Circle')
triangle_left_layout = TreeLayout('triangle_left', 'triangle', 270, 'gold', 'gold', 5, 'Triangle_left')
triangle_right_layout = TreeLayout('triangle_right', 'triangle', 90, 'green', 'green', 7, 'Triangle_right')

# Explore tree with all layouts
tree.explore(layouts=[BASIC_LAYOUT, 
# square_layout, 
# circle_layout, 
triangle_left_layout, 
triangle_right_layout
])
input()