예제 #1
0
def read_parameters_ncinifile(inifile):
    '''
    Reads general and variable parameters in an inifile
    Stores it and returns in a dictionary form
    '''
    general_params = general_class.General()  #dictionary
    src_params = []  #list of dictionaries

    check_file(inifile)

    # Open file with all parameters
    fp = open(inifile, 'r')

    # Determine all the information per line
    type_option = None
    for line in fp.readlines():
        option = [elem.strip() for elem in line.split('#')[0].split('=')]
        if (len(option[0]) < 1):
            # Empty line
            continue

        # Check whether the first element is start of a new class
        # (first character is a '[' )
        if option[0].startswith('['):
            # Start of a new class
            type_option = option[0].strip().upper()
            print('\n' + type_option)
            if (type_option == '[SOURCE]'):
                src_params.append(general_class.General())
            elif (type_option != '[GENERAL]'):
                raise MyError(
                    "File '%s' contains errors. Class %s is not known." %
                    (inifile, option[0]))

        else:
            # Here information from a class is given
            print(option[0] + ' = ' + option[1])
            try:
                if (type_option == '[GENERAL]'):
                    general_params.add_item(option[0], option[1])
                elif (type_option == '[SOURCE]'):
                    src_params[-1].add_item(option[0], option[1])
            except IndexError:
                raise MyError("File '%s' contains errors." % inifile,\
                              "Can not handle line: %s" % line)
    # Close input file
    fp.close()

    return general_params, src_params
예제 #2
0
def make_arguments_strahler():
    args_strahler = general_class.General()
    args_strahler.add_item("slope",None)
    args_strahler.add_item("riverlength",None)
    args_strahler.add_item("riverarea",None) 
    args_strahler.add_item("number_of_rivers",None) 
    args_strahler.add_item("load_distribution",None)
    args_strahler.add_item("flowpath_distribution",None) 
    return args_strahler
예제 #3
0
def convert(obj):
    '''
    Conversion from General() to another type of class.
    '''
    try:
        if (obj.get_val("type") == "GENERAL"):
            # It is a General class.
            outobj = general_class.General()
        elif (obj.get_val("type") == "SRC"):
            # It is a General class.
            outobj = general_class.General()
        elif (obj.get_val("type") == "SPEC"):
            # It is a Spec class.
            outobj = specie.Spec()
        else:
            raise MyError(
                "It is not possible to convert class of type '%s' . Class is not known."
                % obj.get_val("type"))
    except AttributeError:
        raise MyError("It is not possible to convert this object.")

    # Put all attributes in the output class
    for name in obj.get_attrib():
        if (obj.get_val("type") == "GENERAL") or (obj.get_val("type")
                                                  == "SRC"):
            outobj.add_item(name, str2num(obj.get_val(name)))
        else:
            try:
                # Check whether this attribute is part of this class.
                dum = getattr(outobj, name)
            except AttributeError:
                raise MyError(
                    "Attribute %s is unknown for class of type: %s." %
                    (name, obj.get_val("type")))
            setattr(outobj, name, str2num(obj.get_val(name)))
    try:
        # Change all attributes from strings into the type that is required.
        #outobj.check_type()
        pass
    except AttributeError:
        raise MyError("Method check_type is not found for class of type: %s." %
                      obj.get_val("type"))

    return outobj
예제 #4
0
def do():
    arguments = general_class.General()
    arguments.add_item("vol", None)
    arguments.add_item("dvoldt", None)
    arguments.add_item("globrad", None)
    arguments.add_item("temperature", None)
    arguments.add_item("windspeed", None)
    arguments.add_item("flow_velocity", None)  #WJ201708
    arguments.add_item("width", None)
    arguments.add_item("depth", None)
    arguments.add_item("slope", None)
    arguments.add_item("discharge", None)
    arguments.add_item("length", None)
    arguments.add_item("residence_time", None)
    arguments.add_item("area", None)
    arguments.add_item("icell", None)
    arguments.add_item("next_cell", None)
    return arguments
