Generating adder-trees using 'gen -b -A <num> -N <num> <file>.v'.

1 files changed, 49 insertions, 0 deletions

diff --git a/src/base/abci/abcGen.c b/src/base/abci/abcGen.c
index 26ee3f86..d8fbded3 100644
--- a/src/base/abci/abcGen.c
+++ b/src/base/abci/abcGen.c
@@ -821,6 +821,55 @@ void Abc_GenFsm( char * pFileName, int nPis, int nPos, int nStates, int nLines,
     Vec_StrFree( vCond );
 }
 
+/**Function*************************************************************
+
+  Synopsis    []
+
+  Description []
+               
+  SideEffects []
+
+  SeeAlso     []
+
+***********************************************************************/
+void Abc_AdderTree( FILE * pFile, int nArgs, int nBits )
+{
+    int i, k, nDigits = Abc_Base10Log( nBits ), Log2 = Abc_Base2Log( nArgs );
+    assert( nArgs > 1 && nBits > 1 );
+    fprintf( pFile, "module adder_tree_%d_%d (\n   ", nArgs, nBits );
+    for ( i = 0; i < nBits; i++, fprintf(pFile, "\n   ") )
+        for ( k = 0; k < nArgs; k++ )
+            fprintf( pFile, " %c%0*d,", 'a'+k, nDigits, nBits-1-i );
+    fprintf( pFile, " z\n" );
+    fprintf( pFile, "  );\n" );
+    for ( i = 0; i < nBits; i++ )
+    {
+        fprintf( pFile, "  input" );
+        for ( k = 0; k < nArgs; k++ )
+            fprintf( pFile, " %c%0*d%s", 'a'+k, nDigits, nBits-1-i, k==nArgs-1 ? "":"," );
+        fprintf( pFile, ";\n" );
+    }
+    fprintf( pFile, "  output [%d:0] z;\n", nBits+Log2-1 );
+    for ( i = 0; i < nArgs; i++ )
+    {
+        fprintf( pFile, "  wire [%d:0] t%d = {", nBits-1, i );
+        for ( k = 0; k < nBits; k++ )
+            fprintf( pFile, " %c%0*d%s", 'a'+i, nDigits, nBits-1-k, k==nBits-1 ? "":"," );
+        fprintf( pFile, " };\n" );
+    }
+    for ( i = 0; i < nArgs-1; i++ )
+        fprintf( pFile, "  wire [%d:0] s%d = t%d + %s%d;\n", nBits+Log2-1, i+1, i+1, i ? "s":"t", i );
+    fprintf( pFile, "  assign z = s%d;\n", nArgs-1 );
+    fprintf( pFile, "endmodule\n\n" ); 
+}
+void Abc_GenAdderTree( char * pFileName, int nArgs, int nBits )
+{
+    FILE * pFile = fopen( pFileName, "w" );
+    fprintf( pFile, "// %d-argument %d-bit adder-tree generated by ABC on %s\n", nArgs, nBits, Extra_TimeStamp() );
+    Abc_AdderTree( pFile, nArgs, nBits );
+    fclose( pFile );
+}
+
 ////////////////////////////////////////////////////////////////////////
 ///                       END OF FILE                                ///
 ////////////////////////////////////////////////////////////////////////