designs.py

2.8 kB · python · 98 lines

1from typing import TYPE_CHECKING2from mrlypy.three.designs import (3    carpet_3d, net_3d, tree_3d, void_3d,4    level_set_3d, xtree_3d, ytree_3d, ztree_3d,5)6from .geometry import iso, pro, cut78if TYPE_CHECKING:9    from .models import Cell6d1011# CARPET1213def carpet_iso(number: int, level: int = 1) -> "Cell6d":14    return iso(carpet_3d(number, level))1516def carpet_pro(number: int, level: int = 1) -> "Cell6d":17    return pro(carpet_3d(number, level))1819def carpet_cut(number: int, level: int = 1) -> "Cell6d":20    return cut(carpet_3d(number, level))2122# NET2324def net_iso(number: int, level: int = 1) -> "Cell6d":25    return iso(net_3d(number, level))2627def net_pro(number: int, level: int = 1) -> "Cell6d":28    return pro(net_3d(number, level))2930def net_cut(number: int, level: int = 1) -> "Cell6d":31    return cut(net_3d(number, level))3233# TREE3435def tree_iso(number: int, level: int = 1) -> "Cell6d":36    return iso(tree_3d(number, level))3738def tree_pro(number: int, level: int = 1) -> "Cell6d":39    return pro(tree_3d(number, level))4041def tree_cut(number: int, level: int = 1) -> "Cell6d":42    return cut(tree_3d(number, level))4344# VOID4546def void_iso(number: int, level: int = 1) -> "Cell6d":47    return iso(void_3d(number, level))4849def void_pro(number: int, level: int = 1) -> "Cell6d":50    return pro(void_3d(number, level))5152def void_cut(number: int, level: int = 1) -> "Cell6d":53    return cut(void_3d(number, level))5455def xtree_iso(number: int, level: int = 1) -> "Cell6d":56    return iso(xtree_3d(number, level))5758def xtree_pro(number: int, level: int = 1) -> "Cell6d":59    return pro(xtree_3d(number, level))6061def xtree_cut(number: int, level: int = 1) -> "Cell6d":62    return cut(xtree_3d(number, level))6364def ytree_iso(number: int, level: int = 1) -> "Cell6d":65    return iso(ytree_3d(number, level))6667def ytree_pro(number: int, level: int = 1) -> "Cell6d":68    return pro(ytree_3d(number, level))6970def ytree_cut(number: int, level: int = 1) -> "Cell6d":71    return cut(ytree_3d(number, level))7273def ztree_iso(number: int, level: int = 1) -> "Cell6d":74    return iso(ztree_3d(number, level))7576def ztree_pro(number: int, level: int = 1) -> "Cell6d":77    return pro(ztree_3d(number, level))7879def ztree_cut(number: int, level: int = 1) -> "Cell6d":80    return cut(ztree_3d(number, level))8182def level_set_iso(number: int, levels, level: int = 1) -> "Cell6d":83    return iso(level_set_3d(number, levels, level))8485def level_set_pro(number: int, levels, level: int = 1) -> "Cell6d":86    return pro(level_set_3d(number, levels, level))8788def level_set_cut(number: int, levels, level: int = 1) -> "Cell6d":89    return cut(level_set_3d(number, levels, level))9091def anti_iso(design_3d) -> "Cell6d":92    return iso(design_3d.copy().invert())9394def anti_pro(design_3d) -> "Cell6d":95    return pro(design_3d.copy().invert())9697def anti_cut(design_3d) -> "Cell6d":98    return cut(design_3d.copy().invert())