from block import *
from stmt import *

basic_blocks = [ProgramBlock,PythonModuleBlock,ModuleBlock,BlockDataBlock,
                SubroutineBlock,FunctionBlock,InterfaceBlock,TypeBlock]
stmt_blocks = [DoBlock,IfThenBlock,SelectBlock]

Block.classes['free90'] = [ProgramBlock,ModuleBlock,BlockDataBlock,
                           SubroutineBlock,FunctionBlock,InterfaceBlock,TypeBlock] + stmt_blocks
Block.classes['fix90'] = Block.classes['free90']
Block.classes['fix77'] = [ProgramBlock,BlockDataBlock,SubroutineBlock,FunctionBlock] + stmt_blocks
Block.classes['pyf'] = [PythonModuleBlock] + Block.classes['free90']

ProgramBlock.classes['free90'] = [ModuleBlock,SubroutineBlock,FunctionBlock,InterfaceBlock,TypeBlock] + stmt_blocks
ProgramBlock.classes['fix90'] = ProgramBlock.classes['free90']
ProgramBlock.classes['fix77'] = [SubroutineBlock,FunctionBlock] + stmt_blocks
ProgramBlock.classes['pyf'] = ProgramBlock.classes['free90']

ModuleBlock.classes['free90'] = [ModuleBlock,SubroutineBlock,FunctionBlock,InterfaceBlock,TypeBlock]
ModuleBlock.classes['fix90'] = ModuleBlock.classes['free90']
ModuleBlock.classes['fix77'] = []
ModuleBlock.classes['pyf'] = ModuleBlock.classes['free90']

BlockDataBlock.classes['free90'] = [TypeBlock]
BlockDataBlock.classes['fix90'] = BlockDataBlock.classes['free90']
BlockDataBlock.classes['fix77'] = []
BlockDataBlock.classes['pyf'] = BlockDataBlock.classes['free90']


PythonModuleBlock.classes['free90'] = [ModuleBlock,SubroutineBlock,FunctionBlock,InterfaceBlock,TypeBlock]
PythonModuleBlock.classes['fix90'] = PythonModuleBlock.classes['free90']
PythonModuleBlock.classes['fix77'] = []
PythonModuleBlock.classes['pyf'] = PythonModuleBlock.classes['free90']

InterfaceBlock.classes['free90'] = [ModuleBlock,SubroutineBlock,FunctionBlock,InterfaceBlock,TypeBlock] + stmt_blocks
InterfaceBlock.classes['fix90'] = InterfaceBlock.classes['free90']
InterfaceBlock.classes['fix77'] = []
InterfaceBlock.classes['pyf'] = InterfaceBlock.classes['free90']

SubroutineBlock.classes['free90'] = [InterfaceBlock,TypeBlock] + stmt_blocks
SubroutineBlock.classes['fix90'] = SubroutineBlock.classes['free90']
SubroutineBlock.classes['fix77'] = stmt_blocks
SubroutineBlock.classes['pyf'] = SubroutineBlock.classes['free90']

FunctionBlock.classes = SubroutineBlock.classes

TypeBlock.classes['free90'] = [ModuleBlock, SubroutineBlock, FunctionBlock, InterfaceBlock, TypeBlock] + stmt_blocks
TypeBlock.classes['fix90'] = TypeBlock.classes['free90']
TypeBlock.classes['fix77'] = []
TypeBlock.classes['pyf'] = TypeBlock.classes['free90']

StatementBlock.classes['free90'] = stmt_blocks
StatementBlock.classes['fix90'] = StatementBlock.classes['free90']
StatementBlock.classes['fix77'] = stmt_blocks
StatementBlock.classes['pyf'] = StatementBlock.classes['free90']


# Initialize stmt_cls attributes

ProgramBlock.stmt_cls = Program
ModuleBlock.stmt_cls = Module
PythonModuleBlock.stmt_cls = PythonModule
BlockDataBlock.stmt_cls = BlockData
InterfaceBlock.stmt_cls = Interface
SubroutineBlock.stmt_cls = Subroutine
FunctionBlock.stmt_cls = Function
TypeBlock.stmt_cls = Type

IfThenBlock.stmt_cls = IfThen
DoBlock.stmt_cls = Do
SelectBlock.stmt_cls = Select

ProgramBlock.end_stmt_cls = EndProgram
ModuleBlock.end_stmt_cls = EndModule
PythonModuleBlock.end_stmt_cls = EndPythonModule
BlockDataBlock.end_stmt_cls = EndBlockData
InterfaceBlock.end_stmt_cls = EndInterface
SubroutineBlock.end_stmt_cls = EndSubroutine
FunctionBlock.end_stmt_cls = EndFunction
TypeBlock.end_stmt_cls = EndType

IfThenBlock.end_stmt_cls = EndIfThen
DoBlock.end_stmt_cls = EndDo
SelectBlock.end_stmt_cls = EndSelect