Beispiel #1
0
    def add_writenames_in_use_public(self, node):
        parent = node.kgen_parent

        for gentype, reqlist in node.kgen_stmt.geninfo.iteritems():
            for uname, req in reqlist:
                if len(req.res_stmts) > 0 and isinstance(
                        req.res_stmts[0], block_statements.Type):
                    subrname = get_dtype_writename(req.res_stmts[0])

                    # add use
                    checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.isonly and \
                        subrname in n.kgen_stmt.items
                    if (
                            id(parent), subrname
                    ) not in self.state_created_use_items and not part_has_node(
                            parent, USE_PART, checks):
                        attrs = {
                            'name': node.kgen_stmt.name,
                            'isonly': True,
                            'items': [subrname]
                        }
                        part_append_gensnode(parent,
                                             USE_PART,
                                             statements.Use,
                                             attrs=attrs)
                        self.state_created_use_items.append(
                            (id(parent), subrname))
                        parent.kgen_stmt.top.used4genstate = True

                    # add access
                    checks = lambda n: isinstance(
                        n.kgen_stmt, statements.Public
                    ) and n.kgen_stmt.items and subrname in n.kgen_stmt.items
                    if (id(parent),subrname) not in self.state_created_public_items and isinstance(parent.kgen_stmt, block_statements.Module) and \
                        not part_has_node(parent, DECL_PART, checks):
                        attrs = {'items': [subrname]}
                        part_append_gensnode(parent,
                                             DECL_PART,
                                             statements.Public,
                                             attrs=attrs)
                        self.state_created_public_items.append(
                            (id(parent), subrname))
                        parent.kgen_stmt.top.used4genstate = True
Beispiel #2
0
    def add_writenames_in_use_public(self, node):
        parent = node.kgen_parent

        for gentype, reqlist in node.kgen_stmt.geninfo.iteritems():
            for uname, req in reqlist:
                if len(req.res_stmts)>0 and isinstance(req.res_stmts[0], block_statements.Type):
                    subrname = get_dtype_writename(req.res_stmts[0])
                    checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.isonly and \
                        subrname in n.kgen_stmt.items
                    if (id(parent),subrname) not in self.state_created_use_items and not part_has_node(parent, USE_PART, checks):
                        attrs = {'name':node.kgen_stmt.name, 'isonly': True, 'items':[subrname]}
                        part_append_gensnode(parent, USE_PART, statements.Use, attrs=attrs)
                        self.state_created_use_items.append((id(parent),subrname))
                        parent.kgen_stmt.top.used4genstate = True

                    checks = lambda n: isinstance(n.kgen_stmt, statements.Public) and n.kgen_stmt.items and subrname in n.kgen_stmt.items
                    if (id(parent),subrname) not in self.state_created_public_items and isinstance(parent.kgen_stmt, block_statements.Module) and \
                        not part_has_node(parent, DECL_PART, checks):
                        attrs = {'items':[subrname]}
                        part_append_gensnode(parent, DECL_PART, statements.Public, attrs=attrs)
                        self.state_created_public_items.append((id(parent),subrname))
                        parent.kgen_stmt.top.used4genstate = True
