Пример #1
0
    def create_parentblock_parts(self, node):

        namedpart_link_part(node, KERNEL_PBLOCK_USE_PART, USE_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_DECL_PART, DECL_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_EXEC_PART, EXEC_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_CONTAINS_PART, CONTAINS_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_SUBP_PART, SUBP_PART)

        attrs = {'name':'kgen_utils_mod', 'isonly': True, 'items':['kgen_dp', 'kgen_array_sumcheck']}
        part_append_genknode(node, USE_PART, statements.Use, attrs=attrs)
           
        #attrs = {'name':'IEEE_ARITHMETIC', 'nature': 'INTRINSIC', 'isonly': True, 'items':['ieee_is_normal']}
        #part_append_genknode(node, USE_PART, statements.Use, attrs=attrs)

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

        attrs = {'type_spec': 'REAL', 'attrspec': ['INTENT(OUT)'], 'entity_decls': ['kgen_elapsed_time'], \
            'selector': (None, 'kgen_dp')}
        part_append_genknode(node, DECL_PART, typedecl_statements.Integer, attrs=attrs)

        attrs = {'type_spec': 'LOGICAL', 'attrspec': ['INTENT(OUT)'], 'entity_decls': ['kgen_isverified']}
        part_append_genknode(node, DECL_PART, typedecl_statements.Logical, attrs=attrs)

        attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
        part_append_genknode(node, DECL_PART, typedecl_statements.Logical, attrs=attrs)

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

        part_append_comment(node, DECL_PART, '')

        kernel_stmts = getinfo('callsite_stmts')
        if len(kernel_stmts)!=1 or not isinstance(kernel_stmts[0], statements.Call):
            # ensure contains
            checks = lambda n: n.kgen_isvalid and n.kgen_match_class==statements.Contains
            if not node in kernel_gencore_contains and not part_has_node(node, CONTAINS_PART, checks):
                part_append_comment(node, CONTAINS_PART, '')
                part_append_genknode(node, CONTAINS_PART, statements.Contains)
                part_append_comment(node, CONTAINS_PART, '')
                kernel_gencore_contains.append(node)


            part_append_comment(node, SUBP_PART, 'kgen kernel subroutine')
            attrs = {'name': 'kgen_kernel'}
            subrobj = part_append_genknode(node, SUBP_PART, block_statements.Subroutine, attrs=attrs)
            part_append_comment(node, SUBP_PART, '')

            start = kernel_stmts[0].item.span[0]-1
            end = kernel_stmts[-1].item.span[1]
            lines = kernel_stmts[0].top.prep[start:end]
            lines_str = '\n'.join(lines)
            dummy_node = part_append_genknode(subrobj, EXEC_PART, statements.Call)
            dummy_node.kgen_stmt = getinfo('dummy_stmt')
            dummy_node.kgen_forced_line = lines_str