예제 #5
0
def readfile(filename):
    '''
    Read all the specs for the species and processes.
    Returns a list of species and a list of processes.
    '''
    proclist = []
    srclist = []  #added LV 22-07-2016: description of load sources in inifile
    speclist = []
    paramlist = []
    if not os.path.exists(filename):
        raise MyError("File '%s' does not exist" % filename)

    # Open file with all parameters.
    fp = open(filename, "r")

    # Determine all the information per line.
    obj = None
    for line in fp.readlines():
        option = [elem.strip() for elem in line.split("#")[0].split("=")]
        if (len(option[0]) < 1):
            # Empty line
            continue

        # Check whether the first element is start of a new class
        # (first character is a "[" ).
        if option[0].startswith("["):
            # Start of a new class
            if (option[0].strip().upper() == "[PROCES]"):
                # Save old obj
                if (obj != None):
                    if (obj.get_val("type") == "GENERAL"):
                        proclist.append(convert(obj))
                    elif (obj.get_val("type") == "SRC"):
                        srclist.append(convert(obj))
                    else:
                        speclist.append(convert(obj))
                obj = general_class.General()
                obj.add_item("type", "GENERAL")
            elif (option[0].strip().upper() == "[SOURCE]"
                  ):  #added LV 22-07-2016
                # Save old obj
                if (obj != None):
                    if (obj.get_val("type") == "GENERAL"):
                        proclist.append(convert(obj))
                    elif (obj.get_val("type") == "SRC"):
                        srclist.append(convert(obj))
                    else:
                        speclist.append(convert(obj))
                obj = general_class.General()
                obj.add_item("type", "SRC")
            elif (option[0].strip().upper() == "[SPECIE]"):
                # Save old obj
                if (obj != None):
                    if (obj.get_val("type") == "GENERAL"):
                        proclist.append(convert(obj))
                    elif (obj.get_val("type") == "SRC"):
                        srclist.append(convert(obj))
                    else:
                        speclist.append(convert(obj))
                obj = general_class.General()
                obj.add_item("type", "SPEC")
            elif (option[0].strip().upper() == "[PARAMETERS]"):
                # Save old obj
                if (obj != None):
                    if (obj.get_val("type") == "GENERAL"):
                        proclist.append(convert(obj))
                    elif (obj.get_val("type") == "SRC"):
                        srclist.append(convert(obj))
                    else:
                        speclist.append(convert(obj))
                obj = general_class.General()
                obj.add_item("type", "GENERAL")
                # Set extra item, so this entries can easily extracted out of proclist
                obj.add_item("subtype", "PARAMETERS")

            else:
                raise MyError(
                    "File '%s' contains errors. Class %s is not known." %
                    (filename, option[0]))

        else:
            # Here information from a class is given.
            try:
                obj.add_item(option[0], option[1])
            except IndexError:
                raise MyError("File '%s' contains errors." % filename,\
                              "Can not handle line: %s" % line)
    # Close input file
    fp.close()

    # Add last class to output list
    if (obj != None):
        if (obj.get_val("type") == "GENERAL"):
            proclist.append(convert(obj))
        elif (obj.get_val("type") == "SRC"):
            srclist.append(convert(obj))
        else:
            speclist.append(convert(obj))

    # Check whether the processes are refering to the right species and extract the general parameters.
    remove_list = []
    for iproc in range(len(proclist)):
        obj = proclist[iproc]
        try:
            qq = obj.get_val("subtype")
            # This is a parameter entry
            remove_list.append(iproc)
            continue
        except:
            pass
        for name in obj.get_attrib():
            if (name.startswith("dy_")):
                try:
                    ind = general_func.find_spec(speclist, name[3:])
                except MyError:
                    raise MyError("Process %s refers to specie which does not exist." % (obj.get_val("name")),\
                                  "Speciesname found: %s" % name[3:])

    # Check whether the sources are refering to the right species
    for isrc in range(len(srclist)):
        obj = srclist[isrc]
        for name in obj.get_attrib():
            if (name.startswith("fr_")):
                try:
                    ind = general_func.find_spec(speclist, name[3:])
                except MyError:
                    raise MyError("Source %s refers to specie that does not exist." % (obj.get_val("name")),\
                                  "Speciesname found: %s" % name[3:])

    # Make one general object for the parameters.
    params_local = general_class.General()
    for item in remove_list:
        params_local.merge(proclist[item])
    # Correct the names list of general class
    params_local.names = list(set(params_local.names))

    # Delete the parameters entries in the proclist
    for item in range(len(remove_list) - 1, -1, -1):
        del proclist[remove_list[item]]

    #return speclist,proclist,params_local
    return speclist, srclist, proclist, params_local