"{ 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

    self codeNl.

    self codeOnLine: aStringOrBlockOrRBParseNode

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

    "Modified: / 03-06-2015 / 05:52:39 / 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>"

!

codeNl

    self add: ''.

!

codeOnLine: aStringOrBlockOrRBParseNode

    aStringOrBlockOrRBParseNode isString ifTrue:[ 

        self emitCodeAsString: aStringOrBlockOrRBParseNode

    ] ifFalse:[

        (aStringOrBlockOrRBParseNode isKindOf: RBProgramNode) ifTrue:[ 

            self emitCodeAsRBNode: aStringOrBlockOrRBParseNode.

        ] ifFalse:[  

            self emitCodeAsBlock: aStringOrBlockOrRBParseNode

        ].

    ].

! !

!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."

    buffer := String new writeStream.

    indentation := 1.

    temporaries := OrderedCollection new.

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

    "Modified (comment): / 18-06-2015 / 06:04:21 / Jan Vrany <jan.vrany@fit.cvut.cz>"

! !

!PPCCodeBlock methodsFor:'printing and storing'!

sourceOn: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:[

                aStream 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>"

! !

!PPCCodeBlock methodsFor:'private'!

emitCodeAsBlock: aBlock

    aBlock value

!

emitCodeAsRBNode: anRBNode

    anRBNode isSequence ifTrue:[

        anRBNode temporaries do:[:e |  

            (temporaries includes: e name) ifFalse:[ 

                temporaries add: e name

            ].

        ].

        anRBNode statements do:[:e|

            self 	add: (self formatRBNode: e); 

                    addOnLine: '.'.

        ].

    ] ifFalse:[  

        buffer nextPutAll: anRBNode formattedCode.  

    ].

!

emitCodeAsString: aString

    self addOnLine: aString

!

formatRBNode: anRBNode

    | formatter |

    formatter := anRBNode formatterClass new.

    formatter indent: indentation.

    ^ formatter format: anRBNode 

! !