"{ Package: 'stx:goodies/petitparser/compiler' }"

"{ NameSpace: Smalltalk }"

Object subclass:#PPCCodeBlock

	instanceVariableNames:'buffer indentation temporaries'

	classVariableNames:''

	poolDictionaries:''

	category:'PetitCompiler-Compiler-Codegen'

!

!PPCCodeBlock class methodsFor:'instance creation'!

new

    "return an initialized instance"

    ^ self basicNew initialize.

! !

!PPCCodeBlock methodsFor:'as yet unclassified'!

add: string

    self nl.

    self codeIndent.

    self addOnLine: string.

    "Modified: / 01-06-2015 / 22:58:42 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

addOnLine: string

    buffer nextPutAll: string.

!

nl

    ^ buffer nextPut: Character cr

! !

!PPCCodeBlock methodsFor:'code generation'!

code: aStringOrBlockOrRBParseNode

    aStringOrBlockOrRBParseNode isString ifTrue:[ 

        buffer nextPutAll: aStringOrBlockOrRBParseNode

    ] ifFalse:[

        (aStringOrBlockOrRBParseNode isKindOf: RBProgramNode) ifTrue:[ 

            aStringOrBlockOrRBParseNode isSequence ifTrue:[

                aStringOrBlockOrRBParseNode temporaries do:[:e |  

                ].

                aStringOrBlockOrRBParseNode statements do:[:e|

                    buffer nextPutAll: e formattedCode; nextPut: $..  

                    self nl; codeIndent.

                ].

            ] ifFalse:[  

                buffer nextPutAll: aStringOrBlockOrRBParseNode formattedCode.  

            ].

        ] ifFalse:[  

            aStringOrBlockOrRBParseNode value

        ].

    ].

    "Created: / 01-06-2015 / 21:07:10 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

codeIndent

    self codeIndent:indentation

    "Created: / 01-06-2015 / 22:58:00 / Jan Vrany <jan.vrany@fit.cvut.cz>"

!

codeIndent: level

    ((Smalltalk respondsTo:#isSmalltalkX) and:[ Smalltalk isSmalltalkX ]) ifTrue:[ 

        level * 4 timesRepeat: [ buffer nextPut: Character space  ].

    ] ifFalse:[ 

        level timesRepeat: [ buffer nextPut: Character tab  ].

    ].

    "Created: / 01-06-2015 / 22:58:07 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!PPCCodeBlock methodsFor:'code generation - variables'!

allocateTemporaryVariableNamed:preferredName 

    "Allocate a new variable with (preferably) given name.

     Returns a real variable name that should be used."

    (temporaries includes:preferredName) ifFalse:[

        temporaries add:preferredName.

        ^ preferredName

    ] ifTrue:[

        | name |

        name := preferredName , '_' , (temporaries size + 1) printString.

        temporaries add:name.

        ^ name

    ].

    "Created: / 23-04-2015 / 17:37:55 / Jan Vrany <jan.vrany@fit.cvut.cz>"

    "Modified: / 01-06-2015 / 21:03:39 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!PPCCodeBlock methodsFor:'indentation'!

dedent

    indentation := indentation - 1

!

indent 

    indentation := indentation + 1

!

indentationLevel

    ^ indentation

!

indentationLevel: value

    indentation := value

! !

!PPCCodeBlock methodsFor:'initialization'!

initialize

    "Invoked when a new instance is created."

    "/ please change as required (and remove this comment)

    buffer := String new writeStream.

    indentation := 1.

    temporaries := OrderedCollection new.          

    "/ super initialize.   -- commented since inherited method does nothing

    "Modified: / 01-06-2015 / 20:57:08 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!PPCCodeBlock methodsFor:'printing and storing'!

codeOn: aStream

    "Dumps generated code on given stream"

    temporaries notEmpty ifTrue:[

        ((Smalltalk respondsTo:#isSmalltalkX) and:[ Smalltalk isSmalltalkX ]) ifTrue:[ 

            indentation * 4 timesRepeat: [ aStream nextPut: Character space  ].

        ] ifFalse:[ 

            indentation timesRepeat: [ buffer nextPut: Character tab  ].

        ].

        aStream nextPut: $|.

        temporaries do:[:e | aStream space; nextPutAll: e  ].

        aStream space.

        aStream nextPut: $|. 

        self nl.

        "In Smalltalk/X, there should be a blank line after temporaries"

        ((Smalltalk respondsTo:#isSmalltalkX) and:[ Smalltalk isSmalltalkX ]) ifTrue:[ 

            self nl.

        ].

    ].

    aStream nextPutAll: buffer contents

    "Created: / 01-06-2015 / 21:26:03 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !