12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028 |
- Object subclass: #Node
- instanceVariableNames: 'nodes'
- category: 'Compiler'!
- !Node methodsFor: 'accessing'!
- nodes
- ^nodes ifNil: [nodes := Array new]
- !
- addNode: aNode
- self nodes add: aNode
- ! !
- !Node methodsFor: 'building'!
- nodes: aCollection
- nodes := aCollection
- ! !
- !Node methodsFor: 'testing'!
- isValueNode
- ^false
- !
- isBlockNode
- ^false
- !
- isBlockSequenceNode
- ^false
- ! !
- !Node methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitNode: self
- ! !
- Node subclass: #MethodNode
- instanceVariableNames: 'selector arguments source'
- category: 'Compiler'!
- !MethodNode methodsFor: 'accessing'!
- selector
- ^selector
- !
- selector: aString
- selector := aString
- !
- arguments
- ^arguments ifNil: [#()]
- !
- arguments: aCollection
- arguments := aCollection
- !
- source
- ^source
- !
- source: aString
- source := aString
- ! !
- !MethodNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitMethodNode: self
- ! !
- Node subclass: #SendNode
- instanceVariableNames: 'selector arguments receiver'
- category: 'Compiler'!
- !SendNode methodsFor: 'accessing'!
- selector
- ^selector
- !
- selector: aString
- selector := aString
- !
- arguments
- ^arguments ifNil: [arguments := #()]
- !
- arguments: aCollection
- arguments := aCollection
- !
- receiver
- ^receiver
- !
- receiver: aNode
- receiver := aNode
- !
- valueForReceiver: anObject
- ^SendNode new
- receiver: (self receiver
- ifNil: [anObject]
- ifNotNil: [self receiver valueForReceiver: anObject]);
- selector: self selector;
- arguments: self arguments;
- yourself
- !
- cascadeNodeWithMessages: aCollection
- | first |
- first := SendNode new
- selector: self selector;
- arguments: self arguments;
- yourself.
- ^CascadeNode new
- receiver: self receiver;
- nodes: (Array with: first), aCollection;
- yourself
- ! !
- !SendNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitSendNode: self
- ! !
- Node subclass: #CascadeNode
- instanceVariableNames: 'receiver'
- category: 'Compiler'!
- !CascadeNode methodsFor: 'accessing'!
- receiver
- ^receiver
- !
- receiver: aNode
- receiver := aNode
- ! !
- !CascadeNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitCascadeNode: self
- ! !
- Node subclass: #AssignmentNode
- instanceVariableNames: 'left right'
- category: 'Compiler'!
- !AssignmentNode methodsFor: 'accessing'!
- left
- ^left
- !
- left: aNode
- left := aNode
- !
- right
- ^right
- !
- right: aNode
- right := aNode
- ! !
- !AssignmentNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitAssignmentNode: self
- ! !
- Node subclass: #BlockNode
- instanceVariableNames: 'parameters inlined'
- category: 'Compiler'!
- !BlockNode methodsFor: 'accessing'!
- parameters
- ^parameters ifNil: [parameters := Array new]
- !
- parameters: aCollection
- parameters := aCollection
- !
- inlined
- ^inlined ifNil: [false]
- !
- inlined: aBoolean
- inlined := aBoolean
- ! !
- !BlockNode methodsFor: 'testing'!
- isBlockNode
- ^true
- ! !
- !BlockNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitBlockNode: self
- ! !
- Node subclass: #SequenceNode
- instanceVariableNames: 'temps'
- category: 'Compiler'!
- !SequenceNode methodsFor: 'accessing'!
- temps
- ^temps ifNil: [#()]
- !
- temps: aCollection
- temps := aCollection
- ! !
- !SequenceNode methodsFor: 'testing'!
- asBlockSequenceNode
- ^BlockSequenceNode new
- nodes: self nodes;
- temps: self temps;
- yourself
- ! !
- !SequenceNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitSequenceNode: self
- ! !
- SequenceNode subclass: #BlockSequenceNode
- instanceVariableNames: ''
- category: 'Compiler'!
- !BlockSequenceNode methodsFor: 'testing'!
- isBlockSequenceNode
- ^true
- ! !
- !BlockSequenceNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitBlockSequenceNode: self
- ! !
- Node subclass: #ReturnNode
- instanceVariableNames: ''
- category: 'Compiler'!
- !ReturnNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitReturnNode: self
- ! !
- Node subclass: #ValueNode
- instanceVariableNames: 'value'
- category: 'Compiler'!
- !ValueNode methodsFor: 'accessing'!
- value
- ^value
- !
- value: anObject
- value := anObject
- ! !
- !ValueNode methodsFor: 'testing'!
- isValueNode
- ^true
- ! !
- !ValueNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitValueNode: self
- ! !
- ValueNode subclass: #VariableNode
- instanceVariableNames: ''
- category: 'Compiler'!
- !VariableNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitVariableNode: self
- ! !
- VariableNode subclass: #ClassReferenceNode
- instanceVariableNames: ''
- category: 'Compiler'!
- !ClassReferenceNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitClassReferenceNode: self
- ! !
- Node subclass: #JSStatementNode
- instanceVariableNames: 'source'
- category: 'Compiler'!
- !JSStatementNode methodsFor: 'accessing'!
- source
- ^source ifNil: ['']
- !
- source: aString
- source := aString
- ! !
- !JSStatementNode methodsFor: 'visiting'!
- accept: aVisitor
- aVisitor visitJSStatementNode: self
- ! !
- Object subclass: #NodeVisitor
- instanceVariableNames: ''
- category: 'Compiler'!
- !NodeVisitor methodsFor: 'visiting'!
- visit: aNode
- aNode accept: self
- !
- visitNode: aNode
- !
- visitMethodNode: aNode
- self visitNode: aNode
- !
- visitSequenceNode: aNode
- self visitNode: aNode
- !
- visitBlockSequenceNode: aNode
- self visitSequenceNode: aNode
- !
- visitBlockNode: aNode
- self visitNode: aNode
- !
- visitReturnNode: aNode
- self visitNode: aNode
- !
- visitSendNode: aNode
- self visitNode: aNode
- !
- visitCascadeNode: aNode
- self visitNode: aNode
- !
- visitValueNode: aNode
- self visitNode: aNode
- !
- visitVariableNode: aNode
- !
- visitAssignmentNode: aNode
- self visitNode: aNode
- !
- visitClassReferenceNode: aNode
- self
- nextPutAll: 'smalltalk.';
- nextPutAll: aNode value
- !
- visitJSStatementNode: aNode
- self
- nextPutAll: 'function(){';
- nextPutAll: aNode source;
- nextPutAll: '})()'
- ! !
- NodeVisitor subclass: #Compiler
- instanceVariableNames: 'stream nestedBlocks earlyReturn currentClass currentSelector unknownVariables tempVariables messageSends referencedClasses'
- category: 'Compiler'!
- !Compiler methodsFor: 'accessing'!
- parser
- ^SmalltalkParser new
- !
- currentClass
- ^currentClass
- !
- currentClass: aClass
- currentClass := aClass
- !
- unknownVariables
- ^unknownVariables copy
- !
- pseudoVariables
- ^#('self' 'super' 'true' 'false' 'nil' 'thisContext')
- !
- tempVariables
- ^tempVariables copy
- !
- knownVariables
- ^self pseudoVariables
- addAll: self tempVariables;
- yourself
- !
- classNameFor: aClass
- ^aClass isMetaclass
- ifTrue: [aClass instanceClass name, '.klass']
- ifFalse: [
- aClass isNil
- ifTrue: ['nil']
- ifFalse: [aClass name]]
- ! !
- !Compiler methodsFor: 'compiling'!
- loadExpression: aString
- DoIt addCompiledMethod: (self eval: (self compileExpression: aString)).
- ^DoIt new doIt
- !
- load: aString forClass: aClass
- | compiled |
- compiled := self eval: (self compile: aString forClass: aClass).
- self setupClass: aClass.
- ^compiled
- !
- compile: aString forClass: aClass
- self currentClass: aClass.
- ^self compile: aString
- !
- compileExpression: aString
- self currentClass: DoIt.
- ^self compileNode: (self parseExpression: aString)
- !
- eval: aString
- <return eval(aString)>
- !
- compile: aString
- ^self compileNode: (self parse: aString)
- !
- compileNode: aNode
- stream := '' writeStream.
- self visit: aNode.
- ^stream contents
- !
- parse: aString
- ^self parser parse: aString readStream
- !
- parseExpression: aString
- ^self parse: 'doIt ^[', aString, '] value'
- !
- recompile: aClass
- aClass methodDictionary do: [:each || method |
- method := self load: each source forClass: aClass.
- method category: each category.
- aClass addCompiledMethod: method].
- aClass isMetaclass ifFalse: [self recompile: aClass class]
- !
- recompileAll
- Smalltalk current classes do: [:each |
- Transcript show: each; cr.
- [self recompile: each] valueWithTimeout: 100]
- !
- setupClass: aClass
- <smalltalk.init(aClass)>
- ! !
- !Compiler methodsFor: 'initialization'!
- initialize
- super initialize.
- stream := '' writeStream.
- unknownVariables := #().
- tempVariables := #().
- messageSends := #().
- classReferenced := #()
- ! !
- !Compiler methodsFor: 'optimizations'!
- checkClass: aClassName for: receiver
- stream nextPutAll: '(($receiver = ', receiver, ').klass === smalltalk.', aClassName, ') ? '
- !
- inlineLiteral: aSelector receiverNode: anObject argumentNodes: aCollection
- | inlined |
- inlined := false.
- "-- BlockClosures --"
- (aSelector = 'whileTrue:') ifTrue: [
- (anObject isBlockNode and: [aCollection first isBlockNode]) ifTrue: [
- stream nextPutAll: '(function(){while('.
- self visit: anObject.
- stream nextPutAll: '()) {'.
- self visit: aCollection first.
- stream nextPutAll: '()}})()'.
- inlined := true]].
- (aSelector = 'whileFalse:') ifTrue: [
- (anObject isBlockNode and: [aCollection first isBlockNode]) ifTrue: [
- stream nextPutAll: '(function(){while(!!'.
- self visit: anObject.
- stream nextPutAll: '()) {'.
- self visit: aCollection first.
- stream nextPutAll: '()}})()'.
- inlined := true]].
- (aSelector = 'whileTrue') ifTrue: [
- anObject isBlockNode ifTrue: [
- stream nextPutAll: '(function(){while('.
- self visit: anObject.
- stream nextPutAll: '()) {}})()'.
- inlined := true]].
- (aSelector = 'whileFalse') ifTrue: [
- anObject isBlockNode ifTrue: [
- stream nextPutAll: '(function(){while(!!'.
- self visit: anObject.
- stream nextPutAll: '()) {}})()'.
- inlined := true]].
- "-- Numbers --"
- (aSelector = '+') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' + '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '-') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' - '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '*') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' * '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '/') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' / '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '<') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' < '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '<=') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' <= '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '=') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' = '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '>') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' > '.
- self visit: aCollection first.
- inlined := true]].
- (aSelector = '>=') ifTrue: [
- (self isNode: anObject ofClass: Number) ifTrue: [
- self visit: anObject.
- stream nextPutAll: ' >= '.
- self visit: aCollection first.
- inlined := true]].
-
- "-- UndefinedObject --"
- (aSelector = 'ifNil:') ifTrue: [
- aCollection first isBlockNode ifTrue: [
- stream nextPutAll: '(($receiver = '.
- self visit: anObject.
- stream nextPutAll: ') == nil || $receiver == undefined) ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : $receiver'.
- inlined := true]].
- (aSelector = 'ifNotNil:') ifTrue: [
- aCollection first isBlockNode ifTrue: [
- stream nextPutAll: '(($receiver = '.
- self visit: anObject.
- stream nextPutAll: ') !!= nil && $receiver !!= undefined) ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : nil'.
- inlined := true]].
- (aSelector = 'ifNil:ifNotNil:') ifTrue: [
- (aCollection first isBlockNode and: [aCollection second isBlockNode]) ifTrue: [
- stream nextPutAll: '(($receiver = '.
- self visit: anObject.
- stream nextPutAll: ') == nil || $receiver == undefined) ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : '.
- self visit: aCollection second.
- stream nextPutAll: '()'.
- inlined := true]].
- (aSelector = 'ifNotNil:ifNil:') ifTrue: [
- (aCollection first isBlockNode and: [aCollection second isBlockNode]) ifTrue: [
- stream nextPutAll: '(($receiver = '.
- self visit: anObject.
- stream nextPutAll: ') == nil || $receiver == undefined) ? '.
- self visit: aCollection second.
- stream nextPutAll: '() : '.
- self visit: aCollection first.
- stream nextPutAll: '()'.
- inlined := true]].
-
- ^inlined
- !
- isNode: aNode ofClass: aClass
- ^aNode isValueNode and: [
- aNode value class = aClass or: [
- aNode value = 'self' and: [self currentClass = aClass]]]
- !
- inline: aSelector receiver: receiver argumentNodes: aCollection
- | inlined |
- inlined := false.
- "-- Booleans --"
- (aSelector = 'ifFalse:') ifTrue: [
- aCollection first isBlockNode ifTrue: [
- self checkClass: 'Boolean' for: receiver.
- stream nextPutAll: '(!! $receiver ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : nil)'.
- inlined := true]].
- (aSelector = 'ifTrue:') ifTrue: [
- aCollection first isBlockNode ifTrue: [
- self checkClass: 'Boolean' for: receiver.
- stream nextPutAll: '($receiver ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : nil)'.
- inlined := true]].
- (aSelector = 'ifTrue:ifFalse:') ifTrue: [
- (aCollection first isBlockNode and: [aCollection second isBlockNode]) ifTrue: [
- self checkClass: 'Boolean' for: receiver.
- stream nextPutAll: '($receiver ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : '.
- self visit: aCollection second.
- stream nextPutAll: '())'.
- inlined := true]].
- (aSelector = 'ifFalse:ifTrue:') ifTrue: [
- (aCollection first isBlockNode and: [aCollection second isBlockNode]) ifTrue: [
- self checkClass: 'Boolean' for: receiver.
- stream nextPutAll: '(!! $receiver ? '.
- self visit: aCollection first.
- stream nextPutAll: '() : '.
- self visit: aCollection second.
- stream nextPutAll: '())'.
- inlined := true]].
- "-- Numbers --"
- (aSelector = '<') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver <'.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '<=') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver <='.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '=') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver =='.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '>') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver >'.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '>=') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver >='.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '+') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver +'.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '-') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver -'.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '*') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver *'.
- self visit: aCollection first.
- inlined := true].
- (aSelector = '/') ifTrue: [
- self checkClass: 'Number' for: receiver.
- stream nextPutAll: '$receiver /'.
- self visit: aCollection first.
- inlined := true].
- ^inlined
- ! !
- !Compiler methodsFor: 'testing'!
- performOptimizations
- ^self class performOptimizations
- ! !
- !Compiler methodsFor: 'visiting'!
- visit: aNode
- aNode accept: self
- !
- visitMethodNode: aNode
- | str currentSelector |
- currentSelector := aNode selector asSelector.
- nestedBlocks := 0.
- earlyReturn := false.
- messageSends := #().
- referencedClasses := #().
- unknownVariables := #().
- tempVariables := #().
- stream
- nextPutAll: 'smalltalk.method({'; lf;
- nextPutAll: 'selector: "', aNode selector, '",'; lf.
- stream nextPutAll: 'source: unescape("', aNode source escaped, '"),';lf.
- stream nextPutAll: 'fn: function('.
- aNode arguments
- do: [:each |
- tempVariables add: each.
- stream nextPutAll: each]
- separatedBy: [stream nextPutAll: ', '].
- stream
- nextPutAll: '){'; lf;
- nextPutAll: 'var self=this;'; lf.
- str := stream.
- stream := '' writeStream.
- aNode nodes do: [:each |
- self visit: each].
- earlyReturn ifTrue: [
- str nextPutAll: 'try{'].
- str nextPutAll: stream contents.
- stream := str.
- stream
- lf;
- nextPutAll: 'return self;'.
- earlyReturn ifTrue: [
- stream lf; nextPutAll: '} catch(e) {if(e.name === ''stReturn'' && e.selector === ', currentSelector printString, '){return e.fn()} throw(e)}'].
- stream nextPutAll: '}'.
- stream
- nextPutAll: ',', String lf, 'messageSends: ';
- nextPutAll: messageSends asJavascript, ','; lf;
- nextPutAll: 'referencedClasses: ['.
- referencedClasses
- do: [:each | stream nextPutAll: each]
- separatedBy: [stream nextPutAll: ','].
- stream nextPutAll: ']'.
- stream nextPutAll: '})'
- !
- visitBlockNode: aNode
- stream nextPutAll: '(function('.
- aNode parameters
- do: [:each |
- tempVariables add: each.
- stream nextPutAll: each]
- separatedBy: [stream nextPutAll: ', '].
- stream nextPutAll: '){'.
- aNode nodes do: [:each | self visit: each].
- stream nextPutAll: '})'
- !
- visitSequenceNode: aNode
- aNode temps do: [:each |
- tempVariables add: each.
- stream nextPutAll: 'var ', each, '=nil;'; lf].
- aNode nodes do: [:each |
- self visit: each.
- stream nextPutAll: ';']
- separatedBy: [stream lf]
- !
- visitBlockSequenceNode: aNode
- | index |
- nestedBlocks := nestedBlocks + 1.
- aNode nodes isEmpty
- ifTrue: [
- stream nextPutAll: 'return nil;']
- ifFalse: [
- aNode temps do: [:each |
- tempVariables add: each.
- stream nextPutAll: 'var ', each, '=nil;'; lf].
- index := 0.
- aNode nodes do: [:each |
- index := index + 1.
- index = aNode nodes size ifTrue: [
- stream nextPutAll: 'return '].
- self visit: each.
- stream nextPutAll: ';']].
- nestedBlocks := nestedBlocks - 1
- !
- visitReturnNode: aNode
- nestedBlocks > 0 ifTrue: [
- earlyReturn := true].
- earlyReturn
- ifTrue: [
- stream
- nextPutAll: '(function(){throw(';
- nextPutAll: '{name: ''stReturn'', selector: ';
- nextPutAll: currentSelector printString;
- nextPutAll: ', fn: function(){return ']
- ifFalse: [stream nextPutAll: 'return '].
- aNode nodes do: [:each |
- self visit: each].
- earlyReturn ifTrue: [
- stream nextPutAll: '}})})()']
- !
- visitSendNode: aNode
- | str receiver superSend inlined |
- str := stream.
- (messageSends includes: aNode selector) ifFalse: [
- messageSends add: aNode selector].
- stream := '' writeStream.
- self visit: aNode receiver.
- superSend := stream contents = 'super'.
- receiver := superSend ifTrue: ['self'] ifFalse: [stream contents].
- stream := str.
-
- self performOptimizations
- ifTrue: [
- (self inlineLiteral: aNode selector receiverNode: aNode receiver argumentNodes: aNode arguments) ifFalse: [
- (self inline: aNode selector receiver: receiver argumentNodes: aNode arguments)
- ifTrue: [stream nextPutAll: ' : ', (self send: aNode selector to: '$receiver' arguments: aNode arguments superSend: superSend)]
- ifFalse: [stream nextPutAll: (self send: aNode selector to: receiver arguments: aNode arguments superSend: superSend)]]]
- ifFalse: [stream nextPutAll: (self send: aNode selector to: receiver arguments: aNode arguments superSend: superSend)]
- !
- visitCascadeNode: aNode
- | index |
- index := 0.
- (tempVariables includes: '$rec') ifFalse: [
- tempVariables add: '$rec'].
- stream nextPutAll: '(function($rec){'.
- aNode nodes do: [:each |
- index := index + 1.
- index = aNode nodes size ifTrue: [
- stream nextPutAll: 'return '].
- each receiver: (VariableNode new value: '$rec').
- self visit: each.
- stream nextPutAll: ';'].
- stream nextPutAll: '})('.
- self visit: aNode receiver.
- stream nextPutAll: ')'
- !
- visitValueNode: aNode
- stream nextPutAll: aNode value asJavascript
- !
- visitAssignmentNode: aNode
- self visit: aNode left.
- stream nextPutAll: '='.
- self visit: aNode right
- !
- visitClassReferenceNode: aNode
- | klass |
- klass := 'smalltalk.', aNode value.
- (Smalltalk current at: aNode value) isClass ifTrue: [
- (referencedClasses includes: klass)
- ifFalse: [referencedClasses add: klass]].
- stream nextPutAll: klass
- !
- visitVariableNode: aNode
- (self currentClass allInstanceVariableNames includes: aNode value)
- ifTrue: [stream nextPutAll: 'self[''@', aNode value, ''']']
- ifFalse: [
- (self knownVariables includes: aNode value) ifFalse: [
- unknownVariables add: aNode value].
- stream nextPutAll: aNode value]
- !
- visitJSStatementNode: aNode
- stream nextPutAll: (aNode source replace: '>>' with: '>')
- !
- visitFailure: aFailure
- self error: aFailure asString
- !
- send: aSelector to: aReceiver arguments: aCollection superSend: aBoolean
- ^String streamContents: [:str || tmp |
- tmp := stream.
- str nextPutAll: 'smalltalk.send('.
- str nextPutAll: aReceiver.
- str nextPutAll: ', "', aSelector asSelector, '", ['.
- stream := str.
- aCollection
- do: [:each | self visit: each]
- separatedBy: [stream nextPutAll: ', '].
- stream := tmp.
- str nextPutAll: ']'.
- aBoolean ifTrue: [
- str nextPutAll: ', smalltalk.', (self classNameFor: self currentClass superclass)].
- str nextPutAll: ')']
- ! !
- Compiler class instanceVariableNames: 'performOptimizations'!
- !Compiler class methodsFor: 'accessing'!
- performOptimizations
- ^performOptimizations ifNil: [true]
- !
- performOptimizations: aBoolean
- performOptimizations := aBoolean
- ! !
- !Compiler class methodsFor: 'compiling'!
- recompile: aClass
- aClass methodDictionary do: [:each || method |
- method := self new load: each source forClass: aClass.
- method category: each category.
- aClass addCompiledMethod: method].
- aClass isMetaclass ifFalse: [self recompile: aClass class]
- !
- recompileAll
- Smalltalk current classes do: [:each |
- self recompile: each]
- ! !
- Object subclass: #DoIt
- instanceVariableNames: ''
- category: 'Compiler'!
- !DoIt methodsFor: ''!
- doIt ^[Compiler performOptimizations: false.] value
- ! !
|