ABC: A System for Sequential Synthesis and Verification
 
Loading...
Searching...
No Matches
Functions
ioJson.c File Reference
#include "ioAbc.h"
#include "misc/vec/vecWec.h"
#include "misc/util/utilNam.h"
#include "misc/extra/extra.h"
Include dependency graph for ioJson.c:

Go to the source code of this file.

Functions

char * Nnc_LayerType2Str (char *pStr)
 FUNCTION DEFINITIONS ///.
 
void Json_Extract_rec (FILE *pFile, Abc_Nam_t *pStr, Vec_Wec_t *vObjs, Vec_Int_t *vArray, int fWrite, int *pCount)
 
void Json_Extract (char *pFileName, Abc_Nam_t *pStr, Vec_Wec_t *vObjs)
 
void Json_Write_rec (FILE *pFile, Abc_Nam_t *pStr, Vec_Wec_t *vObjs, Vec_Int_t *vArray, int Level, int fAddComma, int fSpaces)
 
void Json_Write (char *pFileName, Abc_Nam_t *pStr, Vec_Wec_t *vObjs)
 
char * Json_ReadPreprocess (char *pIn, int nFileSize)
 
Vec_Wec_t * Json_Read (char *pFileName, Abc_Nam_t **ppStrs)
 
void Json_ReadTest (char *pFileName)
 

Function Documentation

◆ Json_Extract()

void Json_Extract ( char * pFileName,
Abc_Nam_t * pStr,
Vec_Wec_t * vObjs )

Definition at line 151 of file ioJson.c.

152{
153 int Count = 0;
154 FILE * pFile = fopen( pFileName, "wb" );
155 if ( pFile == NULL )
156 {
157 printf( "Cannot open file \"%s\" for writing.\n", pFileName );
158 return;
159 }
160 fprintf( pFile, "# Data extracted from JSON file:\n" );
161 Json_Extract_rec( pFile, pStr, vObjs, Vec_WecEntry(vObjs, 0), 0, &Count );
162 fprintf( pFile, "\n" );
163 fclose( pFile );
164}
Json_Extract_rec
void Json_Extract_rec(FILE *pFile, Abc_Nam_t *pStr, Vec_Wec_t *vObjs, Vec_Int_t *vArray, int fWrite, int *pCount)
Definition ioJson.c:77
Here is the call graph for this function:

◆ Json_Extract_rec()

void Json_Extract_rec ( FILE * pFile,
Abc_Nam_t * pStr,
Vec_Wec_t * vObjs,
Vec_Int_t * vArray,
int fWrite,
int * pCount )

Definition at line 77 of file ioJson.c.

78{
79 int i, Entry1, Entry2;
80 if ( Vec_IntEntry(vArray, 0) ) // array
81 {
82 if ( Vec_IntSize(vArray) == 1 )
83 return;
84 if ( Vec_IntSize(vArray) == 2 && Json_EntryIsName(Vec_IntEntry(vArray,1)) )
85 {
86 if ( fWrite )
87 fprintf( pFile, "%s", Json_EntryName(pStr, Vec_IntEntry(vArray,1)) );
88 return;
89 }
90 else
91 {
92 Vec_IntForEachEntryStart( vArray, Entry1, i, 1 )
93 {
94 if ( Json_EntryIsName(Entry1) )
95 {
96 int Digit = Json_EntryName(pStr, Entry1)[0];
97 if ( fWrite && Digit != '0' )
98 fprintf( pFile, "%s%s", Json_EntryName(pStr, Entry1), Digit >= '0' && Digit <= '9' ? "" : " " );
99 }
100 else
101 Json_Extract_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry1), fWrite, pCount );
102 }
103 return;
104 }
105 }
106 else // list of pairs
107 {
108 int fHaveConfig = 0;
109 assert( Vec_IntSize(vArray) % 2 != 0 );
110 Vec_IntForEachEntryDoubleStart( vArray, Entry1, Entry2, i, 1 )
111 {
112 char * pName1 = Json_EntryIsName(Entry1) ? Json_EntryName(pStr, Entry1) : NULL;
113 char * pName2 = Json_EntryIsName(Entry2) ? Json_EntryName(pStr, Entry2) : NULL;
114 char * pName3 = pName2 ? Nnc_LayerType2Str(pName2) : NULL;
115 if ( pName1 == NULL )
116 continue;
117 if ( !strcmp(pName1, "class_name") )
118 {
119 if ( pName3 )
120 fprintf( pFile, "\n%3d : %-8s ", (*pCount)++, pName3 );
121 }
122 else if ( !strcmp(pName1, "name") )
123 {
124 if ( fHaveConfig )
125 fprintf( pFile, " N=%s ", pName2 ? pName2 : "???" );
126 }
127 else if ( !strcmp(pName1, "kernel_size") )
128 {
129 fprintf( pFile, " K=" );
130 Json_Extract_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry2), 1, pCount );
131 }
132 else if ( !strcmp(pName1, "strides") )
133 {
134 fprintf( pFile, " S=" );
135 Json_Extract_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry2), 1, pCount );
136 }
137 else if ( !strcmp(pName1, "filters") )
138 fprintf( pFile, " C=%s", pName2 );
139 else if ( !strcmp(pName1, "inbound_nodes") )
140 Json_Extract_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry2), 1, pCount );
141 else if ( !strcmp(pName1, "layers") )
142 Json_Extract_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry2), 1, pCount );
143 else if ( !strcmp(pName1, "config") )
144 {
145 fHaveConfig = 1;
146 Json_Extract_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry2), 0, pCount );
147 }
148 }
149 }
150}
Nnc_LayerType2Str
char * Nnc_LayerType2Str(char *pStr)
FUNCTION DEFINITIONS ///.
Definition ioJson.c:52
assert
#define assert(ex)
Definition util_old.h:213
strcmp
int strcmp()
Vec_IntForEachEntryStart
#define Vec_IntForEachEntryStart(vVec, Entry, i, Start)
Definition vecInt.h:56
Vec_IntForEachEntryDoubleStart
#define Vec_IntForEachEntryDoubleStart(vVec, Entry1, Entry2, i, Start)
Definition vecInt.h:74
Here is the call graph for this function:
Here is the caller graph for this function:

