Compiler-AST.st 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768
  1. Smalltalk createPackage: 'Compiler-AST'!
  2. DagParentNode subclass: #ASTNode
  3. instanceVariableNames: 'parent position source shouldBeAliased'
  4. package: 'Compiler-AST'!
  5. !ASTNode commentStamp!
  6. I am the abstract root class of the abstract syntax tree.
  7. Concrete classes should implement `#accept:` to allow visiting.
  8. `position` holds a point containing line and column number of the symbol location in the original source file.!
  9. !ASTNode methodsFor: 'accessing'!
  10. location: aLocation
  11. self position: aLocation start line @ aLocation start column
  12. !
  13. method
  14. ^ self parent ifNotNil: [ :node | node method ]
  15. !
  16. navigationNodeAt: aPoint ifAbsent: aBlock
  17. "Answer the navigation node in the receiver's tree at aPoint
  18. or nil if no navigation node was found.
  19. See `node >> isNaviationNode`"
  20. | children |
  21. children := self allDagChildren select: [ :each |
  22. each isNavigationNode and: [ each inPosition: aPoint ] ].
  23. children ifEmpty: [ ^ aBlock value ].
  24. ^ (children asArray sort: [ :a :b |
  25. (a positionStart dist: aPoint) <=
  26. (b positionStart dist: aPoint) ]) first
  27. !
  28. parent
  29. ^ parent
  30. !
  31. parent: aNode
  32. parent := aNode
  33. !
  34. position
  35. "answer the line and column of the receiver in the source code"
  36. ^ position ifNil: [
  37. self parent ifNotNil: [ :node | node position ] ]
  38. !
  39. position: aPosition
  40. position := aPosition
  41. !
  42. positionEnd
  43. ^ self positionStart + ((self source lines size - 1) @ (self source lines last size - 1))
  44. !
  45. positionStart
  46. ^ self position
  47. !
  48. shouldBeAliased
  49. ^ shouldBeAliased ifNil: [ false ]
  50. !
  51. shouldBeAliased: aBoolean
  52. shouldBeAliased := aBoolean
  53. !
  54. size
  55. ^ self source size
  56. !
  57. source
  58. ^ source ifNil: [ '' ]
  59. !
  60. source: aString
  61. source := aString
  62. ! !
  63. !ASTNode methodsFor: 'building'!
  64. withTail: aCollection
  65. ^ aCollection inject: self into: [
  66. :receiver :send | SendNode new
  67. position: send position;
  68. source: send source;
  69. receiver: receiver;
  70. selector: send selector;
  71. arguments: send arguments;
  72. yourself ]
  73. ! !
  74. !ASTNode methodsFor: 'testing'!
  75. inPosition: aPoint
  76. ^ (self positionStart <= aPoint and: [
  77. self positionEnd >= aPoint ])
  78. !
  79. isAssignmentNode
  80. ^ false
  81. !
  82. isBlockNode
  83. ^ false
  84. !
  85. isBlockSequenceNode
  86. ^ false
  87. !
  88. isCascadeNode
  89. ^ false
  90. !
  91. isImmutable
  92. ^ false
  93. !
  94. isJSStatementNode
  95. ^ false
  96. !
  97. isNavigationNode
  98. "Answer true if the node can be navigated to"
  99. ^ false
  100. !
  101. isReturnNode
  102. ^ false
  103. !
  104. isSendNode
  105. ^ false
  106. !
  107. isSequenceNode
  108. ^ false
  109. !
  110. isSuperKeyword
  111. ^ false
  112. !
  113. isValueNode
  114. ^ false
  115. !
  116. isVariableNode
  117. ^ false
  118. !
  119. requiresSmalltalkContext
  120. "Answer true if the receiver requires a smalltalk context.
  121. Only send nodes require a context.
  122. If no node requires a context, the method will be compiled without one.
  123. See `IRJSTranslator` and `JSStream` for context creation"
  124. ^ (self dagChildren
  125. detect: [ :each | each requiresSmalltalkContext ]
  126. ifNone: [ nil ]) notNil
  127. ! !
  128. ASTNode subclass: #AssignmentNode
  129. instanceVariableNames: 'left right'
  130. package: 'Compiler-AST'!
  131. !AssignmentNode commentStamp!
  132. I represent an assignment node.!
  133. !AssignmentNode methodsFor: 'accessing'!
  134. dagChildren
  135. ^ Array with: self left with: self right
  136. !
  137. left
  138. ^ left
  139. !
  140. left: aNode
  141. left := aNode
  142. !
  143. right
  144. ^ right
  145. !
  146. right: aNode
  147. right := aNode
  148. ! !
  149. !AssignmentNode methodsFor: 'testing'!
  150. isAssignmentNode
  151. ^ true
  152. ! !
  153. !AssignmentNode methodsFor: 'visiting'!
  154. acceptDagVisitor: aVisitor
  155. ^ aVisitor visitAssignmentNode: self
  156. ! !
  157. ASTNode subclass: #BlockNode
  158. instanceVariableNames: 'parameters scope'
  159. package: 'Compiler-AST'!
  160. !BlockNode commentStamp!
  161. I represent an block closure node.!
  162. !BlockNode methodsFor: 'accessing'!
  163. parameters
  164. ^ parameters ifNil: [ parameters := Array new ]
  165. !
  166. parameters: aCollection
  167. parameters := aCollection
  168. !
  169. scope
  170. ^ scope
  171. !
  172. scope: aLexicalScope
  173. scope := aLexicalScope
  174. ! !
  175. !BlockNode methodsFor: 'testing'!
  176. isBlockNode
  177. ^ true
  178. ! !
  179. !BlockNode methodsFor: 'visiting'!
  180. acceptDagVisitor: aVisitor
  181. ^ aVisitor visitBlockNode: self
  182. ! !
  183. ASTNode subclass: #CascadeNode
  184. instanceVariableNames: 'receiver'
  185. package: 'Compiler-AST'!
  186. !CascadeNode commentStamp!
  187. I represent an cascade node.!
  188. !CascadeNode methodsFor: 'accessing'!
  189. receiver
  190. ^ receiver
  191. !
  192. receiver: aNode
  193. receiver := aNode
  194. ! !
  195. !CascadeNode methodsFor: 'testing'!
  196. isCascadeNode
  197. ^ true
  198. ! !
  199. !CascadeNode methodsFor: 'visiting'!
  200. acceptDagVisitor: aVisitor
  201. ^ aVisitor visitCascadeNode: self
  202. ! !
  203. ASTNode subclass: #DynamicArrayNode
  204. instanceVariableNames: ''
  205. package: 'Compiler-AST'!
  206. !DynamicArrayNode commentStamp!
  207. I represent an dynamic array node.!
  208. !DynamicArrayNode methodsFor: 'visiting'!
  209. acceptDagVisitor: aVisitor
  210. ^ aVisitor visitDynamicArrayNode: self
  211. ! !
  212. ASTNode subclass: #DynamicDictionaryNode
  213. instanceVariableNames: ''
  214. package: 'Compiler-AST'!
  215. !DynamicDictionaryNode commentStamp!
  216. I represent an dynamic dictionary node.!
  217. !DynamicDictionaryNode methodsFor: 'visiting'!
  218. acceptDagVisitor: aVisitor
  219. ^ aVisitor visitDynamicDictionaryNode: self
  220. ! !
  221. ASTNode subclass: #JSStatementNode
  222. instanceVariableNames: ''
  223. package: 'Compiler-AST'!
  224. !JSStatementNode commentStamp!
  225. I represent an JavaScript statement node.!
  226. !JSStatementNode methodsFor: 'testing'!
  227. isJSStatementNode
  228. ^ true
  229. !
  230. requiresSmalltalkContext
  231. ^ true
  232. ! !
  233. !JSStatementNode methodsFor: 'visiting'!
  234. acceptDagVisitor: aVisitor
  235. ^ aVisitor visitJSStatementNode: self
  236. ! !
  237. ASTNode subclass: #MethodNode
  238. instanceVariableNames: 'selector arguments source scope classReferences sendIndexes'
  239. package: 'Compiler-AST'!
  240. !MethodNode commentStamp!
  241. I represent an method node.
  242. A method node must be the root and only method node of a valid AST.!
  243. !MethodNode methodsFor: 'accessing'!
  244. arguments
  245. ^ arguments ifNil: [ #() ]
  246. !
  247. arguments: aCollection
  248. arguments := aCollection
  249. !
  250. classReferences
  251. ^ classReferences
  252. !
  253. classReferences: aCollection
  254. classReferences := aCollection
  255. !
  256. messageSends
  257. ^ self sendIndexes keys
  258. !
  259. method
  260. ^ self
  261. !
  262. scope
  263. ^ scope
  264. !
  265. scope: aMethodScope
  266. scope := aMethodScope
  267. !
  268. selector
  269. ^ selector
  270. !
  271. selector: aString
  272. selector := aString
  273. !
  274. sendIndexes
  275. ^ sendIndexes
  276. !
  277. sendIndexes: aDictionary
  278. sendIndexes := aDictionary
  279. !
  280. sequenceNode
  281. self dagChildren do: [ :each |
  282. each isSequenceNode ifTrue: [ ^ each ] ].
  283. ^ nil
  284. !
  285. source
  286. ^ source
  287. !
  288. source: aString
  289. source := aString
  290. ! !
  291. !MethodNode methodsFor: 'visiting'!
  292. acceptDagVisitor: aVisitor
  293. ^ aVisitor visitMethodNode: self
  294. ! !
  295. ASTNode subclass: #ReturnNode
  296. instanceVariableNames: 'scope'
  297. package: 'Compiler-AST'!
  298. !ReturnNode commentStamp!
  299. I represent an return node. At the AST level, there is not difference between a local return or non-local return.!
  300. !ReturnNode methodsFor: 'accessing'!
  301. scope
  302. ^ scope
  303. !
  304. scope: aLexicalScope
  305. scope := aLexicalScope
  306. ! !
  307. !ReturnNode methodsFor: 'testing'!
  308. isReturnNode
  309. ^ true
  310. !
  311. nonLocalReturn
  312. ^ self scope isMethodScope not
  313. ! !
  314. !ReturnNode methodsFor: 'visiting'!
  315. acceptDagVisitor: aVisitor
  316. ^ aVisitor visitReturnNode: self
  317. ! !
  318. ASTNode subclass: #SendNode
  319. instanceVariableNames: 'selector arguments receiver index shouldBeInlined'
  320. package: 'Compiler-AST'!
  321. !SendNode commentStamp!
  322. I represent an message send node.!
  323. !SendNode methodsFor: 'accessing'!
  324. arguments
  325. ^ arguments ifNil: [ arguments := #() ]
  326. !
  327. arguments: aCollection
  328. arguments := aCollection
  329. !
  330. dagChildren
  331. self receiver ifNil: [ ^ self arguments copy ].
  332. ^ (Array with: self receiver)
  333. addAll: self arguments;
  334. yourself
  335. !
  336. index
  337. ^ index
  338. !
  339. index: anInteger
  340. index := anInteger
  341. !
  342. navigationLink
  343. ^ self selector
  344. !
  345. receiver
  346. ^ receiver
  347. !
  348. receiver: aNode
  349. receiver := aNode
  350. !
  351. selector
  352. ^ selector
  353. !
  354. selector: aString
  355. selector := aString
  356. !
  357. shouldBeInlined
  358. ^ shouldBeInlined ifNil: [ false ]
  359. !
  360. shouldBeInlined: aBoolean
  361. shouldBeInlined := aBoolean
  362. !
  363. superSend
  364. ^ self receiver notNil and: [ self receiver isSuperKeyword ]
  365. ! !
  366. !SendNode methodsFor: 'testing'!
  367. isNavigationNode
  368. ^ true
  369. !
  370. isSendNode
  371. ^ true
  372. !
  373. requiresSmalltalkContext
  374. ^ true
  375. ! !
  376. !SendNode methodsFor: 'visiting'!
  377. acceptDagVisitor: aVisitor
  378. ^ aVisitor visitSendNode: self
  379. ! !
  380. ASTNode subclass: #SequenceNode
  381. instanceVariableNames: 'temps scope'
  382. package: 'Compiler-AST'!
  383. !SequenceNode commentStamp!
  384. I represent an sequence node. A sequence represent a set of instructions inside the same scope (the method scope or a block scope).!
  385. !SequenceNode methodsFor: 'accessing'!
  386. scope
  387. ^ scope
  388. !
  389. scope: aLexicalScope
  390. scope := aLexicalScope
  391. !
  392. temps
  393. ^ temps ifNil: [ #() ]
  394. !
  395. temps: aCollection
  396. temps := aCollection
  397. ! !
  398. !SequenceNode methodsFor: 'building'!
  399. asBlockSequenceNode
  400. ^ BlockSequenceNode new
  401. position: self position;
  402. source: self source;
  403. dagChildren: self dagChildren;
  404. temps: self temps;
  405. yourself
  406. ! !
  407. !SequenceNode methodsFor: 'testing'!
  408. isSequenceNode
  409. ^ true
  410. ! !
  411. !SequenceNode methodsFor: 'visiting'!
  412. acceptDagVisitor: aVisitor
  413. ^ aVisitor visitSequenceNode: self
  414. ! !
  415. SequenceNode subclass: #BlockSequenceNode
  416. instanceVariableNames: ''
  417. package: 'Compiler-AST'!
  418. !BlockSequenceNode commentStamp!
  419. I represent an special sequence node for block scopes.!
  420. !BlockSequenceNode methodsFor: 'testing'!
  421. isBlockSequenceNode
  422. ^ true
  423. ! !
  424. !BlockSequenceNode methodsFor: 'visiting'!
  425. acceptDagVisitor: aVisitor
  426. ^ aVisitor visitBlockSequenceNode: self
  427. ! !
  428. ASTNode subclass: #ValueNode
  429. instanceVariableNames: 'value'
  430. package: 'Compiler-AST'!
  431. !ValueNode commentStamp!
  432. I represent a value node.!
  433. !ValueNode methodsFor: 'accessing'!
  434. value
  435. ^ value
  436. !
  437. value: anObject
  438. value := anObject
  439. ! !
  440. !ValueNode methodsFor: 'testing'!
  441. isImmutable
  442. ^ self value isImmutable
  443. !
  444. isValueNode
  445. ^ true
  446. ! !
  447. !ValueNode methodsFor: 'visiting'!
  448. acceptDagVisitor: aVisitor
  449. ^ aVisitor visitValueNode: self
  450. ! !
  451. ValueNode subclass: #VariableNode
  452. instanceVariableNames: 'assigned binding'
  453. package: 'Compiler-AST'!
  454. !VariableNode commentStamp!
  455. I represent an variable node.!
  456. !VariableNode methodsFor: 'accessing'!
  457. alias
  458. ^ self binding alias
  459. !
  460. assigned
  461. ^ assigned ifNil: [ false ]
  462. !
  463. assigned: aBoolean
  464. assigned := aBoolean
  465. !
  466. beAssigned
  467. self binding validateAssignment.
  468. assigned := true
  469. !
  470. binding
  471. ^ binding
  472. !
  473. binding: aScopeVar
  474. binding := aScopeVar
  475. !
  476. navigationLink
  477. ^ self value
  478. ! !
  479. !VariableNode methodsFor: 'testing'!
  480. isArgument
  481. ^ self binding isArgVar
  482. !
  483. isImmutable
  484. ^ self binding isImmutable
  485. !
  486. isNavigationNode
  487. ^ true
  488. !
  489. isSuperKeyword
  490. ^ self value = 'super'
  491. !
  492. isVariableNode
  493. ^ true
  494. ! !
  495. !VariableNode methodsFor: 'visiting'!
  496. acceptDagVisitor: aVisitor
  497. ^ aVisitor visitVariableNode: self
  498. ! !
  499. PathDagVisitor subclass: #ParentFakingPathDagVisitor
  500. instanceVariableNames: 'setParentSelector'
  501. package: 'Compiler-AST'!
  502. !ParentFakingPathDagVisitor commentStamp!
  503. I am base class of `DagNode` visitor.
  504. I hold the path of ancestors up to actual node
  505. in `self path`.!
  506. !ParentFakingPathDagVisitor methodsFor: 'visiting'!
  507. visit: aNode
  508. self path ifNotEmpty: [ :p | aNode parent: p last ].
  509. ^ super visit: aNode
  510. ! !
  511. ParentFakingPathDagVisitor subclass: #NodeVisitor
  512. instanceVariableNames: ''
  513. package: 'Compiler-AST'!
  514. !NodeVisitor commentStamp!
  515. I am the abstract super class of all AST node visitors.!
  516. !NodeVisitor methodsFor: 'visiting'!
  517. visitAssignmentNode: aNode
  518. ^ self visitDagNode: aNode
  519. !
  520. visitBlockNode: aNode
  521. ^ self visitDagNode: aNode
  522. !
  523. visitBlockSequenceNode: aNode
  524. ^ self visitSequenceNode: aNode
  525. !
  526. visitCascadeNode: aNode
  527. ^ self visitDagNode: aNode
  528. !
  529. visitDagNode: aNode
  530. ^ self visitDagNodeVariantSimple: aNode
  531. !
  532. visitDynamicArrayNode: aNode
  533. ^ self visitDagNode: aNode
  534. !
  535. visitDynamicDictionaryNode: aNode
  536. ^ self visitDagNode: aNode
  537. !
  538. visitJSStatementNode: aNode
  539. ^ self visitDagNode: aNode
  540. !
  541. visitMethodNode: aNode
  542. ^ self visitDagNode: aNode
  543. !
  544. visitReturnNode: aNode
  545. ^ self visitDagNode: aNode
  546. !
  547. visitSendNode: aNode
  548. ^ self visitDagNode: aNode
  549. !
  550. visitSequenceNode: aNode
  551. ^ self visitDagNode: aNode
  552. !
  553. visitValueNode: aNode
  554. ^ self visitDagNode: aNode
  555. !
  556. visitVariableNode: aNode
  557. ^ self visitDagNode: aNode
  558. ! !
  559. !CompiledMethod methodsFor: '*Compiler-AST'!
  560. ast
  561. self source ifEmpty: [ self error: 'Method source is empty' ].
  562. ^ Smalltalk parse: self source
  563. ! !