"{ Package: 'jv:tea/compiler' }"

"{ NameSpace: Smalltalk }"

RGAbstractContainer subclass:#TCompilationUnitDefinition
	instanceVariableNames:''
	classVariableNames:''
	poolDictionaries:''
	category:'Languages-Tea-Compiler-Model'
!

!TCompilationUnitDefinition methodsFor:'accessing'!

classes
    "Return a list of classes defined in this compilation unit"

    ^ elements at: #classes ifAbsentPut:[ OrderedCollection new ]

    "Created: / 13-09-2015 / 08:01:38 / Jan Vrany <jan.vrany@fit.cvut.cz>"
!

name
    ^ name ? '<in memory>'

    "Created: / 14-09-2015 / 10:56:12 / Jan Vrany <jan.vrany@fit.cvut.cz>"
! !

!TCompilationUnitDefinition methodsFor:'adding & removing'!

addClass: class 
    self addElement: class in: self classes

    "Created: / 13-09-2015 / 08:00:00 / Jan Vrany <jan.vrany@fit.cvut.cz>"
!

addMethod: method
    | className classIsMeta class |

    classIsMeta := false.
    className := method className.
    (className endsWith: ' class') ifTrue:[
        className := className copyTo: (className size - "' class' size"6).
        classIsMeta := true.
    ].
    class := self classes detect:[:class | class name = className ] ifNone:[ nil ].
    class isNil ifTrue:[ 
        self error: 'No class %1 found in compilation unit'
    ].
    classIsMeta ifTrue:[ 
        class := class theMetaclass.
    ].
    class addMethod: method.

    "Created: / 13-09-2015 / 08:02:33 / Jan Vrany <jan.vrany@fit.cvut.cz>"
! !

!TCompilationUnitDefinition methodsFor:'testing types'!

isCompilationUnit

    ^true

    "Created: / 14-09-2015 / 10:29:51 / Jan Vrany <jan.vrany@fit.cvut.cz>"
! !

!TCompilationUnitDefinition methodsFor:'visiting'!

acceptVisitor: visitor
    ^ visitor acceptCompilationUnitDefinition: self

    "Created: / 14-09-2015 / 10:32:34 / Jan Vrany <jan.vrany@fit.cvut.cz>"
! !