◆ Json_Read()

Vec_Wec_t * Json_Read ( char * pFileName,
Abc_Nam_t ** ppStrs )

Definition at line 307 of file ioJson.c.

308{
309 Abc_Nam_t * pStrs;
310 Vec_Wec_t * vObjs;
311 Vec_Int_t * vStack, * vTemp;
312 char * pContents, * pCur, * pNext, * pCur2, * pNext2;
313 int nFileSize, RetValue, iToken;
314 FILE * pFile;
315
316 // read the file into the buffer
317 pFile = fopen( pFileName, "rb" );
318 if ( pFile == NULL )
319 {
320 printf( "Cannot open file \"%s\" for reading.\n", pFileName );
321 return NULL;
322 }
323 nFileSize = Extra_FileSize( pFileName );
324 pContents = pCur = ABC_ALLOC( char, nFileSize+1 );
325 RetValue = fread( pContents, nFileSize, 1, pFile );
326 pContents[nFileSize] = 0;
327 fclose( pFile );
328
329 pContents = Json_ReadPreprocess( pCur = pContents, nFileSize );
330 nFileSize = strlen(pContents);
331 ABC_FREE( pCur );
332 pCur = pContents;
333
334 // start data-structures
335 vObjs = Vec_WecAlloc( 1000 );
336 vStack = Vec_IntAlloc( 100 );
337 pStrs = Abc_NamStart( 1000, 24 );
338 //Json_AddTypes( pStrs );
339
340 // read lines
341 assert( Vec_WecSize(vObjs) == 0 );
342 assert( Vec_IntSize(vStack) == 0 );
343 while ( pCur < pContents + nFileSize )
344 {
345 pCur = Json_SkipSpaces( pCur );
346 if ( *pCur == '\0' )
347 break;
348 pNext = Json_SkipNonSpaces( pCur );
349 if ( *pCur == '{' || *pCur == '[' )
350 {
351 // add fanin to node on the previous level
352 if ( Vec_IntSize(vStack) > 0 )
353 Vec_IntPush( Vec_WecEntry(vObjs, Vec_IntEntryLast(vStack)), Abc_Var2Lit(Vec_WecSize(vObjs), 0) );
354 // add node to the stack
355 Vec_IntPush( vStack, Vec_WecSize(vObjs) );
356 vTemp = Vec_WecPushLevel( vObjs );
357 Vec_IntGrow( vTemp, 4 );
358 // remember it as an array
359 Vec_IntPush( vTemp, (int)(*pCur == '[') );
360 pCur++;
361 continue;
362 }
363 if ( *pCur == '}' || *pCur == ']' )
364 {
365 Vec_IntPop( vStack );
366 pCur++;
367 continue;
368 }
369 if ( *pCur == ',' || *pCur == ':' )
370 {
371 pCur++;
372 continue;
373 }
374 iToken = Json_TokenCompare( pCur, pNext, &pCur2, &pNext2 );
375 if ( iToken == 0 )
376 iToken = Abc_NamStrFindOrAddLim( pStrs, pCur2, pNext2, NULL );
377 Vec_IntPush( Vec_WecEntry(vObjs, Vec_IntEntryLast(vStack)), Abc_Var2Lit(iToken, 1) );
378 pCur = pNext;
379 }
380 Vec_IntFree( vStack );
381 ABC_FREE( pContents );
382 *ppStrs = pStrs;
383 return vObjs;
384}
ABC_ALLOC
#define ABC_ALLOC(type, num)
Definition abc_global.h:264
ABC_FREE
#define ABC_FREE(obj)
Definition abc_global.h:267
Vec_Int_t
typedefABC_NAMESPACE_IMPL_START struct Vec_Int_t_ Vec_Int_t
DECLARATIONS ///.
Definition bblif.c:37
Extra_FileSize
int Extra_FileSize(char *pFileName)
Definition extraUtilFile.c:327
Json_ReadPreprocess
char * Json_ReadPreprocess(char *pIn, int nFileSize)
Definition ioJson.c:292
Abc_NamStrFindOrAddLim
int Abc_NamStrFindOrAddLim(Abc_Nam_t *p, char *pStr, char *pLim, int *pfFound)
Definition utilNam.c:492
Abc_NamStart
Abc_Nam_t * Abc_NamStart(int nObjs, int nAveSize)
FUNCTION DEFINITIONS ///.
Definition utilNam.c:80
Abc_Nam_t
typedefABC_NAMESPACE_HEADER_START struct Abc_Nam_t_ Abc_Nam_t
INCLUDES ///.
Definition utilNam.h:39
strlen
int strlen()
Vec_Wec_t
typedefABC_NAMESPACE_HEADER_START struct Vec_Wec_t_ Vec_Wec_t
INCLUDES ///.
Definition vecWec.h:42
Here is the call graph for this function:
Here is the caller graph for this function:

