Ejemplo n.º 1
0
def set_parameters(**kwargs):
    """ Set the values of parameters input from the command line.
    """
    print(kwargs)
    logger.info("Parse arguments ...")

    ## Create a Parameters object to store parameters.
    params = Parameters()

    ## Process arguments.
    #
    if kwargs.get(Args.CENTERLINES_FILE):
        params.centerlines_file_name = kwargs.get(Args.CENTERLINES_FILE)
        logger.info("Centerlines file: %s" % params.centerlines_file_name)
        if not os.path.exists(params.centerlines_file_name):
            logger.error("The centerlines file '%s' was not found." %
                         params.centerlines_file_name)
            return None

    if kwargs.get(Args.RADIUS):
        params.radisu = float(kwargs.get(Args.radius))

    if kwargs.get(Args.SOLVER_FILE):
        params.solver_file_name = kwargs.get(Args.SOLVER_FILE)
        logger.info("Solver file: %s" % params.solver_file_name)
        if not os.path.exists(params.solver_file_name):
            logger.error("The solver file '%s' was not found." %
                         params.solver_file_name)
            return None

    if kwargs.get(Args.SURFACE_FILE):
        params.surface_file_name = kwargs.get(Args.SURFACE_FILE)
        logger.info("Surface file: %s" % params.surface_file_name)
        if not os.path.exists(params.surface_file_name):
            logger.error("The surface file '%s' was not found." %
                         params.surface_file_name)
            return None

    return params