36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
import networks.geometry.curve_tools as curve_tools
|
|
import networks.geometry.segment_tools as segment_tools
|
|
import random
|
|
|
|
|
|
class Line:
|
|
def __init__(self, coordinates, line_materials):
|
|
self.coordinates = coordinates # Full lines coordinates, not just endpoints
|
|
self.line_materials = line_materials # From lines.json
|
|
self.coordinates_with_blocks = [] # Output
|
|
|
|
def get_blocks(self):
|
|
pattern_length = 0
|
|
pattern_materials = []
|
|
|
|
# Create the pattern materials list with correct materials depending on the selected pattern.
|
|
for pattern in self.line_materials:
|
|
pattern_length += pattern[1]
|
|
for _ in range(pattern[1]):
|
|
pattern_materials.append(pattern[0])
|
|
|
|
pattern_iteration = 0
|
|
for coordinate in self.coordinates:
|
|
block = random.choices(
|
|
list(pattern_materials[pattern_iteration].keys()),
|
|
weights=pattern_materials[pattern_iteration].values(),
|
|
k=1)[0]
|
|
if block != 'None':
|
|
self.coordinates_with_blocks.append((coordinate, block))
|
|
|
|
pattern_iteration += 1
|
|
if pattern_iteration >= pattern_length:
|
|
pattern_iteration = 0
|
|
|
|
return self.coordinates_with_blocks
|