adjust comments
[fleet.git] / src / edu / berkeley / fleet / fpga / Fpga.java
1 package edu.berkeley.fleet.fpga;
2 import edu.berkeley.fleet.fpga.*;
3 import edu.berkeley.fleet.api.*;
4 import edu.berkeley.fleet.two.*;
5 import edu.berkeley.fleet.*;
6 import java.lang.reflect.*;
7 import edu.berkeley.sbp.chr.*;
8 import edu.berkeley.sbp.misc.*;
9 import edu.berkeley.sbp.meta.*;
10 import edu.berkeley.sbp.util.*;
11 import java.util.*;
12 import java.io.*;
13 import edu.berkeley.fleet.two.*;
14 import static edu.berkeley.fleet.two.FleetTwoFleet.*;
15 import static edu.berkeley.fleet.fpga.verilog.Verilog.*;
16
17
18 public class Fpga extends FleetTwoFleet {
19
20     public  Module top;
21     public  FabricElement top_horn;
22     Ship debugShip;
23
24     public LinkedHashMap<String,FpgaShip> ships = new LinkedHashMap<String,FpgaShip>();
25     public Iterator<Ship> iterator() { return (Iterator<Ship>)(Object)ships.values().iterator(); }
26
27     public Ship getShip(String type, int ordinal) {
28         for(Ship s : this)
29             if (s.getType().equals(type))
30                 if (--ordinal < 0)
31                     return s;
32         return null;
33     }
34
35     public static void main(String[] s) throws Exception { 
36         new Fpga(new Module("root")).top.dump(s[0]);
37         PrintWriter pw;
38
39         pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(s[0]+"/timescale.v")));
40         pw.println("`timescale 1ns / 10ps");
41         pw.close();
42
43         pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(s[0]+"/bram14.v")));
44         pw.println("`define BRAM_ADDR_WIDTH 14");
45         pw.println("`define BRAM_DATA_WIDTH `WORDWIDTH");
46         pw.println("`define BRAM_SIZE (1<<(`BRAM_ADDR_WIDTH))");
47         pw.println("`define BRAM_NAME bram14");
48         pw.println("`include \"bram.inc\"");
49         pw.close();
50
51         pw = new PrintWriter(new OutputStreamWriter(new FileOutputStream(s[0]+"/vram.v")));
52         pw.println("`define BRAM_ADDR_WIDTH 19");
53         pw.println("`define BRAM_DATA_WIDTH 3");
54         pw.println("`define BRAM_SIZE (640*480)");
55         pw.println("`define BRAM_NAME vram");
56         pw.println("`include \"bram.inc\"");
57         pw.close();
58     }
59
60     public Module getVerilogModule() { return top; }
61
62     public FleetProcess run(Instruction[] instructions) {
63         try {
64             return new Client(this, "none", instructions);
65         } catch (Exception e) { throw new RuntimeException(e); }
66     }
67
68     public BitVector getDestAddr(Path path) {
69         return ((FpgaPath)path).toBitVector();
70     }
71
72     // Setup //////////////////////////////////////////////////////////////////////////////
73
74     public Ship createShip(String type, String name) throws IOException {
75         ShipDescription sd = new ShipDescription(type, new BufferedReader(new InputStreamReader(new FileInputStream("ships/"+type+".ship"))));
76         FpgaShip ship = new FpgaShip(this, sd);
77         ships.put(name, ship);
78         return ship;
79     }
80
81     public Fpga() throws Exception { this(new Module("root")); }
82     public Fpga(Module top) throws Exception {
83         this.top = top;
84         debugShip = createShip("Debug",     "debug");
85
86         //boolean small = false;
87         boolean small = true;
88
89         createShip("Memory",      "memory1");
90
91         if (small) {
92             for(int i=0; i<2; i++)
93                 createShip("Fifo",           "fifo"+i);
94             for(int i=0; i<2; i++)
95                 createShip("Alu",            "alu"+i);
96             createShip("Counter",        "counter");
97             createShip("CarrySaveAdder", "csa1");
98             createShip("Rotator",        "rotator");
99             createShip("Lut3",           "lut");
100         } else {
101             createShip("Memory",    "memory2");
102             createShip("Memory",    "memory3");
103
104             for(int i=0; i<3; i++)
105                 createShip("Alu",       "alu"+i);
106
107             for(int i=0; i<1; i++)
108                 createShip("Fifo",      "fifo"+i);
109
110             for(int i=0; i<14; i++)
111                 createShip("Counter",  "counter"+i);
112
113             /*
114             createShip("CarrySaveAdder",  "csa1");
115             createShip("Rotator",         "rotator");
116             createShip("Lut3",            "lut");
117             */
118             //createShip("DDR2",    "ddr2");
119         }
120         createShip("DRAM",    "dram");
121         createShip("Video",   "video");
122
123         //Module.SourcePort  debug_in    = top.createWireSourcePort("debug_in", WIDTH_PACKET);
124         Module.SourcePort  debug_out   = null;
125         for(FpgaShip ship : (Iterable<FpgaShip>)(Object)this) {
126             if (ship.getType().toLowerCase().equals("debug"))
127                 debug_out = ship.getVerilogModule().getOutputPort("debug_out");
128         }
129
130         // for FifoShip
131         new Module.InstantiatedModule(top, new FifoModule(8, WIDTH_WORD));
132
133         Module.SourcePort  in          = top.createInputPort("in", 8);
134         Module.SinkPort    out         = top.createOutputPort("out", 8, "");
135         Module.Latch       temp_in     = top.new Latch("temp", WIDTH_PACKET);
136         Module.Latch       count       = top.new Latch("count", 8);
137         Module.Latch       count_out   = top.new Latch("count_out", 8);
138
139         ArrayList inbox_sources = new ArrayList<FabricElement>();
140         ArrayList inbox_dests   = new ArrayList<FabricElement>();
141         ArrayList outbox_sources = new ArrayList<FabricElement>();
142         ArrayList outbox_dests   = new ArrayList<FabricElement>();
143         ArrayList instruction_dests   = new ArrayList<FabricElement>();
144         int numdocks = 0;
145         for(FpgaShip ship : (Iterable<FpgaShip>)(Object)this) {
146             if (ship.getType().toLowerCase().equals("debug"))
147                 debug_out = ship.getVerilogModule().getOutputPort("debug_out");
148             for(Dock port : ship) {
149                 if (port.isInputDock()) {
150                     inbox_sources.add(((FpgaDock)port));
151                     instruction_dests.add(port.getInstructionDestination());
152                     inbox_dests.add(port.getDataDestination());
153                 } else {
154                     outbox_sources.add(((FpgaDock)port));
155                     instruction_dests.add(port.getInstructionDestination());
156                     outbox_dests.add(port.getDataDestination());
157                 }
158                 numdocks++;
159             }
160         }
161         //System.err.println("dock count = " + numdocks);
162         ArrayList dests   = new ArrayList<FabricElement>();
163         ArrayList sources = new ArrayList<FabricElement>();
164         sources.addAll(inbox_sources);
165         sources.addAll(outbox_sources);
166         dests.addAll(inbox_dests);
167         dests.addAll(instruction_dests);
168         dests.addAll(outbox_dests);
169         top_horn = mkNode((FabricElement[])dests.toArray(new FabricElement[0]), true);
170         FabricElement   source  = mkNode((FabricElement[])sources.toArray(new FabricElement[0]), false);
171         FunnelModule.FunnelInstance top_funnel = new FunnelModule.FunnelInstance(this, top, null, source.getOutputPort());
172         ((FunnelModule.FunnelInstance)source).out = top_funnel;
173         //top_horn.addInput(top_funnel, top_funnel.getOutputPort());
174         top_funnel.addOutput(top_horn, top_horn.getInputPort());
175
176         //Module.SourcePort  debug_in    = top.createWireSourcePort("debug_in", WIDTH_PACKET);
177         Module.SinkPort debug_in = top_funnel.getInputPort("in1");
178
179         top.new Event(new Object[] { in, "count<=7" },
180                       new Object[] { new SimpleAction(temp_in.getVerilogName()+" <= {" + temp_in.getVerilogName() + "["+(WIDTH_PACKET-(1+8))+":0], in[7:0] };"),
181                                      new SimpleAction("count <= count+1;"),
182                                      in
183                       });
184         top.new Event(new Object[] { debug_in, "count>7" },
185                       new Object[] { new SimpleAction(" count <= 0; "),
186                                      new AssignAction(debug_in, temp_in),
187                                      debug_in
188                       });
189         top.new Event(new Object[] { out, debug_out },
190                       new Object[] { new SimpleAction(out.getVerilogName()+" <= ("+debug_out.getVerilogName()+">> (count_out*8));"),
191                                      new SimpleAction("if (count_out >= 5) begin "+
192                                                           "count_out <= 0; "+debug_out.getVerilogName()+"_a <= 1; end"+
193                                                           " else count_out <= count_out+1; "),
194                                      out });
195
196     }
197
198     public FabricElement mkNode(FabricElement[] ports, boolean is_horn) { return mkNode(ports, is_horn, 0, ports.length); }
199     public FabricElement mkNode(FabricElement[] ports, boolean is_horn, int start, int end) {
200         switch(end-start) {
201             case 0: return null;
202             case 1: return ports[start];
203             default: {
204                 FabricElement leftPort  = mkNode(ports, is_horn,  start,         (end+start)/2);
205                 FabricElement rightPort = mkNode(ports, is_horn,  (end+start)/2, end);
206                 return is_horn
207                     ? new HornModule.HornInstance(this, top,     leftPort, rightPort)
208                     : new FunnelModule.FunnelInstance(this, top, leftPort, rightPort);
209             }
210         }
211     }
212
213
214     // Expand //////////////////////////////////////////////////////////////////////////////
215
216     public void expand(ShipDescription sd) {
217         try {
218             if (sd.getSection("fpga")==null) return;
219             String filename = sd.getName().toLowerCase();
220             File outf = new File("build/fpga/"+filename+".v");
221             new File(outf.getParent()).mkdirs();
222             System.err.println("writing to " + outf);
223             FileOutputStream out = new FileOutputStream(outf);
224             PrintWriter pw = new PrintWriter(out);
225
226             boolean debug = "debug".equals(filename);
227
228             pw.println("`define WORDWIDTH                "+WIDTH_WORD);
229             pw.println("`define CODEBAG_SIZE_BITS        "+CBD_SIZE.valmaskwidth);
230             pw.println();
231
232             for(DockDescription dd : sd) {
233                 String name = dd.getName();
234                 pw.println("`define "+name+"_full    ("+name+"_r && !"+name+"_a)");
235                 pw.println("`define "+name+"_empty  (!"+name+"_r && !"+name+"_a)");
236                 if (dd.isInputDock()) {
237                     pw.println("`define drain_"+name+"  "+name+"_a <= 1;");
238                 } else {
239                     pw.println("`define fill_"+name+"  "+name+"_r <= 1;");
240                     pw.println("`define "+name+"_draining ("+name+"_r && "+name+"_a)");
241                 }
242             }
243             if (debug) {
244                 String name = "out";
245                 pw.println("`define "+name+"_full    ("+name+"_r && !"+name+"_a)");
246                 pw.println("`define "+name+"_empty  (!"+name+"_r && !"+name+"_a)");
247                 pw.println("`define fill_"+name+"  "+name+"_r <= 1;");
248                 pw.println("`define "+name+"_draining ("+name+"_r && "+name+"_a)");
249             }
250
251             pw.print("`define reset ");
252             for(DockDescription bb : sd) {
253                 String bb_name = bb.getName();
254                 if (bb.isInputDock()) pw.print(bb_name+"_a <= 1; "+bb_name+"_f <= 0; ");
255                 else                  pw.print(bb_name+"_r <= 0; ");
256             }
257             if (debug) {
258                 String bb_name = "out";
259                 pw.print(bb_name+"_r <= 0; ");
260             }
261             pw.println();
262
263             pw.print("`define cleanup ");
264             for(DockDescription bb : sd) {
265                 String bb_name = bb.getName();
266                 if (bb.isInputDock()) pw.print("if (!"+bb_name+"_r && "+bb_name+"_a) "+bb_name+"_a <= 0; ");
267                 else                  pw.print("if ( "+bb_name+"_r && "+bb_name+"_a) "+bb_name+"_r <= 0; ");
268             }
269             if (debug) {
270                 String bb_name = "out";
271                 pw.print("if ( "+bb_name+"_r && "+bb_name+"_a) "+bb_name+"_r <= 0; ");
272             }
273             pw.println();
274
275             // FIXME: this corresponds to something
276             /*
277             pw.print("`define flush_happening (1");
278             for(DockDescription bb : sd)
279                 if (bb.isInputDock())
280                     pw.print(" && "+bb.getName()+"_r_ && !"+bb.getName()+"_a && "+bb.getName()+"_d["+WIDTH_WORD+"]");
281             pw.println(")");
282             */
283
284             pw.print("`define flush ");
285             for(DockDescription bb : sd)
286                 if (bb.isInputDock())
287                     pw.print(" if (!"+bb.getName()+"_r_) "+bb.getName()+"_f <= 0; ");
288             pw.print("if (1");
289             for(DockDescription bb : sd)
290                 if (bb.isInputDock())
291                     pw.print(" && "+bb.getName()+"_r_ && !"+bb.getName()+"_a");
292             pw.print(") begin ");
293             if (true) {
294                 pw.print("if (1");
295                 for(DockDescription bb : sd)
296                     if (bb.isInputDock())
297                         pw.print(" && "+bb.getName()+"_d["+WIDTH_WORD+"] ");
298                 pw.print(") begin ");
299                 if (true) {
300                     for(DockDescription bb : sd)
301                         if (bb.isInputDock())
302                             pw.print(bb.getName()+"_f <= 1; ");
303                 }
304                 pw.print(" end else if (0");
305                 for(DockDescription bb : sd)
306                     if (bb.isInputDock())
307                         pw.print(" || "+bb.getName()+"_d["+WIDTH_WORD+"] ");
308                 pw.print(") begin ");
309                 if (true) {
310                     for(DockDescription bb : sd)
311                         if (bb.isInputDock())
312                             pw.print(" if (!"+bb.getName()+"_d["+WIDTH_WORD+"]) "+bb.getName()+"_f <= 1; ");
313                 }
314                 pw.print(" end ");
315             }
316             pw.print(" end ");
317             pw.println();
318             
319             pw.println("module " + filename + "( clk, rst ");
320             for(DockDescription bb : sd) {
321                 String bb_name = bb.getName();
322                 pw.print("        ");
323                 if (bb.isInputDock()) {
324                     pw.print(", " + bb_name+"_r_");
325                     pw.print(", " + bb_name+"_a_");
326                     pw.print(", " + bb_name+"_d");
327                 } else {
328                     pw.print(", " + bb_name+"_r_");
329                     pw.print(", " + bb_name+"_a");
330                     pw.print(", " + bb_name+"_d_");
331                 }
332                 pw.println();
333             }
334             if (filename.equals("debug")) {
335                 pw.println("    , out_r_");
336                 pw.println("    , out_a");
337                 pw.println("    , out_d_");
338             }
339             if (filename.equals("dram")) {
340                 pw.println("    , dram_addr_");
341                 pw.println("    , dram_addr_r_");
342                 pw.println("    , dram_addr_a");
343                 pw.println("    , dram_isread_");
344                 pw.println("    , dram_write_data_");
345                 pw.println("    , dram_write_data_push_");
346                 pw.println("    , dram_write_data_full");
347                 pw.println("    , dram_read_data");
348                 pw.println("    , dram_read_data_pop_");
349                 pw.println("    , dram_read_data_empty");
350                 pw.println("    , dram_read_data_latency");
351             }
352             if (filename.equals("ddr2")) {
353                 pw.println("    , ddr2_addr_");
354                 pw.println("    , ddr2_addr_r_");
355                 pw.println("    , ddr2_addr_a");
356                 pw.println("    , ddr2_isread_");
357                 pw.println("    , ddr2_write_data_");
358                 pw.println("    , ddr2_write_data_push_");
359                 pw.println("    , ddr2_write_data_full");
360                 pw.println("    , ddr2_read_data");
361                 pw.println("    , ddr2_read_data_pop_");
362                 pw.println("    , ddr2_read_data_empty");
363                 pw.println("    , ddr2_read_data_latency");
364             }
365             if (filename.equals("video")) {
366                 pw.println("    , vga_clk");
367                 pw.println("    , vga_psave");
368                 pw.println("    , vga_hsync");
369                 pw.println("    , vga_vsync");
370                 pw.println("    , vga_sync");
371                 pw.println("    , vga_blank");
372                 pw.println("    , vga_r");
373                 pw.println("    , vga_g");
374                 pw.println("    , vga_b");
375                 pw.println("    , vga_clkout");
376             }
377             pw.println("        );");
378             pw.println();
379             pw.println("    input clk;");
380             pw.println("    input rst;");
381             if (filename.equals("dram")) {
382                 pw.println("output  [31:0] dram_addr_;");
383                 pw.println("output         dram_addr_r_;");
384                 pw.println("input          dram_addr_a;");
385                 pw.println("output         dram_isread_;");
386                 pw.println("output  [63:0] dram_write_data_;");
387                 pw.println("output         dram_write_data_push_;");
388                 pw.println("input          dram_write_data_full;");
389                 pw.println("input   [63:0] dram_read_data;");
390                 pw.println("output         dram_read_data_pop_;");
391                 pw.println("input          dram_read_data_empty;");
392                 pw.println("input   [1:0]  dram_read_data_latency;");
393             }
394             if (filename.equals("ddr2")) {
395                 pw.println("output  [31:0] ddr2_addr_;");
396                 pw.println("output         ddr2_addr_r_;");
397                 pw.println("input          ddr2_addr_a;");
398                 pw.println("output         ddr2_isread_;");
399                 pw.println("output  [63:0] ddr2_write_data_;");
400                 pw.println("output         ddr2_write_data_push_;");
401                 pw.println("input          ddr2_write_data_full;");
402                 pw.println("input   [63:0] ddr2_read_data;");
403                 pw.println("output         ddr2_read_data_pop_;");
404                 pw.println("input          ddr2_read_data_empty;");
405                 pw.println("input   [1:0]  ddr2_read_data_latency;");
406             }
407             if (filename.equals("video")) {
408                 pw.println("input          vga_clk;");
409                 pw.println("output         vga_psave;");
410                 pw.println("output         vga_hsync;");
411                 pw.println("output         vga_vsync;");
412                 pw.println("output         vga_sync;");
413                 pw.println("output         vga_blank;");
414                 pw.println("output   [7:0] vga_r;");
415                 pw.println("output   [7:0] vga_g;");
416                 pw.println("output   [7:0] vga_b;");
417                 pw.println("output         vga_clkout;");
418             }
419
420             for(DockDescription bb : sd) {
421                 String bb_name = bb.getName();
422                 if (bb.isInputDock()) {
423                     pw.println("        input   ["+WIDTH_WORD+":0] "+bb_name+"_d;");
424                     pw.println("        input   "+bb_name+"_r_;");
425                     pw.println("        wire    "+bb_name+"_r;");
426                     pw.println("        assign  "+bb_name+"_r = "+bb_name+"_r_ & ~"+bb_name+"_d["+WIDTH_WORD+"];");
427                     pw.println("        output  "+bb_name+"_a_;");
428                     pw.println("        reg     "+bb_name+"_a;");
429                     pw.println("        initial "+bb_name+"_a  = 0;");
430                     pw.println("        reg     "+bb_name+"_f;");
431                     pw.println("        initial "+bb_name+"_f  = 0;");
432                     pw.println("        assign  "+bb_name+"_a_ = "+bb_name+"_a || "+bb_name+"_f;");
433                 } else {
434                     pw.println("        output  ["+WIDTH_WORD+":0] "+bb_name+"_d_;");
435                     pw.println("        input   "+bb_name+"_a;");
436                     pw.println("        output  "+bb_name+"_r_;");
437                     pw.println("        reg     "+bb_name+"_r;");
438                     pw.println("        initial "+bb_name+"_r  = 0;");
439                     pw.println("        assign  "+bb_name+"_r_ = "+bb_name+"_r;");
440                 }
441                 pw.println();
442             }
443             if (filename.equals("debug")) {
444                 String bb_name = "out";
445                 pw.println("        output  ["+WIDTH_WORD+":0] "+bb_name+"_d_;");
446                 pw.println("        input   "+bb_name+"_a;");
447                 pw.println("        output  "+bb_name+"_r_;");
448                 pw.println("        reg     "+bb_name+"_r;");
449                 pw.println("        initial "+bb_name+"_r  = 0;");
450                 pw.println("        assign  "+bb_name+"_r_ = "+bb_name+"_r;");
451             }
452
453             if (filename.equals("fifo")) {
454                 pw.println("  wire in_a__;");
455                 pw.println("  wire out_r__;");
456                 pw.println("  fifo8x37 fifo8x37(clk, rst,");
457                 pw.println("                    in_r,    in_a__, in_d,");
458                 pw.println("                    out_r__, out_a,  out_d_);");
459                 pw.println("  always @(posedge clk) begin");
460                 pw.println("    if (!rst) begin");
461                 pw.println("      `reset");
462                 pw.println("    end else begin");
463                 pw.println("      `flush");
464                 pw.println("      out_r <= out_r__;");
465                 pw.println("      in_a  <= in_a__;");
466                 pw.println("    end");
467                 pw.println("  end");
468             } else {
469                 pw.println(sd.getSection("fpga"));
470             }
471
472             pw.println("endmodule");
473
474             pw.flush();
475             pw.close();
476         } catch (Exception e) { throw new RuntimeException(e); }
477     }
478
479 }