Beispiel #3
0
    def create_subr_write_typedecl_in_module(self, node):
        parent = node.kgen_parent
        stmt = node.kgen_stmt

        raw_entity_names = set([
            uname.firstpartname()
            for uname, req in KGGenType.get_state(stmt.geninfo)
        ])
        entity_names = [
            e for e in raw_entity_names
            if not stmt.get_variable(e).is_parameter()
        ]

        raw_out_entity_names = set([
            uname.firstpartname()
            for uname, req in KGGenType.get_state_out(stmt.geninfo)
        ])
        out_entity_names = [
            e for e in raw_out_entity_names
            if not stmt.get_variable(e).is_parameter()
        ]

        #entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state(stmt.geninfo)])
        #out_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state_out(stmt.geninfo)])

        is_class_derived = check_class_derived(stmt)

        for entity_name, entity_decl in zip(entity_names, stmt.entity_decls):
            if node.kgen_parent.name + entity_name in self.state_extern_writes:
                continue
            if is_remove_state(entity_name, stmt): continue

            self.state_extern_writes.append(node.kgen_parent.name +
                                            entity_name)

            var = stmt.get_variable(entity_name)
            subrname = get_typedecl_writename(stmt, entity_name)

            if var.is_array():
                if is_zero_array(var, stmt): continue
                if stmt.is_derived() or is_class_derived:
                    self.create_write_call(
                        self.state_externs_subrs[node.kgen_parent][0],
                        subrname, entity_name, stmt, var)
                    if entity_name in out_entity_names:
                        self.create_write_call(
                            self.state_externs_subrs[node.kgen_parent][1],
                            subrname, entity_name, stmt, var)
                    if subrname not in self.state_created_subrs:
                        create_write_subr(subrname, entity_name, parent, var,
                                          stmt)
                        self.state_created_subrs.append(subrname)
                else:  # intrinsic type
                    if var.is_explicit_shape_array():
                        self.create_write_intrinsic(
                            self.state_externs_subrs[node.kgen_parent][0],
                            entity_name, stmt, var)
                        if entity_name in out_entity_names:
                            self.create_write_intrinsic(
                                self.state_externs_subrs[node.kgen_parent][1],
                                entity_name, stmt, var)
                    else:  # implicit array
                        self.create_write_call(
                            self.state_externs_subrs[node.kgen_parent][0],
                            subrname, entity_name, stmt, var)
                        if entity_name in out_entity_names:
                            self.create_write_call(
                                self.state_externs_subrs[node.kgen_parent][1],
                                subrname, entity_name, stmt, var)
                        if subrname not in self.state_created_subrs:
                            create_write_subr(subrname, entity_name, parent,
                                              var, stmt)
                            self.state_created_subrs.append(subrname)
            else:  # scalar
                if stmt.is_derived() or is_class_derived:
                    if var.is_allocatable() or var.is_pointer():
                        self.create_write_call(
                            self.state_externs_subrs[node.kgen_parent][0],
                            subrname, entity_name, stmt, var)
                        if entity_name in out_entity_names:
                            self.create_write_call(
                                self.state_externs_subrs[node.kgen_parent][1],
                                subrname, entity_name, stmt, var)
                        if subrname not in self.state_created_subrs:
                            create_write_subr(subrname, entity_name, parent,
                                              var, stmt)
                            self.state_created_subrs.append(subrname)
                    else:
                        subrname = None
                        for uname, req in stmt.unknowns.iteritems():
                            if (is_class_derived and uname.firstpartname()
                                    == stmt.selector[1]
                                ) or uname.firstpartname() == stmt.name:
                                #if uname.firstpartname()==stmt.name:
                                if len(req.res_stmts) > 0:
                                    res = req.res_stmts[0]
                                    subrname = get_dtype_writename(res)
                                    break
                        if subrname is None:
                            print 'WARNING: Can not find Type resolver for %s' % stmt.name
                            namedpart_append_comment(self.state_externs_subrs[node.kgen_parent][0], EXEC_PART, \
                                'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                                (stmt.name, stmt.name))
                        else:
                            self.create_write_call(
                                self.state_externs_subrs[node.kgen_parent][0],
                                subrname, entity_name, stmt, var)
                            if entity_name in out_entity_names:
                                self.create_write_call(
                                    self.state_externs_subrs[node.kgen_parent]
                                    [1], subrname, entity_name, stmt, var)
                else:  # intrinsic type
                    self.create_write_intrinsic(
                        self.state_externs_subrs[node.kgen_parent][0],
                        entity_name, stmt, var)
                    if entity_name in out_entity_names:
                        self.create_write_intrinsic(
                            self.state_externs_subrs[node.kgen_parent][1],
                            entity_name, stmt, var)
Beispiel #4
0
    def create_dtype_write_subr(self, node):
        assert node.kgen_stmt, 'None kgen statement'

        subrname = get_dtype_writename(node.kgen_stmt)
        if subrname is None: return

        parent = node.kgen_parent
        checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine
                                      ) and n.name == subrname
        if not part_has_node(parent, SUBP_PART, checks):

            checks = lambda n: n.kgen_match_class == statements.Contains
            if not parent in state_gencore_contains and not part_has_node(
                    parent, CONTAINS_PART, checks):
                part_append_comment(parent, CONTAINS_PART, '')
                part_append_gensnode(parent, CONTAINS_PART,
                                     statements.Contains)
                part_append_comment(parent, CONTAINS_PART, '')
                state_gencore_contains.append(parent)

            part_append_comment(parent, SUBP_PART,
                                'read state subroutine for %s' % subrname)
            attrs = {
                'prefix': 'RECURSIVE',
                'name': subrname,
                'args': ['var', 'kgen_unit', 'printname', 'printvar']
            }
            subrobj = part_append_gensnode(parent,
                                           SUBP_PART,
                                           block_statements.Subroutine,
                                           attrs=attrs)
            part_append_comment(parent, SUBP_PART, '')

            # variable
            attrs = {
                'type_spec': 'TYPE',
                'attrspec': ['INTENT(IN)'],
                'selector': (None, node.name),
                'entity_decls': ['var']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Type,
                                 attrs=attrs)

            # kgen_unit
            attrs = {
                'type_spec': 'INTEGER',
                'attrspec': ['INTENT(IN)'],
                'entity_decls': ['kgen_unit']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Integer,
                                 attrs=attrs)

            # printname
            attrs = {
                'type_spec': 'CHARACTER',
                'attrspec': ['INTENT(IN)'],
                'selector': ('*', None),
                'entity_decls': ['printname']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Character,
                                 attrs=attrs)

            # printvar
            attrs = {
                'type_spec': 'LOGICAL',
                'attrspec': ['INTENT(IN)', 'OPTIONAL'],
                'selector': (None, None),
                'entity_decls': ['printvar']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Character,
                                 attrs=attrs)

            # kgen_istrue
            attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Logical,
                                 attrs=attrs)

            attrs = {
                'type_spec': 'REAL',
                'entity_decls': ['kgen_array_sum'],
                'selector': (None, '8')
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Real,
                                 attrs=attrs)

            part_append_comment(subrobj, DECL_PART, '')

            parent_names = []
            getinfo('traverse')(node.kgen_stmt.f2003, self.get_extends,
                                parent_names)

            extends = []
            for parent_typename in parent_names:
                for uname, req in node.kgen_stmt.unknowns.iteritems():
                    if uname.firstpartname() == parent_typename:
                        if len(req.res_stmts) > 0:
                            extends.extend(
                                get_part(req.res_stmts[0].genspair, TYPE_PART))
                            break

            comp_part = get_part(node, TYPE_PART)
            for item in extends + comp_part:
                if not hasattr(item, 'kgen_stmt'): continue
                if not isinstance(
                        item.kgen_stmt,
                        typedecl_statements.TypeDeclarationStatement):
                    continue

                stmt = item.kgen_stmt
                entity_names = [
                    get_entity_name(decl) for decl in stmt.entity_decls
                ]

                for entity_name, entity_decl in zip(entity_names,
                                                    stmt.entity_decls):

                    if is_remove_state(entity_name, stmt): continue

                    node.kgen_stmt.top.used4genstate = True

                    var = stmt.get_variable(entity_name)
                    callname = get_typedecl_writename(stmt, entity_name)

                    if var.is_array():
                        if stmt.is_derived():
                            self.create_write_call(subrobj, callname,
                                                   entity_name, stmt, var)
                        else:  # intrinsic type
                            if var.is_explicit_shape_array():
                                self.create_write_intrinsic(
                                    subrobj, entity_name, stmt, var)
                                callname = None
                            else:  # implicit array
                                self.create_write_call(subrobj, callname,
                                                       entity_name, stmt, var)
                    else:  # scalar
                        if stmt.is_derived():
                            if var.is_allocatable() or var.is_pointer():
                                self.create_write_call(subrobj, callname,
                                                       entity_name, stmt, var)
                            else:
                                callname = None
                                for uname, req in stmt.unknowns.iteritems():
                                    if uname.firstpartname(
                                    ) == stmt.name and len(req.res_stmts) > 0:
                                        res = req.res_stmts[0]
                                        callname = get_dtype_writename(res)
                                        break
                                if callname is None:
                                    print 'WARNING: Can not find Type resolver for %s' % stmt.name
                                    part_append_comment(subrobj, EXEC_PART, \
                                        'ERROR: "%s" is not resolved. Call statement to write "%s" is not created here.'%\
                                        (stmt.name, stmt.name))
                                else:
                                    self.create_write_call(
                                        subrobj, callname, entity_name, stmt,
                                        var)
                        else:  # intrinsic type
                            self.create_write_intrinsic(
                                subrobj, entity_name, stmt, var)
                            callname = None

                    if node.kgen_stmt.ancestors(
                    )[0] != item.kgen_stmt.ancestors()[0] and callname:
                        # add use
                        pnode = node.kgen_parent
                        checks = lambda n: n.kgen_match_class==statements.Use and n.kgen_stmt and n.kgen_stmt.isonly and \
                            callname in n.kgen_stmt.items
                        if (
                                id(pnode), callname
                        ) not in self.state_created_use_items and not part_has_node(
                                pnode, USE_PART, checks):
                            attrs = {
                                'name': item.kgen_stmt.ancestors()[0].name,
                                'isonly': True,
                                'items': [callname]
                            }
                            part_append_gensnode(pnode,
                                                 USE_PART,
                                                 statements.Use,
                                                 attrs=attrs)
                            self.state_created_use_items.append(
                                (id(pnode), callname))
                            pnode.kgen_stmt.top.used4genstate = True

                part_append_comment(subrobj, EXEC_PART, '')

            # create public stmt
            if parent.kgen_match_class in [
                    block_statements.Program, block_statements.Module
            ]:
                attrs = {'items': [subrname]}
                part_append_gensnode(parent,
                                     DECL_PART,
                                     statements.Public,
                                     attrs=attrs)
