Added workable KEY.
[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 function raw_mode!(mode::Bool)
624     if ccall(:jl_tty_set_mode, Int32, (Ptr{Void}, Int32), STDIN.handle, mode) != 0
625         throw("FATAL: Terminal unable to enter raw mode.")
626     end
627 end
628
629 KEY = defPrimWord("KEY", () -> begin
630     raw_mode!(true)
631     pushPS(Int(readbytes(STDIN, 1)[1]))
632     raw_mode!(false)
633     return NEXT
634 end)
635
636 SPAN, SPAN_CFA = defNewVar("SPAN", 0)
637 EXPECT = defPrimWord("EXPECT", () -> begin
638     maxLen = popPS()
639     addr = popPS()
640
641     if !eof(currentSource())
642         line = chomp(readline(currentSource()))
643         mem[SPAN] = min(length(line), maxLen)
644         putString(line[1:mem[SPAN]], addr)
645     else
646         mem[SPAN] = 1
647         mem[addr] = 4 # eof
648     end
649
650     return NEXT
651 end)
652
653 BASE, BASE_CFA = defNewVar("BASE", 10)
654 NUMBER = defPrimWord("NUMBER", () -> begin
655     wordAddr = popPS()+1
656     wordLen = mem[wordAddr-1]
657
658     s = getString(wordAddr, wordLen)
659
660     pushPS(parse(Int64, s, mem[BASE]))
661
662     return NEXT
663 end)
664
665 # Dictionary searches
666
667 TOCFA = defPrimWord(">CFA", () -> begin
668
669     addr = popPS()
670     lenAndFlags = mem[addr+1]
671     len = lenAndFlags & F_LENMASK
672
673     pushPS(addr + 2 + len)
674
675     return NEXT
676 end)
677
678 TOBODY = defWord(">BODY", [INCR, EXIT])
679
680 FIND = defPrimWord("FIND", () -> begin
681
682     countedAddr = popPS()
683     wordAddr = countedAddr + 1
684     wordLen = mem[countedAddr]
685     word = lowercase(getString(wordAddr, wordLen))
686
687     latest = LATEST
688     lenAndFlags = 0
689     
690     i = 0
691     while (latest = mem[latest]) > 0
692         lenAndFlags = mem[latest+1]
693         len = lenAndFlags & F_LENMASK
694         hidden = (lenAndFlags & F_HIDDEN) == F_HIDDEN
695
696         if hidden || len != wordLen
697             continue
698         end
699         
700         thisAddr = latest+2
701         thisWord = lowercase(getString(thisAddr, len))
702
703         if lowercase(thisWord) == lowercase(word)
704             break
705         end
706     end
707
708     if latest > 0
709         pushPS(latest)
710         callPrim(mem[TOCFA])
711         if (lenAndFlags & F_IMMED) == F_IMMED
712             pushPS(1)
713         else
714             pushPS(-1)
715         end
716     else
717         pushPS(countedAddr)
718         pushPS(0)
719     end
720
721     return NEXT
722 end)
723
724
725 # Branching
726
727 BRANCH = defPrimWord("BRANCH", () -> begin
728     reg.IP += mem[reg.IP]
729     return NEXT
730 end)
731
732 ZBRANCH = defPrimWord("0BRANCH", () -> begin
733     if (popPS() == 0)
734         reg.IP += mem[reg.IP]
735     else
736         reg.IP += 1
737     end
738
739     return NEXT
740 end)
741
742 # Strings
743
744 LITSTRING = defPrimWord("LITSTRING", () -> begin
745     len = mem[reg.IP]
746     reg.IP += 1
747     pushPS(reg.IP)
748     pushPS(len)
749     reg.IP += len
750
751     return NEXT
752 end)
753
754 TYPE = defPrimWord("TYPE", () -> begin
755     len = popPS()
756     addr = popPS()
757     str = getString(addr, len)
758     print(str)
759     return NEXT
760 end)
761
762 # Outer interpreter
763
764 TRACE = defPrimWord("TRACE", () -> begin
765     println("reg.W: $(reg.W) reg.IP: $(reg.IP)")
766     print("PS: "); printPS()
767     print("RS: "); printRS()
768     print("[paused]")
769     readline()
770
771     return NEXT
772 end)
773
774 COMMA = defPrimWord(",", () -> begin
775     mem[mem[H]] = popPS()
776     mem[H] += 1
777
778     return NEXT
779 end)
780
781 BTICK = defWord("[']",
782     [FROMR, DUP, INCR, TOR, FETCH, EXIT])
783
784 EXECUTE = defPrimWord("EXECUTE", () -> begin
785     reg.W = popPS()
786     return mem[reg.W]
787 end)
788
789 TIB_CFA = defConst("TIB", TIB)
790 NUMTIB, NUMTIB_CFA = defNewVar("#TIB", 0)
791 TOIN, TOIN_CFA = defNewVar(">IN", 0)
792
793 QUERY = defWord("QUERY",
794     [TIB_CFA, LIT, 160, EXPECT,
795     SPAN_CFA, FETCH, NUMTIB_CFA, STORE,
796     LIT, 0, TOIN_CFA, STORE,
797     EXIT])
798
799 WORD = defPrimWord("WORD", () -> begin
800     delim = popPS()
801
802     # Chew up initial occurrences of delim
803     while (mem[TOIN]<mem[NUMTIB] && mem[TIB+mem[TOIN]] == delim)
804         mem[TOIN] += 1
805     end
806
807     countAddr = mem[H]
808     addr = mem[H]+1
809
810     # Start reading in word
811     count = 0
812     while (mem[TOIN]<mem[NUMTIB])
813         mem[addr] = mem[TIB+mem[TOIN]]
814         mem[TOIN] += 1
815
816         if (mem[addr] == delim)
817             break
818         end
819
820         count += 1
821         addr += 1
822     end
823
824     # Record count
825     mem[countAddr] = count
826     pushPS(countAddr)
827
828     return NEXT
829 end)
830
831 PARSE = defPrimWord("PARSE", () -> begin
832     delim = popPS()
833
834     # Chew up initial occurrences of delim
835     addr = mem[H]
836
837     # Start reading input stream
838     count = 0
839     while (mem[TOIN]<mem[NUMTIB])
840         mem[addr] = mem[TIB+mem[TOIN]]
841         mem[TOIN] += 1
842
843         if (mem[addr] == delim)
844             break
845         end
846
847         count += 1
848         addr += 1
849     end
850
851     pushPS(addr)
852     pushPS(count)
853
854     return NEXT
855 end)
856
857 BYE = defPrimWord("BYE", () -> begin
858     println("Bye!")
859     return 0
860 end)
861
862 STATE, STATE_CFA = defNewVar("STATE", 0)
863
864 INTERPRET = defWord("INTERPRET",
865     [LIT, 32, WORD, # Read next space-delimited word
866
867     DUP, FETCH, ZE, ZBRANCH, 3,
868         DROP, EXIT, # Exit if TIB is exhausted
869
870     STATE_CFA, FETCH, ZBRANCH, 24,
871         # Compiling
872         FIND, QDUP, ZBRANCH, 13,
873
874             # Found word. 
875             LIT, -1, EQ, INVERT, ZBRANCH, 4,
876
877                 # Immediate: Execute!
878                 EXECUTE, BRANCH, -26,
879
880                 # Not immediate: Compile!
881                 COMMA, BRANCH, -29,
882
883             # No word found, parse number
884             NUMBER, BTICK, LIT, COMMA, COMMA, BRANCH, -36,
885         
886        # Interpreting
887         FIND, QDUP, ZBRANCH, 5,
888
889             # Found word. Execute!
890             DROP, EXECUTE, BRANCH, -44,
891
892             # No word found, parse number and leave on stack
893             NUMBER, BRANCH, -47,
894     EXIT]
895 )
896
897 PROMPT = defPrimWord("PROMPT", () -> begin
898     if (mem[STATE] == 0 && currentSource() == STDIN)
899         println(" ok")
900     end
901
902     return NEXT
903 end)
904
905 QUIT = defWord("QUIT",
906     [LIT, 0, STATE_CFA, STORE,
907     LIT, 0, NUMTIB_CFA, STORE,
908     RSP0_CFA, RSPSTORE,
909     QUERY,
910     INTERPRET, PROMPT,
911     BRANCH,-4])
912
913 ABORT = defWord("ABORT",
914     [PSP0_CFA, PSPSTORE, QUIT])
915
916 INCLUDE = defPrimWord("INCLUDE", () -> begin
917     pushPS(32)
918     callPrim(mem[WORD])
919     wordAddr = popPS()+1
920     wordLen = mem[wordAddr-1]
921     word = getString(wordAddr, wordLen)
922
923     push!(sources, open(word, "r"))
924
925     # Clear input buffer
926     mem[NUMTIB] = 0
927
928     return NEXT
929 end)
930
931 # Compilation
932
933 HERE = defWord("HERE",
934     [H_CFA, FETCH, EXIT])
935
936 HEADER = defPrimWord("HEADER", () -> begin
937     wordAddr = popPS()+1
938     wordLen = mem[wordAddr-1]
939     word = getString(wordAddr, wordLen)
940
941     createHeader(word, 0)
942
943     return NEXT
944 end)
945
946 CREATE = defWord("CREATE",
947     [LIT, 32, WORD, HEADER,
948     LIT, DOVAR, COMMA,
949     EXIT])
950
951 DODOES = defPrim(() -> begin
952     pushRS(reg.IP)
953     reg.IP = popPS()
954     pushPS(reg.W + 1)
955     return NEXT
956 end, name="DODOES")
957
958 DOES_HELPER = defPrimWord("(DOES>)", () -> begin
959
960     pushPS(mem[LATEST])
961     callPrim(mem[TOCFA])
962     cfa = popPS()
963
964     runtimeAddr = popPS()
965
966     mem[cfa] = defPrim(eval(:(() -> begin
967         pushPS($(runtimeAddr))
968         return DODOES
969     end)), name="doesPrim")
970
971     return NEXT
972 end, flags=F_IMMED)
973
974 DOES = defWord("DOES>",
975     [BTICK, LIT, COMMA, HERE, LIT, 3, ADD, COMMA,
976     BTICK, DOES_HELPER, COMMA, BTICK, EXIT, COMMA, EXIT],
977     flags=F_IMMED)
978
979 LBRAC = defPrimWord("[", () -> begin
980     mem[STATE] = 0
981     return NEXT
982 end, flags=F_IMMED)
983
984 RBRAC = defPrimWord("]", () -> begin
985     mem[STATE] = 1
986     return NEXT
987 end, flags=F_IMMED)
988
989 HIDDEN = defPrimWord("HIDDEN", () -> begin
990     lenAndFlagsAddr = mem[LATEST] + 1
991     mem[lenAndFlagsAddr] = mem[lenAndFlagsAddr] $ F_HIDDEN
992     return NEXT
993 end)
994
995 COLON = defWord(":",
996     [LIT, 32, WORD,
997     HEADER,
998     LIT, DOCOL, COMMA,
999     HIDDEN,
1000     RBRAC,
1001     EXIT])
1002
1003 SEMICOLON = defWord(";",
1004     [LIT, EXIT, COMMA,
1005     HIDDEN,
1006     LBRAC,
1007     EXIT], flags=F_IMMED)
1008
1009 IMMEDIATE = defPrimWord("IMMEDIATE", () -> begin
1010     lenAndFlagsAddr = mem[LATEST] + 1
1011     mem[lenAndFlagsAddr] = mem[lenAndFlagsAddr] $ F_IMMED
1012     return NEXT
1013 end, flags=F_IMMED)
1014
1015
1016 #### VM loop ####
1017
1018 initialized = false
1019 initFileName = nothing
1020 if isfile("lib.4th")
1021     initFileName = "lib.4th"
1022 elseif isfile(Pkg.dir("forth/src/lib.4th"))
1023     initFileName = Pkg.dir("forth/src/lib.4th")
1024 end
1025
1026 function run(;initialize=true)
1027     # Begin with STDIN as source
1028     push!(sources, STDIN)
1029
1030     global initialized, initFileName
1031     if !initialized && initialize
1032         if initFileName != nothing
1033             print("Including definitions from $initFileName...")
1034             push!(sources, open(initFileName, "r"))
1035             initialized = true
1036         else
1037             println("No library file found. Only primitive words available.")
1038         end
1039     end
1040
1041     # Start with IP pointing to first instruction of outer interpreter
1042     reg.IP = QUIT + 1
1043
1044     # Primitive processing loop.
1045     # Everyting else is simply a consequence of this loop!
1046     jmp = NEXT
1047     while jmp != 0
1048         try
1049             #println("Entering prim $(getPrimName(jmp))")
1050             jmp = callPrim(jmp)
1051
1052         catch ex
1053             showerror(STDOUT, ex)
1054             println()
1055
1056             while !isempty(sources) && currentSource() != STDIN
1057                 close(pop!(sources))
1058             end
1059
1060             # QUIT
1061             reg.IP = ABORT + 1
1062             jmp = NEXT
1063         end
1064     end
1065 end
1066
1067 # Debugging tools
1068
1069 function dump(startAddr::Int64; count::Int64 = 100, cellsPerLine::Int64 = 10)
1070     chars = Array{Char,1}(cellsPerLine)
1071
1072     lineStartAddr = cellsPerLine*div((startAddr-1),cellsPerLine) + 1
1073     endAddr = startAddr + count - 1
1074
1075     q, r = divrem((endAddr-lineStartAddr+1), cellsPerLine)
1076     numLines = q + (r > 0 ? 1 : 0)
1077
1078     i = lineStartAddr
1079     for l in 1:numLines
1080         print(i,":")
1081
1082         for c in 1:cellsPerLine
1083             if i >= startAddr && i <= endAddr
1084                 print("\t",mem[i])
1085                 if mem[i]>=32 && mem[i]<128
1086                     chars[c] = Char(mem[i])
1087                 else
1088                     chars[c] = '.'
1089                 end
1090             else
1091                 print("\t")
1092                 chars[c] = ' '
1093             end
1094
1095             i += 1
1096         end
1097
1098         println("\t", ASCIIString(chars))
1099     end
1100 end
1101
1102 function printPS()
1103     count = reg.PSP - PSP0
1104
1105     if count > 0
1106         print("<$count>")
1107         for i in (PSP0+1):reg.PSP
1108             print(" $(mem[i])")
1109         end
1110         println()
1111     else
1112         println("Parameter stack empty")
1113     end
1114 end
1115
1116 function printRS()
1117     count = reg.RSP - RSP0
1118
1119     if count > 0
1120         print("<$count>")
1121         for i in (RSP0+1):reg.RSP
1122             print(" $(mem[i])")
1123         end
1124         println()
1125     else
1126         println("Return stack empty")
1127     end
1128 end
1129
1130 DUMP = defPrimWord("DUMP", () -> begin
1131     count = popPS()
1132     addr = popPS()
1133
1134     dump(addr, count=count)
1135
1136     return NEXT
1137 end)
1138
1139 end