◆ Json_ReadPreprocess()

char * Json_ReadPreprocess ( char * pIn,
int nFileSize )

Function*************************************************************

Synopsis [Reads JSON from a file.]

Description []

SideEffects []

SeeAlso []

Definition at line 292 of file ioJson.c.

293{
294 char * pOut = ABC_ALLOC( char, 3*nFileSize ); int i, k = 0;
295 for ( i = 0; i < nFileSize; i++ )
296 if ( pIn[i] == '{' || pIn[i] == '}' || pIn[i] == '[' || pIn[i] == ']' )
297 {
298 pOut[k++] = ' ';
299 pOut[k++] = pIn[i];
300 pOut[k++] = ' ';
301 }
302 else
303 pOut[k++] = pIn[i];
304 pOut[k++] = '\0';
305 return pOut;
306}
Here is the caller graph for this function:

◆ Json_ReadTest()

void Json_ReadTest ( char * pFileName)

Function*************************************************************

Synopsis []

Description []

SideEffects []

SeeAlso []

Definition at line 397 of file ioJson.c.

398{
399 Abc_Nam_t * pStrs;
400 Vec_Wec_t * vObjs;
401 vObjs = Json_Read( pFileName, &pStrs );
402 if ( vObjs == NULL )
403 return;
404 Json_Write( "test.json", pStrs, vObjs );
405 Abc_NamDeref( pStrs );
406 Vec_WecFree( vObjs );
407}
Json_Write
void Json_Write(char *pFileName, Abc_Nam_t *pStr, Vec_Wec_t *vObjs)
Definition ioJson.c:269
Json_Read
Vec_Wec_t * Json_Read(char *pFileName, Abc_Nam_t **ppStrs)
Definition ioJson.c:307
Abc_NamDeref
void Abc_NamDeref(Abc_Nam_t *p)
Definition utilNam.c:212
Here is the call graph for this function:

◆ Json_Write()

void Json_Write ( char * pFileName,
Abc_Nam_t * pStr,
Vec_Wec_t * vObjs )

Definition at line 269 of file ioJson.c.

270{
271 FILE * pFile = fopen( pFileName, "wb" );
272 if ( pFile == NULL )
273 {
274 printf( "Cannot open file \"%s\" for writing.\n", pFileName );
275 return;
276 }
277 Json_Write_rec( pFile, pStr, vObjs, Vec_WecEntry(vObjs, 0), 1, 0, 1 );
278 fclose( pFile );
279}
Json_Write_rec
void Json_Write_rec(FILE *pFile, Abc_Nam_t *pStr, Vec_Wec_t *vObjs, Vec_Int_t *vArray, int Level, int fAddComma, int fSpaces)
Definition ioJson.c:221
Here is the call graph for this function:
Here is the caller graph for this function:

