main.py

from edge.edge import reconstruct_edges
from exports import export_reconstructed_model
from preprocess.parse import parse_two_D_projections
from type import Raw2DProjections, Reconstructed3DModel
from preprocess.validate import validateTwoDProjections
from vertex.vertex import reconstruct_vertices


def two_D_to_three_D(two_D_projections: Raw2DProjections) -> Reconstructed3DModel:
    assert validateTwoDProjections(two_D_projections), "invalid twoDProjections"
    parsed_two_D_projections = parse_two_D_projections(two_D_projections)

    three_D_reconstruction = {
        'reconstructed': {},
        'dandling': {}
    }

    reconstructed_vertices, dandling_vertices = reconstruct_vertices(parsed_two_D_projections)

    three_D_reconstruction['reconstructed']['vertices'] = reconstructed_vertices
    three_D_reconstruction['dandling']['vertices'] = dandling_vertices

    reconstructed_edges, dandling_edges = reconstruct_edges(parsed_two_D_projections)

    three_D_reconstruction['reconstructed']['edges'] = reconstructed_edges
    three_D_reconstruction['dandling']['edges'] = dandling_edges

    return export_reconstructed_model(three_D_reconstruction)