ext-sample/utils.cpp at master · NTU-ALComLab/ext-sample · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#include "ext-sample/utils.h"
#include <iostream>
#include <cassert>
#include <random>
#include <ctime>
#include <algorithm>


#include "ext-sample/SampleCircuit.h"

using namespace std;

extern "C"{
Abc_Ntk_t * Abc_NtkDC2( Abc_Ntk_t * pNtk, int fBalance, int fUpdateLevel, int fFanout, int fPower, int fVerbose );
Abc_Ntk_t * Abc_NtkDarSeqSweep2( Abc_Ntk_t * pNtk, Ssw_Pars_t * pPars );
Abc_Ntk_t * Abc_NtkDarFraig( Abc_Ntk_t * pNtk, int nConfLimit, int fDoSparse, int fProve, int fTransfer, int fSpeculate, int fChoicing, int fVerbose );
void Io_WriteAiger( Abc_Ntk_t * pNtk, char * pFileName, int fWriteSymbols, int fCompact, int fUnique );
}

Abc_Ntk_t * Ntk_Optimize(Abc_Ntk_t* pNtk)
{
    Ssw_Pars_t Pars, * pPars = &Pars;
    Ssw_ManSetDefaultParams( pPars );
    Abc_Ntk_t * pNtkOpt = Abc_NtkDup( pNtk );
    strcat(pNtkOpt->pName, "Opt");
    for (int j = 0; j < 5; j++)
    {
        if ( !Abc_NtkIsComb(pNtkOpt) )
            pNtkOpt = Abc_NtkDarSeqSweep2( pNtkOpt, pPars );
        pNtkOpt = Abc_NtkDC2( pNtkOpt, 0, 0, 1, 0, 0 );
        pNtkOpt = Abc_NtkDarFraig( pNtkOpt, 100, 1, 0, 0, 0, 0, 0 );
    }

    return pNtkOpt;
}

Abc_Ntk_t * Ntk_StuckGen(Abc_Ntk_t * pNtk)
{
    int i, nAnd=0, picked;
    Abc_Obj_t * pObj;
    Abc_Ntk_t * pNtkRes;
    vector<Abc_Obj_t*> vAnd;

    assert(Abc_NtkIsStrash(pNtk) && Abc_NtkHasAig(pNtk));
    pNtkRes = Abc_NtkDup(pNtk);

    // const node
    Abc_Obj_t * pAigOne  = Abc_AigConst1(pNtkRes);
    Abc_Obj_t * pAigZero = Abc_ObjNot(pAigOne);

    // collect all nodes
    Abc_AigForEachAnd( pNtkRes, pObj, i )
    {
        nAnd++;
        vAnd.push_back(pObj);
    }

    // randomly choose a node
    random_device rd;
    default_random_engine gen = default_random_engine(rd());
    uniform_int_distribution<int> dis(0,nAnd-1);
    picked = dis(gen);

    // stuck to 0 or 1
    srand((unsigned)time(NULL));
    if (rand()%2 == 0)
        Abc_AigReplace( (Abc_Aig_t*)pNtkRes->pManFunc, vAnd[picked], pAigZero, 1 );
    else
        Abc_AigReplace( (Abc_Aig_t*)pNtkRes->pManFunc, vAnd[picked], pAigOne, 1 );

    return pNtkRes;
}

void split(vector<string> &vs, string str, char delim)
{
    size_t pos;
    string s;

    vs.clear();
    while ( (pos=str.find(delim)) != string::npos)
    {
        if (pos!=0)
        {
            s = str.substr(0, pos);
            vs.push_back(s);
        }
        str = str.substr(pos+1);
    }
    if (str.length()!=0)
        vs.push_back(str);

    return;
}