◆ Json_Write_rec()

void Json_Write_rec ( FILE * pFile,
Abc_Nam_t * pStr,
Vec_Wec_t * vObjs,
Vec_Int_t * vArray,
int Level,
int fAddComma,
int fSpaces )

Function*************************************************************

Synopsis [Writes JSON into a file.]

Description []

SideEffects []

SeeAlso []

Definition at line 221 of file ioJson.c.

222{
223 int i, Entry1, Entry2, fComma;
224 if ( Vec_IntEntry(vArray, 0) ) // array
225 {
226 if ( Vec_IntSize(vArray) == 1 )
227 fprintf( pFile, "[]" );
228 else if ( Vec_IntSize(vArray) == 2 && Json_EntryIsName(Vec_IntEntry(vArray,1)) )
229 fprintf( pFile, "[ \"%s\" ]", Json_EntryName(pStr, Vec_IntEntry(vArray,1)) );
230 else
231 {
232 if ( fSpaces )
233 fprintf( pFile, "%*s", 3*(Level-1), "" );
234 fprintf( pFile, "[\n" );
235 Vec_IntForEachEntryStart( vArray, Entry1, i, 1 )
236 {
237 fComma = (i < Vec_IntSize(vArray) - 1);
238 if ( Json_EntryIsName(Entry1) )
239 fprintf( pFile, "%*s\"%s\"%s\n", 3*Level, "", Json_EntryName(pStr, Entry1), fComma ? ",":"" );
240 else
241 Json_Write_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry1), Level+1, fComma, 1 );
242 }
243 fprintf( pFile, "%*s]", 3*(Level-1), "" );
244 }
245 fprintf( pFile, "%s\n", fAddComma ? ",":"" );
246 }
247 else // list of pairs
248 {
249 if ( fSpaces )
250 fprintf( pFile, "%*s", 3*(Level-1), "" );
251 fprintf( pFile, "{\n" );
252 assert( Vec_IntSize(vArray) % 2 != 0 );
253 Vec_IntForEachEntryDoubleStart( vArray, Entry1, Entry2, i, 1 )
254 {
255 fComma = (i < Vec_IntSize(vArray) - 3);
256 if ( Json_EntryIsName(Entry1) )
257 fprintf( pFile, "%*s\"%s\"", 3*Level, "", Json_EntryName(pStr, Entry1) );
258 else
259 Json_Write_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry1), Level+1, 0, 1 );
260 fprintf( pFile, " : " );
261 if ( Json_EntryIsName(Entry2) )
262 fprintf( pFile, "\"%s\"%s\n", Json_EntryName(pStr, Entry2), fComma ? ",":"" );
263 else
264 Json_Write_rec( pFile, pStr, vObjs, Json_EntryNode(vObjs, Entry2), Level+1, fComma, 0 );
265 }
266 fprintf( pFile, "%*s}%s\n", 3*(Level-1), "", fAddComma ? ",":"" );
267 }
268}
Here is the call graph for this function:
Here is the caller graph for this function:

◆ Nnc_LayerType2Str()

char * Nnc_LayerType2Str ( char * pStr)

FUNCTION DEFINITIONS ///.

Function*************************************************************

Synopsis [Writes JSON into a file.]

Description []

SideEffects []

SeeAlso []

Definition at line 52 of file ioJson.c.

53{
54 if ( !strcmp(pStr, "InputLayer") )
55 return "input ";
56 if ( !strcmp(pStr, "Conv2D") )
57 return "convo ";
58 if ( !strcmp(pStr, "BatchNormalization") )
59 return "batch ";
60 if ( !strcmp(pStr, "Activation") )
61 return "relu ";
62 if ( !strcmp(pStr, "Add") )
63 return "eltwise";
64 if ( !strcmp(pStr, "MaxPooling2D") )
65 return "pool ";
66 if ( !strcmp(pStr, "GlobalAveragePooling2D") )
67 return "pool ";
68 if ( !strcmp(pStr, "Dense") )
69 return "fullcon";
70 if ( !strcmp(pStr, "ZeroPadding2D") )
71 return "pad";
72// if ( !strcmp(pStr, "InputLayer") )
73// return "softmax";
74 return NULL;
75}
Here is the call graph for this function:
Here is the caller graph for this function: