2017-10-20 11:30:47 +00:00
|
|
|
import os
|
2017-10-24 09:19:14 +00:00
|
|
|
import sys
|
|
|
|
import random
|
|
|
|
import json
|
|
|
|
import product_gen.generate_product
|
2017-10-24 19:18:12 +00:00
|
|
|
import shapes.shape_description as sd
|
2017-10-24 09:19:14 +00:00
|
|
|
from argparse import ArgumentParser
|
2017-10-20 11:30:47 +00:00
|
|
|
|
2017-10-24 09:19:14 +00:00
|
|
|
if __name__ == "__main__":
|
|
|
|
|
2017-10-24 14:05:25 +00:00
|
|
|
shape = "pendant"
|
2017-10-24 09:19:14 +00:00
|
|
|
output = "test"
|
|
|
|
|
|
|
|
if not os.path.exists(os.getcwd() + "/" + output):
|
|
|
|
os.makedirs(os.getcwd() + "/" + output)
|
|
|
|
|
2017-10-24 19:18:12 +00:00
|
|
|
# Global bounding box
|
|
|
|
bbox_xmax = 4.0
|
|
|
|
bbox_ymax = 5.0
|
|
|
|
|
|
|
|
# Minimum feature sizes
|
|
|
|
feature_xmin = 0.4
|
|
|
|
feature_ymin = 1.0
|
|
|
|
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)
|
2017-10-24 09:19:14 +00:00
|
|
|
|
|
|
|
with open(os.getcwd() + "/" + output + "/"+ output + '.json', 'w') as outfile:
|
|
|
|
json.dump(shape_parameters, outfile)
|