From 063289f4f872d642078d70757685c728e3fdfc81 Mon Sep 17 00:00:00 2001 From: Samo Penic <samo.penic@gmail.com> Date: Thu, 06 Mar 2014 07:48:56 +0000 Subject: [PATCH] Fixed a bug. added cells --- src/io.c | 318 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 313 insertions(+), 5 deletions(-) diff --git a/src/io.c b/src/io.c index 8b51793..b4e879f 100644 --- a/src/io.c +++ b/src/io.c @@ -11,6 +11,311 @@ #include "initial_distribution.h" #include "poly.h" + + +/** DUMP STATE TO DISK DRIVE **/ + +ts_bool dump_state(ts_vesicle *vesicle){ + + /* save current state with wrong pointers. Will fix that later */ + ts_uint i,j,k; + FILE *fh=fopen("dump.bin","wb"); + + /* dump vesicle */ + fwrite(vesicle, sizeof(vesicle),1,fh); + /* dump vertex list */ + fwrite(vesicle->vlist, sizeof(ts_vertex_list),1,fh); + /* dump bond list */ + fwrite(vesicle->blist, sizeof(ts_bond_list),1,fh); + /* dump triangle list */ + fwrite(vesicle->tlist, sizeof(ts_triangle_list),1,fh); + /* dump cell list */ + fwrite(vesicle->clist, sizeof(ts_cell_list),1,fh); + /* dump poly list */ + fwrite(vesicle->poly_list, sizeof(ts_poly_list),1,fh); + /* level 1 complete */ + + /*dump vertices*/ + for(i=0;i<vesicle->vlist->n;i++){ + fwrite(vesicle->vlist->vtx[i],sizeof(ts_vertex),1,fh); + /* dump pointer offsets for: + neigh + bond + tria + cell is ignored + */ + for(j=0;j<vesicle->vlist->vtx[i]->neigh_no;j++){ + fwrite(&vesicle->vlist->vtx[i]->neigh[j]->idx,sizeof(ts_uint),1,fh); + } + for(j=0;j<vesicle->vlist->vtx[i]->bond_no;j++){ + fwrite(&vesicle->vlist->vtx[i]->bond[j]->idx,sizeof(ts_uint),1,fh); + } + for(j=0;j<vesicle->vlist->vtx[i]->tristar_no;j++){ + fwrite(&vesicle->vlist->vtx[i]->tristar[j]->idx,sizeof(ts_uint),1,fh); + } + } + + /*dump bonds*/ + for(i=0;i<vesicle->blist->n;i++){ + fwrite(vesicle->blist->bond[i],sizeof(ts_bond),1,fh); + /* dump pointer offsets for vtx1 and vtx2 */ + //off=(ts_ulong)(vesicle->blist->bond[i]->vtx1-vesicle->vlist->vtx[0]); + fwrite(&vesicle->blist->bond[i]->vtx1->idx,sizeof(ts_uint),1,fh); + //off=(ts_ulong)(vesicle->blist->bond[i]->vtx2-vesicle->vlist->vtx[0]); + fwrite(&vesicle->blist->bond[i]->vtx2->idx,sizeof(ts_uint),1,fh); + } + + /*dump triangles*/ + for(i=0;i<vesicle->tlist->n;i++){ + fwrite(vesicle->tlist->tria[i],sizeof(ts_triangle),1,fh); + /* dump pointer offsets for vertex */ + fwrite(&vesicle->tlist->tria[i]->vertex[0]->idx,sizeof(ts_uint),1,fh); + fwrite(&vesicle->tlist->tria[i]->vertex[1]->idx,sizeof(ts_uint),1,fh); + fwrite(&vesicle->tlist->tria[i]->vertex[2]->idx,sizeof(ts_uint),1,fh); + /* dump pointer offsets for neigh */ + for(j=0;j<vesicle->tlist->tria[i]->neigh_no;j++){ + fwrite(&vesicle->tlist->tria[i]->neigh[j]->idx,sizeof(ts_uint),1,fh); + } + } + + + /*dump polymeres */ + for(i=0;i<vesicle->poly_list->n;i++){ + fwrite(vesicle->poly_list->poly[i],sizeof(ts_poly),1,fh); + fwrite(vesicle->poly_list->poly[i]->vlist,sizeof(ts_vertex_list),1,fh); + fwrite(vesicle->poly_list->poly[i]->blist,sizeof(ts_bond_list),1,fh); + } + + /* dump poly vertex(monomer) list*/ + for(i=0;i<vesicle->poly_list->n;i++){ + for(j=0;j<vesicle->poly_list->poly[i]->vlist->n;j++){ + fwrite(vesicle->poly_list->poly[i]->vlist->vtx[j],sizeof(ts_vertex),1,fh); + /* dump offset for neigh and bond */ + for(k=0;k<vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh_no;k++){ + // off=(ts_ulong)(vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh[k]-vesicle->poly_list->poly[i]->vlist->vtx[0]); + fwrite(&vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh[k]->idx,sizeof(ts_uint),1,fh); + } + for(k=0;k<vesicle->poly_list->poly[i]->vlist->vtx[j]->bond_no;k++){ + //off=(ts_ulong)(vesicle->poly_list->poly[i]->vlist->vtx[j]->bond[k]-vesicle->poly_list->poly[i]->blist->bond[0]); + fwrite(&vesicle->poly_list->poly[i]->vlist->vtx[j]->bond[k]->idx,sizeof(ts_uint),1,fh); + } + } + // grafted vtx on vesicle data dump + fwrite(&vesicle->poly_list->poly[i]->grafted_vtx->idx, sizeof(ts_uint),1,fh); + } + /* dump poly bonds between monomers list*/ + for(i=0;i<vesicle->poly_list->n;i++){ + for(j=0;j<vesicle->poly_list->poly[i]->blist->n;j++){ + fwrite(vesicle->poly_list->poly[i]->blist->bond[j],sizeof(ts_bond),1,fh); + /* dump vtx1 and vtx2 offsets */ + //off=(ts_ulong)(vesicle->poly_list->poly[i]->blist->bond[j]->vtx1-vesicle->poly_list->poly[i]->vlist->vtx[0]); + fwrite(&vesicle->poly_list->poly[i]->blist->bond[j]->vtx1->idx,sizeof(ts_uint),1,fh); +// off=(ts_ulong)(vesicle->poly_list->poly[i]->blist->bond[j]->vtx2-vesicle->poly_list->poly[i]->vlist->vtx[0]); + fwrite(&vesicle->poly_list->poly[i]->blist->bond[j]->vtx2->idx,sizeof(ts_uint),1,fh); + } + } + +/* pointer offsets for fixing the restored pointers */ +/* need pointers for + vlist->vtx + blist->bond + tlist->tria + clist->cell + poly_list->poly + and for each poly: + poly_list->poly->vtx + poly_list->poly->bond +*/ + + fwrite(vesicle->clist, sizeof(ts_cell_list),1, fh); + + fclose(fh); + return TS_SUCCESS; +} + + +/** RESTORE DUMP FROM DISK **/ +ts_vesicle *restore_state(){ + ts_uint i,j,k; + FILE *fh=fopen("dump.bin","rb"); + ts_uint retval; + ts_uint idx; + +/* we restore all the data from the dump */ + /* restore vesicle */ + ts_vesicle *vesicle=(ts_vesicle *)calloc(1,sizeof(ts_vesicle)); + retval=fread(vesicle, sizeof(vesicle),1,fh); + /* restore vertex list */ + vesicle->vlist=(ts_vertex_list *)malloc(sizeof(ts_vertex_list)); + retval=fread(vesicle->vlist, sizeof(ts_vertex_list),1,fh); + /* restore bond list */ + vesicle->blist=(ts_bond_list *)malloc(sizeof(ts_bond_list)); + retval=fread(vesicle->blist, sizeof(ts_bond_list),1,fh); + /* restore triangle list */ + vesicle->tlist=(ts_triangle_list *)malloc(sizeof(ts_triangle_list)); + retval=fread(vesicle->tlist, sizeof(ts_triangle_list),1,fh); + /* restore cell list */ + vesicle->clist=(ts_cell_list *)malloc(sizeof(ts_cell_list)); + retval=fread(vesicle->clist, sizeof(ts_cell_list),1,fh); + /* restore poly list */ + vesicle->poly_list=(ts_poly_list *)calloc(1,sizeof(ts_poly_list)); + retval=fread(vesicle->poly_list, sizeof(ts_poly_list),1,fh); + /* level 1 complete */ + +/* prerequisity. Bonds must be malloced before vertexes are recreated */ + vesicle->blist->bond=(ts_bond **)calloc(vesicle->blist->n,sizeof(ts_bond *)); + for(i=0;i<vesicle->blist->n;i++){ + vesicle->blist->bond[i]=(ts_bond *)malloc(sizeof(ts_bond)); + } +/* prerequisity. Triangles must be malloced before vertexes are recreated */ + vesicle->tlist->tria=(ts_triangle **)calloc(vesicle->tlist->n,sizeof(ts_triangle *)); + for(i=0;i<vesicle->tlist->n;i++){ + vesicle->tlist->tria[i]=(ts_triangle *)malloc(sizeof(ts_triangle)); +} + + /*restore vertices*/ + vesicle->vlist->vtx=(ts_vertex **)calloc(vesicle->vlist->n,sizeof(ts_vertex *)); + for(i=0;i<vesicle->vlist->n;i++){ + vesicle->vlist->vtx[i]=(ts_vertex *)malloc(sizeof(ts_vertex)); + retval=fread(vesicle->vlist->vtx[i],sizeof(ts_vertex),1,fh); + /*restore neigh, bond, tristar. Ignoring cell */ + vesicle->vlist->vtx[i]->neigh=(ts_vertex **)calloc(vesicle->vlist->vtx[i]->neigh_no, sizeof(ts_vertex *)); + for(j=0;j<vesicle->vlist->vtx[i]->neigh_no;j++){ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->vlist->vtx[i]->neigh[j]=vesicle->vlist->vtx[idx]; + } + vesicle->vlist->vtx[i]->bond=(ts_bond **)calloc(vesicle->vlist->vtx[i]->bond_no, sizeof(ts_bond *)); + for(j=0;j<vesicle->vlist->vtx[i]->bond_no;j++){ + retval=fread(&idx,sizeof(ts_uint),1,fh); +/* pointer can be assigned only when list of bonds is fully initialized in memory. Thus bondlist popularization must be done before vertex can reference to it */ + vesicle->vlist->vtx[i]->bond[j]=vesicle->blist->bond[idx]; + } + + vesicle->vlist->vtx[i]->tristar=(ts_triangle **)calloc(vesicle->vlist->vtx[i]->tristar_no, sizeof(ts_triangle *)); + for(j=0;j<vesicle->vlist->vtx[i]->tristar_no;j++){ + retval=fread(&idx,sizeof(ts_uint),1,fh); +/* same comment as above */ + vesicle->vlist->vtx[i]->tristar[j]=vesicle->tlist->tria[idx]; + } + + } + + /*restore bonds*/ + // vesicle->blist->bond=(ts_bond **)calloc(vesicle->blist->n,sizeof(ts_bond *)); // done before. + for(i=0;i<vesicle->blist->n;i++){ + // vesicle->blist->bond[i]=(ts_bond *)malloc(sizeof(ts_bond)); //done before. + retval=fread(vesicle->blist->bond[i],sizeof(ts_bond),1,fh); + /* restore vtx1 and vtx2 */ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->blist->bond[i]->vtx1=vesicle->vlist->vtx[idx]; + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->blist->bond[i]->vtx2=vesicle->vlist->vtx[idx]; + } + + /*restore triangles*/ +// vesicle->tlist->tria=(ts_triangle **)calloc(vesicle->tlist->n,sizeof(ts_triangle *)); // done before + for(i=0;i<vesicle->tlist->n;i++){ + // vesicle->tlist->tria[i]=(ts_triangle *)malloc(sizeof(ts_triangle)); // done before + retval=fread(vesicle->tlist->tria[i],sizeof(ts_triangle),1,fh); + /* restore pointers for vertices */ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->tlist->tria[i]->vertex[0]=vesicle->vlist->vtx[idx]; + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->tlist->tria[i]->vertex[1]=vesicle->vlist->vtx[idx]; + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->tlist->tria[i]->vertex[2]=vesicle->vlist->vtx[idx]; + /* restore pointers for neigh */ + vesicle->tlist->tria[i]->neigh=(ts_triangle **)malloc(vesicle->tlist->tria[i]->neigh_no*sizeof(ts_triangle *)); + for(j=0;j<vesicle->tlist->tria[i]->neigh_no;j++){ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->tlist->tria[i]->neigh[j]=vesicle->tlist->tria[idx]; + } + + } + + /*restore cells */ +/*TODO: do we need to recalculate cells here? */ +/* vesicle->clist->cell=(ts_cell **)malloc(vesicle->clist->cellno*sizeof(ts_cell *)); + for(i=0;i<vesicle->clist->cellno;i++){ + vesicle->clist->cell[i]=(ts_cell *)malloc(sizeof(ts_cell)); + retval=fread(vesicle->clist->cell[i],sizeof(ts_cell),1,fh); + } +*/ + /*restore polymeres */ + vesicle->poly_list->poly = (ts_poly **)calloc(vesicle->poly_list->n,sizeof(ts_poly *)); + for(i=0;i<vesicle->poly_list->n;i++){ + vesicle->poly_list->poly[i]=(ts_poly *)calloc(1,sizeof(ts_poly)); + retval=fread(vesicle->poly_list->poly[i],sizeof(ts_poly),1,fh); + vesicle->poly_list->poly[i]->vlist=(ts_vertex_list *)calloc(1,sizeof(ts_vertex_list)); + retval=fread(vesicle->poly_list->poly[i]->vlist,sizeof(ts_vertex_list),1,fh); + vesicle->poly_list->poly[i]->blist=(ts_bond_list *)calloc(1,sizeof(ts_bond_list)); + retval=fread(vesicle->poly_list->poly[i]->blist,sizeof(ts_bond_list),1,fh); + /* initialize adress space for pointers that will hold specific vertices (monomers) and bonds */ + vesicle->poly_list->poly[i]->vlist->vtx=(ts_vertex **)calloc(vesicle->poly_list->poly[i]->vlist->n,sizeof(ts_vertex *)); + vesicle->poly_list->poly[i]->blist->bond=(ts_bond **)calloc(vesicle->poly_list->poly[i]->blist->n,sizeof(ts_bond *)); + for(j=0;j<vesicle->poly_list->poly[i]->vlist->n;j++){ + vesicle->poly_list->poly[i]->vlist->vtx[j]=(ts_vertex *)malloc(sizeof(ts_vertex)); + } + for(j=0;j<vesicle->poly_list->poly[i]->blist->n;j++){ + vesicle->poly_list->poly[i]->blist->bond[j]=(ts_bond *)malloc(sizeof(ts_bond)); + } + + } + + + /* restore poly vertex(monomer) list*/ + for(i=0;i<vesicle->poly_list->n;i++){ + for(j=0;j<vesicle->poly_list->poly[i]->vlist->n;j++){ + retval=fread(vesicle->poly_list->poly[i]->vlist->vtx[j],sizeof(ts_vertex),1,fh); + + /* restore neigh and bonds */ + vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh=(ts_vertex **)calloc(vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh_no, sizeof(ts_vertex *)); + for(k=0;k<vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh_no;k++){ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->poly_list->poly[i]->vlist->vtx[j]->neigh[k]=vesicle->poly_list->poly[i]->vlist->vtx[idx]; + } + vesicle->poly_list->poly[i]->vlist->vtx[j]->bond=(ts_bond **)calloc(vesicle->poly_list->poly[i]->vlist->vtx[j]->bond_no, sizeof(ts_bond *)); + for(k=0;k<vesicle->poly_list->poly[i]->vlist->vtx[j]->bond_no;k++){ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->poly_list->poly[i]->vlist->vtx[j]->bond[k]=vesicle->poly_list->poly[i]->blist->bond[idx]; + } + + } + /* restore grafted vtx on vesicle and grafted_poly */ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->vlist->vtx[idx]->grafted_poly=vesicle->poly_list->poly[i]; + vesicle->poly_list->poly[i]->grafted_vtx=vesicle->vlist->vtx[idx]; + } + + /* restore poly bonds between monomers list*/ + for(i=0;i<vesicle->poly_list->n;i++){ + for(j=0;j<vesicle->poly_list->poly[i]->blist->n;j++){ + // vesicle->poly_list->poly[i]->blist->bond[j]=(ts_bond *)malloc(sizeof(ts_bond)); + retval=fread(vesicle->poly_list->poly[i]->blist->bond[j],sizeof(ts_bond),1,fh); + /* restore vtx1 and vtx2 */ + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->poly_list->poly[i]->blist->bond[j]->vtx1=vesicle->poly_list->poly[i]->vlist->vtx[idx]; + retval=fread(&idx,sizeof(ts_uint),1,fh); + vesicle->poly_list->poly[i]->blist->bond[j]->vtx2=vesicle->poly_list->poly[i]->vlist->vtx[idx]; + } + } + +// recreating space for cells // + vesicle->clist=(ts_cell_list *)malloc(sizeof(ts_cell_list)); + retval=fread(vesicle->clist, sizeof(ts_cell_list), 1,fh); + vesicle->clist->cell=(ts_cell **)malloc(sizeof(ts_cell *)*vesicle->clist->ncmax[0]*vesicle->clist->ncmax[1]*vesicle->clist->ncmax[2]); + for(i=0;i<vesicle->clist->ncmax[0]*vesicle->clist->ncmax[1]*vesicle->clist->ncmax[2];i++){ + vesicle->clist->cell[i]=(ts_cell *)calloc(1,sizeof(ts_cell)); + vesicle->clist->cell[i]->idx=i+1; // We enumerate cells! Probably never required! + } + + if(retval); + fclose(fh); + return vesicle; +} + + + ts_bool print_vertex_list(ts_vertex_list *vlist){ ts_uint i; printf("Number of vertices: %u\n",vlist->n); @@ -220,7 +525,7 @@ /* Here comes header of the file */ //find number of extra vtxs and bonds of polymeres - ts_uint monono=0, polyno=0; + ts_uint monono=0, polyno=0, poly_idx=0; ts_bool poly=0; if(vesicle->poly_list!=NULL){ if(vesicle->poly_list->poly[0]!=NULL){ @@ -237,9 +542,10 @@ } //polymeres if(poly){ + poly_idx=vlist->n; for(i=0;i<vesicle->poly_list->n;i++){ - for(j=0;j<vesicle->poly_list->poly[i]->vlist->n;j++){ - fprintf(fh,"%u ", vesicle->poly_list->poly[i]->vlist->vtx[j]->idx); + for(j=0;j<vesicle->poly_list->poly[i]->vlist->n;j++,poly_idx++){ + fprintf(fh,"%u ", poly_idx); } } } @@ -263,12 +569,14 @@ } //polymeres if(poly){ + poly_idx=vlist->n; for(i=0;i<vesicle->poly_list->n;i++){ for(j=0;j<vesicle->poly_list->poly[i]->blist->n;j++){ - fprintf(fh,"%u %u\n", vesicle->poly_list->poly[i]->blist->bond[j]->vtx1->idx,vesicle->poly_list->poly[i]->blist->bond[j]->vtx2->idx); +// fprintf(fh,"%u %u\n", vesicle->poly_list->poly[i]->blist->bond[j]->vtx1->idx,vesicle->poly_list->poly[i]->blist->bond[j]->vtx2->idx); + fprintf(fh,"%u %u\n", vesicle->poly_list->poly[i]->blist->bond[j]->vtx1->idx+vlist->n+i*monono,vesicle->poly_list->poly[i]->blist->bond[j]->vtx2->idx+vlist->n+i*monono); } //grafted bonds - fprintf(fh,"%u %u\n", vesicle->poly_list->poly[i]->grafted_vtx->idx, vesicle->poly_list->poly[i]->vlist->vtx[0]->idx); + fprintf(fh,"%u %u\n", vesicle->poly_list->poly[i]->grafted_vtx->idx, vesicle->poly_list->poly[i]->vlist->vtx[0]->idx+vlist->n+i*monono); } } -- Gitblit v1.9.3