import time
import networkx as nx
import talus.morse as morse
import numpy as np


def build_nodes(n):
    sampled = np.random.random((n, 3))
    points = []
    for i, row in enumerate(sampled):
        points.append(morse.MorseNode(identifier=i, value=row[0], vector=list(row[1:])))
    return points


for size in [500000]:
    print(size)
    points = build_nodes(size)
    result = morse.persistence_by_approximate_knn(points, 8, 0.8, 0.001)
    1/0
    start = time.time()
    result = morse.persistence_by_knn(points, 8)
    print(time.time() - start)
    print(size)
    start = time.time()
    result = morse.persistence_by_approximate_knn(points, 8, 0.5, 0.001)
    print(time.time() - start)
    print(len(result))