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