54 lines
2.2 KiB
Python
54 lines
2.2 KiB
Python
|
from __future__ import print_function
|
||
|
|
import numpy as np
|
||
|
|
import math
|
||
|
|
import poisson
|
||
|
|
import array
|
||
|
|
|
||
|
|
def main():
|
||
|
|
# user defined options
|
||
|
|
disk = True # this parameter defines if we look for Poisson-like distribution on a disk/sphere (center at 0, radius 1) or in a square/box (0-1 on x and y)
|
||
|
|
repeatPattern = True # this parameter defines if we look for "repeating" pattern so if we should maximize distances also with pattern repetitions
|
||
|
|
num_points = 2 # number of points we are looking for
|
||
|
|
num_iterations = 4 # number of iterations in which we take average minimum squared distances between points and try to maximize them
|
||
|
|
first_point_zero = False # should be first point zero (useful if we already have such sample) or random
|
||
|
|
iterations_per_point = 128 # iterations per point trying to look for a new point with larger distance
|
||
|
|
sorting_buckets = 0 # if this option is > 0, then sequence will be optimized for tiled cache locality in n x n tiles (x followed by y)
|
||
|
|
num_dim = 3 # 1, 2, 3 dimensional version
|
||
|
|
num_rotations = 1 # number of rotations of pattern to check against
|
||
|
|
|
||
|
|
|
||
|
|
poisson_generator = poisson.PoissonGenerator(num_dim, disk, repeatPattern, first_point_zero)
|
||
|
|
points = poisson_generator.find_point_set(num_points, num_iterations, iterations_per_point, num_rotations)
|
||
|
|
print(points)
|
||
|
|
print("")
|
||
|
|
print("")
|
||
|
|
print("")
|
||
|
|
#normalize vectors
|
||
|
|
norm_points = []
|
||
|
|
for vector in points:
|
||
|
|
norm_points.append(vector/np.linalg.norm(vector))
|
||
|
|
print(norm_points)
|
||
|
|
#ensure all vectors are pointing upwards
|
||
|
|
final_points = []
|
||
|
|
for vector in norm_points:
|
||
|
|
if np.dot(vector, [0,1,0]) < 0:
|
||
|
|
final_points.append(-vector)
|
||
|
|
else:
|
||
|
|
final_points.append(vector)
|
||
|
|
print("")
|
||
|
|
print("")
|
||
|
|
print("")
|
||
|
|
print(final_points)
|
||
|
|
print("")
|
||
|
|
print("")
|
||
|
|
print("")
|
||
|
|
#format output
|
||
|
|
print("vec3 starPositions[{}] = vec3[](".format(num_points))
|
||
|
|
for vector in final_points:
|
||
|
|
print(" vec3({}, {}, {}),".format(vector[0], vector[1], vector[2]))
|
||
|
|
print(");")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|