Add line object

This commit is contained in:
2024-05-26 18:37:33 +02:00
parent 0d4bd5906b
commit 2e66a539cf
7 changed files with 166 additions and 67 deletions

41
networks/lines/Line.py Normal file
View File

@@ -0,0 +1,41 @@
import networks.geometry.curve as curve
import networks.geometry.segment as segment
import random
class Line:
def __init__(self, coordinates, line_type):
self.coordinates = coordinates
self.line_type = line_type
self.surface = []
def get_surface(self):
resolution, distance = curve.resolution_distance(self.coordinates, 6)
curve_points = curve.curve(self.coordinates, resolution)
# Compute the line
pattern_length = 0
pattern_materials = []
for key, value in self.line_type.items():
pattern_length += int(key)
for _ in range(int(key)):
pattern_materials.append(value)
pattern_iteration = 0
for i in range(len(curve_points)-1):
line = segment.discrete_segment(curve_points[i], curve_points[i+1])
for coordinate in line:
block = random.choices(
list(pattern_materials[pattern_iteration].keys()),
weights=pattern_materials[pattern_iteration].values(),
k=1)[0]
if block != 'None':
self.surface.append((coordinate, block))
pattern_iteration += 1
if pattern_iteration >= pattern_length:
pattern_iteration = 0
return self.surface

9
networks/lines/line.json Normal file
View File

@@ -0,0 +1,9 @@
{
"solid_white": {
"1": {"white_concrete": 3, "white_concrete_powder": 1}
},
"broken_white": {
"3": {"white_concrete": 3, "white_concrete_powder": 1},
"1": {"None": 1}
}
}