From 4aaab8f3953e44b51b84ca0895b68745be26344b Mon Sep 17 00:00:00 2001
From: Miodrag Milanovic <mmicko@gmail.com>
Date: Wed, 22 May 2024 15:45:18 +0200
Subject: [PATCH] start adding wfg model

---
 techlibs/nanoxplore/cells_bb_u.v  | 21 -------------
 techlibs/nanoxplore/cells_sim_u.v | 51 +++++++++++++++++++++++++++++--
 2 files changed, 49 insertions(+), 23 deletions(-)

diff --git a/techlibs/nanoxplore/cells_bb_u.v b/techlibs/nanoxplore/cells_bb_u.v
index 95cb7158f..56559fd23 100644
--- a/techlibs/nanoxplore/cells_bb_u.v
+++ b/techlibs/nanoxplore/cells_bb_u.v
@@ -539,27 +539,6 @@ module NX_PLL_U(R, REF, FBK, OSC, VCO, LDFO, REFO, CLK_DIV1, CLK_DIV2, CLK_DIV3,
     parameter use_pll = 1'b1;
 endmodule
 
-(* blackbox *)
-module NX_WFG_U(R, SI, ZI, SO, ZO);
-    input R;
-    input SI;
-    output SO;
-    input ZI;
-    output ZO;
-    parameter delay = 0;
-    parameter delay_on = 1'b0;
-    parameter div_phase = 1'b0;
-    parameter div_ratio = 0;
-    parameter location = "";
-    parameter mode = 0;
-    parameter pattern = 16'b0000000000000000;
-    parameter pattern_end = 0;
-    parameter reset_on_cal_lock_n = 1'b0;
-    parameter reset_on_pll_lock_n = 1'b0;
-    parameter reset_on_pll_locka_n = 1'b0;
-    parameter wfg_edge = 1'b0;
-endmodule
-
 (* blackbox *)
 module NX_XCDC_U(CK1, CK2, ASRSTI, ADRSTI, ASRSTO, ADRSTO, BSRSTI, BDRSTI, BSRSTO, BDRSTO, CSRSTI, CDRSTI, CSRSTO, CDRSTO, DSRSTI, DDRSTI, DSRSTO, DDRSTO, BI, CI, CO
 , AO, BO, AI, DI, DO);
diff --git a/techlibs/nanoxplore/cells_sim_u.v b/techlibs/nanoxplore/cells_sim_u.v
index 094adcaf8..209da3b5f 100644
--- a/techlibs/nanoxplore/cells_sim_u.v
+++ b/techlibs/nanoxplore/cells_sim_u.v
@@ -142,7 +142,7 @@ module NX_RFB_U(WCK, I1, I2, I3, I4, I5, I6, I7, I8, I9, I10, I11, I12, I13, I14
     integer i;
     initial begin
         for (i = 0; i < MEM_SIZE; i = i + 1)
-            mem[i] = MEM_WIDTH'b0;
+            mem[i] = 36'b0;
     end
 
     wire [ADDR_WIDTH-1:0] WA = (mode==2) ? { WA6, WA5, WA4, WA3, WA2, WA1 } : { WA5, WA4, WA3, WA2, WA1 };
@@ -177,4 +177,51 @@ module NX_RFB_U(WCK, I1, I2, I3, I4, I5, I6, I7, I8, I9, I10, I11, I12, I13, I14
     always @(posedge clock)
         if (WE)
             mem[WA] <= I[MEM_WIDTH-1:0];
-endmodule
\ No newline at end of file
+endmodule
+
+(* abc9_box, lib_whitebox *)
+module NX_WFG_U(R, SI, ZI, SO, ZO);
+    input R;
+    input SI;
+    output SO;
+    input ZI;
+    output ZO;
+    parameter delay = 0;
+    parameter delay_on = 1'b0;
+    parameter div_phase = 1'b0;
+    parameter div_ratio = 0;
+    parameter location = "";
+    parameter mode = 0;
+    parameter pattern = 16'b0000000000000000;
+    parameter pattern_end = 0;
+    parameter reset_on_cal_lock_n = 1'b0;
+    parameter reset_on_pll_lock_n = 1'b0;
+    parameter reset_on_pll_locka_n = 1'b0;
+    parameter wfg_edge = 1'b0;
+
+    generate
+        if (mode==0) begin
+            assign SO = SI;
+        end
+        else if (mode==1) begin
+            wire clock = ZI ^ wfg_edge;
+            wire reset = R || SI;
+            reg [3:0] counter = 0;
+            reg [15:0] rom = pattern;
+
+            always @(posedge clock)
+            begin
+                if (reset)
+                    counter <= 4'b0;
+                else
+                    counter <= counter + 1;
+            end
+            assign SO = counter == pattern_end;
+            assign ZO = rom[counter];
+        end
+        else if (mode==2) begin
+        end
+        else
+            $error("Unknown NX_WFG_U mode");
+    endgenerate
+endmodule