Get constraints from class

Signed-off-by: Ryszard Rozak <[email protected]>
antmicro · Dec 4, 2024 · bd2f27c · bd2f27c
1 parent e120993
commit bd2f27c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/V3Randomize.cpp b/src/V3Randomize.cpp
@@ -2111,6 +2111,17 @@ class RandomizeVisitor final : public VNVisitor {
         AstVar* const randModeVarp = getRandModeVar(classp);
         if (randModeVarp) addSetRandMode(randomizeFuncp, localGenp, randModeVarp);
 
+        // Get constraints from the class
+        classp->foreachMember([&](AstClass* const clp, AstConstraint* const constrp) {
+            AstTask* const setupTaskp = VN_AS(constrp->user2p(), Task);
+            UASSERT_OBJ(setupTaskp, constrp, "Constraint should have setup task generated before");
+            AstTaskRef* const setupRefp = new AstTaskRef{
+                fl, setupTaskp->name(),
+                new AstArg{fl, "constraint", new AstVarRef{fl, localGenp, VAccess::READWRITE}}};
+            setupRefp->taskp(setupTaskp);
+            randomizeFuncp->addStmtsp(
+                wrapIfConstraintMode(classp, constrp, new AstStmtExpr{fl, setupRefp}));
+        });
         // Generate constraint setup code and a hardcoded call to the solver
         AstNode* const capturedTreep = withp->exprp()->unlinkFrBackWithNext();
         randomizeFuncp->addStmtsp(capturedTreep);