Beispiel #5
0
    def create_write_subr(self, subrname, entity_name, parent, var, stmt):

        checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine
                                      ) and n.name == subrname
        if subrname not in self.state_created_subrs and not part_has_node(
                parent, SUBP_PART, checks):

            if is_remove_state(entity_name, stmt): return

            self.state_created_subrs.append(subrname)

            checks = lambda n: n.kgen_match_class == statements.Contains
            if not parent in state_gencore_contains and not part_has_node(
                    parent, CONTAINS_PART, checks):
                part_append_comment(parent, CONTAINS_PART, '')
                part_append_gensnode(parent, CONTAINS_PART,
                                     statements.Contains)
                part_append_comment(parent, CONTAINS_PART, '')
                state_gencore_contains.append(parent)

            part_append_comment(parent, SUBP_PART,
                                'write state subroutine for %s' % subrname)
            attrs = {
                'name': subrname,
                'args': ['var', 'kgen_unit', 'printvar']
            }
            subrobj = part_append_gensnode(parent,
                                           SUBP_PART,
                                           block_statements.Subroutine,
                                           attrs=attrs)
            part_append_comment(parent, SUBP_PART, '')

            parent.kgen_stmt.top.used4genstate = True

            # variable A
            #import pdb; pdb.set_trace()
            attrspec = ['INTENT(IN)']
            if var.is_allocatable(): attrspec.append('ALLOCATABLE')
            if var.is_pointer(): attrspec.append('POINTER')
            if var.is_array():
                attrspec.append('DIMENSION(%s)' % ','.join(':' * var.rank))
            attrs = {
                'type_spec': stmt.__class__.__name__.upper(),
                'attrspec': attrspec,
                'selector': stmt.selector,
                'entity_decls': ['var']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 stmt.__class__,
                                 attrs=attrs)

            # kgen_unit
            attrs = {
                'type_spec': 'INTEGER',
                'attrspec': ['INTENT(IN)'],
                'entity_decls': ['kgen_unit']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Integer,
                                 attrs=attrs)

            # printvar
            attrs = {
                'type_spec': 'CHARACTER',
                'attrspec': ['INTENT(IN)', 'OPTIONAL'],
                'selector': ('*', None),
                'entity_decls': ['printvar']
            }
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Character,
                                 attrs=attrs)

            # kgen_istrue
            attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
            part_append_gensnode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Logical,
                                 attrs=attrs)

            # array index A
            if var.is_array():
                attrs = {
                    'type_spec': 'INTEGER',
                    'entity_decls':
                    ['idx%d' % (d + 1) for d in range(var.rank)]
                }
                part_append_gensnode(subrobj,
                                     DECL_PART,
                                     typedecl_statements.Integer,
                                     attrs=attrs)

            attrs = {
                'type_spec': 'REAL',
                'entity_decls': ['kgen_array_sum'],
                'selector': (None, '8')
            }
            part_append_genknode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Real,
                                 attrs=attrs)

            part_append_comment(subrobj, DECL_PART, '')

            pobj = gen_write_istrue(subrobj, var, 'var')

            if var.is_array():

                for dim in range(var.rank):
                    attrs = {
                        'items': ['LBOUND(var, %d)' % (dim + 1)],
                        'specs': ['UNIT = kgen_unit']
                    }
                    part_append_gensnode(pobj,
                                         EXEC_PART,
                                         statements.Write,
                                         attrs=attrs)

                    attrs = {
                        'items': ['UBOUND(var, %d)' % (dim + 1)],
                        'specs': ['UNIT = kgen_unit']
                    }
                    part_append_gensnode(pobj,
                                         EXEC_PART,
                                         statements.Write,
                                         attrs=attrs)

                if stmt.is_derived():
                    indexes = ['idx%d' % (d + 1) for d in range(var.rank)]
                    str_indexes = ','.join(indexes)
                    #tostr_indexes = ','.join([ 'kgen_tostr(idx)' for idx in indexes])

                    prevobj = pobj
                    doobjs = []
                    for d in range(var.rank):
                        attrs = {
                            'loopcontrol':
                            'idx%(d)d=LBOUND(var,%(d)d), UBOUND(var,%(d)d)' % {
                                'd': d + 1
                            }
                        }
                        doobj = part_append_gensnode(prevobj,
                                                     EXEC_PART,
                                                     block_statements.Do,
                                                     attrs=attrs)

                        doobjs.append(doobj)
                        prevobj = doobj

                    attrs = {'expr': 'PRESENT( printvar )'}
                    ifpvarobj = part_append_gensnode(doobjs[-1],
                                                     EXEC_PART,
                                                     block_statements.IfThen,
                                                     attrs=attrs)

                    callname = None
                    for uname, req in stmt.unknowns.iteritems():
                        if uname.firstpartname() == stmt.name and len(
                                req.res_stmts) > 0:
                            res = req.res_stmts[0]
                            callname = get_dtype_writename(res)
                            break
                    if callname is None:
                        print 'WARNING: Can not find Type resolver for %s' % stmt.name
                        part_append_comment(ifpvarobj, EXEC_PART, \
                            'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                            (stmt.name, stmt.name))
                    else:
                        #attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', 'printvar // "(", %s, ")"'%tostr_indexes]}
                        attrs = {
                            'designator':
                            callname,
                            'items': [
                                'var(%s)' % str_indexes, 'kgen_unit',
                                'printvar // "(%s)"' % str_indexes
                            ]
                        }
                        part_append_gensnode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Call,
                                             attrs=attrs)

                        part_append_gensnode(ifpvarobj, EXEC_PART,
                                             statements.Else)

                        if any(
                                match_namepath(pattern,
                                               pack_exnamepath(
                                                   stmt, entity_name),
                                               internal=False)
                                for pattern in getinfo('print_var_names')):
                            #attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', '"%s(", %s, ")"'%(entity_name, tostr_indexes)]}
                            attrs = {
                                'designator':
                                callname,
                                'items': [
                                    'var(%s)' % str_indexes, 'kgen_unit',
                                    '"%s(%s)"' % (entity_name, str_indexes)
                                ]
                            }
                            part_append_gensnode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)
                        else:
                            attrs = {
                                'designator': callname,
                                'items':
                                ['var(%s)' % str_indexes, 'kgen_unit']
                            }
                            part_append_gensnode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)

                else:  # intrinsic type
                    attrs = {'items': ['var'], 'specs': ['UNIT = kgen_unit']}
                    part_append_gensnode(pobj,
                                         EXEC_PART,
                                         statements.Write,
                                         attrs=attrs)

                    if any(
                            match_namepath(pattern,
                                           pack_exnamepath(stmt, entity_name),
                                           internal=False)
                            for pattern in getinfo('print_var_names')):
                        if stmt.is_numeric() and var.is_array():
                            attrs = {
                                'items': [
                                    '"** KGEN DEBUG: " // "REAL(SUM(%s), 8) **"'
                                    % entity_name, 'REAL(SUM(var), 8)'
                                ]
                            }
                        else:
                            attrs = {
                                'items': [
                                    '"** KGEN DEBUG: " // "%s **" // NEW_LINE("A")'
                                    % entity_name, 'var'
                                ]
                            }
                        part_append_gensnode(pobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)
                    else:
                        attrs = {'expr': 'PRESENT( printvar )'}
                        ifpvarobj = part_append_gensnode(
                            pobj,
                            EXEC_PART,
                            block_statements.IfThen,
                            attrs=attrs)

                        if stmt.is_numeric() and var.is_array():
                            attrs = {
                                'items': [
                                    '"** KGEN DEBUG: REAL(SUM(" // printvar // "), 8) **"',
                                    'REAL(SUM(var), 8)'
                                ]
                            }
                        else:
                            attrs = {
                                'items': [
                                    '"** KGEN DEBUG: " // printvar // " **" // NEW_LINE("A")',
                                    'var'
                                ]
                            }
                        part_append_gensnode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)

            else:  # scalar
                if stmt.is_derived():
                    attrs = {'expr': 'PRESENT( printvar )'}
                    ifpvarobj = part_append_gensnode(pobj,
                                                     EXEC_PART,
                                                     block_statements.IfThen,
                                                     attrs=attrs)

                    callname = None
                    for uname, req in stmt.unknowns.iteritems():
                        if uname.firstpartname() == stmt.name and len(
                                req.res_stmts) > 0:
                            res = req.res_stmts[0]
                            callname = get_dtype_writename(res)
                            break
                    if callname is None:
                        print 'WARNING: Can not find Type resolver for %s' % stmt.name
                        part_append_comment(ifpvarobj, EXEC_PART, \
                            'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                            (stmt.name, stmt.name))
                    else:
                        attrs = {
                            'designator':
                            callname,
                            'items': [
                                'var', 'kgen_unit',
                                'printvar // " %s "' % entity_name
                            ]
                        }
                        part_append_gensnode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Call,
                                             attrs=attrs)

                        part_append_gensnode(ifpvarobj, EXEC_PART,
                                             statements.Else)

                        if any(
                                match_namepath(pattern,
                                               pack_exnamepath(
                                                   stmt, entity_name),
                                               internal=False)
                                for pattern in getinfo('print_var_names')):
                            attrs = {
                                'designator': callname,
                                'items':
                                ['var', 'kgen_unit',
                                 '"%s"' % entity_name]
                            }
                            part_append_gensnode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)
                        else:
                            attrs = {
                                'designator': callname,
                                'items': ['var', 'kgen_unit']
                            }
                            part_append_gensnode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)
                else:  # intrinsic type
                    attrs = {'items': ['var'], 'specs': ['UNIT = kgen_unit']}
                    part_append_gensnode(pobj,
                                         EXEC_PART,
                                         statements.Write,
                                         attrs=attrs)

                    if any(
                            match_namepath(pattern,
                                           pack_exnamepath(stmt, entity_name),
                                           internal=False)
                            for pattern in getinfo('print_var_names')):
                        attrs = {
                            'items': [
                                '"** KGEN DEBUG: " // "%s **"' % entity_name,
                                'var'
                            ]
                        }
                        part_append_gensnode(pobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)
                    else:
                        attrs = {'expr': 'PRESENT( printvar )'}
                        ifpvarobj = part_append_gensnode(
                            pobj,
                            EXEC_PART,
                            block_statements.IfThen,
                            attrs=attrs)

                        attrs = {
                            'items':
                            ['"** KGEN DEBUG: " // printvar // " **"', 'var']
                        }
                        part_append_gensnode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)
