From 459ff94855b2659de14ac2ac902b9658d28e19d5 Mon Sep 17 00:00:00 2001
From: Samo Penic <samo@CAE-linux.(none)>
Date: Mon, 21 Apr 2014 11:11:37 +0000
Subject: [PATCH] Added complex spherical harmonics.

---
 src/initial_distribution.c |  279 +++++++++++++++++++++++++++++++++++--------------------
 1 files changed, 178 insertions(+), 101 deletions(-)

diff --git a/src/initial_distribution.c b/src/initial_distribution.c
index 4881e11..bff9f05 100644
--- a/src/initial_distribution.c
+++ b/src/initial_distribution.c
@@ -9,26 +9,103 @@
 #include "triangle.h"
 #include "initial_distribution.h"
 #include "energy.h"
+#include "poly.h"
+#include "io.h"
+#include "sh.h"
+#include "shcomplex.h"
 
 ts_vesicle *initial_distribution_dipyramid(ts_uint nshell, ts_uint ncmax1, ts_uint ncmax2, ts_uint ncmax3, ts_double stepsize){
-    ts_fprintf(stderr,"Starting initial_distribution on vesicle with %u shells!...\n",nshell);
+	ts_fprintf(stdout,"Starting initial_distribution on vesicle with %u shells!...\n",nshell);
 	ts_bool retval;
-    ts_uint no_vertices=5*nshell*nshell+2;
-	
-    ts_vesicle *vesicle=init_vesicle(no_vertices,ncmax1,ncmax2,ncmax3,stepsize);
-    vesicle->nshell=nshell;
-    retval = vtx_set_global_values(vesicle);
-    retval = pentagonal_dipyramid_vertex_distribution(vesicle->vlist);
-    retval = init_vertex_neighbours(vesicle->vlist);
-    retval = init_sort_neighbours(vesicle->vlist);
-    retval = init_vesicle_bonds(vesicle);
-    retval = init_triangles(vesicle);
-    retval = init_triangle_neighbours(vesicle);
-    retval = init_common_vertex_triangle_neighbours(vesicle);
-    retval = mean_curvature_and_energy(vesicle);
- ts_fprintf(stderr,"initial_distribution finished!\n");
+	ts_uint no_vertices=5*nshell*nshell+2;	
+	ts_vesicle *vesicle=init_vesicle(no_vertices,ncmax1,ncmax2,ncmax3,stepsize);
+	vesicle->nshell=nshell;
+	//retval = vtx_set_global_values(vesicle);
+	retval = pentagonal_dipyramid_vertex_distribution(vesicle->vlist);
+	retval = init_vertex_neighbours(vesicle->vlist);
+	vesicle->vlist = init_sort_neighbours(vesicle->blist,vesicle->vlist);
+   // retval = init_vesicle_bonds(vesicle); // bonds are created in sort_neigh
+	retval = init_triangles(vesicle);
+	retval = init_triangle_neighbours(vesicle);
+	retval = init_common_vertex_triangle_neighbours(vesicle);
+	retval = init_normal_vectors(vesicle->tlist);
+	retval = mean_curvature_and_energy(vesicle);
+	ts_fprintf(stdout,"initial_distribution finished!\n");
+	if(retval);
 	return vesicle;
 } 
