FIND behaviour now standard. Closes #5.
[forth.jl.git] / src / forth.jl
1 module forth
2
3 # VM mem size
4 size_mem = 1000000 # 1 mega-int
5
6 # Buffer sizes
7 size_RS = 1000   # Return stack size
8 size_PS = 1000   # Parameter stack size
9 size_TIB = 1000  # Terminal input buffer size
10
11 # The mem array constitutes the memory of the VM. It has the following geography:
12 #
13 # mem = +-----------------------+
14 #       | Built-in Variables    |
15 #       +-----------------------+
16 #       | Return Stack          |
17 #       +-----------------------+
18 #       | Parameter Stack       |
19 #       +-----------------------+
20 #       | Terminal Input Buffer |
21 #       +-----------------------+
22 #       | Dictionary            |
23 #       +-----------------------+
24 #
25 # Note that all words (user-defined, primitive, variables, etc) are included in
26 # the dictionary.
27 #
28 # Simple linear addressing is used with one exception: references to primitive code
29 # blocks, which are represented as anonymous functions, appear as negative indicies
30 # into the primitives array which contains these functions.
31
32 mem = Array{Int64,1}(size_mem)
33 primitives = Array{Function,1}()
34 primNames = Array{ASCIIString,1}()
35
36 # Built-in variables
37
38 nextVarAddr = 1
39 H = nextVarAddr; nextVarAddr += 1
40 LATEST = nextVarAddr; nextVarAddr += 1
41
42 RSP0 = nextVarAddr                  # bottom of RS
43 PSP0 = RSP0 + size_RS               # bottom of PS
44 TIB = PSP0 + size_PS                # address of terminal input buffer
45 mem[H] = TIB + size_TIB          # location of bottom of dictionary
46 mem[LATEST] = 0                     # no previous definition
47
48 DICT = mem[H] # Save bottom of dictionary as constant
49
50 # VM registers
51 type Reg
52     RSP::Int64  # Return stack pointer
53     PSP::Int64  # Parameter/data stack pointer
54     IP::Int64   # Instruction pointer
55     W::Int64    # Working register
56 end
57 reg = Reg(RSP0, PSP0, 0, 0)
58
59 # Stack manipulation functions
60
61 type ParamStackUnderflow <: Exception end
62 type ReturnStackUnderflow <: Exception end
63
64 Base.showerror(io::IO, ex::ParamStackUnderflow) = print(io, "Parameter stack underflow.")
65 Base.showerror(io::IO, ex::ReturnStackUnderflow) = print(io, "Return stack underflow.")
66
67 getRSDepth() = reg.RSP - RSP0
68 getPSDepth() = reg.PSP - PSP0
69
70 function ensurePSDepth(depth::Int64)
71     if getPSDepth()<depth
72         throw(ParamStackUnderflow())
73     end
74 end
75
76 function ensureRSDepth(depth::Int64)
77     if getRSDepth()<depth
78         throw(ReturnStackUnderflow())
79     end
80 end
81
82 function pushRS(val::Int64)
83     mem[reg.RSP+=1] = val
84 end
85
86 function popRS()
87     ensureRSDepth(1)
88
89     val = mem[reg.RSP]
90     reg.RSP -= 1
91     return val
92 end
93
94 function pushPS(val::Int64)
95     mem[reg.PSP += 1] = val
96 end
97
98 function popPS()
99     ensurePSDepth(1)
100
101     val = mem[reg.PSP]
102     reg.PSP -= 1
103     return val
104 end
105
106 # Handy functions for adding/retrieving strings to/from memory.
107
108 getString(addr::Int64, len::Int64) = ASCIIString([Char(c) for c in mem[addr:(addr+len-1)]])
109
110 function putString(str::ASCIIString, addr::Int64)
111     mem[addr:(addr+length(str)-1)] = [Int64(c) for c in str]
112 end
113
114 # Primitive creation and calling functions
115
116 function defPrim(f::Function; name="nameless")
117     push!(primitives, f)
118     push!(primNames, replace(name, "\004", "EOF"))
119
120     return -length(primitives)
121 end
122
123 function callPrim(addr::Int64)
124     if addr >=0 || -addr>length(primitives)
125         error("Attempted to execute non-existent primitive at address $addr.")
126     else
127         primitives[-addr]()
128     end
129 end
130 getPrimName(addr::Int64) = primNames[-addr]
131
132 # Word creation functions
133
134 F_LENMASK = 31
135 F_IMMED = 32
136 F_HIDDEN = 64
137 NFA_MARK = 128
138
139 function createHeader(name::AbstractString, flags::Int64)
140     mem[mem[H]] = mem[LATEST]
141     mem[LATEST] = mem[H]
142     mem[H] += 1
143
144     mem[mem[H]] = length(name) | flags | NFA_MARK; mem[H] += 1
145     putString(name, mem[H]); mem[H] += length(name)
146 end
147
148 function defPrimWord(name::AbstractString, f::Function; flags::Int64=0)
149     createHeader(name, flags)
150
151     codeWordAddr = mem[H]
152     mem[codeWordAddr] = defPrim(f, name=name)
153     mem[H] += 1
154
155     return codeWordAddr
156 end
157
158 function defWord(name::AbstractString, wordAddrs::Array{Int64,1}; flags::Int64=0)
159     createHeader(name, flags)
160
161     addr = mem[H]
162     mem[mem[H]] = DOCOL
163     mem[H] += 1
164
165     for wordAddr in wordAddrs
166         mem[mem[H]] = wordAddr
167         mem[H] += 1
168     end
169
170     return addr
171 end
172
173 # Variable creation functions
174
175 function defExistingVar(name::AbstractString, varAddr::Int64; flags::Int64=0)
176
177     defPrimWord(name, eval(:(() -> begin
178         pushPS($(varAddr))
179         return NEXT
180     end)))
181 end
182
183 function defNewVar(name::AbstractString, initial::Int64; flags::Int64=0)
184     createHeader(name, flags)
185     
186     codeWordAddr = mem[H]
187     varAddr = mem[H] + 1
188
189     mem[mem[H]] = DOVAR; mem[H] += 1
190     mem[mem[H]] = initial; mem[H] += 1
191
192     return varAddr, codeWordAddr
193 end
194
195 function defConst(name::AbstractString, val::Int64; flags::Int64=0)
196     createHeader(name, flags)
197
198     codeWordAddr = mem[H]
199
200     mem[mem[H]] = DOCON; mem[H] += 1
201     mem[mem[H]] = val; mem[H] += 1
202
203     return codeWordAddr
204 end
205
206 # Threading Primitives (inner interpreter)
207
208 NEXT = defPrim(() -> begin
209     reg.W = mem[reg.IP]
210     reg.IP += 1
211     return mem[reg.W]
212 end, name="NEXT")
213
214 DOCOL = defPrim(() -> begin
215     pushRS(reg.IP)
216     reg.IP = reg.W + 1
217     return NEXT
218 end, name="DOCOL")
219
220 DOVAR = defPrim(() -> begin
221     pushPS(reg.W + 1)
222     return NEXT
223 end, name="DOVAR")
224
225 DOCON = defPrim(() -> begin
226     pushPS(mem[reg.W + 1])
227     return NEXT
228 end, name="DOVAR")
229
230 EXIT = defPrimWord("EXIT", () -> begin
231     reg.IP = popRS()
232     return NEXT
233 end)
234
235 # Dictionary entries for core built-in variables, constants
236
237 H_CFA = defExistingVar("H", H)
238 LATEST_CFA = defExistingVar("LATEST", LATEST)
239
240 PSP0_CFA = defConst("PSP0", PSP0)
241 RSP0_CFA = defConst("RSP0", RSP0)
242
243 defConst("DOCOL", DOCOL)
244 defConst("DOCON", DOCON)
245 defConst("DOVAR", DOVAR)
246
247 defConst("DICT", DICT)
248 defConst("MEMSIZE", size_mem)
249
250 F_IMMED_CFA = defConst("F_IMMED", F_IMMED)
251 F_HIDDEN_CFA = defConst("F_HIDDEN", F_HIDDEN)
252 F_LENMASK_CFA = defConst("F_LENMASK", F_LENMASK)
253 NFA_MARK_CFA = defConst("NFA_MARK", NFA_MARK)
254
255 # Basic forth primitives
256
257 DROP = defPrimWord("DROP", () -> begin
258     popPS()
259     return NEXT
260 end)
261
262 SWAP = defPrimWord("SWAP", () -> begin
263     a = popPS()
264     b = popPS()
265     pushPS(a)
266     pushPS(b)
267     return NEXT
268 end)
269
270 DUP = defPrimWord("DUP", () -> begin
271     ensurePSDepth(1)
272     pushPS(mem[reg.PSP])
273     return NEXT
274 end)
275
276 OVER = defPrimWord("OVER", () -> begin
277     ensurePSDepth(2)
278     pushPS(mem[reg.PSP-1])
279     return NEXT
280 end)
281
282 ROT = defPrimWord("ROT", () -> begin
283     a = popPS()
284     b = popPS()
285     c = popPS()
286     pushPS(b)
287     pushPS(a)
288     pushPS(c)
289     return NEXT
290 end)
291
292 NROT = defPrimWord("-ROT", () -> begin
293     a = popPS()
294     b = popPS()
295     c = popPS()
296     pushPS(a)
297     pushPS(c)
298     pushPS(b)
299     return NEXT
300 end)
301
302
303 TWODROP = defPrimWord("2DROP", () -> begin
304     popPS()
305     popPS()
306     return NEXT
307 end)
308
309 TWODUP = defPrimWord("2DUP", () -> begin
310     ensurePSDepth(2)
311     a = mem[reg.PSP-1]
312     b = mem[reg.PSP]
313     pushPS(a)
314     pushPS(b)
315     return NEXT
316 end)
317
318 TWOSWAP = defPrimWord("2SWAP", () -> begin
319     a = popPS()
320     b = popPS()
321     c = popPS()
322     d = popPS()
323     pushPS(b)
324     pushPS(a)
325     pushPS(d)
326     pushPS(c)
327     return NEXT
328 end)
329
330 TWOOVER = defPrimWord("2OVER", () -> begin
331     ensurePSDepth(4)
332     a = mem[reg.PSP-3]
333     b = mem[reg.PSP-2]
334     pushPS(a)
335     pushPS(b)
336     return NEXT
337 end)
338
339 QDUP = defPrimWord("?DUP", () -> begin
340     ensurePSDepth(1)
341     val = mem[reg.PSP]
342     if val != 0
343         pushPS(val)
344     end
345     return NEXT
346 end)
347
348 INCR = defPrimWord("1+", () -> begin
349     ensurePSDepth(1)
350     mem[reg.PSP] += 1
351     return NEXT
352 end)
353
354 DECR = defPrimWord("1-", () -> begin
355     ensurePSDepth(1)
356     mem[reg.PSP] -= 1
357     return NEXT
358 end)
359
360 INCR2 = defPrimWord("2+", () -> begin
361     ensurePSDepth(1)
362     mem[reg.PSP] += 2
363     return NEXT
364 end)
365
366 DECR2 = defPrimWord("2-", () -> begin
367     ensurePSDepth(1)
368     mem[reg.PSP] -= 2
369     return NEXT
370 end)
371
372 ADD = defPrimWord("+", () -> begin
373     b = popPS()
374     a = popPS()
375     pushPS(a+b)
376     return NEXT
377 end)
378
379 SUB = defPrimWord("-", () -> begin
380     b = popPS()
381     a = popPS()
382     pushPS(a-b)
383     return NEXT
384 end)
385
386 MUL = defPrimWord("*", () -> begin
387     b = popPS()
388     a = popPS()
389     pushPS(a*b)
390     return NEXT
391 end)
392
393 DIVMOD = defPrimWord("/MOD", () -> begin
394     b = popPS()
395     a = popPS()
396     q,r = divrem(a,b)
397     pushPS(r)
398     pushPS(q)
399     return NEXT
400 end)
401
402 TWOMUL = defPrimWord("2*", () -> begin
403     pushPS(popPS() << 1)
404     return NEXT
405 end)
406
407 TWODIV = defPrimWord("2/", () -> begin
408     pushPS(popPS() >> 1)
409     return NEXT
410 end)
411
412 EQ = defPrimWord("=", () -> begin
413     b = popPS()
414     a = popPS()
415     pushPS(a==b ? -1 : 0)
416     return NEXT
417 end)
418
419 NE = defPrimWord("<>", () -> begin
420     b = popPS()
421     a = popPS()
422     pushPS(a!=b ? -1 : 0)
423     return NEXT
424 end)
425
426 LT = defPrimWord("<", () -> begin
427     b = popPS()
428     a = popPS()
429     pushPS(a<b ? -1 : 0)
430     return NEXT
431 end)
432
433 GT = defPrimWord(">", () -> begin
434     b = popPS()
435     a = popPS()
436     pushPS(a>b ? -1 : 0)
437     return NEXT
438 end)
439
440 LE = defPrimWord("<=", () -> begin
441     b = popPS()
442     a = popPS()
443     pushPS(a<=b ? -1 : 0)
444     return NEXT
445 end)
446
447 GE = defPrimWord(">=", () -> begin
448     b = popPS()
449     a = popPS()
450     pushPS(a>=b ? -1 : 0)
451     return NEXT
452 end)
453
454 ZE = defPrimWord("0=", () -> begin
455     pushPS(popPS() == 0 ? -1 : 0)
456     return NEXT
457 end)
458
459 ZNE = defPrimWord("0<>", () -> begin
460     pushPS(popPS() != 0 ? -1 : 0)
461     return NEXT
462 end)
463
464 ZLT = defPrimWord("0<", () -> begin
465     pushPS(popPS() < 0 ? -1 : 0)
466     return NEXT
467 end)
468
469 ZGT = defPrimWord("0>", () -> begin
470     pushPS(popPS() > 0 ? -1 : 0)
471     return NEXT
472 end)
473
474 ZLE = defPrimWord("0<=", () -> begin
475     pushPS(popPS() <= 0 ? -1 : 0)
476     return NEXT
477 end)
478
479 ZGE = defPrimWord("0>=", () -> begin
480     pushPS(popPS() >= 0 ? -1 : 0)
481     return NEXT
482 end)
483
484 AND = defPrimWord("AND", () -> begin
485     b = popPS()
486     a = popPS()
487     pushPS(a & b)
488     return NEXT
489 end)
490
491 OR = defPrimWord("OR", () -> begin
492     b = popPS()
493     a = popPS()
494     pushPS(a | b)
495     return NEXT
496 end)
497
498 XOR = defPrimWord("XOR", () -> begin
499     b = popPS()
500     a = popPS()
501     pushPS(a $ b)
502     return NEXT
503 end)
504
505 INVERT = defPrimWord("INVERT", () -> begin
506     pushPS(~popPS())
507     return NEXT
508 end)
509
510 # Literals
511
512 LIT = defPrimWord("LIT", () -> begin
513     pushPS(mem[reg.IP])
514     reg.IP += 1
515     return NEXT
516 end)
517
518 # Memory primitives
519
520 STORE = defPrimWord("!", () -> begin
521     addr = popPS()
522     dat = popPS()
523     mem[addr] = dat
524     return NEXT
525 end)
526
527 FETCH = defPrimWord("@", () -> begin
528     addr = popPS()
529     pushPS(mem[addr])
530     return NEXT
531 end)
532
533 ADDSTORE = defPrimWord("+!", () -> begin
534     addr = popPS()
535     toAdd = popPS()
536     mem[addr] += toAdd
537     return NEXT
538 end)
539
540 SUBSTORE = defPrimWord("-!", () -> begin
541     addr = popPS()
542     toSub = popPS()
543     mem[addr] -= toSub
544     return NEXT
545 end)
546
547
548 # Return Stack
549
550 TOR = defPrimWord(">R", () -> begin
551     pushRS(popPS())
552     return NEXT
553 end)
554
555 FROMR = defPrimWord("R>", () -> begin
556     pushPS(popRS())
557     return NEXT
558 end)
559
560 RFETCH = defPrimWord("R@", () -> begin
561     pushPS(mem[reg.RSP])
562     return NEXT
563 end)
564
565 RSPFETCH = defPrimWord("RSP@", () -> begin
566     pushPS(reg.RSP)
567     return NEXT
568 end)
569
570 RSPSTORE = defPrimWord("RSP!", () -> begin
571     reg.RSP = popPS()
572     return NEXT
573 end)
574
575 RDROP = defPrimWord("RDROP", () -> begin
576     popRS()
577     return NEXT
578 end)
579
580 # Parameter Stack
581
582 PSPFETCH = defPrimWord("PSP@", () -> begin
583     pushPS(reg.PSP)
584     return NEXT
585 end)
586
587 PSPSTORE = defPrimWord("PSP!", () -> begin
588     reg.PSP = popPS()
589     return NEXT
590 end)
591
592 # Working Register
593
594 WFETCH = defPrimWord("W@", () -> begin
595     pushPS(reg.W)
596     return NEXT
597 end)
598
599 WSTORE = defPrimWord("W!", () -> begin
600     reg.W = popPS()
601     return NEXT
602 end)
603
604 # I/O
605
606 sources = Array{Any,1}()
607 currentSource() = sources[length(sources)]
608
609 EOF = defPrimWord("\x04", () -> begin
610     close(pop!(sources))
611     if !isempty(sources)
612         return NEXT
613     else
614         return 0
615     end
616 end)
617
618 EMIT = defPrimWord("EMIT", () -> begin
619     print(Char(popPS()))
620     return NEXT
621 end)
622
623 SPAN, SPAN_CFA = defNewVar("SPAN", 0)
624 EXPECT = defPrimWord("EXPECT", () -> begin
625     maxLen = popPS()
626     addr = popPS()
627
628     if !eof(currentSource())
629         line = chomp(readline(currentSource()))
630         mem[SPAN] = min(length(line), maxLen)
631         putString(line[1:mem[SPAN]], addr)
632     else
633         mem[SPAN] = 1
634         mem[addr] = 4 # eof
635     end
636
637     return NEXT
638 end)
639
640 BASE, BASE_CFA = defNewVar("BASE", 10)
641 NUMBER = defPrimWord("NUMBER", () -> begin
642     wordAddr = popPS()+1
643     wordLen = mem[wordAddr-1]
644
645     s = getString(wordAddr, wordLen)
646
647     pushPS(parse(Int64, s, mem[BASE]))
648
649     return NEXT
650 end)
651
652 # Dictionary searches
653
654 TOCFA = defPrimWord(">CFA", () -> begin
655
656     addr = popPS()
657     lenAndFlags = mem[addr+1]
658     len = lenAndFlags & F_LENMASK
659
660     pushPS(addr + 2 + len)
661
662     return NEXT
663 end)
664
665 TOBODY = defWord(">BODY", [INCR, EXIT])
666
667 FIND = defPrimWord("FIND", () -> begin
668
669     countedAddr = popPS()
670     wordAddr = countedAddr + 1
671     wordLen = mem[countedAddr]
672     word = lowercase(getString(wordAddr, wordLen))
673
674     latest = LATEST
675     lenAndFlags = 0
676     
677     i = 0
678     while (latest = mem[latest]) > 0
679         lenAndFlags = mem[latest+1]
680         len = lenAndFlags & F_LENMASK
681         hidden = (lenAndFlags & F_HIDDEN) == F_HIDDEN
682
683         if hidden || len != wordLen
684             continue
685         end
686         
687         thisAddr = latest+2
688         thisWord = lowercase(getString(thisAddr, len))
689
690         if lowercase(thisWord) == lowercase(word)
691             break
692         end
693     end
694
695     if latest > 0
696         pushPS(latest)
697         callPrim(mem[TOCFA])
698         if (lenAndFlags & F_IMMED) == F_IMMED
699             pushPS(1)
700         else
701             pushPS(-1)
702         end
703     else
704         pushPS(countedAddr)
705         pushPS(0)
706     end
707
708     return NEXT
709 end)
710
711
712 # Branching
713
714 BRANCH = defPrimWord("BRANCH", () -> begin
715     reg.IP += mem[reg.IP]
716     return NEXT
717 end)
718
719 ZBRANCH = defPrimWord("0BRANCH", () -> begin
720     if (popPS() == 0)
721         reg.IP += mem[reg.IP]
722     else
723         reg.IP += 1
724     end
725
726     return NEXT
727 end)
728
729 # Strings
730
731 LITSTRING = defPrimWord("LITSTRING", () -> begin
732     len = mem[reg.IP]
733     reg.IP += 1
734     pushPS(reg.IP)
735     pushPS(len)
736     reg.IP += len
737
738     return NEXT
739 end)
740
741 TYPE = defPrimWord("TYPE", () -> begin
742     len = popPS()
743     addr = popPS()
744     str = getString(addr, len)
745     print(str)
746     return NEXT
747 end)
748
749 # Outer interpreter
750
751 TRACE = defPrimWord("TRACE", () -> begin
752     println("reg.W: $(reg.W) reg.IP: $(reg.IP)")
753     print("PS: "); printPS()
754     print("RS: "); printRS()
755     print("[paused]")
756     readline()
757
758     return NEXT
759 end)
760
761 COMMA = defPrimWord(",", () -> begin
762     mem[mem[H]] = popPS()
763     mem[H] += 1
764
765     return NEXT
766 end)
767
768 BTICK = defWord("[']",
769     [FROMR, DUP, INCR, TOR, FETCH, EXIT])
770
771 EXECUTE = defPrimWord("EXECUTE", () -> begin
772     reg.W = popPS()
773     return mem[reg.W]
774 end)
775
776 TIB_CFA = defConst("TIB", TIB)
777 NUMTIB, NUMTIB_CFA = defNewVar("#TIB", 0)
778 TOIN, TOIN_CFA = defNewVar(">IN", 0)
779
780 QUERY = defWord("QUERY",
781     [TIB_CFA, LIT, 160, EXPECT,
782     SPAN_CFA, FETCH, NUMTIB_CFA, STORE,
783     LIT, 0, TOIN_CFA, STORE,
784     EXIT])
785
786 WORD = defPrimWord("WORD", () -> begin
787     delim = popPS()
788
789     # Chew up initial occurrences of delim
790     while (mem[TOIN]<mem[NUMTIB] && mem[TIB+mem[TOIN]] == delim)
791         mem[TOIN] += 1
792     end
793
794     countAddr = mem[H]
795     addr = mem[H]+1
796
797     # Start reading in word
798     count = 0
799     while (mem[TOIN]<mem[NUMTIB])
800         mem[addr] = mem[TIB+mem[TOIN]]
801         mem[TOIN] += 1
802
803         if (mem[addr] == delim)
804             break
805         end
806
807         count += 1
808         addr += 1
809     end
810
811     # Record count
812     mem[countAddr] = count
813     pushPS(countAddr)
814
815     return NEXT
816 end)
817
818 PARSE = defPrimWord("PARSE", () -> begin
819     delim = popPS()
820
821     # Chew up initial occurrences of delim
822     addr = mem[H]
823
824     # Start reading input stream
825     count = 0
826     while (mem[TOIN]<mem[NUMTIB])
827         mem[addr] = mem[TIB+mem[TOIN]]
828         mem[TOIN] += 1
829
830         if (mem[addr] == delim)
831             break
832         end
833
834         count += 1
835         addr += 1
836     end
837
838     pushPS(addr)
839     pushPS(count)
840
841     return NEXT
842 end)
843
844 BYE = defPrimWord("BYE", () -> begin
845     println("Bye!")
846     return 0
847 end)
848
849 STATE, STATE_CFA = defNewVar("STATE", 0)
850
851 INTERPRET = defWord("INTERPRET",
852     [LIT, 32, WORD, # Read next space-delimited word
853
854     DUP, FETCH, ZE, ZBRANCH, 3,
855         DROP, EXIT, # Exit if TIB is exhausted
856
857     STATE_CFA, FETCH, ZBRANCH, 24,
858         # Compiling
859         FIND, QDUP, ZBRANCH, 13,
860
861             # Found word. 
862             LIT, -1, EQ, INVERT, ZBRANCH, 4,
863
864                 # Immediate: Execute!
865                 EXECUTE, BRANCH, -26,
866
867                 # Not immediate: Compile!
868                 COMMA, BRANCH, -29,
869
870             # No word found, parse number
871             NUMBER, BTICK, LIT, COMMA, COMMA, BRANCH, -36,
872         
873        # Interpreting
874         FIND, QDUP, ZBRANCH, 5,
875
876             # Found word. Execute!
877             DROP, EXECUTE, BRANCH, -44,
878
879             # No word found, parse number and leave on stack
880             NUMBER, BRANCH, -47,
881     EXIT]
882 )
883
884 PROMPT = defPrimWord("PROMPT", () -> begin
885     if (mem[STATE] == 0 && currentSource() == STDIN)
886         println(" ok")
887     end
888
889     return NEXT
890 end)
891
892 QUIT = defWord("QUIT",
893     [LIT, 0, STATE_CFA, STORE,
894     LIT, 0, NUMTIB_CFA, STORE,
895     RSP0_CFA, RSPSTORE,
896     QUERY,
897     INTERPRET, PROMPT,
898     BRANCH,-4])
899
900 ABORT = defWord("ABORT",
901     [PSP0_CFA, PSPSTORE, QUIT])
902
903 INCLUDE = defPrimWord("INCLUDE", () -> begin
904     pushPS(32)
905     callPrim(mem[WORD])
906     wordAddr = popPS()+1
907     wordLen = mem[wordAddr-1]
908     word = getString(wordAddr, wordLen)
909
910     push!(sources, open(word, "r"))
911
912     # Clear input buffer
913     mem[NUMTIB] = 0
914
915     return NEXT
916 end)
917
918 # Compilation
919
920 HERE = defWord("HERE",
921     [H_CFA, FETCH, EXIT])
922
923 HEADER = defPrimWord("HEADER", () -> begin
924     wordAddr = popPS()+1
925     wordLen = mem[wordAddr-1]
926     word = getString(wordAddr, wordLen)
927
928     createHeader(word, 0)
929
930     return NEXT
931 end)
932
933 CREATE = defWord("CREATE",
934     [LIT, 32, WORD, HEADER,
935     LIT, DOVAR, COMMA,
936     EXIT])
937
938 DODOES = defPrim(() -> begin
939     pushRS(reg.IP)
940     reg.IP = popPS()
941     pushPS(reg.W + 1)
942     return NEXT
943 end, name="DODOES")
944
945 DOES_HELPER = defPrimWord("(DOES>)", () -> begin
946
947     pushPS(mem[LATEST])
948     callPrim(mem[TOCFA])
949     cfa = popPS()
950
951     runtimeAddr = popPS()
952
953     mem[cfa] = defPrim(eval(:(() -> begin
954         pushPS($(runtimeAddr))
955         return DODOES
956     end)), name="doesPrim")
957
958     return NEXT
959 end, flags=F_IMMED)
960
961 DOES = defWord("DOES>",
962     [BTICK, LIT, COMMA, HERE, LIT, 3, ADD, COMMA,
963     BTICK, DOES_HELPER, COMMA, BTICK, EXIT, COMMA, EXIT],
964     flags=F_IMMED)
965
966 LBRAC = defPrimWord("[", () -> begin
967     mem[STATE] = 0
968     return NEXT
969 end, flags=F_IMMED)
970
971 RBRAC = defPrimWord("]", () -> begin
972     mem[STATE] = 1
973     return NEXT
974 end, flags=F_IMMED)
975
976 HIDDEN = defPrimWord("HIDDEN", () -> begin
977     lenAndFlagsAddr = mem[LATEST] + 1
978     mem[lenAndFlagsAddr] = mem[lenAndFlagsAddr] $ F_HIDDEN
979     return NEXT
980 end)
981
982 COLON = defWord(":",
983     [LIT, 32, WORD,
984     HEADER,
985     LIT, DOCOL, COMMA,
986     HIDDEN,
987     RBRAC,
988     EXIT])
989
990 SEMICOLON = defWord(";",
991     [LIT, EXIT, COMMA,
992     HIDDEN,
993     LBRAC,
994     EXIT], flags=F_IMMED)
995
996 IMMEDIATE = defPrimWord("IMMEDIATE", () -> begin
997     lenAndFlagsAddr = mem[LATEST] + 1
998     mem[lenAndFlagsAddr] = mem[lenAndFlagsAddr] $ F_IMMED
999     return NEXT
1000 end, flags=F_IMMED)
1001
1002
1003 #### VM loop ####
1004
1005 initialized = false
1006 initFileName = nothing
1007 if isfile("lib.4th")
1008     initFileName = "lib.4th"
1009 elseif isfile(Pkg.dir("forth/src/lib.4th"))
1010     initFileName = Pkg.dir("forth/src/lib.4th")
1011 end
1012
1013 function run(;initialize=true)
1014     # Begin with STDIN as source
1015     push!(sources, STDIN)
1016
1017     global initialized, initFileName
1018     if !initialized && initialize
1019         if initFileName != nothing
1020             print("Including definitions from $initFileName...")
1021             push!(sources, open(initFileName, "r"))
1022             initialized = true
1023         else
1024             println("No library file found. Only primitive words available.")
1025         end
1026     end
1027
1028     # Start with IP pointing to first instruction of outer interpreter
1029     reg.IP = QUIT + 1
1030
1031     # Primitive processing loop.
1032     # Everyting else is simply a consequence of this loop!
1033     jmp = NEXT
1034     while jmp != 0
1035         try
1036             #println("Entering prim $(getPrimName(jmp))")
1037             jmp = callPrim(jmp)
1038
1039         catch ex
1040             showerror(STDOUT, ex)
1041             println()
1042
1043             while !isempty(sources) && currentSource() != STDIN
1044                 close(pop!(sources))
1045             end
1046
1047             # QUIT
1048             reg.IP = ABORT + 1
1049             jmp = NEXT
1050         end
1051     end
1052 end
1053
1054 # Debugging tools
1055
1056 function dump(startAddr::Int64; count::Int64 = 100, cellsPerLine::Int64 = 10)
1057     chars = Array{Char,1}(cellsPerLine)
1058
1059     lineStartAddr = cellsPerLine*div((startAddr-1),cellsPerLine) + 1
1060     endAddr = startAddr + count - 1
1061
1062     q, r = divrem((endAddr-lineStartAddr+1), cellsPerLine)
1063     numLines = q + (r > 0 ? 1 : 0)
1064
1065     i = lineStartAddr
1066     for l in 1:numLines
1067         print(i,":")
1068
1069         for c in 1:cellsPerLine
1070             if i >= startAddr && i <= endAddr
1071                 print("\t",mem[i])
1072                 if mem[i]>=32 && mem[i]<128
1073                     chars[c] = Char(mem[i])
1074                 else
1075                     chars[c] = '.'
1076                 end
1077             else
1078                 print("\t")
1079                 chars[c] = ' '
1080             end
1081
1082             i += 1
1083         end
1084
1085         println("\t", ASCIIString(chars))
1086     end
1087 end
1088
1089 function printPS()
1090     count = reg.PSP - PSP0
1091
1092     if count > 0
1093         print("<$count>")
1094         for i in (PSP0+1):reg.PSP
1095             print(" $(mem[i])")
1096         end
1097         println()
1098     else
1099         println("Parameter stack empty")
1100     end
1101 end
1102
1103 function printRS()
1104     count = reg.RSP - RSP0
1105
1106     if count > 0
1107         print("<$count>")
1108         for i in (RSP0+1):reg.RSP
1109             print(" $(mem[i])")
1110         end
1111         println()
1112     else
1113         println("Return stack empty")
1114     end
1115 end
1116
1117 DUMP = defPrimWord("DUMP", () -> begin
1118     count = popPS()
1119     addr = popPS()
1120
1121     dump(addr, count=count)
1122
1123     return NEXT
1124 end)
1125
1126 end