Beispiel #6
0
def create_write_subr(subrname, entity_name, parent, var, stmt, implicit=False):
    checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine) and n.name==subrname

    is_class_derived = check_class_derived(stmt)

    if not part_has_node(parent, SUBP_PART, checks):

        checks = lambda n: n.kgen_match_class==statements.Contains
        if not parent in state_gencore_contains and not part_has_node(parent, CONTAINS_PART, checks):
            part_append_comment(parent, CONTAINS_PART, '')
            part_append_gensnode(parent, CONTAINS_PART, statements.Contains)
            part_append_comment(parent, CONTAINS_PART, '')
            state_gencore_contains.append(parent)

        part_append_comment(parent, SUBP_PART, 'write state subroutine for %s'%subrname)
        attrs = {'name': subrname, 'args': ['var', 'kgen_unit', 'printvar']}
        subrobj = part_append_gensnode(parent, SUBP_PART, block_statements.Subroutine, attrs=attrs)
        part_append_comment(parent, SUBP_PART, '')

        parent.kgen_stmt.top.used4genstate = True

        # variable A
        #import pdb; pdb.set_trace()
        attrspec = ['INTENT(IN)']
        if var.is_pointer(): attrspec.append('POINTER')
        if var.is_allocatable(): attrspec.append('ALLOCATABLE')
        if var.is_array(): attrspec.append('DIMENSION(%s)'% ','.join(':'*var.rank))
        attrs = {'type_spec': stmt.__class__.__name__.upper(), 'attrspec': attrspec, 'selector':stmt.selector, 'entity_decls': ['var']}
        part_append_gensnode(subrobj, DECL_PART, stmt.__class__, attrs=attrs)

        # kgen_unit
        attrs = {'type_spec': 'INTEGER', 'attrspec': ['INTENT(IN)'], 'entity_decls': ['kgen_unit']}
        part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Integer, attrs=attrs)

        # printvar
        attrs = {'type_spec': 'CHARACTER', 'attrspec': ['INTENT(IN)', 'OPTIONAL'], 'selector':('*', None), 'entity_decls': ['printvar']}
        part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Character, attrs=attrs)

        # kgen_istrue
        attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
        part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Logical, attrs=attrs)

        attrs = {'type_spec': 'REAL', 'entity_decls': ['kgen_array_sum'], 'selector': (None, '8')}
        part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Real, attrs=attrs)

        # array index A
        if var.is_array():
            attrs = {'type_spec': 'INTEGER', 'entity_decls': [ 'idx%d'%(d+1) for d in range(var.rank) ]}
            part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Integer, attrs=attrs)

        part_append_comment(subrobj, DECL_PART, '')

        pobj = gen_write_istrue(subrobj, var, 'var')
        part_append_comment(subrobj, EXEC_PART, '')

        if var.is_array():
            for dim in range(var.rank):
                attrs = {'items': ['LBOUND(var, %d)'%(dim+1)], 'specs': ['UNIT = kgen_unit']}
                part_append_gensnode(pobj, EXEC_PART, statements.Write, attrs=attrs)

                attrs = {'items': ['UBOUND(var, %d)'%(dim+1)], 'specs': ['UNIT = kgen_unit']}
                part_append_gensnode(pobj, EXEC_PART, statements.Write, attrs=attrs)

            if stmt.is_derived() or is_class_derived:
                indexes = [ 'idx%d'%(d+1) for d in range(var.rank) ]
                str_indexes = ','.join(indexes)
                #tostr_indexes = ','.join([ 'kgen_tostr(idx)' for idx in indexes])

                prevobj = pobj
                doobjs = []
                for d in range(var.rank):
                    attrs = {'loopcontrol': 'idx%(d)d=LBOUND(var,%(d)d), UBOUND(var,%(d)d)'%{'d':d+1}}
                    doobj = part_append_gensnode(prevobj, EXEC_PART, block_statements.Do, attrs=attrs)

                    doobjs.append(doobj)
                    prevobj = doobj

                attrs = {'expr': 'PRESENT( printvar )'}
                ifpvarobj = part_append_gensnode(doobjs[-1], EXEC_PART, block_statements.IfThen, attrs=attrs)

                callname = None
                for uname, req in stmt.unknowns.iteritems():
                    if ( is_class_derived and uname.firstpartname()==stmt.selector[1]) or uname.firstpartname()==stmt.name:
                    #if uname.firstpartname()==stmt.name:
                        if len(req.res_stmts)>0:
                            res = req.res_stmts[0]
                            callname = get_dtype_writename(res)
                            break
                if callname is None:
                    print 'WARNING: Can not find Type resolver for %s'%stmt.name
                    part_append_comment(ifpvarobj, EXEC_PART, \
                        'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                        (stmt.name, stmt.name))
                else:
                    #attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', 'printvar // "(", %s, ")"'%tostr_indexes]}
                    attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', 'printvar // "(%s)"'%str_indexes]}
                    part_append_gensnode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

                    part_append_gensnode(ifpvarobj, EXEC_PART, statements.Else)

                    if any(match_namepath(pattern, pack_exnamepath(stmt, entity_name), internal=False) for pattern in getinfo('print_var_names')):
                        #attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', '"%s(", %s, ")"'%(entity_name, tostr_indexes)]}
                        attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', '"%s(%s)"'%(entity_name, str_indexes)]}
                        part_append_gensnode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)
                    else:
                        attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit']}
                        part_append_gensnode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

            else: # intrinsic type
                attrs = {'items': ['var'], 'specs': ['UNIT = kgen_unit']}
                part_append_gensnode(pobj, EXEC_PART, statements.Write, attrs=attrs)

                attrs = {'expr': 'PRESENT( printvar )'}
                ifpvarobj = part_append_gensnode(pobj, EXEC_PART, block_statements.IfThen, attrs=attrs)

                attrs = {'items': ['"** KGEN DEBUG: " // printvar // "**" // NEW_LINE("A")', 'var']}
                part_append_gensnode(ifpvarobj, EXEC_PART, statements.Write, attrs=attrs)