Пример #2
0
    def create_kernel_module_parts(self, node):
        in_subrobj = None
        in_subrname = get_module_in_readname(node.kgen_stmt)
        checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine
                                      ) and n.name == in_subrname
        if not part_has_node(node, SUBP_PART, checks):

            checks = lambda n: n.kgen_isvalid and n.kgen_match_class == statements.Contains
            if not node in kernel_gencore_contains and not part_has_node(
                    node, CONTAINS_PART, checks):
                part_append_comment(node, CONTAINS_PART, '')
                part_append_genknode(node, CONTAINS_PART, statements.Contains)
                part_append_comment(node, CONTAINS_PART, '')
                kernel_gencore_contains.append(node)

            attrs = {'name': in_subrname, 'args': ['kgen_unit']}
            part_append_comment(node, SUBP_PART,
                                'read state subroutine for %s' % in_subrname)
            in_subrobj = part_append_genknode(node,
                                              SUBP_PART,
                                              block_statements.Subroutine,
                                              attrs=attrs)
            part_append_comment(node, SUBP_PART, '')

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

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

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

            part_append_comment(in_subrobj, DECL_PART, '')

            # add public stmt
            attrs = {'items': [in_subrname]}
            part_append_genknode(node,
                                 DECL_PART,
                                 statements.Public,
                                 attrs=attrs)

        out_subrobj = None
        if hasattr(node.kgen_stmt, 'geninfo') and KGGenType.has_state_out(
                node.kgen_stmt.geninfo):
            out_subrname = get_module_out_readname(node.kgen_stmt)
            checks = lambda n: isinstance(n.kgen_stmt, block_statements.
                                          Subroutine
                                          ) and n.name == out_subrname
            if not part_has_node(node, SUBP_PART, checks):

                checks = lambda n: n.kgen_isvalid and n.kgen_match_class == statements.Contains
                if not node in kernel_gencore_contains and not part_has_node(
                        node, CONTAINS_PART, checks):
                    part_append_comment(node, CONTAINS_PART, '')
                    part_append_genknode(node, CONTAINS_PART,
                                         statements.Contains)
                    part_append_comment(node, CONTAINS_PART, '')
                    kernel_gencore_contains.append(node)

                attrs = {'name': out_subrname, 'args': ['kgen_unit']}
                part_append_comment(
                    node, SUBP_PART,
                    'read state subroutine for %s' % out_subrname)
                out_subrobj = part_append_genknode(node,
                                                   SUBP_PART,
                                                   block_statements.Subroutine,
                                                   attrs=attrs)
                part_append_comment(node, SUBP_PART, '')

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

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

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

                # add public stmt
                attrs = {'items': [out_subrname]}
                part_append_genknode(node,
                                     DECL_PART,
                                     statements.Public,
                                     attrs=attrs)

        if in_subrobj or out_subrobj:
            self.kernel_externs_subrs[node] = (in_subrobj, out_subrobj)

            # register event per typedecl
            self.frame_msg.add_event(KERNEL_SELECTION.ALL, FILE_TYPE.KERNEL, GENERATION_STAGE.BEGIN_PROCESS, \
                typedecl_statements.TypeDeclarationStatement, self.is_extern_in_kernel_module, self.create_subr_read_typedecl_in_module)

            # register event per module
            self.frame_msg.add_event(KERNEL_SELECTION.ALL, FILE_TYPE.KERNEL, GENERATION_STAGE.BEGIN_PROCESS, \
                block_statements.Module, self.has_externs_in_module, self.create_kernel_stmts_in_callsite)
Пример #3
0
    def create_parentblock_parts(self, node):

        namedpart_link_part(node, KERNEL_PBLOCK_USE_PART, USE_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_DECL_PART, DECL_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_EXEC_PART, EXEC_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_CONTAINS_PART, CONTAINS_PART)
        namedpart_link_part(node, KERNEL_PBLOCK_SUBP_PART, SUBP_PART)

        attrs = {
            'name': 'kgen_utils_mod',
            'isonly': True,
            'items': ['kgen_dp', 'kgen_array_sumcheck']
        }
        part_append_genknode(node, USE_PART, statements.Use, attrs=attrs)

        #attrs = {'name':'IEEE_ARITHMETIC', 'nature': 'INTRINSIC', 'isonly': True, 'items':['ieee_is_normal']}
        #part_append_genknode(node, USE_PART, statements.Use, attrs=attrs)

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

        attrs = {'type_spec': 'REAL', 'attrspec': ['INTENT(OUT)'], 'entity_decls': ['kgen_elapsed_time'], \
            'selector': (None, 'kgen_dp')}
        part_append_genknode(node,
                             DECL_PART,
                             typedecl_statements.Integer,
                             attrs=attrs)

        attrs = {
            'type_spec': 'LOGICAL',
            'attrspec': ['INTENT(OUT)'],
            'entity_decls': ['kgen_isverified']
        }
        part_append_genknode(node,
                             DECL_PART,
                             typedecl_statements.Logical,
                             attrs=attrs)

        attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
        part_append_genknode(node,
                             DECL_PART,
                             typedecl_statements.Logical,
                             attrs=attrs)

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

        part_append_comment(node, DECL_PART, '')

        kernel_stmts = getinfo('callsite_stmts')
        if len(kernel_stmts) != 1 or not isinstance(kernel_stmts[0],
                                                    statements.Call):
            # ensure contains
            checks = lambda n: n.kgen_isvalid and n.kgen_match_class == statements.Contains
            if not node in kernel_gencore_contains and not part_has_node(
                    node, CONTAINS_PART, checks):
                part_append_comment(node, CONTAINS_PART, '')
                part_append_genknode(node, CONTAINS_PART, statements.Contains)
                part_append_comment(node, CONTAINS_PART, '')
                kernel_gencore_contains.append(node)

            part_append_comment(node, SUBP_PART, 'kgen kernel subroutine')
            attrs = {'name': 'kgen_kernel'}
            subrobj = part_append_genknode(node,
                                           SUBP_PART,
                                           block_statements.Subroutine,
                                           attrs=attrs)
            part_append_comment(node, SUBP_PART, '')

            start = kernel_stmts[0].item.span[0] - 1
            end = kernel_stmts[-1].item.span[1]
            lines = kernel_stmts[0].top.prep[start:end]
            lines_str = '\n'.join(lines)
            dummy_node = part_append_genknode(subrobj, EXEC_PART,
                                              statements.Call)
            dummy_node.kgen_stmt = getinfo('dummy_stmt')
            dummy_node.kgen_forced_line = lines_str
