Skip to content

Commit

Permalink
Simplify testing of attributes for python nodes (#15)
Browse files Browse the repository at this point in the history
  • Loading branch information
@mgeplf @pramodk
mgeplf authored and pramodk committed Feb 25, 2019
1 parent 0d28f30 commit a0589ae
Showing 1 changed file with 50 additions and 73 deletions.
123 changes: 50 additions & 73 deletions src/language/nodes.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@
import textwrap
from utils import *


class BaseNode:
"""base class for all node types (parent + child) """

Expand All @@ -39,27 +40,27 @@ def get_data_type_name(self):

@property
def is_statement_block_node(self):
return True if self.class_name == STATEMENT_BLOCK_NODE else False
return self.class_name == STATEMENT_BLOCK_NODE

@property
def is_statement_node(self):
return True if self.class_name in STATEMENT_TYPES else False
return self.class_name in STATEMENT_TYPES

@property
def is_global_block_node(self):
return True if self.class_name in GLOBAL_BLOCKS else False
return self.class_name in GLOBAL_BLOCKS

@property
def is_prime_node(self):
return True if self.class_name == PRIME_NAME_NODE else False
return self.class_name == PRIME_NAME_NODE

@property
def is_program_node(self):
"""
check if current node is main program container node
:return: True if main program block otherwise False
"""
return True if self.class_name == PROGRAM_BLOCK else False
return self.class_name == PROGRAM_BLOCK

@property
def is_block_node(self):
Expand All @@ -70,81 +71,76 @@ def is_block_node(self):
:return: True or False
"""
return True if self.class_name in BLOCK_TYPES else False
return self.class_name in BLOCK_TYPES

@property
def is_unit_block(self):
return True if self.class_name == UNIT_BLOCK else False
return self.class_name == UNIT_BLOCK

@property
def is_data_type_node(self):
return True if self.class_name in DATA_TYPES.keys() else False
return self.class_name in DATA_TYPES

@property
def is_symbol_var_node(self):
return True if self.class_name in SYMBOL_VAR_TYPES else False
return self.class_name in SYMBOL_VAR_TYPES

@property
def is_symbol_helper_node(self):
return True if self.class_name in SYMBOL_TABLE_HELPER_NODES else False
return self.class_name in SYMBOL_TABLE_HELPER_NODES

@property
def is_symbol_block_node(self):
return True if self.class_name in SYMBOL_BLOCK_TYPES else False
return self.class_name in SYMBOL_BLOCK_TYPES

@property
def is_base_type_node(self):
return True if self.class_name in BASE_TYPES else False
return self.class_name in BASE_TYPES

@property
def is_string_node(self):
return True if self.class_name == STRING_NODE else False
return self.class_name == STRING_NODE

@property
def is_integer_node(self):
return True if self.class_name == INTEGER_NODE else False
return self.class_name == INTEGER_NODE

@property
def is_number_node(self):
return True if self.class_name == NUMBER_NODE else False
return self.class_name == NUMBER_NODE

@property
def is_boolean_node(self):
return True if self.class_name == BOOLEAN_NODE else False
return self.class_name == BOOLEAN_NODE

@property
def is_identifier_node(self):
return True if self.class_name == IDENTIFIER_NODE else False
return self.class_name == IDENTIFIER_NODE

@property
def is_name_node(self):
return True if self.class_name == NAME_NODE else False
return self.class_name == NAME_NODE

@property
def is_enum_node(self):
data_type = DATA_TYPES[self.class_name]
return True if data_type in ENUM_BASE_TYPES else False
return data_type in ENUM_BASE_TYPES

@property
def is_pointer_node(self):
if self.class_name in PTR_EXCLUDE_TYPES:
return False
if self.is_base_type_node:
return False
return True
return not (self.class_name in PTR_EXCLUDE_TYPES or
self.is_base_type_node)

@property
def is_ptr_excluded_node(self):
if self.class_name in PTR_EXCLUDE_TYPES:
return True
return False
return self.class_name in PTR_EXCLUDE_TYPES

@property
def requires_default_constructor(self):
if self.class_name in LEXER_DATA_TYPES or self.is_program_node or self.is_ptr_excluded_node:
return True
else:
return False
return (self.class_name in LEXER_DATA_TYPES or
self.is_program_node or
self.is_ptr_excluded_node)


class ChildNode(BaseNode):
"""represent member variable for a Node"""
Expand Down Expand Up @@ -228,6 +224,7 @@ def get_setter_method(self):
reference = "" if self.is_base_type_node else "&&"
return f"void {setter_method}({setter_type}{reference} {self.varname}) {{ this->{self.varname} = {self.varname}; }}"


class Node(BaseNode):
"""represent a class for every rule in language specification"""

Expand All @@ -252,30 +249,30 @@ def add_child(self, args):
self.children.append(node)

def has_children(self):
return True if self.children else False
return bool(self.children)

@property
def is_block_scoped_node(self):
"""
Check if node is derived from BASE_BLOCK
:return: True / False
"""
return True if self.base_class == BASE_BLOCK else False
return self.base_class == BASE_BLOCK

def has_parent_block_node(self):
"""
check is base or parent is structured base block
:return: True if parent is BASE_BLOCK otherwise False
"""
return True if self.base_class == BASE_BLOCK else False
return self.base_class == BASE_BLOCK

@property
def is_base_block_node(self):
"""
check if node is Block
:return: True if node type/name is BASE_BLOCK
"""
return True if self.class_name == BASE_BLOCK else False
return self.class_name == BASE_BLOCK

@property
def is_symtab_needed(self):
Expand All @@ -286,9 +283,7 @@ def is_symtab_needed(self):
:return: True or False
"""
# block scope nodes have symtab pointer
if self.is_program_node or self.is_block_scoped_node:
return True
return False
return self.is_program_node or self.is_block_scoped_node

@property
def is_symtab_method_required(self):
Expand Down Expand Up @@ -322,72 +317,55 @@ def is_base_class_number_node(self):
"""
Check if node is of type Number
"""
return True if self.base_class == NUMBER_NODE else False
return self.base_class == NUMBER_NODE

def ctor_declaration(self):
args = []
for c in self.children:
args.append(f'{c.get_typename()} {c.varname}')
args = [f'{c.get_typename()} {c.varname}' for c in self.children]
return f"{self.class_name}({', '.join(args)});"

def ctor_definition(self):
args = []
for c in self.children:
args.append(f'{c.get_typename()} {c.varname}')
initlist = []
for c in self.children:
initlist.append(f'{c.varname}({c.varname})')
args = [f'{c.get_typename()} {c.varname}' for c in self.children]
initlist = [f'{c.varname}({c.varname})' for c in self.children]

s = f"""{self.class_name}::{self.class_name}({', '.join(args)})
: {', '.join(initlist)} {{}}
"""
return textwrap.dedent(s)

def ctor_shrptr_declaration(self):
args = []
for c in self.children:
args.append(f'{c.member_typename} {c.varname}')
args = [f'{c.member_typename} {c.varname}' for c in self.children]
return f"{self.class_name}({', '.join(args)});"

def ctor_shrptr_definition(self):
args = []
for c in self.children:
args.append(f'{c.member_typename} {c.varname}')
initlist = []
for c in self.children:
initlist.append(f'{c.varname}({c.varname})')
args = [f'{c.member_typename} {c.varname}' for c in self.children]
initlist = [f'{c.varname}({c.varname})' for c in self.children]

s = f"""{self.class_name}::{self.class_name}({', '.join(args)})
: {', '.join(initlist)} {{}}
"""
return textwrap.dedent(s)

def has_ptr_children(self):
for c in self.children:
if not (c.is_vector or c.is_base_type_node or c.is_ptr_excluded_node):
return True
return False

return any(not (c.is_vector or c.is_base_type_node or c.is_ptr_excluded_node)
for c in self.children)

def public_members(self):
"""
Return public members of the node
"""
members = []

for child in self.children:
if child.is_public:
members.append([child.member_typename, child.varname, child.description])
members = [[child.member_typename, child.varname, child.description]
for child in self.children
if child.is_public]

return members

def private_members(self):
"""
Return private members of the node
"""
members = []

for child in self.children:
if not child.is_public:
members.append([child.member_typename, child.varname, child.description])
members = [[child.member_typename, child.varname, child.description]
for child in self.children
if not child.is_public]

if self.has_token:
members.append(["std::shared_ptr<ModToken>", "token", "token with location information"])
Expand All @@ -400,7 +378,6 @@ def private_members(self):

return members


@property
def non_base_members(self):
return [child for child in self.children if not child.is_base_type_node]

0 comments on commit a0589ae

Please sign in to comment.