From ab798b28b9f3b836db6c9eb9e4a25d57981022ae Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@fe.uni-lj.si> Date: Tue, 01 Mar 2016 08:51:46 +0000 Subject: [PATCH] Added resotration from vtu files by using --restore-from-vtk flag. Requires filename --- src/restore.c | 253 +++++++++++++++++++++++++++++++++++++++++++++++-- 1 files changed, 239 insertions(+), 14 deletions(-) diff --git a/src/restore.c b/src/restore.c index 647b871..19d8cad 100644 --- a/src/restore.c +++ b/src/restore.c @@ -9,11 +9,17 @@ #include <zlib.h> #include "vesicle.h" #include "vertex.h" -ts_bool parseDump(char *dumpfname) { - xmlDocPtr doc; - xmlNodePtr cur; - ts_vesicle *vesicle; +#include "triangle.h" +#include "bond.h" +#include "energy.h" +#include "poly.h" +#include "initial_distribution.h" +#include "io.h" +ts_vesicle *parseDump(char *dumpfname) { + xmlDocPtr doc; + xmlNodePtr cur, cur1,cur2; + ts_vesicle *vesicle=NULL; doc = xmlParseFile(dumpfname); if (doc == NULL ) { @@ -32,20 +38,66 @@ cur = cur->xmlChildrenNode; while (cur != NULL) { + + if ((!xmlStrcmp(cur->name, (const xmlChar *)"tape"))){ + setGlobalTapeTXTfromTapeTag(doc, cur); + } + if ((!xmlStrcmp(cur->name, (const xmlChar *)"trisurf"))){ vesicle=parseTrisurfTag(doc, cur); } - + // START Point Position data & Bonds + if ((!xmlStrcmp(cur->name, (const xmlChar *)"UnstructuredGrid"))){ + cur1 = cur->xmlChildrenNode; + while(cur1!=NULL){ + if ((!xmlStrcmp(cur1->name, (const xmlChar *)"Piece"))){ + cur2=cur1->xmlChildrenNode; + while(cur2!=NULL){ + if ((!xmlStrcmp(cur2->name, (const xmlChar *)"Points"))){ + fprintf(stderr,"Found point data\n"); + if(vesicle!=NULL) + parseXMLVertexPosition(vesicle, doc, cur2); + } + if ((!xmlStrcmp(cur2->name, (const xmlChar *)"Cells"))){ + fprintf(stderr,"Found cell(Bonds) data\n"); + if(vesicle!=NULL) + parseXMLBonds(vesicle, doc, cur2); + } + cur2=cur2->next; + } + } + cur1 = cur1->next; + } + } + // END Point Position data & Bonds cur = cur->next; } xmlFreeDoc(doc); + +// vesicle->poly_list=init_poly_list(0, 0, vesicle->vlist, vesicle); + + init_normal_vectors(vesicle->tlist); + mean_curvature_and_energy(vesicle); + +/* TODO: filaments */ + fprintf(stderr,"Restoration completed\n"); - exit(0); - vesicle_free(vesicle); +// write_vertex_xml_file(vesicle,999); +// vesicle_free(vesicle); +// exit(0); + return vesicle; +} + +ts_bool setGlobalTapeTXTfromTapeTag(xmlDocPtr doc, xmlNodePtr cur){ + xmlChar *tape = xmlNodeListGetString(doc, cur->xmlChildrenNode, 1); + strcpy(tapetxt,(char *)tape); + xmlFree(tape); return TS_SUCCESS; } + +/* this is a parser of additional data in xml */ ts_vesicle *parseTrisurfTag(xmlDocPtr doc, xmlNodePtr cur){ fprintf(stderr,"Parsing trisurf tag\n"); xmlNodePtr child; @@ -84,32 +136,49 @@ free(subtree); #endif /*parse xml subtree */ - xmlChar *nvtx, *npoly, *nfono; + xmlChar *nvtx, *npoly, *nmono; nvtx = xmlGetProp(cur, (xmlChar *)"nvtx"); npoly=xmlGetProp(cur, (xmlChar *)"npoly"); - nfono=xmlGetProp(cur, (xmlChar *)"nfono"); - fprintf(stderr,"nvtx=%u\n",atoi((char *)nvtx)); - ts_vesicle *vesicle=init_vesicle(atoi((char *)nvtx),10,10,10,0.1); + nmono=xmlGetProp(cur, (xmlChar *)"nmono"); + ts_tape *tape=parsetapebuffer(tapetxt); + //fprintf(stderr,"nvtx=%u\n",atoi((char *)nvtx)); + //TODO: check if nvtx is in agreement with nshell from tape + ts_vesicle *vesicle=init_vesicle(atoi((char *)nvtx),tape->ncxmax,tape->ncymax,tape->nczmax,tape->stepsize); //vesicle->poly_list=init_poly_list(atoi((char *)npoly),atoi((char *)nmono), vesicle->vlist, vesicle); xmlFree(nvtx); xmlFree(npoly); - xmlFree(nfono); + xmlFree(nmono); child = cur->xmlChildrenNode; while (child != NULL) { if ((!xmlStrcmp(child->name, (const xmlChar *)"vtxn"))){ parseTrisurfVtxn(vesicle->vlist, doc, child); } - + if ((!xmlStrcmp(child->name, (const xmlChar *)"tria"))){ + parseTrisurfTria(vesicle, doc, child); + } + if ((!xmlStrcmp(child->name, (const xmlChar *)"trianeigh"))){ + parseTrisurfTriaNeigh(vesicle, doc, child); + } + if ((!xmlStrcmp(child->name, (const xmlChar *)"tristar"))){ + parseTrisurfTristar(vesicle, doc, child); + } + child = child->next; } + vesicle->tape=tape; + set_vesicle_values_from_tape(vesicle); return vesicle; } -ts_bool *parseTrisurfVtxn(ts_vertex_list *vlist, xmlDocPtr doc, xmlNodePtr cur){ + + +/* Low level tags parsers */ + +ts_bool parseTrisurfVtxn(ts_vertex_list *vlist, xmlDocPtr doc, xmlNodePtr cur){ xmlChar *chari; xmlChar *neighs; @@ -136,3 +205,159 @@ return TS_SUCCESS; } +ts_bool parseTrisurfTria(ts_vesicle *vesicle, xmlDocPtr doc, xmlNodePtr cur){ + xmlChar *triangles; + char *tria; + char *vtx[3]; + + ts_uint i,j; + triangles = xmlNodeListGetString(doc, cur->xmlChildrenNode, 1); + tria=(char *)triangles; + vtx[0]=strtok(tria," "); + for(i=1;i<3;i++) vtx[i]=strtok(NULL," "); + j=0; + while(vtx[2]!=NULL){ + triangle_add(vesicle->tlist, vesicle->vlist->vtx[atoi(vtx[0])],vesicle->vlist->vtx[atoi(vtx[1])],vesicle->vlist->vtx[atoi(vtx[2])]); + for(i=0;i<3;i++) vtx[i]=strtok(NULL," "); + j++; + } + fprintf(stderr,"Parsing triangles %s j=%d\n",triangles,j); + + xmlFree(triangles); + return TS_SUCCESS; +} + + +ts_bool parseTrisurfTriaNeigh(ts_vesicle *vesicle, xmlDocPtr doc, xmlNodePtr cur){ + xmlChar *triangles; + char *tria; + char *ntria[3]; + ts_uint i,j; + triangles = xmlNodeListGetString(doc, cur->xmlChildrenNode, 1); + tria=(char *)triangles; + ntria[0]=strtok(tria," "); + for(i=1;i<3;i++) ntria[i]=strtok(NULL," "); + j=0; + while(ntria[2]!=NULL){ + triangle_add_neighbour(vesicle->tlist->tria[j],vesicle->tlist->tria[atoi(ntria[0])]); + triangle_add_neighbour(vesicle->tlist->tria[j],vesicle->tlist->tria[atoi(ntria[1])]); + triangle_add_neighbour(vesicle->tlist->tria[j],vesicle->tlist->tria[atoi(ntria[2])]); + j++; + for(i=0;i<3;i++) ntria[i]=strtok(NULL," "); + } + fprintf(stderr,"Parsing triangle neighbors j=%d\n",j); + + xmlFree(triangles); + return TS_SUCCESS; +} + + +ts_bool parseTrisurfTristar(ts_vesicle *vesicle, xmlDocPtr doc, xmlNodePtr cur){ + + xmlChar *chari; + xmlChar *tristar; + char *t; + char *token; + ts_uint neighi; + ts_uint i; + chari = xmlGetProp(cur, (xmlChar *)"idx"); + i=atoi((char *)chari); + xmlFree(chari); + ts_vertex *vtx=vesicle->vlist->vtx[i]; + tristar = xmlNodeListGetString(doc, cur->xmlChildrenNode, 1); +// fprintf(stderr,"Found tristar for vtx %u that seems to have index %u with tristar=%s\n",i,vtx->idx,tristar); + + t=(char *)tristar; + token=strtok(t," "); + while(token!=NULL){ + neighi=atoi(token); + //fprintf(stderr,"%u", neighi); + vertex_add_tristar(vtx,vesicle->tlist->tria[neighi]); + token=strtok(NULL," "); + } + xmlFree(tristar); + return TS_SUCCESS; +} + + +/* this is a parser of vertex positions and bonds from main xml data */ +ts_bool parseXMLVertexPosition(ts_vesicle *vesicle,xmlDocPtr doc, xmlNodePtr cur){ + xmlNodePtr child = cur->xmlChildrenNode; + xmlChar *points; + char *pts; + int i, idx, polyidx, monoidx; + char *token[3]; + while (child != NULL) { + if ((!xmlStrcmp(child->name, (const xmlChar *)"DataArray"))){ + points = xmlNodeListGetString(doc, child->xmlChildrenNode, 1); + pts=(char *)points; + token[0]=strtok(pts," "); + token[1]=strtok(NULL," "); + token[2]=strtok(NULL,"\n"); + idx=0; + while(token[0]!=NULL){ + if(idx<vesicle->vlist->n){ + vesicle->vlist->vtx[idx]->x=atof(token[0]); + vesicle->vlist->vtx[idx]->y=atof(token[1]); + vesicle->vlist->vtx[idx]->z=atof(token[2]); + } else { + polyidx=(idx-vesicle->vlist->n)/vesicle->tape->nmono; + monoidx=(idx-vesicle->vlist->n)%vesicle->tape->nmono; + vesicle->poly_list->poly[polyidx]->vlist->vtx[monoidx]->x=atof(token[0]); + vesicle->poly_list->poly[polyidx]->vlist->vtx[monoidx]->y=atof(token[1]); + vesicle->poly_list->poly[polyidx]->vlist->vtx[monoidx]->z=atof(token[2]); + } + for(i=0;i<2;i++) token[i]=strtok(NULL," "); + token[2]=strtok(NULL,"\n"); + idx++; + } + xmlFree(points); + } + child=child->next; + } + fprintf(stderr,"Vertices position j=%d\n",idx); + + return TS_SUCCESS; +} +ts_bool parseXMLBonds(ts_vesicle *vesicle,xmlDocPtr doc, xmlNodePtr cur){ + xmlNodePtr child = cur->xmlChildrenNode; + xmlChar *bonds, *conname; + char *b; + int idx, polyidx; + char *token[2]; + while (child != NULL) { + conname=xmlGetProp(child, (xmlChar *)"Name"); + if ((!xmlStrcmp(child->name, (const xmlChar *)"DataArray")) && !xmlStrcmp(conname, (const xmlChar *)"connectivity") ){ + bonds = xmlNodeListGetString(doc, child->xmlChildrenNode, 1); + b=(char *)bonds; + token[0]=strtok(b," "); + token[1]=strtok(NULL,"\n"); + idx=0; + while(token[0]!=NULL){ + if(idx<3*(vesicle->vlist->n-2)){ + bond_add(vesicle->blist, vesicle->vlist->vtx[atoi(token[0])], vesicle->vlist->vtx[atoi(token[1])]); + } + else { + //find grafted vtx + if((vesicle->tape->nmono-1)==(idx-3*(vesicle->vlist->n-2))%(vesicle->tape->nmono)){ + polyidx=(idx-3*(vesicle->vlist->n-2))/(vesicle->tape->nmono); + fprintf(stderr,"poly=%d, vertex=%d\n",polyidx,atoi(token[0])); + vesicle->poly_list->poly[polyidx]->grafted_vtx=vesicle->vlist->vtx[atoi(token[0])]; + vesicle->vlist->vtx[atoi(token[0])]->grafted_poly=vesicle->poly_list->poly[polyidx]; + } + } + token[0]=strtok(NULL," "); + token[1]=strtok(NULL,"\n"); + idx++; + } + xmlFree(bonds); + } + xmlFree(conname); + child=child->next; + } + fprintf(stderr,"Bond data j=%d\n",idx); + return TS_SUCCESS; +} + + + -- Gitblit v1.9.3