40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
"""
|
|
First pass shape generation.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import random
|
|
import json
|
|
import product_gen.generate_product
|
|
import shapes.shape_description as sd
|
|
from argparse import ArgumentParser
|
|
|
|
if __name__ == "__main__":
|
|
|
|
argv = sys.argv
|
|
argv = argv[argv.index("--") + 1:]
|
|
shape_params = json.loads(argv[0])
|
|
|
|
shape = shape_params['shape']
|
|
output = shape_params['output']
|
|
|
|
if not os.path.exists(os.getcwd() + "/" + output):
|
|
os.makedirs(os.getcwd() + "/" + output)
|
|
|
|
# Global bounding box
|
|
bbox_xmax = 5.0
|
|
bbox_ymax = 4.0
|
|
|
|
# Minimum feature sizes
|
|
feature_xmin = 2.0
|
|
feature_ymin = 0.5
|
|
shape_parameters = sd.get_random_shape_description(shape,
|
|
[bbox_xmax, bbox_ymax],
|
|
[feature_xmin, feature_ymin])
|
|
|
|
shape_parameters = product_gen.generate_product.generate(shape_parameters, output)
|
|
shape_parameters["output"] = output
|
|
|
|
with open(os.getcwd() + "/" + output + "/"+ output + '.json', 'w') as outfile:
|
|
json.dump(shape_parameters, outfile)
|