misc-template.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685
  1. /* LIBDGL -- a Directed Graph Library implementation
  2. * Copyright (C) 2002 Roberto Micarelli
  3. *
  4. * This program is free software; you can redistribute it and/or modify
  5. * it under the terms of the GNU General Public License as published by
  6. * the Free Software Foundation; either version 2 of the License, or
  7. * (at your option) any later version.
  8. *
  9. * This program is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. * GNU General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU General Public License
  15. * along with this program; if not, write to the Free Software
  16. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17. */
  18. /*
  19. * best view with tabstop=4
  20. */
  21. /*
  22. * Edge Traversing
  23. */
  24. int DGL_EDGE_T_INITIALIZE_FUNC(dglGraph_s * pGraph, dglEdgeTraverser_s * pT,
  25. dglEdgePrioritizer_s * pEP)
  26. {
  27. #if defined(_DGL_V1)
  28. pGraph->iErrno = DGL_ERR_NotSupported;
  29. return -pGraph->iErrno;
  30. #else
  31. if (pGraph->Flags & DGL_GS_FLAT) {
  32. if (pEP && pEP->pvAVL) {
  33. if ((pT->pvAVLT = malloc(sizeof(struct avl_traverser))) == NULL) {
  34. pGraph->iErrno = DGL_ERR_MemoryExhausted;
  35. return -pGraph->iErrno;
  36. }
  37. avl_t_init(pT->pvAVLT, pEP->pvAVL);
  38. pT->pnEdge = NULL;
  39. pT->pEdgePrioritizer = pEP;
  40. }
  41. else {
  42. pT->pvAVLT = NULL;
  43. pT->pnEdge = NULL;
  44. pT->pEdgePrioritizer = NULL;
  45. }
  46. }
  47. else {
  48. if ((pT->pvAVLT = malloc(sizeof(struct avl_traverser))) == NULL) {
  49. pGraph->iErrno = DGL_ERR_MemoryExhausted;
  50. return -pGraph->iErrno;
  51. }
  52. if (pEP && pEP->pvAVL) {
  53. avl_t_init(pT->pvAVLT, pEP->pvAVL);
  54. pT->pnEdge = NULL;
  55. pT->pEdgePrioritizer = pEP;
  56. }
  57. else {
  58. avl_t_init(pT->pvAVLT, pGraph->pEdgeTree);
  59. pT->pnEdge = NULL;
  60. pT->pEdgePrioritizer = NULL;
  61. }
  62. }
  63. pT->pGraph = pGraph;
  64. return 0;
  65. #endif
  66. }
  67. void DGL_EDGE_T_RELEASE_FUNC(dglEdgeTraverser_s * pT)
  68. {
  69. #if defined(_DGL_V1)
  70. pT->pGraph->iErrno = DGL_ERR_NotSupported;
  71. #else
  72. if (pT->pvAVLT)
  73. free(pT->pvAVLT);
  74. pT->pvAVLT = NULL;
  75. pT->pnEdge = NULL;
  76. pT->pEdgePrioritizer = NULL;
  77. #endif
  78. }
  79. dglInt32_t *DGL_EDGE_T_FIRST_FUNC(dglEdgeTraverser_s * pT)
  80. {
  81. #if defined(_DGL_V1)
  82. pT->pGraph->iErrno = DGL_ERR_NotSupported;
  83. return NULL;
  84. #else
  85. dglGraph_s *pG = pT->pGraph;
  86. pT->pnEdge = NULL;
  87. if (pT->pvAVLT && pT->pEdgePrioritizer) {
  88. dglEdgePrioritizer_s *pPri = pT->pEdgePrioritizer;
  89. dglTreeEdgePri32_s *pItem;
  90. pItem = avl_t_first(pT->pvAVLT, pPri->pvAVL);
  91. if (pItem) {
  92. /*
  93. printf("edge_t_first: cEdge=%ld\n", pItem->cnData);
  94. */
  95. pPri->cEdge = pItem->cnData;
  96. pPri->iEdge = 0;
  97. if (pPri->iEdge < pPri->cEdge) {
  98. pT->pnEdge =
  99. DGL_GET_EDGE_FUNC(pG, pItem->pnData[pPri->iEdge]);
  100. pPri->iEdge++;
  101. }
  102. }
  103. pPri->pEdgePri32Item = pItem;
  104. }
  105. else if (pT->pvAVLT) {
  106. dglTreeEdge_s *pEdgeItem;
  107. if ((pEdgeItem = avl_t_first(pT->pvAVLT, pG->pEdgeTree)) == NULL) {
  108. pT->pnEdge = NULL;
  109. }
  110. else {
  111. pT->pnEdge = pEdgeItem->pv;
  112. }
  113. }
  114. else {
  115. if (pG->cEdge > 0)
  116. pT->pnEdge = (dglInt32_t *) pG->pEdgeBuffer;
  117. else
  118. pT->pnEdge = NULL;
  119. }
  120. return pT->pnEdge;
  121. #endif
  122. }
  123. dglInt32_t *DGL_EDGE_T_NEXT_FUNC(dglEdgeTraverser_s * pT)
  124. {
  125. #if defined(_DGL_V1)
  126. pT->pGraph->iErrno = DGL_ERR_NotSupported;
  127. return NULL;
  128. #else
  129. dglGraph_s *pG = pT->pGraph;
  130. pT->pnEdge = NULL;
  131. if (pT->pvAVLT && pT->pEdgePrioritizer) {
  132. dglEdgePrioritizer_s *pPri = pT->pEdgePrioritizer;
  133. dglTreeEdgePri32_s *pItem = pPri->pEdgePri32Item;
  134. if (pItem && pPri->iEdge < pPri->cEdge) {
  135. pT->pnEdge = DGL_GET_EDGE_FUNC(pG, pItem->pnData[pPri->iEdge]);
  136. pPri->iEdge++;
  137. }
  138. else {
  139. if ((pItem = avl_t_next(pT->pvAVLT)) != NULL) {
  140. pPri->cEdge = pItem->cnData;
  141. pPri->iEdge = 0;
  142. if (pPri->iEdge < pPri->cEdge) {
  143. pT->pnEdge =
  144. DGL_GET_EDGE_FUNC(pG, pItem->pnData[pPri->iEdge]);
  145. pPri->iEdge++;
  146. }
  147. }
  148. pPri->pEdgePri32Item = pItem;
  149. }
  150. }
  151. else if (pT->pvAVLT) {
  152. dglTreeEdge_s *pItem;
  153. if ((pItem = avl_t_next(pT->pvAVLT)) != NULL) {
  154. pT->pnEdge = pItem->pv;
  155. }
  156. }
  157. else {
  158. pT->pnEdge += DGL_NODE_WSIZE(pG->EdgeAttrSize);
  159. if (pT->pnEdge >= (dglInt32_t *) (pG->pEdgeBuffer + pG->iEdgeBuffer)) {
  160. pT->pnEdge = NULL;
  161. }
  162. }
  163. return pT->pnEdge;
  164. #endif
  165. }
  166. /*
  167. * Node Traversing
  168. */
  169. int DGL_NODE_T_INITIALIZE_FUNC(dglGraph_s * pGraph, dglNodeTraverser_s * pT)
  170. {
  171. if (pGraph->Flags & DGL_GS_FLAT) {
  172. pT->pnNode = NULL;
  173. pT->pvAVLT = NULL;
  174. }
  175. else {
  176. if ((pT->pvAVLT = malloc(sizeof(struct avl_traverser))) == NULL) {
  177. pGraph->iErrno = DGL_ERR_MemoryExhausted;
  178. return -pGraph->iErrno;
  179. }
  180. avl_t_init(pT->pvAVLT, pGraph->pNodeTree);
  181. pT->pnNode = NULL;
  182. }
  183. pT->pGraph = pGraph;
  184. return 0;
  185. }
  186. void DGL_NODE_T_RELEASE_FUNC(dglNodeTraverser_s * pT)
  187. {
  188. if (pT->pvAVLT)
  189. free(pT->pvAVLT);
  190. pT->pvAVLT = NULL;
  191. pT->pnNode = NULL;
  192. }
  193. dglInt32_t *DGL_NODE_T_FIRST_FUNC(dglNodeTraverser_s * pT)
  194. {
  195. DGL_T_NODEITEM_TYPE *pNodeItem;
  196. if (pT->pvAVLT) {
  197. if ((pNodeItem =
  198. avl_t_first(pT->pvAVLT, pT->pGraph->pNodeTree)) == NULL)
  199. pT->pnNode = NULL;
  200. else
  201. pT->pnNode = DGL_T_NODEITEM_NodePTR(pNodeItem);
  202. }
  203. else {
  204. if (pT->pGraph->cNode > 0)
  205. pT->pnNode = (dglInt32_t *) pT->pGraph->pNodeBuffer;
  206. else
  207. pT->pnNode = NULL;
  208. }
  209. return pT->pnNode;
  210. }
  211. dglInt32_t *DGL_NODE_T_NEXT_FUNC(dglNodeTraverser_s * pT)
  212. {
  213. DGL_T_NODEITEM_TYPE *pNodeItem;
  214. if (pT->pvAVLT) {
  215. if ((pNodeItem = avl_t_next(pT->pvAVLT)) == NULL)
  216. pT->pnNode = NULL;
  217. else
  218. pT->pnNode = DGL_T_NODEITEM_NodePTR(pNodeItem);
  219. }
  220. else {
  221. pT->pnNode += DGL_NODE_WSIZE(pT->pGraph->NodeAttrSize);
  222. if (pT->pnNode >=
  223. (dglInt32_t *) (pT->pGraph->pNodeBuffer +
  224. pT->pGraph->iNodeBuffer))
  225. pT->pnNode = NULL;
  226. }
  227. return pT->pnNode;
  228. }
  229. dglInt32_t *DGL_NODE_T_FIND_FUNC(dglNodeTraverser_s * pT, dglInt32_t nNodeId)
  230. {
  231. DGL_T_NODEITEM_TYPE *pNodeItem, findItem;
  232. if (pT->pvAVLT) {
  233. findItem.nKey = nNodeId;
  234. if ((pNodeItem =
  235. avl_t_find(pT->pvAVLT, pT->pGraph->pNodeTree,
  236. &findItem)) == NULL)
  237. pT->pnNode = NULL;
  238. else
  239. pT->pnNode = DGL_T_NODEITEM_NodePTR(pNodeItem);
  240. }
  241. else {
  242. pT->pnNode = DGL_GET_NODE_FUNC(pT->pGraph, nNodeId);
  243. }
  244. return pT->pnNode;
  245. }
  246. /*
  247. * Edgeset Traversing
  248. */
  249. int DGL_EDGESET_T_INITIALIZE_FUNC(dglGraph_s * pGraph,
  250. dglEdgesetTraverser_s * pT,
  251. dglInt32_t * pnEdgeset)
  252. {
  253. pT->pGraph = pGraph;
  254. pT->pnEdgeset = pnEdgeset;
  255. pT->cEdge = (pnEdgeset) ? *pnEdgeset : 0;
  256. pT->iEdge = 0;
  257. return 0;
  258. }
  259. void DGL_EDGESET_T_RELEASE_FUNC(dglEdgesetTraverser_s * pT)
  260. {
  261. }
  262. dglInt32_t *DGL_EDGESET_T_FIRST_FUNC(dglEdgesetTraverser_s * pT)
  263. {
  264. #if defined(_DGL_V2)
  265. dglInt32_t *pnOffset;
  266. dglTreeEdge_s *pEdgeItem, EdgeItem;
  267. #endif
  268. if (pT->cEdge == 0)
  269. return NULL;
  270. pT->iEdge = 1;
  271. #if defined(_DGL_V1)
  272. return pT->pnEdgeset + 1;
  273. #endif
  274. #if defined(_DGL_V2)
  275. pnOffset = pT->pnEdgeset + 1;
  276. if (pT->pGraph->Flags & DGL_GS_FLAT) {
  277. pT->pvCurrentItem = NULL;
  278. return DGL_EDGEBUFFER_SHIFT(pT->pGraph, *pnOffset);
  279. }
  280. else {
  281. EdgeItem.nKey = *pnOffset;
  282. if ((pEdgeItem = avl_find(pT->pGraph->pEdgeTree, &EdgeItem)) != NULL) {
  283. pT->pvCurrentItem = pEdgeItem;
  284. return pEdgeItem->pv;
  285. }
  286. }
  287. #endif
  288. return NULL;
  289. }
  290. dglInt32_t *DGL_EDGESET_T_NEXT_FUNC(dglEdgesetTraverser_s * pT)
  291. {
  292. #if defined(_DGL_V2)
  293. dglInt32_t *pnOffset;
  294. dglTreeEdge_s *pEdgeItem, EdgeItem;
  295. #endif
  296. if (pT->cEdge > 0 && pT->iEdge < pT->cEdge) {
  297. #if defined(_DGL_V1)
  298. return DGL_EDGESET_EDGE_PTR(pT->pnEdgeset, pT->iEdge++,
  299. pT->pGraph->EdgeAttrSize);
  300. #endif
  301. #if defined(_DGL_V2)
  302. pnOffset = pT->pnEdgeset + 1 + pT->iEdge++;
  303. if (pT->pGraph->Flags & DGL_GS_FLAT) {
  304. return DGL_EDGEBUFFER_SHIFT(pT->pGraph, *pnOffset);
  305. }
  306. else {
  307. EdgeItem.nKey = *pnOffset;
  308. if ((pEdgeItem =
  309. avl_find(pT->pGraph->pEdgeTree, &EdgeItem)) != NULL) {
  310. pT->pvCurrentItem = pEdgeItem;
  311. return pEdgeItem->pv;
  312. }
  313. }
  314. #endif
  315. }
  316. return NULL;
  317. }
  318. /*
  319. * Flatten the graph
  320. */
  321. int DGL_FLATTEN_FUNC(dglGraph_s * pgraph)
  322. {
  323. register DGL_T_NODEITEM_TYPE *ptreenode;
  324. #if defined(_DGL_V2)
  325. register dglTreeEdge_s *ptreeEdge;
  326. int i;
  327. #endif
  328. register dglInt32_t *pEdge;
  329. register dglInt32_t *pnode;
  330. register dglInt32_t *pnodescan;
  331. dglInt32_t *pOutEdgeset;
  332. dglInt32_t *pInEdgeset;
  333. int cOutEdgeset;
  334. int cInEdgeset;
  335. struct avl_traverser avlTraverser;
  336. if (pgraph->Flags & DGL_GS_FLAT) {
  337. pgraph->iErrno = DGL_ERR_BadOnFlatGraph;
  338. return -pgraph->iErrno;
  339. }
  340. pgraph->pNodeBuffer = NULL; /* should be already */
  341. pgraph->iNodeBuffer = 0;
  342. pgraph->pEdgeBuffer = NULL;
  343. pgraph->iEdgeBuffer = 0;
  344. #if defined(_DGL_V2)
  345. /*
  346. printf("flatten: traversing edges\n");
  347. */
  348. avl_t_init(&avlTraverser, pgraph->pEdgeTree);
  349. for (ptreeEdge = avl_t_first(&avlTraverser, pgraph->pEdgeTree);
  350. ptreeEdge; ptreeEdge = avl_t_next(&avlTraverser)
  351. ) {
  352. pEdge = ptreeEdge->pv;
  353. /*
  354. printf( "flatten: add edge %ld to edge buffer\n", DGL_EDGE_ID(pEdge) );
  355. */
  356. pgraph->pEdgeBuffer = realloc(pgraph->pEdgeBuffer,
  357. pgraph->iEdgeBuffer +
  358. DGL_EDGE_SIZEOF(pgraph->EdgeAttrSize)
  359. );
  360. if (pgraph->pEdgeBuffer == NULL) {
  361. pgraph->iErrno = DGL_ERR_MemoryExhausted;
  362. return -pgraph->iErrno;
  363. }
  364. memcpy(pgraph->pEdgeBuffer + pgraph->iEdgeBuffer, pEdge,
  365. DGL_EDGE_SIZEOF(pgraph->EdgeAttrSize));
  366. pgraph->iEdgeBuffer += DGL_EDGE_SIZEOF(pgraph->EdgeAttrSize);
  367. }
  368. #endif
  369. /*
  370. printf("flatten: traversing nodes\n");
  371. */
  372. avl_t_init(&avlTraverser, pgraph->pNodeTree);
  373. for (ptreenode = avl_t_first(&avlTraverser, pgraph->pNodeTree);
  374. ptreenode; ptreenode = avl_t_next(&avlTraverser)
  375. ) {
  376. pnode = DGL_T_NODEITEM_NodePTR(ptreenode);
  377. pOutEdgeset = DGL_T_NODEITEM_OutEdgesetPTR(ptreenode);
  378. pInEdgeset = DGL_T_NODEITEM_InEdgesetPTR(ptreenode);
  379. if (!(DGL_NODE_STATUS(pnode) & DGL_NS_ALONE)) {
  380. cOutEdgeset = (pOutEdgeset) ?
  381. DGL_EDGESET_SIZEOF(DGL_EDGESET_EDGECOUNT(pOutEdgeset),
  382. pgraph->EdgeAttrSize) : sizeof(dglInt32_t);
  383. #if !defined(_DGL_V1)
  384. cInEdgeset = (pInEdgeset) ?
  385. DGL_EDGESET_SIZEOF(DGL_EDGESET_EDGECOUNT(pInEdgeset),
  386. pgraph->EdgeAttrSize) : sizeof(dglInt32_t);
  387. #else
  388. cInEdgeset = 0;
  389. #endif
  390. pgraph->pEdgeBuffer = realloc(pgraph->pEdgeBuffer,
  391. pgraph->iEdgeBuffer + cOutEdgeset +
  392. cInEdgeset);
  393. if (pgraph->pEdgeBuffer == NULL) {
  394. pgraph->iErrno = DGL_ERR_MemoryExhausted;
  395. return -pgraph->iErrno;
  396. }
  397. {
  398. dglInt32_t nDummy = 0;
  399. memcpy(pgraph->pEdgeBuffer + pgraph->iEdgeBuffer,
  400. (pOutEdgeset) ? pOutEdgeset : &nDummy, cOutEdgeset);
  401. #if !defined(_DGL_V1)
  402. memcpy(pgraph->pEdgeBuffer + pgraph->iEdgeBuffer +
  403. cOutEdgeset, (pInEdgeset) ? pInEdgeset : &nDummy,
  404. cInEdgeset);
  405. #endif
  406. }
  407. DGL_NODE_EDGESET_OFFSET(pnode) = pgraph->iEdgeBuffer;
  408. pgraph->iEdgeBuffer += cOutEdgeset + cInEdgeset;
  409. }
  410. pgraph->pNodeBuffer =
  411. realloc(pgraph->pNodeBuffer,
  412. pgraph->iNodeBuffer +
  413. DGL_NODE_SIZEOF(pgraph->NodeAttrSize));
  414. if (pgraph->pNodeBuffer == NULL) {
  415. pgraph->iErrno = DGL_ERR_MemoryExhausted;
  416. return -pgraph->iErrno;
  417. }
  418. memcpy(pgraph->pNodeBuffer + pgraph->iNodeBuffer, pnode,
  419. DGL_NODE_SIZEOF(pgraph->NodeAttrSize));
  420. pgraph->iNodeBuffer += DGL_NODE_SIZEOF(pgraph->NodeAttrSize);
  421. }
  422. #if defined(_DGL_V2)
  423. if (pgraph->pEdgeTree) {
  424. avl_destroy(pgraph->pEdgeTree, dglTreeEdgeCancel);
  425. pgraph->pEdgeTree = NULL;
  426. }
  427. #endif
  428. if (pgraph->pNodeTree) {
  429. avl_destroy(pgraph->pNodeTree, dglTreeNodeCancel);
  430. pgraph->pNodeTree = NULL;
  431. }
  432. pgraph->Flags |= DGL_GS_FLAT; /* flattened */
  433. /*
  434. * convert node-id to node-offset
  435. */
  436. DGL_FOREACH_NODE(pgraph, pnodescan) {
  437. if (!(DGL_NODE_STATUS(pnodescan) & DGL_NS_ALONE)) {
  438. pOutEdgeset =
  439. DGL_EDGEBUFFER_SHIFT(pgraph,
  440. DGL_NODE_EDGESET_OFFSET(pnodescan));
  441. #if defined(_DGL_V2)
  442. for (i = 0; i < pOutEdgeset[0]; i++) {
  443. /*
  444. printf("flatten: node %ld: scan out edge %ld/%ld - %ld\n", DGL_NODE_ID(pnodescan), i+1, pOutEdgeset[0], pOutEdgeset[i+1]);
  445. */
  446. pEdge = DGL_GET_EDGE_FUNC(pgraph, pOutEdgeset[i + 1]);
  447. if (pEdge == NULL) {
  448. pgraph->iErrno = DGL_ERR_UnexpectedNullPointer;
  449. return -pgraph->iErrno;
  450. }
  451. /*
  452. printf(" retrieved id %ld\n", DGL_EDGE_ID(pEdge) );
  453. */
  454. pOutEdgeset[i + 1] = DGL_EDGEBUFFER_OFFSET(pgraph, pEdge);
  455. }
  456. pInEdgeset = pOutEdgeset + pOutEdgeset[0] + 1;
  457. for (i = 0; i < pInEdgeset[0]; i++) {
  458. /*
  459. printf("flatten: node %ld: scan in edge %ld/%ld - %ld\n",
  460. DGL_NODE_ID(pnodescan), i+1, pInEdgeset[0], pInEdgeset[i+1]);
  461. */
  462. pEdge = DGL_GET_EDGE_FUNC(pgraph, pInEdgeset[i + 1]);
  463. if (pEdge == NULL) {
  464. pgraph->iErrno = DGL_ERR_UnexpectedNullPointer;
  465. return -pgraph->iErrno;
  466. }
  467. /*
  468. printf(" retrieved id %ld\n", DGL_EDGE_ID(pEdge) );
  469. */
  470. pInEdgeset[i + 1] = DGL_EDGEBUFFER_OFFSET(pgraph, pEdge);
  471. }
  472. #endif
  473. #if defined(_DGL_V2)
  474. {
  475. int iEdge;
  476. DGL_FOREACH_EDGE(pgraph, pOutEdgeset, pEdge, iEdge)
  477. #else
  478. DGL_FOREACH_EDGE(pgraph, pOutEdgeset, pEdge)
  479. #endif
  480. {
  481. if ((pnode =
  482. DGL_GET_NODE_FUNC(pgraph,
  483. DGL_EDGE_HEADNODE_OFFSET(pEdge))) ==
  484. NULL) {
  485. pgraph->iErrno = DGL_ERR_HeadNodeNotFound;
  486. return -pgraph->iErrno;
  487. }
  488. DGL_EDGE_HEADNODE_OFFSET(pEdge) =
  489. DGL_NODEBUFFER_OFFSET(pgraph, pnode);
  490. if ((pnode =
  491. DGL_GET_NODE_FUNC(pgraph,
  492. DGL_EDGE_TAILNODE_OFFSET(pEdge))) ==
  493. NULL) {
  494. pgraph->iErrno = DGL_ERR_TailNodeNotFound;
  495. return -pgraph->iErrno;
  496. }
  497. DGL_EDGE_TAILNODE_OFFSET(pEdge) =
  498. DGL_NODEBUFFER_OFFSET(pgraph, pnode);
  499. }
  500. #if defined(_DGL_V2)
  501. }
  502. #endif
  503. }
  504. }
  505. return 0;
  506. }
  507. int DGL_UNFLATTEN_FUNC(dglGraph_s * pgraph)
  508. {
  509. register dglInt32_t *pHead;
  510. register dglInt32_t *pTail;
  511. register dglInt32_t *pEdge;
  512. register dglInt32_t *pEdgeset;
  513. int nret;
  514. if (!(pgraph->Flags & DGL_GS_FLAT)) {
  515. pgraph->iErrno = DGL_ERR_BadOnTreeGraph;
  516. return -pgraph->iErrno;
  517. }
  518. /*
  519. * unflag it now to avoid DGL_ADD_EDGE_FUNC() failure
  520. */
  521. pgraph->Flags &= ~DGL_GS_FLAT;
  522. pgraph->cNode = 0;
  523. pgraph->cEdge = 0;
  524. pgraph->cHead = 0;
  525. pgraph->cTail = 0;
  526. pgraph->cAlone = 0;
  527. pgraph->nnCost = (dglInt64_t) 0;
  528. if (pgraph->pNodeTree == NULL)
  529. pgraph->pNodeTree =
  530. avl_create(DGL_T_NODEITEM_Compare, NULL, dglTreeGetAllocator());
  531. if (pgraph->pNodeTree == NULL) {
  532. pgraph->iErrno = DGL_ERR_MemoryExhausted;
  533. return -pgraph->iErrno;
  534. }
  535. #if defined(_DGL_V1)
  536. pgraph->pEdgeTree = NULL;
  537. #else
  538. if (pgraph->pEdgeTree == NULL)
  539. pgraph->pEdgeTree =
  540. avl_create(dglTreeEdgeCompare, NULL, dglTreeGetAllocator());
  541. if (pgraph->pEdgeTree == NULL) {
  542. pgraph->iErrno = DGL_ERR_MemoryExhausted;
  543. return -pgraph->iErrno;
  544. }
  545. #endif
  546. DGL_FOREACH_NODE(pgraph, pHead) {
  547. if (DGL_NODE_STATUS(pHead) & DGL_NS_HEAD) {
  548. pEdgeset =
  549. DGL_EDGEBUFFER_SHIFT(pgraph, DGL_NODE_EDGESET_OFFSET(pHead));
  550. #if defined(_DGL_V2)
  551. {
  552. int iEdge;
  553. DGL_FOREACH_EDGE(pgraph, pEdgeset, pEdge, iEdge)
  554. #else
  555. DGL_FOREACH_EDGE(pgraph, pEdgeset, pEdge)
  556. #endif
  557. {
  558. pTail =
  559. DGL_NODEBUFFER_SHIFT(pgraph,
  560. DGL_EDGE_TAILNODE_OFFSET(pEdge));
  561. nret = DGL_ADD_EDGE_FUNC(pgraph,
  562. DGL_NODE_ID(pHead),
  563. DGL_NODE_ID(pTail),
  564. DGL_EDGE_COST(pEdge),
  565. DGL_EDGE_ID(pEdge),
  566. DGL_NODE_ATTR_PTR(pHead),
  567. DGL_NODE_ATTR_PTR(pTail),
  568. DGL_EDGE_ATTR_PTR(pEdge), 0);
  569. if (nret < 0) {
  570. goto error;
  571. }
  572. }
  573. #if defined(_DGL_V2)
  574. }
  575. #endif
  576. }
  577. else
  578. if (DGL_NODE_STATUS(pHead) & DGL_NS_ALONE) {
  579. nret = DGL_ADD_NODE_FUNC(pgraph,
  580. DGL_NODE_ID(pHead), DGL_NODE_ATTR_PTR(pHead), 0);
  581. if (nret < 0) {
  582. goto error;
  583. }
  584. }
  585. }
  586. /* move away flat-state data
  587. */
  588. if (pgraph->pNodeBuffer)
  589. free(pgraph->pNodeBuffer);
  590. if (pgraph->pEdgeBuffer)
  591. free(pgraph->pEdgeBuffer);
  592. pgraph->pNodeBuffer = NULL;
  593. pgraph->pEdgeBuffer = NULL;
  594. return 0;
  595. error:
  596. if (pgraph->pNodeTree)
  597. avl_destroy(pgraph->pNodeTree, dglTreeNodeCancel);
  598. if (pgraph->pEdgeTree)
  599. avl_destroy(pgraph->pEdgeTree, dglTreeEdgeCancel);
  600. pgraph->pNodeTree = NULL;
  601. pgraph->pEdgeTree = NULL;
  602. pgraph->Flags |= DGL_GS_FLAT;
  603. return nret;
  604. }