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 |  110 +++++++++++++++++++++++++++++++++++++++++--------------
 1 files changed, 82 insertions(+), 28 deletions(-)

diff --git a/src/restore.c b/src/restore.c
index 0ca42ec..19d8cad 100644
--- a/src/restore.c
+++ b/src/restore.c
@@ -12,13 +12,14 @@
 #include "triangle.h"
 #include "bond.h"
 #include "energy.h"
+#include "poly.h"
 #include "initial_distribution.h"
+#include "io.h"
 
-ts_bool parseDump(char *dumpfname) {
+ts_vesicle *parseDump(char *dumpfname) {
 	xmlDocPtr doc;
 	xmlNodePtr cur, cur1,cur2;
 	ts_vesicle *vesicle=NULL;
-
 	doc = xmlParseFile(dumpfname);
 	
 	if (doc == NULL ) {
@@ -37,6 +38,11 @@
 	
 	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);
 		}
@@ -50,12 +56,12 @@
 						if ((!xmlStrcmp(cur2->name, (const xmlChar *)"Points"))){
 							fprintf(stderr,"Found point data\n");
 							if(vesicle!=NULL)
-								parseXMLVertexPosition(vesicle, doc, cur);
+								parseXMLVertexPosition(vesicle, doc, cur2);
 						}
 						if ((!xmlStrcmp(cur2->name, (const xmlChar *)"Cells"))){
 						fprintf(stderr,"Found cell(Bonds) data\n");
 							if(vesicle!=NULL)
-								parseXMLBonds(vesicle, doc, cur);
+								parseXMLBonds(vesicle, doc, cur2);
 						}
 						cur2=cur2->next;
 					}	
@@ -68,17 +74,27 @@
 	}
 	
 	xmlFreeDoc(doc);
+
+//	vesicle->poly_list=init_poly_list(0, 0, vesicle->vlist, vesicle);
+
 	init_normal_vectors(vesicle->tlist);
 	mean_curvature_and_energy(vesicle);
 
-/* TODO: cells, polymeres, filaments, core, tape */
+/* TODO: filaments */
 
 	fprintf(stderr,"Restoration completed\n");
-	exit(0);
-	vesicle_free(vesicle);
-	return TS_SUCCESS;
+//	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 */
@@ -120,16 +136,18 @@
 	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) {
@@ -150,6 +168,8 @@
 	}
 
 
+	vesicle->tape=tape;
+	set_vesicle_values_from_tape(vesicle);
 
 	return vesicle;
 }
@@ -190,14 +210,18 @@
 	char *tria;
 	char *vtx[3];
 	
-	ts_uint i;
+	ts_uint i,j;
 	triangles = xmlNodeListGetString(doc, cur->xmlChildrenNode, 1);
 	tria=(char *)triangles;
-	for(i=0;i<3;i++)	vtx[i]=strtok(tria," ");
+	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;
@@ -211,7 +235,8 @@
 	ts_uint i,j;
 	triangles = xmlNodeListGetString(doc, cur->xmlChildrenNode, 1);
 	tria=(char *)triangles;
-	for(i=0;i<3;i++)	ntria[i]=strtok(tria," ");
+	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])]);
@@ -260,48 +285,77 @@
 	xmlNodePtr child = cur->xmlChildrenNode;
 	xmlChar *points;
 	char *pts;
-	int i, idx;
+	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;
-			for(i=0;i<3;i++)	token[i]=strtok(pts," ");
+			token[0]=strtok(pts," ");
+			token[1]=strtok(NULL," ");
+			token[2]=strtok(NULL,"\n");
 			idx=0;
 			while(token[0]!=NULL){
-				vesicle->vlist->vtx[idx]->x=atof(token[0]);
-				vesicle->vlist->vtx[idx]->y=atof(token[1]);
-				vesicle->vlist->vtx[idx]->z=atof(token[2]);
-				for(i=0;i<3;i++)	token[i]=strtok(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;
+	xmlChar *bonds, *conname;
 	char *b;
-	int i, idx;
+	int idx, polyidx;
 	char *token[2];
 	while (child != NULL) {
-		if ((!xmlStrcmp(child->name, (const xmlChar *)"DataArray"))){
+		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;
-			for(i=0;i<2;i++)	token[i]=strtok(b," ");
+			token[0]=strtok(b," ");
+			token[1]=strtok(NULL,"\n");
 			idx=0;
 			while(token[0]!=NULL){
-				bond_add(vesicle->blist, vesicle->vlist->vtx[atoi(token[0])], vesicle->vlist->vtx[atoi(token[1])]);
-				for(i=0;i<2;i++)	token[i]=strtok(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