Пример #4
0
    def create_dtype_read_subr(self, node):
        assert node.kgen_stmt, 'None kgen statement'

        subrname = get_dtype_readname(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_isvalid and n.kgen_match_class == statements.Contains
            if not parent in kernel_gencore_contains and not part_has_node(
                    parent, CONTAINS_PART, checks):
                part_append_comment(parent, CONTAINS_PART, '')
                part_append_genknode(parent, CONTAINS_PART,
                                     statements.Contains)
                part_append_comment(parent, CONTAINS_PART, '')
                kernel_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_genknode(parent,
                                           SUBP_PART,
                                           block_statements.Subroutine,
                                           attrs=attrs)
            part_append_comment(parent, SUBP_PART, '')

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

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

            # printname
            attrs = {
                'type_spec': 'CHARACTER',
                'attrspec': ['INTENT(IN)'],
                'selector': ('*', None),
                'entity_decls': ['printname']
            }
            part_append_genknode(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_genknode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Character,
                                 attrs=attrs)

            # kgen_istrue
            attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
            part_append_genknode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Logical,
                                 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, '')

            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].genkpair, 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

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

                    if var.is_array():
                        if stmt.is_derived():
                            self.create_read_call(subrobj, callname,
                                                  entity_name, stmt, var)
                        else:  # intrinsic type
                            if var.is_explicit_shape_array():
                                self.create_read_intrinsic(
                                    subrobj, entity_name, stmt, var)
                                callname = None
                            else:  # implicit array
                                self.create_read_call(subrobj, callname,
                                                      entity_name, stmt, var)
                    else:  # scalar
                        if stmt.is_derived():
                            if var.is_allocatable() or var.is_pointer():
                                self.create_read_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_readname(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 read "%s" is not created here.'%\
                                        (stmt.name, stmt.name))
                                else:
                                    self.create_read_call(
                                        subrobj, callname, entity_name, stmt,
                                        var)
                        else:  # intrinsic type
                            self.create_read_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_genknode(pnode,
                                                 USE_PART,
                                                 statements.Use,
                                                 attrs=attrs)
                            self.state_created_use_items.append(
                                (id(pnode), callname))

                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_genknode(parent,
                                     DECL_PART,
                                     statements.Public,
                                     attrs=attrs)
Пример #5
0
    def create_read_subr(self,
                         subrname,
                         entity_name,
                         parent,
                         var,
                         stmt,
                         allocate=False):
        checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine
                                      ) and n.name == subrname
        if subrname not in self.kernel_created_subrs and not part_has_node(
                parent, SUBP_PART, checks):

            if is_remove_state(entity_name, stmt): return

            self.kernel_created_subrs.append(subrname)

            checks = lambda n: n.kgen_isvalid and n.kgen_match_class == statements.Contains
            if not parent in kernel_gencore_contains and not part_has_node(
                    parent, CONTAINS_PART, checks):
                part_append_comment(parent, CONTAINS_PART, '')
                part_append_genknode(parent, CONTAINS_PART,
                                     statements.Contains)
                part_append_comment(parent, CONTAINS_PART, '')
                kernel_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_genknode(parent,
                                           SUBP_PART,
                                           block_statements.Subroutine,
                                           attrs=attrs)
            part_append_comment(parent, SUBP_PART, '')

            # variable A
            #import pdb; pdb.set_trace()
            attrspec = ['INTENT(INOUT)']
            if var.is_allocatable() or allocate: 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_genknode(subrobj,
                                 DECL_PART,
                                 stmt.__class__,
                                 attrs=attrs)

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

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

            # kgen_istrue
            attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
            part_append_genknode(subrobj,
                                 DECL_PART,
                                 typedecl_statements.Logical,
                                 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)

            # 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_genknode(subrobj,
                                     DECL_PART,
                                     typedecl_statements.Integer,
                                     attrs=attrs)

                attrs = {
                    'type_spec': 'INTEGER',
                    'attrspec': ['DIMENSION(2,%d)' % var.rank],
                    'entity_decls': ['kgen_bound']
                }
                part_append_genknode(subrobj,
                                     DECL_PART,
                                     typedecl_statements.Integer,
                                     attrs=attrs)

            part_append_comment(subrobj, DECL_PART, '')

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

            if var.is_array():

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

                    attrs = {
                        'items': ['kgen_bound(2, %d)' % (dim + 1)],
                        'specs': ['UNIT = kgen_unit']
                    }
                    part_append_genknode(pobj,
                                         EXEC_PART,
                                         statements.Read,
                                         attrs=attrs)

                    #bound_args.append('kgen_bound(2,%d)-kgen_bound(1,%d)+1'%(dim+1, dim+1))
                    bound_args.append('kgen_bound(1,%d):kgen_bound(2,%d)' %
                                      (dim + 1, dim + 1))

                if var.is_allocatable() or var.is_pointer() or allocate:
                    attrs = {'items': ['var(%s)' % ', '.join(bound_args)]}
                    part_append_genknode(pobj,
                                         EXEC_PART,
                                         statements.Allocate,
                                         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=kgen_bound(1,%(d)d), kgen_bound(2,%(d)d)'
                            % {
                                'd': d + 1
                            }
                        }
                        doobj = part_append_genknode(prevobj,
                                                     EXEC_PART,
                                                     block_statements.Do,
                                                     attrs=attrs)

                        doobjs.append(doobj)
                        prevobj = doobj

                    attrs = {'expr': 'PRESENT( printvar )'}
                    ifpvarobj = part_append_genknode(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_readname(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 read "%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_genknode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Call,
                                             attrs=attrs)

                        part_append_genknode(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_genknode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)
                        else:
                            attrs = {
                                'designator': callname,
                                'items':
                                ['var(%s)' % str_indexes, 'kgen_unit']
                            }
                            part_append_genknode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)

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

                    if stmt.is_numeric():
                        if isinstance(stmt, typedecl_statements.Real):
                            attrs = {
                                'designator':
                                'kgen_array_sumcheck',
                                'items': [
                                    '"%s"' % entity_name, 'kgen_array_sum',
                                    'REAL(SUM(var, mask=(var .eq. var)), 8)',
                                    '.TRUE.'
                                ]
                            }
                        else:
                            attrs = {
                                'designator':
                                'kgen_array_sumcheck',
                                'items': [
                                    '"%s"' % entity_name, 'kgen_array_sum',
                                    'REAL(SUM(var), 8)', '.TRUE.'
                                ]
                            }
                        part_append_genknode(pobj,
                                             EXEC_PART,
                                             statements.Call,
                                             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_genknode(pobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)
                    else:
                        attrs = {'expr': 'PRESENT( printvar )'}
                        ifpvarobj = part_append_genknode(
                            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_genknode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)

            else:  # scalar
                if var.is_allocatable() or var.is_pointer() or allocate:
                    attrs = {'items': ['var']}
                    part_append_genknode(pobj,
                                         EXEC_PART,
                                         statements.Allocate,
                                         attrs=attrs)

                if stmt.is_derived():
                    attrs = {'expr': 'PRESENT( printvar )'}
                    ifpvarobj = part_append_genknode(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_readname(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 read "%s" is not created here.'%\
                            (stmt.name, stmt.name))
                    else:
                        attrs = {
                            'designator':
                            callname,
                            'items': [
                                'var', 'kgen_unit',
                                'printvar // " %s "' % entity_name
                            ]
                        }
                        part_append_genknode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Call,
                                             attrs=attrs)

                        part_append_genknode(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_genknode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)
                        else:
                            attrs = {
                                'designator': callname,
                                'items': ['var', 'kgen_unit']
                            }
                            part_append_genknode(ifpvarobj,
                                                 EXEC_PART,
                                                 statements.Call,
                                                 attrs=attrs)
                else:  # intrinsic type
                    attrs = {'items': ['var'], 'specs': ['UNIT = kgen_unit']}
                    part_append_genknode(pobj,
                                         EXEC_PART,
                                         statements.Read,
                                         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 **" // NEW_LINE("A")'
                                % entity_name, 'var'
                            ]
                        }
                        part_append_genknode(pobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)
                    else:
                        attrs = {'expr': 'PRESENT( printvar )'}
                        ifpvarobj = part_append_genknode(
                            pobj,
                            EXEC_PART,
                            block_statements.IfThen,
                            attrs=attrs)

                        attrs = {
                            'items': [
                                '"** KGEN DEBUG: " // printvar // " **" // NEW_LINE("A")',
                                'var'
                            ]
                        }
                        part_append_genknode(ifpvarobj,
                                             EXEC_PART,
                                             statements.Write,
                                             attrs=attrs)
Пример #6
0
def create_read_subr(subrname, entity_name, parent, var, stmt, allocate=False, ename_prefix=''):

    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_isvalid and n.kgen_match_class==statements.Contains
        if not parent in kernel_gencore_contains and not part_has_node(parent, CONTAINS_PART, checks):
            part_append_comment(parent, CONTAINS_PART, '')
            part_append_genknode(parent, CONTAINS_PART, statements.Contains)
            part_append_comment(parent, CONTAINS_PART, '')
            kernel_gencore_contains.append(parent)

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

        # variable A
        #import pdb; pdb.set_trace()
        attrspec = ['INTENT(INOUT)']
        if var.is_pointer(): attrspec.append('POINTER')
        if var.is_allocatable() or allocate: 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_genknode(subrobj, DECL_PART, stmt.__class__, attrs=attrs)

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

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

        # kgen_istrue
        attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
        part_append_genknode(subrobj, DECL_PART, typedecl_statements.Logical, 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)

        # 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_genknode(subrobj, DECL_PART, typedecl_statements.Integer, attrs=attrs)

            attrs = {'type_spec': 'INTEGER', 'attrspec': ['DIMENSION(2,%d)'%var.rank], 'entity_decls': [ 'kgen_bound' ]}
            part_append_genknode(subrobj, DECL_PART, typedecl_statements.Integer, attrs=attrs)

        part_append_comment(subrobj, DECL_PART, '')

        pobj = gen_read_istrue(subrobj, var, 'var', allocate=allocate)

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

                attrs = {'items': ['kgen_bound(2, %d)'%(dim+1)], 'specs': ['UNIT = kgen_unit']}
                part_append_genknode(pobj, EXEC_PART, statements.Read, attrs=attrs)

                #bound_args.append('kgen_bound(2,%d)-kgen_bound(1,%d)+1'%(dim+1, dim+1))
                bound_args.append('kgen_bound(1,%d):kgen_bound(2,%d)'%(dim+1, dim+1))

            if var.is_allocatable() or var.is_pointer() or allocate:
                attrs = {'items': ['var(%s)'%', '.join(bound_args)]}
                part_append_genknode(pobj, EXEC_PART, statements.Allocate, 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=kgen_bound(1,%(d)d), kgen_bound(2,%(d)d)'%{'d':d+1}}
                    doobj = part_append_genknode(prevobj, EXEC_PART, block_statements.Do, attrs=attrs)
                    doobjs.append(doobj)
                    prevobj = doobj

                attrs = {'expr': 'PRESENT( printvar )'}
                ifpvarobj = part_append_genknode(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 len(req.res_stmts)>0:
                            res = req.res_stmts[0]
                            callname = get_dtype_readname(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 read "%s" is not created here.'%\
                        (stmt.name, stmt.name))
                else:
                    attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', 'printvar // "(%s)"'%str_indexes]}
                    part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

                    part_append_genknode(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(", %s, ")"'%(ename_prefix, entity_name, tostr_indexes)]}
                        attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit', '"%s%s(%s)"'%(ename_prefix, entity_name, str_indexes)]}
                        part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)
                    else:
                        attrs = {'designator': callname, 'items': ['var(%s)'%str_indexes, 'kgen_unit']}
                        part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

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

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

                if stmt.is_numeric():
                    if isinstance(stmt, typedecl_statements.Real):
                        attrs = {'designator': 'kgen_array_sumcheck', 'items': ['printvar', \
                            'kgen_array_sum', 'REAL(SUM(var, mask=(var .eq. var)), 8)', '.TRUE.']}
                    else:
                        attrs = {'designator': 'kgen_array_sumcheck', 'items': ['printvar', \
                            'kgen_array_sum', 'REAL(SUM(var), 8)', '.TRUE.']}
                    part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

                attrs = {'items': ['"** KGEN DEBUG: " // printvar // "**" // NEW_LINE("A")', 'var']}
                part_append_genknode(ifpvarobj, EXEC_PART, statements.Write, attrs=attrs)

                part_append_genknode(ifpvarobj, EXEC_PART, statements.Else)

                if stmt.is_numeric():
                    if isinstance(stmt, typedecl_statements.Real):
                        attrs = {'designator': 'kgen_array_sumcheck', 'items': ['"UNKNOWN"', \
                            'kgen_array_sum', 'REAL(SUM(var, mask=(var .eq. var)), 8)', '.TRUE.']}
                    else:
                        attrs = {'designator': 'kgen_array_sumcheck', 'items': ['"UNKNOWN"', \
                            'kgen_array_sum', 'REAL(SUM(var), 8)', '.TRUE.']}
                    part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

#
#                part_append_genknode(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: " // printvar "%s%s **"'%(ename_prefix, entity_name), 'SUM(var)']}
#                    else:
#                        attrs = {'items': ['"** KGEN DEBUG: " // "%s%s **" // NEW_LINE("A")'%(ename_prefix, entity_name), 'var']}
#                    part_append_genknode(pobj, EXEC_PART, statements.Write, attrs=attrs)
#                else:
#                    attrs = {'expr': 'PRESENT( printvar )'}
#                    ifpvarobj = part_append_genknode(pobj, EXEC_PART, block_statements.IfThen, attrs=attrs)
#
#                    if stmt.is_numeric():
#                        attrs = {'items': ['"** KGEN DEBUG: " // printvar // " %s%s **"'%(ename_prefix, entity_name), 'SUM(var)']}
#                    else:
#                        attrs = {'items': ['"** KGEN DEBUG: " // printvar // " %s%s **" // NEW_LINE("A")'%(ename_prefix, entity_name), 'var']}
#                    part_append_genknode(ifpvarobj, EXEC_PART, statements.Write, attrs=attrs)


        else: # scalar

            if var.is_allocatable() or var.is_pointer() or allocate:
                attrs = {'items': ['var']}
                part_append_genknode(pobj, EXEC_PART, statements.Allocate, attrs=attrs)

            if stmt.is_derived() or is_class_derived:
                attrs = {'expr': 'PRESENT( printvar )'}
                ifpvarobj = part_append_genknode(pobj, 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 len(req.res_stmts)>0:
                            res = req.res_stmts[0]
                            callname = get_dtype_readname(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 read "%s" is not created here.'%\
                        (stmt.name, stmt.name))
                else:
                    attrs = {'designator': callname, 'items': ['var', 'kgen_unit', 'printvar // " %s%s "'%(ename_prefix, entity_name)]}
                    part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)

                    part_append_genknode(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%s"'%(ename_prefix, entity_name)]}
                        part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)
                    else:
                        attrs = {'designator': callname, 'items': ['var', 'kgen_unit']}
                        part_append_genknode(ifpvarobj, EXEC_PART, statements.Call, attrs=attrs)
            else: # intrinsic type
                attrs = {'items': ['var'], 'specs': ['UNIT = kgen_unit']}
                part_append_genknode(pobj, EXEC_PART, statements.Read, attrs=attrs)

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

                attrs = {'items': ['"** KGEN DEBUG: " // printvar // "**" // NEW_LINE("A")', 'var']}
                part_append_genknode(ifpvarobj, EXEC_PART, statements.Write, attrs=attrs)
Пример #7
0
    def create_kernel_module_parts(self, node):
        in_subrobj = None
        in_subrname = get_module_in_readname(node.kgen_stmt)
        checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine) and n.name==in_subrname
        if not part_has_node(node, SUBP_PART, checks):

            checks = lambda n: n.kgen_isvalid and n.kgen_match_class==statements.Contains
            if not node in kernel_gencore_contains and not part_has_node(node, CONTAINS_PART, checks):
                part_append_comment(node, CONTAINS_PART, '')
                part_append_genknode(node, CONTAINS_PART, statements.Contains)
                part_append_comment(node, CONTAINS_PART, '')
                kernel_gencore_contains.append(node)

            attrs = {'name': in_subrname, 'args': ['kgen_unit']}
            part_append_comment(node, SUBP_PART, 'read state subroutine for %s'%in_subrname)
            in_subrobj = part_append_genknode(node, SUBP_PART, block_statements.Subroutine, attrs=attrs)
            part_append_comment(node, SUBP_PART, '')

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

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

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

            part_append_comment(in_subrobj, DECL_PART, '')

            # add public stmt
            attrs = {'items':[in_subrname]}
            part_append_genknode(node, DECL_PART, statements.Public, attrs=attrs)

        out_subrobj = None
        if hasattr(node.kgen_stmt, 'geninfo') and KGGenType.has_state_out(node.kgen_stmt.geninfo):
            out_subrname = get_module_out_readname(node.kgen_stmt)
            checks = lambda n: isinstance(n.kgen_stmt, block_statements.Subroutine) and n.name==out_subrname
            if not part_has_node(node, SUBP_PART, checks):

                checks = lambda n: n.kgen_isvalid and n.kgen_match_class==statements.Contains
                if not node in kernel_gencore_contains and not part_has_node(node, CONTAINS_PART, checks):
                    part_append_comment(node, CONTAINS_PART, '')
                    part_append_genknode(node, CONTAINS_PART, statements.Contains)
                    part_append_comment(node, CONTAINS_PART, '')
                    kernel_gencore_contains.append(node)

                attrs = {'name': out_subrname, 'args': ['kgen_unit']}
                part_append_comment(node, SUBP_PART, 'read state subroutine for %s'%out_subrname)
                out_subrobj = part_append_genknode(node, SUBP_PART, block_statements.Subroutine, attrs=attrs)
                part_append_comment(node, SUBP_PART, '')

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

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

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

                # add public stmt
                attrs = {'items':[out_subrname]}
                part_append_genknode(node, DECL_PART, statements.Public, attrs=attrs)


        if in_subrobj or out_subrobj:
            self.kernel_externs_subrs[node] = ( in_subrobj, out_subrobj )

            # register event per typedecl 
            self.frame_msg.add_event(KERNEL_SELECTION.ALL, FILE_TYPE.KERNEL, GENERATION_STAGE.BEGIN_PROCESS, \
                typedecl_statements.TypeDeclarationStatement, self.is_extern_in_kernel_module, self.create_subr_read_typedecl_in_module) 

            # register event per module
            self.frame_msg.add_event(KERNEL_SELECTION.ALL, FILE_TYPE.KERNEL, GENERATION_STAGE.BEGIN_PROCESS, \
                block_statements.Module, self.has_externs_in_module, self.create_kernel_stmts_in_callsite) 
Пример #8
0
    def create_dtype_read_subr(self, node):
        assert node.kgen_stmt, 'None kgen statement'

        subrname = get_dtype_readname(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_isvalid and n.kgen_match_class==statements.Contains
            if not parent in kernel_gencore_contains and not part_has_node(parent, CONTAINS_PART, checks):
                part_append_comment(parent, CONTAINS_PART, '')
                part_append_genknode(parent, CONTAINS_PART, statements.Contains)
                part_append_comment(parent, CONTAINS_PART, '')
                kernel_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_genknode(parent, SUBP_PART, block_statements.Subroutine, attrs=attrs)
            part_append_comment(parent, SUBP_PART, '')

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

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

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

            # kgen_istrue
            attrs = {'type_spec': 'LOGICAL', 'entity_decls': ['kgen_istrue']}
            part_append_genknode(subrobj, DECL_PART, typedecl_statements.Logical, 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, '')

            #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

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

                    if var.is_array():
                        if stmt.is_derived():
                            self.create_read_call(subrobj, callname, entity_name, stmt, var)
                        else: # intrinsic type
                            if var.is_explicit_shape_array():
                                self.create_read_intrinsic(subrobj, entity_name, stmt, var)
                            else: # implicit array
                                self.create_read_call(subrobj, callname, entity_name, stmt, var)
                    else: # scalar
                        if stmt.is_derived():
                            if var.is_allocatable() or var.is_pointer():
                                self.create_read_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_readname(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 read "%s" is not created here.'%\
                                        (stmt.name, stmt.name))
                                else:
                                    self.create_read_call(subrobj, callname, entity_name, stmt, var)
                        else: # intrinsic type
                            self.create_read_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_genknode(parent, DECL_PART, statements.Public, attrs=attrs)