#
#                part_append_gensnode(ifpvarobj, EXEC_PART, statements.Else)
#
#                if any(match_namepath(pattern, pack_exnamepath(stmt, entity_name), internal=False) for pattern in getinfo('print_var_names')):
#                    if stmt.is_numeric():
#                        attrs = {'items': ['"** KGEN DEBUG: " // "%s **"'%entity_name, 'SUM(var)']}
#                    else:
#                        attrs = {'items': ['"** KGEN DEBUG: " // "%s **" // NEW_LINE("A")'%entity_name, 'var']}
#                    part_append_gensnode(pobj, EXEC_PART, statements.Write, attrs=attrs)
#                else:
#                    attrs = {'expr': 'PRESENT( printvar )'}
#                    ifpvarobj = part_append_gensnode(pobj, EXEC_PART, block_statements.IfThen, attrs=attrs)
#
#                    if stmt.is_numeric():
#                        attrs = {'items': ['"** KGEN DEBUG: " // printvar // " %s **"'%entity_name, 'SUM(var)']}
#                    else:
#                        attrs = {'items': ['"** KGEN DEBUG: " // printvar // " %s **" // NEW_LINE("A")'%entity_name, 'var']}
#                    part_append_gensnode(ifpvarobj, EXEC_PART, statements.Write, attrs=attrs)


        else: # scalar
            if stmt.is_derived() or is_class_derived:

                callname = None
                for uname, req in stmt.unknowns.iteritems():
                    if ( is_class_derived and uname.firstpartname()==stmt.selector[1]) or uname.firstpartname()==stmt.name:
                    #if uname.firstpartname()==stmt.name:
                        if len(req.res_stmts)>0:
                            res = req.res_stmts[0]
                            callname = get_dtype_writename(res)
                            break
                if callname is None:
                    print 'WARNING: Can not find Type resolver for %s'%stmt.name
                    part_append_comment(pobj, EXEC_PART, \
                        'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                        (stmt.name, stmt.name))
                else:
                    attrs = {'expr': 'PRESENT( printvar )'}
                    ifpvarobj = part_append_gensnode(pobj, EXEC_PART, block_statements.IfThen, attrs=attrs)

                    attrs = {'designator': callname, 'items': ['var', 'kgen_unit', 'printvar // " %s "'%entity_name]}
                    part_append_gensnode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

                    part_append_gensnode(ifpvarobj, EXEC_PART, statements.Else)

                    if any(match_namepath(pattern, pack_exnamepath(stmt, entity_name), internal=False) for pattern in getinfo('print_var_names')):
                        attrs = {'designator': callname, 'items': ['var', 'kgen_unit', '"%s"'%entity_name]}
                        part_append_gensnode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)
                    else:
                        attrs = {'designator': callname, 'items': ['var', 'kgen_unit']}
                        part_append_gensnode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)
            else: # intrinsic type
                attrs = {'items': ['var'], 'specs': ['UNIT = kgen_unit']}
                part_append_gensnode(pobj, EXEC_PART, statements.Write, attrs=attrs)

                attrs = {'expr': 'PRESENT( printvar )'}
                ifpvarobj = part_append_gensnode(pobj, EXEC_PART, block_statements.IfThen, attrs=attrs)

                attrs = {'items': ['"** KGEN DEBUG: " // printvar // "**" // NEW_LINE("A")', 'var']}
                part_append_gensnode(ifpvarobj, EXEC_PART, statements.Write, attrs=attrs)
    def create_subr_write_typedecl_in_parentblock(self, node):
        stmt = node.kgen_stmt

        argintype = []
        localintype = []
        localouttype = []
        for uname, req in KGGenType.get_state_in(stmt.geninfo):
            entity_name = uname.firstpartname()
            var = stmt.get_variable(entity_name)

            if var.is_parameter(): continue
            if is_remove_state(entity_name, stmt): continue

            if self.check_intent(entity_name, stmt):
                if (entity_name, STATE_PBLOCK_WRITE_IN_ARGS) not in argintype:
                    argintype.append((entity_name, STATE_PBLOCK_WRITE_IN_ARGS))
            elif (entity_name,
                  STATE_PBLOCK_WRITE_IN_LOCALS) not in localintype and (
                      entity_name,
                      STATE_PBLOCK_WRITE_IN_ARGS) not in argintype:
                localintype.append(
                    (uname.firstpartname(), STATE_PBLOCK_WRITE_IN_LOCALS))

        for uname, req in KGGenType.get_state_out(stmt.geninfo):
            entity_name = uname.firstpartname()
            var = stmt.get_variable(entity_name)

            if var.is_parameter(): continue
            if is_remove_state(entity_name, stmt): continue

            if (entity_name,
                    STATE_PBLOCK_WRITE_OUT_LOCALS) not in localouttype:
                localouttype.append(
                    (uname.firstpartname(), STATE_PBLOCK_WRITE_OUT_LOCALS))

            if (entity_name, STATE_PBLOCK_WRITE_IN_ARGS) in argintype: continue

            if (entity_name, STATE_PBLOCK_WRITE_IN_LOCALS) not in localintype:
                localintype.append(
                    (uname.firstpartname(), STATE_PBLOCK_WRITE_IN_LOCALS))
        vartypes = {
            'argintype': argintype,
            'localintype': localintype,
            'localouttype': localouttype
        }

        # for state
        is_class_derived = check_class_derived(stmt)
        for vartypename, vartype in vartypes.iteritems():
            for entity_name, partid in vartype:
                var = stmt.get_variable(entity_name)
                subrname = get_typedecl_writename(stmt, entity_name)
                if var.is_array():
                    if is_zero_array(var, stmt): continue
                    if stmt.is_derived() or is_class_derived:
                        self.create_write_call(node.kgen_kernel_id, partid,
                                               subrname, entity_name, stmt,
                                               var)
                        if subrname not in self.state_created_subrs:
                            create_write_subr(subrname, entity_name,
                                              node.kgen_parent, var, stmt)
                            self.state_created_subrs.append(subrname)
                    else:  # intrinsic type
                        if var.is_explicit_shape_array():
                            if vartypename == 'argintype' or var.is_pointer():
                                self.create_write_call(node.kgen_kernel_id,
                                                       partid, subrname,
                                                       entity_name, stmt, var)
                                if subrname not in self.state_created_subrs:
                                    create_write_subr(subrname, entity_name,
                                                      node.kgen_parent, var,
                                                      stmt)
                                    self.state_created_subrs.append(subrname)
                            else:
                                self.create_write_intrinsic(
                                    node.kgen_kernel_id, partid, entity_name,
                                    stmt, var)
                        else:  # implicit array
                            self.create_write_call(node.kgen_kernel_id, partid,
                                                   subrname, entity_name, stmt,
                                                   var)
                            if subrname not in self.state_created_subrs:
                                create_write_subr(subrname, entity_name,
                                                  node.kgen_parent, var, stmt)
                                self.state_created_subrs.append(subrname)
                else:  # scalar
                    if stmt.is_derived() or is_class_derived or var.is_pointer(
                    ):
                        if var.is_allocatable() or var.is_pointer(
                        ) or var.is_pointer():
                            self.create_write_call(node.kgen_kernel_id, partid,
                                                   subrname, entity_name, stmt,
                                                   var)
                            if subrname not in self.state_created_subrs:
                                create_write_subr(subrname, entity_name,
                                                  node.kgen_parent, var, stmt)
                                self.state_created_subrs.append(subrname)
                        else:
                            subrname = None
                            for uname, req in stmt.unknowns.iteritems():
                                if uname.firstpartname() == stmt.name and len(
                                        req.res_stmts) > 0:
                                    res = req.res_stmts[0]
                                    subrname = get_dtype_writename(res)
                                    break
                            if subrname is None:
                                print 'WARNING: Can not find Type resolver for %s' % stmt.name
                                namedpart_append_comment(node.kgen_kernel_id, partid, \
                                    'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                                    (stmt.name, stmt.name))
                            else:
                                self.create_write_call(node.kgen_kernel_id,
                                                       partid, subrname,
                                                       entity_name, stmt, var)
                    else:  # intrinsic type
                        self.create_write_intrinsic(node.kgen_kernel_id,
                                                    partid, entity_name, stmt,
                                                    var)
    def create_subr_write_typedecl_in_parentblock(self, node):
        stmt = node.kgen_stmt

        argintype = []
        localintype = []
        localouttype = []
        for uname, req in KGGenType.get_state_in(stmt.geninfo):
            entity_name = uname.firstpartname()
            var = stmt.get_variable(entity_name)

            if var.is_parameter(): continue
            if is_remove_state(entity_name, stmt): continue

            if self.check_intent(entity_name, stmt):
                if (entity_name,STATE_PBLOCK_WRITE_IN_ARGS) not in argintype:
                    argintype.append((entity_name, STATE_PBLOCK_WRITE_IN_ARGS))
            elif (entity_name,STATE_PBLOCK_WRITE_IN_LOCALS) not in localintype and (entity_name,STATE_PBLOCK_WRITE_IN_ARGS) not in argintype:
                localintype.append((uname.firstpartname(), STATE_PBLOCK_WRITE_IN_LOCALS))

        for uname, req in KGGenType.get_state_out(stmt.geninfo):
            entity_name = uname.firstpartname()
            var = stmt.get_variable(entity_name)

            if var.is_parameter(): continue
            if is_remove_state(entity_name, stmt): continue

            if (entity_name,STATE_PBLOCK_WRITE_OUT_LOCALS) not in localouttype:
                localouttype.append((uname.firstpartname(), STATE_PBLOCK_WRITE_OUT_LOCALS))

            if (entity_name,STATE_PBLOCK_WRITE_IN_ARGS) in argintype: continue

            if (entity_name,STATE_PBLOCK_WRITE_IN_LOCALS) not in localintype:
                localintype.append((uname.firstpartname(), STATE_PBLOCK_WRITE_IN_LOCALS))
        vartypes = { 'argintype': argintype, 'localintype': localintype, 'localouttype': localouttype }

        # for state
        is_class_derived = check_class_derived(stmt)
        for vartypename, vartype in vartypes.iteritems():
            for entity_name, partid in vartype:
                var = stmt.get_variable(entity_name)
                subrname = get_typedecl_writename(stmt, entity_name)
                if var.is_array():
                    if is_zero_array(var, stmt): continue
                    if stmt.is_derived() or is_class_derived:
                        self.create_write_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var)
                        if subrname not in self.state_created_subrs:
                            create_write_subr(subrname, entity_name, node.kgen_parent, var, stmt)
                            self.state_created_subrs.append(subrname)
                    else: # intrinsic type
                        if var.is_explicit_shape_array():
                            if vartypename=='argintype' or var.is_pointer():
                                self.create_write_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var)
                                if subrname not in self.state_created_subrs:
                                    create_write_subr(subrname, entity_name, node.kgen_parent, var, stmt)
                                    self.state_created_subrs.append(subrname)
                            else:
                                self.create_write_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var)
                        else: # implicit array
                            self.create_write_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var)
                            if subrname not in self.state_created_subrs:
                                create_write_subr(subrname, entity_name, node.kgen_parent, var, stmt)
                                self.state_created_subrs.append(subrname)
                else: # scalar
                    if stmt.is_derived() or is_class_derived or var.is_pointer():
                        if var.is_allocatable() or var.is_pointer() or var.is_pointer():
                            self.create_write_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var)
                            if subrname not in self.state_created_subrs:
                                create_write_subr(subrname, entity_name, node.kgen_parent, var, stmt)
                                self.state_created_subrs.append(subrname)
                        else:
                            subrname = None
                            for uname, req in stmt.unknowns.iteritems():
                                if uname.firstpartname()==stmt.name and len(req.res_stmts)>0:
                                    res = req.res_stmts[0]
                                    subrname = get_dtype_writename(res)
                                    break
                            if subrname is None:
                                print 'WARNING: Can not find Type resolver for %s'%stmt.name
                                namedpart_append_comment(node.kgen_kernel_id, partid, \
                                    'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                                    (stmt.name, stmt.name))
                            else:
                                self.create_write_call(node.kgen_kernel_id, partid, subrname, entity_name, stmt, var)
                    else: # intrinsic type
                        self.create_write_intrinsic(node.kgen_kernel_id, partid, entity_name, stmt, var)
Beispiel #9
0
    def create_subr_write_typedecl_in_module(self, node):
        parent = node.kgen_parent
        stmt = node.kgen_stmt

        raw_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state(stmt.geninfo)])
        entity_names = [ e for e in raw_entity_names if not stmt.get_variable(e).is_parameter() ]
        
        raw_out_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state_out(stmt.geninfo)])
        out_entity_names = [ e for e in raw_out_entity_names if not stmt.get_variable(e).is_parameter() ]

        #entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state(stmt.geninfo)])
        #out_entity_names = set([ uname.firstpartname() for uname, req in KGGenType.get_state_out(stmt.geninfo)])

        is_class_derived = check_class_derived(stmt)

        for entity_name, entity_decl in zip(entity_names, stmt.entity_decls):
            if node.kgen_parent.name+entity_name in self.state_extern_writes: continue
            if is_remove_state(entity_name, stmt): continue

            self.state_extern_writes.append(node.kgen_parent.name+entity_name)

            var = stmt.get_variable(entity_name)
            subrname = get_typedecl_writename(stmt, entity_name)

            if var.is_array():
                if is_zero_array(var, stmt): continue
                if stmt.is_derived() or is_class_derived:
                    self.create_write_call(self.state_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var)
                    if entity_name in out_entity_names:
                        self.create_write_call(self.state_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var)
                    if subrname not in self.state_created_subrs:
                        create_write_subr(subrname, entity_name, parent, var, stmt)
                        self.state_created_subrs.append(subrname)
                else: # intrinsic type
                    if var.is_explicit_shape_array():
                        self.create_write_intrinsic(self.state_externs_subrs[node.kgen_parent][0], entity_name, stmt, var)
                        if entity_name in out_entity_names:
                            self.create_write_intrinsic(self.state_externs_subrs[node.kgen_parent][1], entity_name, stmt, var)
                    else: # implicit array
                        self.create_write_call(self.state_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var)
                        if entity_name in out_entity_names:
                            self.create_write_call(self.state_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var)
                        if subrname not in self.state_created_subrs:
                            create_write_subr(subrname, entity_name, parent, var, stmt)
                            self.state_created_subrs.append(subrname)
            else: # scalar
                if stmt.is_derived() or is_class_derived:
                    if var.is_allocatable() or var.is_pointer():
                        self.create_write_call(self.state_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var)
                        if entity_name in out_entity_names:
                            self.create_write_call(self.state_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var)
                        if subrname not in self.state_created_subrs:
                            create_write_subr(subrname, entity_name, parent, var, stmt)
                            self.state_created_subrs.append(subrname)
                    else:
                        subrname = None
                        for uname, req in stmt.unknowns.iteritems():
                            if ( is_class_derived and uname.firstpartname()==stmt.selector[1]) or uname.firstpartname()==stmt.name:
                            #if uname.firstpartname()==stmt.name:
                                if len(req.res_stmts)>0:
                                    res = req.res_stmts[0]
                                    subrname = get_dtype_writename(res)
                                    break
                        if subrname is None:
                            print 'WARNING: Can not find Type resolver for %s'%stmt.name
                            namedpart_append_comment(self.state_externs_subrs[node.kgen_parent][0], EXEC_PART, \
                                'ERROR: "%s" is not resolved. Call statements to write "%s" is not created here.'%\
                                (stmt.name, stmt.name))
                        else:
                            self.create_write_call(self.state_externs_subrs[node.kgen_parent][0], subrname, entity_name, stmt, var)
                            if entity_name in out_entity_names:
                                self.create_write_call(self.state_externs_subrs[node.kgen_parent][1], subrname, entity_name, stmt, var)
                else: # intrinsic type
                    self.create_write_intrinsic(self.state_externs_subrs[node.kgen_parent][0], entity_name, stmt, var)
                    if entity_name in out_entity_names:
                        self.create_write_intrinsic(self.state_externs_subrs[node.kgen_parent][1], entity_name, stmt, var)
