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
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
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)
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)
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)
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)
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)
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)