+
+
+
+ts_vesicle *create_vesicle_from_tape(ts_tape *tape){
+	ts_vesicle *vesicle;
+	ts_vertex *vtx;
+
+	vesicle=initial_distribution_dipyramid(tape->nshell,tape->ncxmax,tape->ncymax,tape->nczmax,tape->stepsize);
+	// Nucleus:
+	vesicle->R_nucleus=tape->R_nucleus*tape->R_nucleus;
+
+	vesicle->clist->dmin_interspecies = tape->dmin_interspecies*tape->dmin_interspecies;
+
+	//Initialize grafted polymers (brush):
+	vesicle->poly_list=init_poly_list(tape->npoly,tape->nmono, vesicle->vlist, vesicle);
+	vesicle->spring_constant=tape->kspring;
+	poly_assign_spring_const(vesicle);
+
+	//Initialize filaments (polymers inside the vesicle):
+	vesicle->filament_list=init_poly_list(tape->nfil,tape->nfono, NULL, vesicle);
+	poly_assign_filament_xi(vesicle,tape);
+
+	ts_uint i,j;
+	for(i=0;i<vesicle->filament_list->n;i++){
+		for(j=0;j<vesicle->filament_list->poly[i]->blist->n;j++){
+			bond_vector(vesicle->filament_list->poly[i]->blist->bond[j]);
+			vesicle->filament_list->poly[i]->blist->bond[j]->bond_length = sqrt(vtx_distance_sq(vesicle->filament_list->poly[i]->blist->bond[j]->vtx1,vesicle->filament_list->poly[i]->blist->bond[j]->vtx2));
+		}
+	}
+
+	for(i=0;i<vesicle->filament_list->n;i++){
+		for(j=0;j<vesicle->filament_list->poly[i]->vlist->n;j++){
+			vtx = vesicle->filament_list->poly[i]->vlist->vtx[j];
+			if(vtx->bond_no == 2){
+			vtx->energy = -(vtx->bond[0]->x*vtx->bond[1]->x + vtx->bond[0]->y*vtx->bond[1]->y + vtx->bond[0]->z*vtx->bond[1]->z)/vtx->bond[0]->bond_length/vtx->bond[1]->bond_length;
+			}
+		}
+	}
+
+	for(i=0;i<vesicle->filament_list->n;i++){
+		vertex_list_assign_id(vesicle->filament_list->poly[i]->vlist,TS_ID_FILAMENT);
+	}
+
+//	vesicle->spring_constant=tape->kspring;
+//	poly_assign_spring_const(vesicle);
+
+	
+	vesicle->nshell=tape->nshell;
+	vesicle->dmax=tape->dmax*tape->dmax; /* dmax^2 in the vesicle dmax variable */
+	vesicle->bending_rigidity=tape->xk0;
+	vtx_set_global_values(vesicle); /* make xk0 default value for every vertex */ 
+	ts_fprintf(stdout, "Tape setting: xk0=%e\n",tape->xk0);
+	vesicle->stepsize=tape->stepsize;
+	vesicle->clist->ncmax[0]=tape->ncxmax;
+	vesicle->clist->ncmax[1]=tape->ncymax;
+	vesicle->clist->ncmax[2]=tape->nczmax;
+	vesicle->clist->max_occupancy=8; /* hard coded max occupancy? */
+
+	vesicle->pressure= tape->pressure;
+	vesicle->pswitch=tape->pswitch;
+    if(tape->shc>0){
+	    vesicle->sphHarmonics=complex_sph_init(vesicle->vlist,tape->shc);
+    }
+    else {
+        vesicle->sphHarmonics=NULL;
+    }
+    return vesicle;
+
+}
+
+
+
 
 
 ts_bool pentagonal_dipyramid_vertex_distribution(ts_vertex_list *vlist){
@@ -62,40 +139,40 @@
 	ts_double dx,dy; // end loop prereq
 
 	/* topmost vertex */
-	vtx[1]->data->x=0.0;
-	vtx[1]->data->y=0.0;
-	vtx[1]->data->z=z0*(ts_double)nshell;
+	vtx[1]->x=0.0;
+	vtx[1]->y=0.0;
+	vtx[1]->z=z0*(ts_double)nshell;
 	
 	/* starting from to in circular order on pentagrams */	
 	for(i=1;i<=nshell;i++){
 		n0=2+5*i*(i-1)/2; //-1 would be for the reason that C index starts from 0 
-		vtx[n0]->data->x=0.0;
-		vtx[n0]->data->y=(ts_double)i*xl0;
-		vtx[n0+i]->data->x=vtx[n0]->data->y*s1;
-		vtx[n0+i]->data->y=vtx[n0]->data->y*c1;
-		vtx[n0+2*i]->data->x=vtx[n0]->data->y*s2;
-		vtx[n0+2*i]->data->y=vtx[n0]->data->y*c2;
-		vtx[n0+3*i]->data->x=-vtx[n0+2*i]->data->x;
-		vtx[n0+3*i]->data->y=vtx[n0+2*i]->data->y;
-		vtx[n0+4*i]->data->x=-vtx[n0+i]->data->x;
-		vtx[n0+4*i]->data->y=vtx[n0+i]->data->y;
+		vtx[n0]->x=0.0;
+		vtx[n0]->y=(ts_double)i*xl0;
+		vtx[n0+i]->x=vtx[n0]->y*s1;
+		vtx[n0+i]->y=vtx[n0]->y*c1;
+		vtx[n0+2*i]->x=vtx[n0]->y*s2;
+		vtx[n0+2*i]->y=vtx[n0]->y*c2;
+		vtx[n0+3*i]->x=-vtx[n0+2*i]->x;
+		vtx[n0+3*i]->y=vtx[n0+2*i]->y;
+		vtx[n0+4*i]->x=-vtx[n0+i]->x;
+		vtx[n0+4*i]->y=vtx[n0+i]->y;
 	}
 
 	/* vertexes on the faces of the dipyramid */
 	for(i=1;i<=nshell;i++){
 		n0=2+5*i*(i-1)/2; // -1 would be because of C!
 		for(j=1;j<=i-1;j++){
-			dx=(vtx[n0]->data->x-vtx[n0+4*i]->data->x)/(ts_double)i;
-			dy=(vtx[n0]->data->y-vtx[n0+4*i]->data->y)/(ts_double)i;
-			vtx[n0+4*i+j]->data->x=(ts_double)j*dx+vtx[n0+4*i]->data->x;
-			vtx[n0+4*i+j]->data->y=(ts_double)j*dy+vtx[n0+4*i]->data->y;
+			dx=(vtx[n0]->x-vtx[n0+4*i]->x)/(ts_double)i;
+			dy=(vtx[n0]->y-vtx[n0+4*i]->y)/(ts_double)i;
+			vtx[n0+4*i+j]->x=(ts_double)j*dx+vtx[n0+4*i]->x;
+			vtx[n0+4*i+j]->y=(ts_double)j*dy+vtx[n0+4*i]->y;
 		}
 		for(k=0;k<=3;k++){ // I would be worried about zero starting of for
-			dx=(vtx[n0+(k+1)*i]->data->x - vtx[n0+k*i]->data->x)/(ts_double) i;
-			dy=(vtx[n0+(k+1)*i]->data->y - vtx[n0+k*i]->data->y)/(ts_double) i;
+			dx=(vtx[n0+(k+1)*i]->x - vtx[n0+k*i]->x)/(ts_double) i;
+			dy=(vtx[n0+(k+1)*i]->y - vtx[n0+k*i]->y)/(ts_double) i;
 			for(j=1; j<=i-1;j++){
-				vtx[n0+k*i+j]->data->x= (ts_double)j*dx+vtx[n0+k*i]->data->x;
-				vtx[n0+k*i+j]->data->y= (ts_double)j*dy+vtx[n0+k*i]->data->y;
+				vtx[n0+k*i+j]->x= (ts_double)j*dx+vtx[n0+k*i]->x;
+				vtx[n0+k*i+j]->y= (ts_double)j*dy+vtx[n0+k*i]->y;
 			} 
 		} 
 	}
@@ -103,15 +180,15 @@
 	for(i=1;i<=nshell;i++){
 		n0= 2+ 5*i*(i-1)/2;
 		for(j=0;j<=5*i-1;j++){
-		vtx[n0+j]->data->z= z0*(ts_double)(nshell-i);   // I would be worried about zero starting of for
+		vtx[n0+j]->z= z0*(ts_double)(nshell-i);   // I would be worried about zero starting of for
 		}
 	}
 
 /* for botom part of dipyramide we calculate the positions of vertices */
 	for(i=2+5*nshell*(nshell+1)/2;i<=vlist->n;i++){
-		vtx[i]->data->x=vtx[vlist->n - i +1]->data->x;
-		vtx[i]->data->y=vtx[vlist->n - i +1]->data->y;
-		vtx[i]->data->z=-vtx[vlist->n - i +1]->data->z;
+		vtx[i]->x=vtx[vlist->n - i +1]->x;
+		vtx[i]->y=vtx[vlist->n - i +1]->y;
+		vtx[i]->z=-vtx[vlist->n - i +1]->z;
 	}
 
 	for(i=1;i<=vlist->n;i++){
@@ -150,29 +227,30 @@
 	return TS_SUCCESS;
 }
 
-// TODO: with new datastructure can be rewritten.
-ts_bool init_sort_neighbours(ts_vertex_list *vlist){
+// TODO: with new datastructure can be rewritten. Partially it is done, but it is complicated.
+ts_vertex_list *init_sort_neighbours(ts_bond_list *blist,ts_vertex_list *vlist){
 	ts_vertex **vtx=vlist->vtx -1; // take a look at dipyramid function for comment.
 	ts_uint i,l,j,jj,jjj,k=0;   
     ts_double eps=0.001; // Take a look if EPS from math.h can be used
 
 /*lets initialize memory for temporary vertex_list. Should we write a function instead */
-    ts_vertex_list *tvlist=init_vertex_list(vlist->n);
+    ts_vertex_list *tvlist=vertex_list_copy(vlist);
     ts_vertex **tvtx=tvlist->vtx -1;  /* again to compensate for 0-indexing */
 
 	ts_double dist2; // Square of distance of neighbours
     ts_double direct; // Something, dont know what, but could be normal of some kind
 	for(i=1;i<=vlist->n;i++){
 		k++; // WHY i IS NOT GOOD??
-       	vtx_add_neighbour(tvtx[k], tvtx[vtx[i]->data->neigh[0]->idx+1]); //always add 1st
+       	vtx_add_cneighbour(blist,tvtx[k], tvtx[vtx[i]->neigh[0]->idx+1]); //always add 1st
        	jjj=1;
        	jj=1;
-       	for(l=2;l<=vtx[i]->data->neigh_no;l++){
-           	for(j=2;j<=vtx[i]->data->neigh_no;j++){
-               	dist2=vtx_distance_sq(vtx[i]->data->neigh[j-1],vtx[i]->data->neigh[jj-1]);
-               	direct=vtx_direct(vtx[i],vtx[i]->data->neigh[j-1],vtx[i]->data->neigh[jj-1]);
+       	for(l=2;l<=vtx[i]->neigh_no;l++){
+           	for(j=2;j<=vtx[i]->neigh_no;j++){
+               	dist2=vtx_distance_sq(vtx[i]->neigh[j-1],vtx[i]->neigh[jj-1]);
+               	direct=vtx_direct(vtx[i],vtx[i]->neigh[j-1],vtx[i]->neigh[jj-1]);
+// TODO: check if fabs can be used with all floating point types!!
                	if( (fabs(dist2-A0*A0)<=eps) && (direct>0.0) && (j!=jjj) ){
-           			vtx_add_neighbour(tvtx[k],tvtx[vtx[i]->data->neigh[j-1]->idx+1]);
+           			vtx_add_cneighbour(blist,tvtx[k],tvtx[vtx[i]->neigh[j-1]->idx+1]);
            			jjj=jj;
            			jj=j;
            			break;
@@ -180,24 +258,19 @@
        		}
        	}	
 	}
-
-    for(i=1;i<=vlist->n;i++){
-        for(j=1;j<=vtx[i]->data->neigh_no;j++){
-            if(vtx[i]->data->neigh_no!=tvtx[i]->data->neigh_no){ //doesn't work with nshell=1!
-//                fprintf(stderr,"data1=%u data2=%u\n",vtx[i]->data->neigh_no,tvtx[i]->data->neigh_no);
-                fatal("Number of neighbours not the same in init_sort_neighbours.",4);
-            }
-            //we must correct the pointers in original to point to their
-            //neighbours according to indexes. Must be sure not to do it any
-            //other way! Also, we need to repair the collection of bonds...
-            vtx[i]->data->neigh[j-1]=vtx[tvtx[i]->data->neigh[j-1]->idx+1];
-        }
+/* We use the temporary vertex for our main vertices and we abandon main
+ * vertices, because their neighbours are not correctly ordered */
+   // tvtx=vlist->vtx;
+   // vlist->vtx=tvtx;
+   // tvlist->vtx=vtx;
+    vtx_list_free(vlist);
+/* Let's make a check if the number of bonds is correct */
+    if((blist->n)!=3*(tvlist->n-2)){
+        ts_fprintf(stderr,"Number of bonds is %u should be %u!\n", blist->n, 3*(tvlist->n-2));
+        fatal("Number of bonds is not 3*(no_vertex-2).",4);
     }
 
-   // Must free memory for temporary vertex array to avoid memory leak! HERE! NOW!
-//    free_vertex(tvlist.vertex,tvlist.n);
-    vtx_list_free(tvlist);
-	return TS_SUCCESS;
+	return tvlist;
 }
 
 
@@ -209,8 +282,8 @@
 	ts_uint i,j,k;
 	for(i=1;i<=vlist->n;i++){
 		for(j=i+1;j<=vlist->n;j++){
-			for(k=0;k<vtx[i]->data->neigh_no;k++){ // has changed 0 to < instead of 1 and <=
-				if(vtx[i]->data->neigh[k]==vtx[j]){  //if addresses matches it is the same
+			for(k=0;k<vtx[i]->neigh_no;k++){ // has changed 0 to < instead of 1 and <=
+				if(vtx[i]->neigh[k]==vtx[j]){  //if addresses matches it is the same
 					bond_add(blist,vtx[i],vtx[j]);
 					break;
 				}
@@ -235,14 +308,15 @@
 	ts_double eps=0.001; // can we use EPS from math.h?
 	k=0;
 	for(i=1;i<=vesicle->vlist->n;i++){
-		for(j=1;j<=vtx[i]->data->neigh_no;j++){
-			for(jj=1;jj<=vtx[i]->data->neigh_no;jj++){
+		for(j=1;j<=vtx[i]->neigh_no;j++){
+			for(jj=1;jj<=vtx[i]->neigh_no;jj++){
 		//		ts_fprintf(stderr,"%u: (%u,%u) neigh_no=%u ",i,j,jj,vtx[i].neigh_no);
         //      ts_fprintf(stderr,"%e, %e",vtx[i].neigh[j-1]->x,vtx[i].neigh[jj-1]->x);
-				dist=vtx_distance_sq(vtx[i]->data->neigh[j-1],vtx[i]->data->neigh[jj-1]);
-				direct=vtx_direct(vtx[i],vtx[i]->data->neigh[j-1],vtx[i]->data->neigh[jj-1]);				
-				if(fabs(dist-A0*A0)<=eps && direct < 0.0 && vtx[i]->data->neigh[j-1]->idx+1 > i && vtx[i]->data->neigh[jj-1]->idx+1 >i){
-					triangle_add(tlist,vtx[i],vtx[i]->data->neigh[j-1],vtx[i]->data->neigh[jj-1]);
+				dist=vtx_distance_sq(vtx[i]->neigh[j-1],vtx[i]->neigh[jj-1]);
+				direct=vtx_direct(vtx[i],vtx[i]->neigh[j-1],vtx[i]->neigh[jj-1]);				
+// TODO: same as above				
+				if(fabs(dist-A0*A0)<=eps && direct < 0.0 && vtx[i]->neigh[j-1]->idx+1 > i && vtx[i]->neigh[jj-1]->idx+1 >i){
+					triangle_add(tlist,vtx[i],vtx[i]->neigh[j-1],vtx[i]->neigh[jj-1]);
 				}	
 			}	
 		}
@@ -253,11 +327,11 @@
 	for(i=0;i<tlist->n;i++){
         k=0;
 		for(j=0;j<3;j++){
-			if(tlist->tria[i]->data->vertex[j]!=NULL)
+			if(tlist->tria[i]->vertex[j]!=NULL)
             k++;
 		}
             if(k!=3){
-                fatal("Some triangles has less than 3 vertices..",4);
+                fatal("Some triangles have less than 3 vertices..",4);
             }   
 	} 
     if(tlist->n!=2*(vesicle->vlist->n -2)){
@@ -277,14 +351,14 @@
     ts_triangle **tria=tlist->tria -1;
     nobo=0;
     for(i=1;i<=tlist->n;i++){
-        i1=tria[i]->data->vertex[0]; 
-        i2=tria[i]->data->vertex[1]; 
-        i3=tria[i]->data->vertex[2]; 
+        i1=tria[i]->vertex[0]; 
+        i2=tria[i]->vertex[1]; 
+        i3=tria[i]->vertex[2]; 
         for(j=1;j<=tlist->n;j++){
             if(j==i) continue;
-            j1=tria[j]->data->vertex[0]; 
-            j2=tria[j]->data->vertex[1]; 
-            j3=tria[j]->data->vertex[2]; 
+            j1=tria[j]->vertex[0]; 
+            j2=tria[j]->vertex[1]; 
+            j3=tria[j]->vertex[2]; 
             if((i1==j1 && i3==j2) || (i1==j2 && i3==j3) || (i1==j3 && i3==j1)){
                     triangle_add_neighbour(tria[i],tria[j]);
                     nobo++;
@@ -292,14 +366,14 @@
         }
     }
     for(i=1;i<=tlist->n;i++){
-        i1=tria[i]->data->vertex[0]; 
-        i2=tria[i]->data->vertex[1]; 
-        i3=tria[i]->data->vertex[2]; 
+        i1=tria[i]->vertex[0]; 
+        i2=tria[i]->vertex[1]; 
+        i3=tria[i]->vertex[2]; 
         for(j=1;j<=tlist->n;j++){
             if(j==i) continue;
-            j1=tria[j]->data->vertex[0]; 
-            j2=tria[j]->data->vertex[1]; 
-            j3=tria[j]->data->vertex[2]; 
+            j1=tria[j]->vertex[0]; 
+            j2=tria[j]->vertex[1]; 
+            j3=tria[j]->vertex[2]; 
             if((i1==j1 && i2==j3) || (i1==j3 && i2==j2) || (i1==j2 && i2==j1)){
                 triangle_add_neighbour(tria[i],tria[j]);
                 nobo++;
@@ -307,14 +381,14 @@
         }
     }
     for(i=1;i<=tlist->n;i++){
-        i1=tria[i]->data->vertex[0]; 
-        i2=tria[i]->data->vertex[1]; 
-        i3=tria[i]->data->vertex[2]; 
+        i1=tria[i]->vertex[0]; 
+        i2=tria[i]->vertex[1]; 
+        i3=tria[i]->vertex[2]; 
         for(j=1;j<=tlist->n;j++){
             if(j==i) continue;
-            j1=tria[j]->data->vertex[0]; 
-            j2=tria[j]->data->vertex[1]; 
-            j3=tria[j]->data->vertex[2]; 
+            j1=tria[j]->vertex[0]; 
+            j2=tria[j]->vertex[1]; 
+            j3=tria[j]->vertex[2]; 
             if((i2==j1 && i3==j3) || (i2==j3 && i3==j2) || (i2==j2 && i3==j1)){
                 triangle_add_neighbour(tria[i],tria[j]);
                 nobo++;
@@ -337,20 +411,23 @@
     ts_triangle **tria=tlist->tria -1;
 
     for(i=1;i<=vesicle->vlist->n;i++){
-        for(j=1;j<=vtx[i]->data->neigh_no;j++){
-            k1=vtx[i]->data->neigh[j-1];
+        for(j=1;j<=vtx[i]->neigh_no;j++){
+            k1=vtx[i]->neigh[j-1];
             jp=j+1;
-            if(j == vtx[i]->data->neigh_no) jp=1;
-            k2=vtx[i]->data->neigh[jp-1];
+            if(j == vtx[i]->neigh_no) jp=1;
+            k2=vtx[i]->neigh[jp-1];
             for(k=1;k<=tlist->n;k++){		// VERY NON-OPTIMAL!!! too many loops (vlist.n * vtx.neigh * tlist.n )!
-                k3=tria[k]->data->vertex[0];
-                k4=tria[k]->data->vertex[1];
-                k5=tria[k]->data->vertex[2];
+                k3=tria[k]->vertex[0];
+                k4=tria[k]->vertex[1];
+                k5=tria[k]->vertex[2];
 //                ts_fprintf(stderr,"%u %u: k=(%u %u %u)\n",k1,k2,k3,k4,k5);
                 if((vtx[i]==k3 && k1==k4 && k2==k5) ||
                 (vtx[i]==k4 && k1==k5 && k2==k3) ||
                 (vtx[i]==k5 && k1==k3 && k2==k4)){
-          //          ts_fprintf(stderr, "Added to tristar! ");
+
+//TODO: probably something wrong with neighbour distribution.
+//                if(vtx[i]==k3 || vtx[i]==k4 || vtx[i]==k5){
+    //                    if(i==6) ts_fprintf(stdout, "Vtx[%u] > Added to tristar!\n",i);
                     vertex_add_tristar(vtx[i],tria[k]);
                 }
             }

--
Gitblit v1.9.3