def create_subr_write_typedecl_in_type(self, node):
stmt = node.kgen_stmt
parent = node.kgen_parent.kgen_parent
entity_names = [get_entity_name(decl) for decl in stmt.entity_decls]
for entity_name, entity_decl in zip(entity_names, stmt.entity_decls):
var = stmt.get_variable(entity_name)
subrname = get_typedecl_writename(stmt, entity_name)
if subrname is None: raise Exception('Can not get subroutinename')
if var.is_array():
if is_zero_array(var, stmt): continue
if stmt.is_derived():
self.create_write_subr(subrname, entity_name, parent, var,
stmt)
else: # intrinsic type
if var.is_explicit_shape_array():
pass
else: # implicit array
self.create_write_subr(subrname, entity_name, parent,
var, stmt)
else:
if stmt.is_derived():
if var.is_allocatable() or var.is_pointer():
self.create_write_subr(subrname, entity_name, parent,
var, stmt)
def create_subr_write_typedecl_in_type(self, node):
stmt = node.kgen_stmt
parent = node.kgen_parent.kgen_parent
entity_names = [ get_entity_name(decl) for decl in stmt.entity_decls ]
for entity_name, entity_decl in zip(entity_names, stmt.entity_decls):
var = stmt.get_variable(entity_name)
subrname = get_typedecl_writename(stmt, entity_name)
if subrname is None: raise Exception('Can not get subroutinename')
if var.is_array():
if is_zero_array(var, stmt): continue
if stmt.is_derived():
self.create_write_subr(subrname, entity_name, parent, var, stmt)
else: # intrinsic type
if var.is_explicit_shape_array():
pass
else: # implicit array
self.create_write_subr(subrname, entity_name, parent, var, stmt)
else:
if stmt.is_derived():
if var.is_allocatable() or var.is_pointer():
self.create_write_subr(subrname, entity_name, parent, var, stmt)
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_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)
