base foudations
This commit is contained in:
14
buildings/geometry/Point.py
Normal file
14
buildings/geometry/Point.py
Normal file
@@ -0,0 +1,14 @@
|
||||
class Point:
|
||||
def __init__(self, x : int = None, y : int = None, z : int = None, p : tuple[int] = None):
|
||||
if p != None: x,y,z = p
|
||||
self.x = x
|
||||
self.y = y
|
||||
self.z = z
|
||||
self.position = (x,y,z)
|
||||
|
||||
def set_position(self, x : int = None, y : int = None, z : int = None, p : tuple[int] = None):
|
||||
if p != None: x,y,z = p
|
||||
self.x = x if x != None else self.x
|
||||
self.y = y if y != None else self.y
|
||||
self.z = z if z != None else self.z
|
||||
self.position = (self.x,self.y,self.z)
|
||||
104
buildings/geometry/Polygon.py
Normal file
104
buildings/geometry/Polygon.py
Normal file
@@ -0,0 +1,104 @@
|
||||
from Enums import DIRECTION
|
||||
from gdpc import Editor, Block, geometry
|
||||
from buildings.geometry.Tile import Tile
|
||||
from buildings.geometry.Point import Point
|
||||
from buildings.geometry.Rectangle import Rectangle
|
||||
from buildings.geometry.Vertice import Vertice
|
||||
|
||||
class Polygon:
|
||||
def __init__(self, position : Point, size: tuple[int,int], vertices : list[Vertice] = []):
|
||||
self.position = position
|
||||
self.size = size
|
||||
self.compressed = {"shape":[], "vertices":[]}
|
||||
self.vertices = vertices
|
||||
|
||||
def fill_polygon(self, editor : Editor, material : str, y : int, y2 : int = None):
|
||||
if y2 == None: y2 = y
|
||||
for rect in self.compressed["shape"]:
|
||||
rect.fill(editor, material, y, y2)
|
||||
|
||||
def fill_vertice(self, editor : Editor, material : str, y : int, y2 : int = None):
|
||||
if y2 == None: y2 = y
|
||||
for vertice in self.compressed["vertices"]:
|
||||
vertice.fill(editor, Block(material), y, y2)
|
||||
|
||||
def compress(self, tiles : list[Tile]):
|
||||
remaining_tiles = tiles.copy()
|
||||
while len(remaining_tiles) > 0:
|
||||
start = remaining_tiles[0]
|
||||
neightbor = start.get_neighbor(DIRECTION.WEST)
|
||||
row = []
|
||||
|
||||
# Find western border
|
||||
while neightbor:
|
||||
start = neightbor
|
||||
neightbor = start.get_neighbor(DIRECTION.WEST)
|
||||
|
||||
# Find eastern border
|
||||
while True:
|
||||
row.append(start)
|
||||
remaining_tiles.remove(start)
|
||||
neightbor = start.get_neighbor(DIRECTION.EAST)
|
||||
if not neightbor: break
|
||||
start = neightbor
|
||||
|
||||
# Find northern border
|
||||
north_row = self.find_row_border(row.copy(), DIRECTION.NORTH, remaining_tiles)
|
||||
# Find southern border
|
||||
south_row = self.find_row_border(row.copy(), DIRECTION.SOUTH, remaining_tiles)
|
||||
|
||||
area = Rectangle(north_row[0].north_west, south_row[-1].south_east)
|
||||
self.compressed["shape"].append(area)
|
||||
|
||||
remaining_vertices = self.vertices.copy()
|
||||
current = remaining_vertices.pop()
|
||||
while len(remaining_vertices) > 0:
|
||||
neighbors = current.get_neighbors()
|
||||
has_next1 = self.has_next(neighbors[0], current.facing, remaining_vertices)
|
||||
has_next2 = self.has_next(neighbors[1], current.facing, remaining_vertices)
|
||||
|
||||
if has_next1:
|
||||
current = Vertice(has_next1.point1, current.point2, current.facing)
|
||||
elif has_next2:
|
||||
current = Vertice(current.point1, has_next2.point2, current.facing)
|
||||
else:
|
||||
self.compressed["vertices"].append(current)
|
||||
current = remaining_vertices.pop()
|
||||
|
||||
if len(remaining_vertices) == 0: self.compressed["vertices"].append(current)
|
||||
|
||||
def find_row_border(self, line : list[Tile], direction : str, remaining_tiles : list[Tile]) -> list[Tile]:
|
||||
while True:
|
||||
new_line = []
|
||||
for tile in line:
|
||||
neightbor = tile.get_neighbor(direction)
|
||||
if neightbor not in remaining_tiles: return line
|
||||
new_line.append(neightbor)
|
||||
for tile in new_line: remaining_tiles.remove(tile)
|
||||
line = new_line
|
||||
|
||||
def set_vertices_and_neighbors(self, tiles : list[Tile]):
|
||||
for tile in tiles:
|
||||
targets = tile.get_neighbors_coords()
|
||||
for vertice_num,target in enumerate(targets):
|
||||
has_neighbor = self.has_neighbor(target, tiles)
|
||||
if not has_neighbor:
|
||||
self.vertices.append(tile.get_vertice(vertice_num))
|
||||
else :
|
||||
tile.set_neighbor(vertice_num, has_neighbor)
|
||||
|
||||
def has_neighbor(self, target : tuple[int], tiles : list[Tile]) -> bool|Tile:
|
||||
for tile in tiles:
|
||||
if tile.pos.position == target.position:
|
||||
return tile
|
||||
return False
|
||||
|
||||
def has_next(self, target : Point, facing : str, remaining_vertices : list[Vertice]) -> bool|Vertice:
|
||||
for vertice in remaining_vertices:
|
||||
if vertice.facing == facing:
|
||||
if vertice.point1.position == target.position or vertice.point2.position == target.position:
|
||||
remaining_vertices.remove(vertice)
|
||||
return vertice
|
||||
return False
|
||||
|
||||
|
||||
14
buildings/geometry/Rectangle.py
Normal file
14
buildings/geometry/Rectangle.py
Normal file
@@ -0,0 +1,14 @@
|
||||
from gdpc import Editor, Block, geometry
|
||||
from buildings.geometry.Point import Point
|
||||
|
||||
class Rectangle:
|
||||
def __init__(self, point1 : Point, point2 : Point):
|
||||
self.point1 = point1
|
||||
self.point2 = point2
|
||||
|
||||
def get_position(self):
|
||||
return (self.point1.position, self.point2.position)
|
||||
|
||||
def fill(self,editor : Editor, material : str, y : int, y2 : int = None):
|
||||
if y2 == None: y2 = y
|
||||
geometry.placeCuboid(editor, (self.point1.x, y, self.point1.z), (self.point2.x, y2, self.point2.z), Block(material))
|
||||
64
buildings/geometry/Tile.py
Normal file
64
buildings/geometry/Tile.py
Normal file
@@ -0,0 +1,64 @@
|
||||
from gdpc import Editor, Block, geometry
|
||||
from Enums import DIRECTION
|
||||
from buildings.geometry.Point import Point
|
||||
from buildings.geometry.Vertice import Vertice
|
||||
|
||||
class Tile:
|
||||
def __init__(self, size : int, position : tuple[int, int]):
|
||||
self.size = size
|
||||
x,z = position
|
||||
self.pos = Point(x = x, z = z)
|
||||
leng = self.size-1
|
||||
self.north_west = self.pos
|
||||
self.north_east = Point(x = self.pos.x + leng, z =self.pos.z)
|
||||
self.south_west = Point(x = self.pos.x, z = self.pos.z + leng)
|
||||
self.south_east = Point(x = self.pos.x + leng, z = self.pos.z + leng)
|
||||
self.west_neighbor = None
|
||||
self.east_neighbor = None
|
||||
self.north_neighbor = None
|
||||
self.south_neighbor = None
|
||||
|
||||
def fill(self, editor : Editor, material : str, y : int, y2 : int = None) -> list[Point]:
|
||||
if y2 == None: y2 = y
|
||||
geometry.placeCuboid(editor, (self.pos.x, y, self.pos.z), (self.pos.x+self.size-1, y2, self.pos.z+self.size-1), Block(material))
|
||||
|
||||
def get_neighbors_coords(self):
|
||||
return [Point(x = self.pos.x - self.size, z = self.pos.z), # west
|
||||
Point(x = self.pos.x + self.size, z = self.pos.z), # east
|
||||
Point(x = self.pos.x, z = self.pos.z - self.size), # north
|
||||
Point(x = self.pos.x, z = self.pos.z + self.size)] # south
|
||||
|
||||
def get_vertice(self,vertice : int) -> Vertice:
|
||||
# gives the corresponding vertice :
|
||||
# 0 = west, 1 = east, 2 = north, 3 = south
|
||||
match(vertice):
|
||||
case 0 :
|
||||
return Vertice(self.north_west, self.south_west, DIRECTION.WEST)
|
||||
case 1 :
|
||||
return Vertice(self.north_east, self.south_east, DIRECTION.EAST)
|
||||
case 2 :
|
||||
return Vertice(self.north_west, self.north_east, DIRECTION.NORTH)
|
||||
case 3 :
|
||||
return Vertice(self.south_west, self.south_east, DIRECTION.SOUTH)
|
||||
|
||||
def get_neighbor(self, direction) -> Point:
|
||||
match(direction):
|
||||
case DIRECTION.WEST:
|
||||
return self.west_neighbor
|
||||
case DIRECTION.EAST:
|
||||
return self.east_neighbor
|
||||
case DIRECTION.NORTH:
|
||||
return self.north_neighbor
|
||||
case DIRECTION.SOUTH:
|
||||
return self.south_neighbor
|
||||
|
||||
def set_neighbor(self, direction, neighbor : Point):
|
||||
match(direction):
|
||||
case DIRECTION.WEST:
|
||||
self.west_neighbor = neighbor
|
||||
case DIRECTION.EAST:
|
||||
self.east_neighbor = neighbor
|
||||
case DIRECTION.NORTH:
|
||||
self.north_neighbor = neighbor
|
||||
case DIRECTION.SOUTH:
|
||||
self.south_neighbor = neighbor
|
||||
18
buildings/geometry/Vertice.py
Normal file
18
buildings/geometry/Vertice.py
Normal file
@@ -0,0 +1,18 @@
|
||||
from Enums import DIRECTION
|
||||
from buildings.geometry.Point import Point
|
||||
from buildings.geometry.Rectangle import Rectangle
|
||||
|
||||
class Vertice(Rectangle):
|
||||
def __init__(self, point1 : Point, point2 : Point, facing : str):
|
||||
Rectangle.__init__(self, point1, point2)
|
||||
self.facing = facing
|
||||
|
||||
def get_neighbors(self):
|
||||
match self.facing:
|
||||
case DIRECTION.NORTH | DIRECTION.SOUTH:
|
||||
return [Point(x = self.point1.x - 1, z = self.point1.z),
|
||||
Point(x = self.point2.x + 1, z = self.point2.z)]
|
||||
case DIRECTION.EAST | DIRECTION.WEST:
|
||||
return [Point(x = self.point1.x, z = self.point1.z - 1),
|
||||
Point(x = self.point2.x, z = self.point2.z + 1)]
|
||||
|
||||
Reference in New Issue
Block a user