115 lines
2.4 KiB
Python
115 lines
2.4 KiB
Python
import numpy as np
|
|
|
|
import obja.obja as obja
|
|
|
|
|
|
class Edge:
|
|
"""Edge representation"""
|
|
|
|
def __init__(self, a: int, b: int) -> None:
|
|
"""Create an edge
|
|
|
|
Args:
|
|
a (int): first vertex
|
|
b (int): second vertex
|
|
"""
|
|
self.a = min(a, b)
|
|
self.b = max(a, b)
|
|
|
|
self.face1: Face | None = None
|
|
self.face2: Face | None = None
|
|
|
|
self.fold: float = 0.0
|
|
self.curvature: float = 0.0
|
|
|
|
def __eq__(self, __o: object) -> bool:
|
|
"""Check if two edges are equal
|
|
|
|
Args:
|
|
__o (object): other edge
|
|
|
|
Returns:
|
|
bool: True if equal, False otherwise
|
|
"""
|
|
if isinstance(__o, Edge):
|
|
return self.a == __o.a and self.b == __o.b
|
|
|
|
return False
|
|
|
|
|
|
class Face:
|
|
"""Face representation"""
|
|
|
|
def __init__(self, a: int, b: int, c: int) -> None:
|
|
"""Face constructor
|
|
|
|
Args:
|
|
a (int): first vertex index
|
|
b (int): second vertex index
|
|
c (int): third vertex index
|
|
"""
|
|
self.a = a
|
|
self.b = b
|
|
self.c = c
|
|
|
|
self.edges: list[Edge] = []
|
|
self.a = a
|
|
self.b = b
|
|
self.c = c
|
|
|
|
self.normal = np.zeros(3)
|
|
|
|
def to_obja(self) -> obja.Face:
|
|
"""Convert face to obja format
|
|
|
|
Returns:
|
|
obja.Face: face in obja format
|
|
"""
|
|
return obja.Face(self.a, self.b, self.c)
|
|
|
|
def __eq__(self, __o: object) -> bool:
|
|
"""Check if two faces are equal
|
|
|
|
Args:
|
|
__o (object): other face
|
|
|
|
Returns:
|
|
bool: True if equal, False otherwise
|
|
"""
|
|
if isinstance(__o, Face):
|
|
return set((__o.a, __o.b, __o.c)) == set((self.a, self.b, self.c))
|
|
|
|
return False
|
|
|
|
|
|
class Vertex:
|
|
"""Vertex representation"""
|
|
|
|
def __init__(self, pos: np.ndarray[int, float]) -> None:
|
|
"""Vertex constructor
|
|
|
|
Args:
|
|
pos (np.ndarray[int, float]): position of the vertex
|
|
"""
|
|
self.pos = pos
|
|
|
|
self.vertex_ring: list[int] = []
|
|
self.face_ring: list[int] = []
|
|
|
|
self.normal: np.ndarray = np.zeros(3)
|
|
|
|
self.area: float = 0.0
|
|
self.curvature: float = 0.0
|
|
|
|
def to_obja(self) -> np.ndarray:
|
|
"""Convert vertex to obja format
|
|
|
|
Returns:
|
|
np.ndarray: vertex in obja format
|
|
"""
|
|
return self.pos
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pass
|