Beispiel #10
0
    def create_dtype_write_subr(self, node):
        assert node.kgen_stmt, 'None kgen statement'

        subrname = get_dtype_writename(node.kgen_stmt)
        if subrname is None: return

        parent = node.kgen_parent
        checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine) and n.name==subrname
        if  not part_has_node(parent, SUBP_PART, checks):

            checks = lambda n: n.kgen_match_class==statements.Contains
            if not parent in state_gencore_contains and not part_has_node(parent, CONTAINS_PART, checks):
                part_append_comment(parent, CONTAINS_PART, '')
                part_append_gensnode(parent, CONTAINS_PART, statements.Contains)
                part_append_comment(parent, CONTAINS_PART, '')
                state_gencore_contains.append(parent)

            part_append_comment(parent, SUBP_PART, 'read state subroutine for %s'%subrname)
            attrs = {'prefix': 'RECURSIVE', 'name': subrname, 'args': ['var', 'kgen_unit', 'printvar']}
            subrobj = part_append_gensnode(parent, SUBP_PART, block_statements.Subroutine, attrs=attrs)
            part_append_comment(parent, SUBP_PART, '')

            # variable
            attrs = {'type_spec': 'TYPE', 'attrspec': ['INTENT(IN)'], 'selector':(None, node.name), 'entity_decls': ['var']}
            part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Type, attrs=attrs)

            # kgen_unit
            attrs = {'type_spec': 'INTEGER', 'attrspec': ['INTENT(IN)'], 'entity_decls': ['kgen_unit']}
            part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Integer, attrs=attrs)

            # printvar
            attrs = {'type_spec': 'CHARACTER', 'attrspec': ['INTENT(IN)', 'OPTIONAL'], 'selector':('*', None), 'entity_decls': ['printvar']}
            part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Character, attrs=attrs)

            # kgen_istrue
            attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
            part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Logical, attrs=attrs)

            attrs = {'type_spec': 'REAL', 'entity_decls': ['kgen_array_sum'], 'selector': (None, '8')}
            part_append_gensnode(subrobj, DECL_PART, typedecl_statements.Real, attrs=attrs)

            part_append_comment(subrobj, DECL_PART, '')

            #comp_part = get_part(node, TYPE_COMP_PART) 
            comp_part = get_part(node, TYPE_PART) 
            for item in comp_part:
                if not hasattr(item, 'kgen_stmt'): continue
                if not isinstance(item.kgen_stmt, typedecl_statements.TypeDeclarationStatement): continue

                stmt = item.kgen_stmt
                entity_names = [ get_entity_name(decl) for decl in stmt.entity_decls ]

                for entity_name, entity_decl in zip(entity_names, stmt.entity_decls):

                    if is_remove_state(entity_name, stmt): continue

                    node.kgen_stmt.top.used4genstate = True

                    var = stmt.get_variable(entity_name)
                    callname = get_typedecl_writename(stmt, entity_name)

                    if var.is_array():
                        if stmt.is_derived():
                            self.create_write_call(subrobj, callname, entity_name, stmt, var)
                        else: # intrinsic type
                            if var.is_explicit_shape_array():
                                self.create_write_intrinsic(subrobj, entity_name, stmt, var)
                            else: # implicit array
                                self.create_write_call(subrobj, callname, entity_name, stmt, var)
                    else: # scalar
                        if stmt.is_derived():
                            if var.is_allocatable() or var.is_pointer():
                                self.create_write_call(subrobj, callname, entity_name, stmt, var)
                            else:
                                callname = None
                                for uname, req in stmt.unknowns.iteritems():
                                    if uname.firstpartname()==stmt.name and len(req.res_stmts)>0:
                                        res = req.res_stmts[0]
                                        callname = get_dtype_writename(res)
                                        break
                                if callname is None:
                                    print 'WARNING: Can not find Type resolver for %s'%stmt.name
                                    part_append_comment(subrobj, EXEC_PART, \
                                        'ERROR: "%s" is not resolved. Call statement to write "%s" is not created here.'%\
                                        (stmt.name, stmt.name))
                                else:
                                    self.create_write_call(subrobj, callname, entity_name, stmt, var)
                        else: # intrinsic type
                            self.create_write_intrinsic(subrobj, entity_name, stmt, var)

                part_append_comment(subrobj, EXEC_PART, '')

            # create public stmt
            if parent.kgen_match_class in [ block_statements.Program, block_statements.Module ]:
                attrs = {'items': [subrname]}
                part_append_gensnode(parent, DECL_PART, statements.Public, attrs=attrs)