main.py

import argparse

import numpy as np
import random
import time

from plyfile import PlyData, PlyElement

from utils import *

from ball_pivot import *


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument(
        '--seed', '-s', default=int(time.time()), type=int,
        help='random seed'
    )
    
    parser.add_argument(
        '--invert_normal', action='store_true', default=False,
        help='invert the normals from ply file'
    )

    parser.add_argument(
        '--radius', '-r', required=True, type=float, nargs='+',
        help='radius'
    )

    parser.add_argument(
        '--input', '-i', type=str, required=True,
        help='input file'
    )

    parser.add_argument(
        '--output', '-o', type=str, required=True,
        help='output file'
    )
    args = parser.parse_args()

    np.random.seed(args.seed)
    random.seed(args.seed)

    plydata = PlyData.read(args.input)

    vs = build_vertex_set_ply(plydata, args.invert_normal)

    mesh = pivot_ball(vs, args.radius)
    if mesh is None:
        print 'Cannot find seed triangle. Reconstruction failed!'
    else:
        write_collada(vs.vertices, vs.normals, np.array(mesh.faces.keys()), args.output)