12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247 |
- Smalltalk current createPackage: 'Compiler-IR'!
- NodeVisitor subclass: #IRASTTranslator
- instanceVariableNames: 'source theClass method sequence nextAlias'
- package: 'Compiler-IR'!
- !IRASTTranslator commentStamp!
- I am the AST (abstract syntax tree) visitor responsible for building the intermediate representation graph.!
- !IRASTTranslator methodsFor: 'accessing'!
- method
- ^ method
- !
- method: anIRMethod
- method := anIRMethod
- !
- nextAlias
- nextAlias ifNil: [ nextAlias := 0 ].
- nextAlias := nextAlias + 1.
- ^ nextAlias asString
- !
- sequence
- ^ sequence
- !
- sequence: anIRSequence
- sequence := anIRSequence
- !
- source
- ^ source
- !
- source: aString
- source := aString
- !
- theClass
- ^ theClass
- !
- theClass: aClass
- theClass := aClass
- !
- withSequence: aSequence do: aBlock
- | outerSequence |
- outerSequence := self sequence.
- self sequence: aSequence.
- aBlock value.
- self sequence: outerSequence.
- ^ aSequence
- ! !
- !IRASTTranslator methodsFor: 'visiting'!
- alias: aNode
- | variable |
- aNode isImmutable ifTrue: [ ^ self visit: aNode ].
- variable := IRVariable new
- variable: (AliasVar new name: '$', self nextAlias);
- yourself.
- self sequence add: (IRAssignment new
- add: variable;
- add: (self visit: aNode);
- yourself).
- self method internalVariables add: variable.
- ^ variable
- !
- aliasTemporally: aCollection
- "https://github.com/NicolasPetton/amber/issues/296
-
- If a node is aliased, all preceding ones are aliased as well.
- The tree is iterated twice. First we get the aliasing dependency,
- then the aliasing itself is done"
- | threshold result |
- threshold := 0.
-
- aCollection withIndexDo: [ :each :i |
- each subtreeNeedsAliasing
- ifTrue: [ threshold := i ]].
- result := OrderedCollection new.
- aCollection withIndexDo: [ :each :i |
- result add: (i <= threshold
- ifTrue: [ self alias: each ]
- ifFalse: [ self visit: each ])].
- ^result
- !
- visitAssignmentNode: aNode
- | left right assignment |
- right := self visit: aNode right.
- left := self visit: aNode left.
- self sequence add: (IRAssignment new
- add: left;
- add: right;
- yourself).
- ^ left
- !
- visitBlockNode: aNode
- | closure |
- closure := IRClosure new
- arguments: aNode parameters;
- scope: aNode scope;
- yourself.
- aNode scope temps do: [ :each |
- closure add: (IRTempDeclaration new
- name: each name;
- scope: aNode scope;
- yourself) ].
- aNode nodes do: [ :each | closure add: (self visit: each) ].
- ^ closure
- !
- visitBlockSequenceNode: aNode
- ^ self
- withSequence: IRBlockSequence new
- do: [
- aNode nodes ifNotEmpty: [
- aNode nodes allButLast do: [ :each |
- self sequence add: (self visit: each) ].
- aNode nodes last isReturnNode
- ifFalse: [ self sequence add: (IRBlockReturn new add: (self visit: aNode nodes last); yourself) ]
- ifTrue: [ self sequence add: (self visit: aNode nodes last) ]]]
- !
- visitCascadeNode: aNode
- | alias |
- aNode receiver isImmutable ifFalse: [
- alias := self alias: aNode receiver.
- aNode nodes do: [ :each |
- each receiver: (VariableNode new binding: alias variable) ]].
- aNode nodes allButLast do: [ :each |
- self sequence add: (self visit: each) ].
- ^ self alias: aNode nodes last
- !
- visitDynamicArrayNode: aNode
- | array |
- array := IRDynamicArray new.
- (self aliasTemporally: aNode nodes) do: [:each | array add: each].
- ^ array
- !
- visitDynamicDictionaryNode: aNode
- | dictionary |
- dictionary := IRDynamicDictionary new.
- (self aliasTemporally: aNode nodes) do: [:each | dictionary add: each].
- ^ dictionary
- !
- visitJSStatementNode: aNode
- ^ IRVerbatim new
- source: aNode source crlfSanitized;
- yourself
- !
- visitMethodNode: aNode
- self method: (IRMethod new
- source: self source crlfSanitized;
- theClass: self theClass;
- arguments: aNode arguments;
- selector: aNode selector;
- messageSends: aNode messageSends;
- superSends: aNode superSends;
- classReferences: aNode classReferences;
- scope: aNode scope;
- yourself).
- aNode scope temps do: [ :each |
- self method add: (IRTempDeclaration new
- name: each name;
- scope: aNode scope;
- yourself) ].
- aNode nodes do: [ :each | self method add: (self visit: each) ].
- aNode scope hasLocalReturn ifFalse: [
- (self method add: IRReturn new) add: (IRVariable new
- variable: (aNode scope pseudoVars at: 'self');
- yourself) ].
- ^ self method
- !
- visitReturnNode: aNode
- | return |
- return := aNode nonLocalReturn
- ifTrue: [ IRNonLocalReturn new ]
- ifFalse: [ IRReturn new ].
- return scope: aNode scope.
- aNode nodes do: [ :each |
- return add: (self alias: each) ].
- ^ return
- !
- visitSendNode: aNode
- | send all receiver arguments |
- send := IRSend new.
- send
- selector: aNode selector;
- index: aNode index.
- aNode superSend ifTrue: [ send classSend: self theClass superclass ].
-
- all := self aliasTemporally: { aNode receiver }, aNode arguments.
- receiver := all first.
- arguments := all allButFirst.
- send add: receiver.
- arguments do: [ :each | send add: each ].
- ^ send
- !
- visitSequenceNode: aNode
- ^ self
- withSequence: IRSequence new
- do: [
- aNode nodes do: [ :each | | instruction |
- instruction := self visit: each.
- instruction isVariable ifFalse: [
- self sequence add: instruction ]]]
- !
- visitValueNode: aNode
- ^ IRValue new
- value: aNode value;
- yourself
- !
- visitVariableNode: aNode
- ^ IRVariable new
- variable: aNode binding;
- yourself
- ! !
- Object subclass: #IRInstruction
- instanceVariableNames: 'parent instructions'
- package: 'Compiler-IR'!
- !IRInstruction commentStamp!
- I am the abstract root class of the IR (intermediate representation) instructions class hierarchy.
- The IR graph is used to emit JavaScript code using a JSStream.!
- !IRInstruction methodsFor: 'accessing'!
- instructions
- ^ instructions ifNil: [ instructions := OrderedCollection new ]
- !
- method
- ^ self parent method
- !
- parent
- ^ parent
- !
- parent: anIRInstruction
- parent := anIRInstruction
- ! !
- !IRInstruction methodsFor: 'building'!
- add: anObject
- anObject parent: self.
- ^ self instructions add: anObject
- !
- remove
- self parent remove: self
- !
- remove: anIRInstruction
- self instructions remove: anIRInstruction
- !
- replace: anIRInstruction with: anotherIRInstruction
- anotherIRInstruction parent: self.
- self instructions
- at: (self instructions indexOf: anIRInstruction)
- put: anotherIRInstruction
- !
- replaceWith: anIRInstruction
- self parent replace: self with: anIRInstruction
- ! !
- !IRInstruction methodsFor: 'testing'!
- canBeAssigned
- ^ true
- !
- isClosure
- ^ false
- !
- isInlined
- ^ false
- !
- isLocalReturn
- ^ false
- !
- isMethod
- ^ false
- !
- isReturn
- ^ false
- !
- isSend
- ^ false
- !
- isSequence
- ^ false
- !
- isTempDeclaration
- ^ false
- !
- isVariable
- ^ false
- !
- needsBoxingAsReceiver
- ^ true
- ! !
- !IRInstruction methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRInstruction: self
- ! !
- !IRInstruction class methodsFor: 'instance creation'!
- on: aBuilder
- ^ self new
- builder: aBuilder;
- yourself
- ! !
- IRInstruction subclass: #IRAssignment
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRAssignment methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRAssignment: self
- ! !
- IRInstruction subclass: #IRDynamicArray
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRDynamicArray methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRDynamicArray: self
- ! !
- IRInstruction subclass: #IRDynamicDictionary
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRDynamicDictionary methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRDynamicDictionary: self
- ! !
- IRInstruction subclass: #IRScopedInstruction
- instanceVariableNames: 'scope'
- package: 'Compiler-IR'!
- !IRScopedInstruction methodsFor: 'accessing'!
- scope
- ^ scope
- !
- scope: aScope
- scope := aScope
- ! !
- IRScopedInstruction subclass: #IRClosureInstruction
- instanceVariableNames: 'arguments'
- package: 'Compiler-IR'!
- !IRClosureInstruction methodsFor: 'accessing'!
- arguments
- ^ arguments ifNil: [ #() ]
- !
- arguments: aCollection
- arguments := aCollection
- !
- locals
- ^ self arguments copy
- addAll: (self tempDeclarations collect: [ :each | each name ]);
- yourself
- !
- scope: aScope
- super scope: aScope.
- aScope instruction: self
- !
- tempDeclarations
- ^ self instructions select: [ :each |
- each isTempDeclaration ]
- ! !
- IRClosureInstruction subclass: #IRClosure
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRClosure methodsFor: 'accessing'!
- sequence
- ^ self instructions last
- ! !
- !IRClosure methodsFor: 'testing'!
- isClosure
- ^ true
- ! !
- !IRClosure methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRClosure: self
- ! !
- IRClosureInstruction subclass: #IRMethod
- instanceVariableNames: 'theClass source selector classReferences messageSends superSends internalVariables'
- package: 'Compiler-IR'!
- !IRMethod commentStamp!
- I am a method instruction!
- !IRMethod methodsFor: 'accessing'!
- classReferences
- ^ classReferences
- !
- classReferences: aCollection
- classReferences := aCollection
- !
- internalVariables
- ^ internalVariables ifNil: [ internalVariables := Set new ]
- !
- isMethod
- ^ true
- !
- messageSends
- ^ messageSends
- !
- messageSends: aCollection
- messageSends := aCollection
- !
- method
- ^ self
- !
- selector
- ^ selector
- !
- selector: aString
- selector := aString
- !
- source
- ^ source
- !
- source: aString
- source := aString
- !
- superSends
- ^ superSends
- !
- superSends: aCollection
- superSends := aCollection
- !
- theClass
- ^ theClass
- !
- theClass: aClass
- theClass := aClass
- ! !
- !IRMethod methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRMethod: self
- ! !
- IRScopedInstruction subclass: #IRReturn
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRReturn commentStamp!
- I am a local return instruction.!
- !IRReturn methodsFor: 'testing'!
- canBeAssigned
- ^ false
- !
- isBlockReturn
- ^ false
- !
- isLocalReturn
- ^ true
- !
- isNonLocalReturn
- ^ self isLocalReturn not
- !
- isReturn
- ^ true
- ! !
- !IRReturn methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRReturn: self
- ! !
- IRReturn subclass: #IRBlockReturn
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRBlockReturn commentStamp!
- Smalltalk blocks return their last statement. I am a implicit block return instruction.!
- !IRBlockReturn methodsFor: 'testing'!
- isBlockReturn
- ^ true
- ! !
- !IRBlockReturn methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRBlockReturn: self
- ! !
- IRReturn subclass: #IRNonLocalReturn
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRNonLocalReturn commentStamp!
- I am a non local return instruction.
- Non local returns are handled using a try/catch JavaScript statement.
- See `IRNonLocalReturnHandling` class.!
- !IRNonLocalReturn methodsFor: 'testing'!
- isLocalReturn
- ^ false
- ! !
- !IRNonLocalReturn methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRNonLocalReturn: self
- ! !
- IRScopedInstruction subclass: #IRTempDeclaration
- instanceVariableNames: 'name'
- package: 'Compiler-IR'!
- !IRTempDeclaration methodsFor: 'accessing'!
- name
- ^ name
- !
- name: aString
- name := aString
- ! !
- !IRTempDeclaration methodsFor: 'testing'!
- isTempDeclaration
- ^ true
- ! !
- !IRTempDeclaration methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRTempDeclaration: self
- ! !
- IRInstruction subclass: #IRSend
- instanceVariableNames: 'selector classSend index'
- package: 'Compiler-IR'!
- !IRSend commentStamp!
- I am a message send instruction.!
- !IRSend methodsFor: 'accessing'!
- classSend
- ^ classSend
- !
- classSend: aClass
- classSend := aClass
- !
- index
- ^ index
- !
- index: anInteger
- index := anInteger
- !
- selector
- ^ selector
- !
- selector: aString
- selector := aString
- ! !
- !IRSend methodsFor: 'testing'!
- isSend
- ^ true
- ! !
- !IRSend methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRSend: self
- ! !
- IRInstruction subclass: #IRSequence
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRSequence methodsFor: 'testing'!
- isSequence
- ^ true
- ! !
- !IRSequence methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRSequence: self
- ! !
- IRSequence subclass: #IRBlockSequence
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRBlockSequence methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRBlockSequence: self
- ! !
- IRInstruction subclass: #IRValue
- instanceVariableNames: 'value'
- package: 'Compiler-IR'!
- !IRValue commentStamp!
- I am the simplest possible instruction. I represent a value.!
- !IRValue methodsFor: 'accessing'!
- value
- ^value
- !
- value: aString
- value := aString
- ! !
- !IRValue methodsFor: 'testing'!
- needsBoxingAsReceiver
- ^ false
- ! !
- !IRValue methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRValue: self
- ! !
- IRInstruction subclass: #IRVariable
- instanceVariableNames: 'variable'
- package: 'Compiler-IR'!
- !IRVariable commentStamp!
- I am a variable instruction.!
- !IRVariable methodsFor: 'accessing'!
- variable
- ^ variable
- !
- variable: aScopeVariable
- variable := aScopeVariable
- ! !
- !IRVariable methodsFor: 'testing'!
- isVariable
- ^ true
- !
- needsBoxingAsReceiver
- ^ self variable isPseudoVar not
- ! !
- !IRVariable methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRVariable: self
- ! !
- IRInstruction subclass: #IRVerbatim
- instanceVariableNames: 'source'
- package: 'Compiler-IR'!
- !IRVerbatim methodsFor: 'accessing'!
- source
- ^ source
- !
- source: aString
- source := aString
- ! !
- !IRVerbatim methodsFor: 'visiting'!
- accept: aVisitor
- ^ aVisitor visitIRVerbatim: self
- ! !
- Object subclass: #IRVisitor
- instanceVariableNames: ''
- package: 'Compiler-IR'!
- !IRVisitor methodsFor: 'visiting'!
- visit: anIRInstruction
- ^ anIRInstruction accept: self
- !
- visitIRAssignment: anIRAssignment
- ^ self visitIRInstruction: anIRAssignment
- !
- visitIRBlockReturn: anIRBlockReturn
- ^ self visitIRReturn: anIRBlockReturn
- !
- visitIRBlockSequence: anIRBlockSequence
- ^ self visitIRSequence: anIRBlockSequence
- !
- visitIRClosure: anIRClosure
- ^ self visitIRInstruction: anIRClosure
- !
- visitIRDynamicArray: anIRDynamicArray
- ^ self visitIRInstruction: anIRDynamicArray
- !
- visitIRDynamicDictionary: anIRDynamicDictionary
- ^ self visitIRInstruction: anIRDynamicDictionary
- !
- visitIRInlinedClosure: anIRInlinedClosure
- ^ self visitIRClosure: anIRInlinedClosure
- !
- visitIRInlinedSequence: anIRInlinedSequence
- ^ self visitIRSequence: anIRInlinedSequence
- !
- visitIRInstruction: anIRInstruction
- anIRInstruction instructions do: [ :each | self visit: each ].
- ^ anIRInstruction
- !
- visitIRMethod: anIRMethod
- ^ self visitIRInstruction: anIRMethod
- !
- visitIRNonLocalReturn: anIRNonLocalReturn
- ^ self visitIRInstruction: anIRNonLocalReturn
- !
- visitIRNonLocalReturnHandling: anIRNonLocalReturnHandling
- ^ self visitIRInstruction: anIRNonLocalReturnHandling
- !
- visitIRReturn: anIRReturn
- ^ self visitIRInstruction: anIRReturn
- !
- visitIRSend: anIRSend
- ^ self visitIRInstruction: anIRSend
- !
- visitIRSequence: anIRSequence
- ^ self visitIRInstruction: anIRSequence
- !
- visitIRTempDeclaration: anIRTempDeclaration
- ^ self visitIRInstruction: anIRTempDeclaration
- !
- visitIRValue: anIRValue
- ^ self visitIRInstruction: anIRValue
- !
- visitIRVariable: anIRVariable
- ^ self visitIRInstruction: anIRVariable
- !
- visitIRVerbatim: anIRVerbatim
- ^ self visitIRInstruction: anIRVerbatim
- ! !
- IRVisitor subclass: #IRJSTranslator
- instanceVariableNames: 'stream'
- package: 'Compiler-IR'!
- !IRJSTranslator methodsFor: 'accessing'!
- contents
- ^ self stream contents
- !
- stream
- ^ stream
- !
- stream: aStream
- stream := aStream
- ! !
- !IRJSTranslator methodsFor: 'initialization'!
- initialize
- super initialize.
- stream := JSStream new.
- ! !
- !IRJSTranslator methodsFor: 'visiting'!
- visitIRAssignment: anIRAssignment
- self visit: anIRAssignment instructions first.
- self stream nextPutAssignment.
- self visit: anIRAssignment instructions last.
- !
- visitIRClosure: anIRClosure
- self stream
- nextPutClosureWith: [
- self stream nextPutVars: (anIRClosure tempDeclarations collect: [ :each |
- each name asVariableName ]).
- self stream
- nextPutBlockContextFor: anIRClosure
- during: [ super visitIRClosure: anIRClosure ] ]
- arguments: anIRClosure arguments
- !
- visitIRDynamicArray: anIRDynamicArray
- self stream nextPutAll: '['.
- anIRDynamicArray instructions
- do: [ :each | self visit: each ]
- separatedBy: [ self stream nextPutAll: ',' ].
- stream nextPutAll: ']'
- !
- visitIRDynamicDictionary: anIRDynamicDictionary
- self stream nextPutAll: 'smalltalk.HashedCollection._from_(['.
- anIRDynamicDictionary instructions
- do: [ :each | self visit: each ]
- separatedBy: [self stream nextPutAll: ',' ].
- self stream nextPutAll: '])'
- !
- visitIRMethod: anIRMethod
- self stream
- nextPutMethodDeclaration: anIRMethod
- with: [ self stream
- nextPutFunctionWith: [
- self stream nextPutVars: (anIRMethod tempDeclarations collect: [ :each |
- each name asVariableName ]).
- anIRMethod classReferences do: [ :each | self stream nextPutClassRefFunction: each ].
- self stream nextPutContextFor: anIRMethod during: [
- anIRMethod internalVariables notEmpty ifTrue: [
- self stream nextPutVars: (anIRMethod internalVariables asArray collect: [ :each |
- each variable alias ]) ].
- anIRMethod scope hasNonLocalReturn
- ifTrue: [
- self stream nextPutNonLocalReturnHandlingWith: [
- super visitIRMethod: anIRMethod ]]
- ifFalse: [ super visitIRMethod: anIRMethod ]]]
- arguments: anIRMethod arguments ]
- !
- visitIRNonLocalReturn: anIRNonLocalReturn
- self stream nextPutNonLocalReturnWith: [
- super visitIRNonLocalReturn: anIRNonLocalReturn ]
- !
- visitIRReturn: anIRReturn
- self stream nextPutReturnWith: [
- super visitIRReturn: anIRReturn ]
- !
- visitIRSend: anIRSend
- anIRSend classSend
- ifNil: [ self visitSend: anIRSend ]
- ifNotNil: [ self visitSuperSend: anIRSend ]
- !
- visitIRSequence: anIRSequence
- self stream nextPutSequenceWith: [
- anIRSequence instructions do: [ :each |
- self stream nextPutStatementWith: (self visit: each) ]]
- !
- visitIRTempDeclaration: anIRTempDeclaration
- "self stream
- nextPutAll: 'var ', anIRTempDeclaration name asVariableName, ';';
- lf"
- !
- visitIRValue: anIRValue
- self stream nextPutAll: anIRValue value asJavascript
- !
- visitIRVariable: anIRVariable
- anIRVariable variable name = 'thisContext'
- ifTrue: [ self stream nextPutAll: 'smalltalk.getThisContext()' ]
- ifFalse: [ self stream nextPutAll: anIRVariable variable alias ]
- !
- visitIRVerbatim: anIRVerbatim
- self stream nextPutStatementWith: [
- self stream nextPutAll: anIRVerbatim source ]
- !
- visitReceiver: anIRInstruction
- anIRInstruction needsBoxingAsReceiver ifFalse: [ ^ self visit: anIRInstruction ].
-
- self stream nextPutAll: '_st('.
- self visit: anIRInstruction.
- self stream nextPutAll: ')'
- !
- visitSend: anIRSend
- self visitReceiver: anIRSend instructions first.
- self stream nextPutAll: '.', anIRSend selector asSelector, '('.
- anIRSend instructions allButFirst
- do: [ :each | self visit: each ]
- separatedBy: [ self stream nextPutAll: ',' ].
- self stream nextPutAll: ')'
- !
- visitSuperSend: anIRSend
- self stream
- nextPutAll: anIRSend classSend asJavascript, '.fn.prototype.';
- nextPutAll: anIRSend selector asSelector, '.apply(';
- nextPutAll: '_st('.
- self visit: anIRSend instructions first.
- self stream nextPutAll: '), ['.
- anIRSend instructions allButFirst
- do: [ :each | self visit: each ]
- separatedBy: [ self stream nextPutAll: ',' ].
- self stream nextPutAll: '])'
- ! !
- Object subclass: #JSStream
- instanceVariableNames: 'stream'
- package: 'Compiler-IR'!
- !JSStream methodsFor: 'accessing'!
- contents
- ^ stream contents
- ! !
- !JSStream methodsFor: 'initialization'!
- initialize
- super initialize.
- stream := '' writeStream.
- ! !
- !JSStream methodsFor: 'streaming'!
- lf
- stream lf
- !
- nextPut: aString
- stream nextPut: aString
- !
- nextPutAll: aString
- stream nextPutAll: aString
- !
- nextPutAssignment
- stream nextPutAll: '='
- !
- nextPutBlockContextFor: anIRClosure during: aBlock
- self
- nextPutAll: 'return smalltalk.withContext(function(', anIRClosure scope alias, ') {'; lf.
-
- aBlock value.
-
- self
- nextPutAll: '}, function(', anIRClosure scope alias, ') {';
- nextPutAll: anIRClosure scope alias, '.fillBlock({'.
-
- anIRClosure locals
- do: [ :each |
- self
- nextPutAll: each asVariableName;
- nextPutAll: ':';
- nextPutAll: each asVariableName]
- separatedBy: [ self nextPutAll: ',' ].
-
- self
- nextPutAll: '},';
- nextPutAll: anIRClosure scope outerScope alias, ')})'
- !
- nextPutClassRefFunction: aString
- "Creates an inner function $aString into method and called as `$Foo()`whenever the global is accessed.
- This ensures that undefined global access will answer `nil`"
-
- stream
- nextPutAll: 'function $';
- nextPutAll: aString;
- nextPutAll: '(){return smalltalk.';
- nextPutAll: aString;
- nextPutAll: '||(typeof ';
- nextPutAll: aString;
- nextPutAll: '=="undefined"?nil:';
- nextPutAll: aString;
- nextPutAll: ')}';
- lf
- !
- nextPutClosureWith: aBlock arguments: anArray
- stream nextPutAll: '(function('.
- anArray
- do: [ :each | stream nextPutAll: each asVariableName ]
- separatedBy: [ stream nextPut: ',' ].
- stream nextPutAll: '){'; lf.
- aBlock value.
- stream nextPutAll: '})'
- !
- nextPutContextFor: aMethod during: aBlock
- self
- nextPutAll: 'return smalltalk.withContext(function(', aMethod scope alias, ') { '; lf.
- aBlock value.
-
- self
- nextPutAll: '}, function(', aMethod scope alias, ') {', aMethod scope alias;
- nextPutAll: '.fill(self,', aMethod selector asJavascript, ',{'.
- aMethod locals
- do: [ :each |
- self
- nextPutAll: each asVariableName;
- nextPutAll: ':';
- nextPutAll: each asVariableName]
- separatedBy: [ self nextPutAll: ',' ].
-
- self
- nextPutAll: '},';
- nextPutAll: aMethod theClass asJavascript;
- nextPutAll: ')})'
- !
- nextPutFunctionWith: aBlock arguments: anArray
- stream nextPutAll: 'fn: function('.
- anArray
- do: [ :each | stream nextPutAll: each asVariableName ]
- separatedBy: [ stream nextPut: ',' ].
- stream nextPutAll: '){'; lf.
- stream nextPutAll: 'var self=this;'; lf.
- aBlock value.
- stream nextPutAll: '}'
- !
- nextPutIf: aBlock with: anotherBlock
- stream nextPutAll: 'if('.
- aBlock value.
- stream nextPutAll: '){'; lf.
- anotherBlock value.
- stream nextPutAll: '}'
- !
- nextPutIfElse: aBlock with: ifBlock with: elseBlock
- stream nextPutAll: 'if('.
- aBlock value.
- stream nextPutAll: '){'; lf.
- ifBlock value.
- stream nextPutAll: '} else {'; lf.
- elseBlock value.
- stream nextPutAll: '}'
- !
- nextPutMethodDeclaration: aMethod with: aBlock
- stream
- nextPutAll: 'smalltalk.method({'; lf;
- nextPutAll: 'selector: ', aMethod selector asJavascript, ','; lf;
- nextPutAll: 'source: ', aMethod source asJavascript, ',';lf.
- aBlock value.
- stream
- nextPutAll: ',', String lf, 'messageSends: ';
- nextPutAll: aMethod messageSends asArray asJavascript, ','; lf;
- nextPutAll: 'args: ', (aMethod arguments collect: [ :each | each value ]) asArray asJavascript, ','; lf;
- nextPutAll: 'referencedClasses: ['.
- aMethod classReferences
- do: [:each | stream nextPutAll: each asJavascript]
- separatedBy: [stream nextPutAll: ','].
- stream
- nextPutAll: ']';
- nextPutAll: '})'
- !
- nextPutNonLocalReturnHandlingWith: aBlock
- stream
- nextPutAll: 'var $early={};'; lf;
- nextPutAll: 'try {'; lf.
- aBlock value.
- stream
- nextPutAll: '}'; lf;
- nextPutAll: 'catch(e) {if(e===$early)return e[0]; throw e}'; lf
- !
- nextPutNonLocalReturnWith: aBlock
- stream nextPutAll: 'throw $early=['.
- aBlock value.
- stream nextPutAll: ']'
- !
- nextPutReturn
- stream nextPutAll: 'return '
- !
- nextPutReturnWith: aBlock
- self nextPutReturn.
- aBlock value
- !
- nextPutSequenceWith: aBlock
- "stream
- nextPutAll: 'switch(smalltalk.thisContext.pc){'; lf."
- aBlock value.
- "stream
- nextPutAll: '};'; lf"
- !
- nextPutStatement: anInteger with: aBlock
- stream nextPutAll: 'case ', anInteger asString, ':'; lf.
- self nextPutStatementWith: aBlock.
- stream nextPutAll: 'smalltalk.thisContext.pc=', (anInteger + 1) asString, ';'; lf
- !
- nextPutStatementWith: aBlock
- aBlock value.
- stream nextPutAll: ';'; lf
- !
- nextPutVars: aCollection
- aCollection ifEmpty: [ ^self ].
-
- stream nextPutAll: 'var '.
- aCollection
- do: [ :each | stream nextPutAll: each ]
- separatedBy: [ stream nextPutAll: ',' ].
- stream nextPutAll: ';'; lf
- ! !
- !BlockClosure methodsFor: '*Compiler-IR'!
- appendToInstruction: anIRInstruction
- anIRInstruction appendBlock: self
- ! !
- !String methodsFor: '*Compiler-IR'!
- asVariableName
- ^ (Smalltalk current reservedWords includes: self)
- ifTrue: [ self, '_' ]
- ifFalse: [ self ]
- ! !
|