From 0652ee2715994b48a9bbfd1132bf0b65ad206289 Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Wed, 27 Nov 2013 18:46:13 +0000
Subject: [PATCH] commiting all changes.

---
 src/vertex.c |   60 +++++++++++++++++++++++++++++++++++++++++++++++++++---------
 1 files changed, 51 insertions(+), 9 deletions(-)

diff --git a/src/vertex.c b/src/vertex.c
index 694ee84..cd999df 100644
--- a/src/vertex.c
+++ b/src/vertex.c
@@ -23,7 +23,7 @@
     for(i=0;i<N;i++) {
         vlist->vtx[i]=(ts_vertex *)calloc(1,sizeof(ts_vertex));
         vlist->vtx[i]->idx=i;
-
+	vlist->vtx[i]->locked=(ts_uint *)calloc(1,sizeof(ts_vertex));
     /* initialize Ylm for spherical hamonics DONE in sh.c */
 /*    for(i=0;i<l;i++){
         vlist->vtx[i]->Ylm[i]=(ts_double **)calloc(2*i+1,sizeof(ts_double *));
@@ -72,25 +72,25 @@
 ts_bool vtx_remove_neighbour(ts_vertex *vtx, ts_vertex *nvtx){
 /* find a neighbour */
 /* remove it from the list while shifting remaining neighbours up */
-    ts_int i,j=-1;
+    ts_uint i,j=0;
     for(i=0;i<vtx->neigh_no;i++){
+//		fprintf(stderr,"neigh_addr=%ld\n", (long)vtx->neigh[i]);
         if(vtx->neigh[i]!=nvtx){
             vtx->neigh[j]=vtx->neigh[i];
             j++;
         }
     }
-    if(j==0) { 
-        fatal("vtx_remove_neighbour: Error, vertices are not neighbours", 100);
-    }
+//	fprintf(stderr,"remove_neighbour: vtx1_addr=%ld, vtx2_addr=%ld\n",(long)vtx,(long)nvtx);
 /* resize memory. potentionally time consuming */
     vtx->neigh_no--;
     vtx->neigh=(ts_vertex **)realloc(vtx->neigh,vtx->neigh_no*sizeof(ts_vertex *));
     if(vtx->neigh == NULL && vtx->neigh_no!=0)
-        fatal("Reallocation of memory failed during removal of vertex neighbour in vtx_remove_neighbour",100);
-fprintf(stderr,"first alloc");
+        fatal("(1) Reallocation of memory failed during removal of vertex neighbour in vtx_remove_neighbour",100);
+//fprintf(stderr,"first alloc");
 /* repeat for the neighbour */
 /* find a neighbour */
 /* remove it from the list while shifting remaining neighbours up */
+	j=0;
     for(i=0;i<nvtx->neigh_no;i++){
         if(nvtx->neigh[i]!=vtx){
             nvtx->neigh[j]=nvtx->neigh[i];
@@ -98,10 +98,12 @@
         }
     }
 /* resize memory. potentionally time consuming. */
+//	fprintf(stderr,"Neigbours=%d\n",nvtx->neigh_no);
     nvtx->neigh_no--;
-    nvtx->neigh=(ts_vertex **)realloc(nvtx->neigh,nvtx->neigh_no*sizeof(ts_vertex *));
+    		nvtx->neigh=(ts_vertex **)realloc(nvtx->neigh,nvtx->neigh_no*sizeof(ts_vertex *));
+//	fprintf(stderr,"Neigbours=%d\n",nvtx->neigh_no);
     if(nvtx->neigh == NULL && nvtx->neigh_no!=0)
-        fatal("Reallocation of memory failed during removal of vertex neighbour in vtx_remove_neighbour",100);
+        fatal("(2) Reallocation of memory failed during removal of vertex neighbour in vtx_remove_neighbour",100);
 
     return TS_SUCCESS;
 }
@@ -128,6 +130,7 @@
 ts_bool vtx_add_cneighbour(ts_bond_list *blist, ts_vertex *vtx1, ts_vertex *vtx2){
     ts_bool retval;
     retval=vtx_add_neighbour(vtx1,vtx2);
+  //  retval=vtx_add_neighbour(vtx2,vtx1);
     if(retval==TS_SUCCESS)
     retval=vtx_add_bond(blist,vtx1,vtx2); 
     return retval;
@@ -308,3 +311,42 @@
 
     return vlist;
 }
+
+
+
+ts_bool vertex_taint(ts_vertex *vtx, ts_uint level){
+	ts_uint i;
+	(*(vtx->locked))=1; //lock current vertex
+	if(level==0){ //if we reach last in recursion exit
+		return TS_SUCCESS;	
+	}	
+	for(i=0; i<vtx->neigh_no; i++){ //else recursive call self with decreased level
+		vertex_taint(vtx->neigh[i], level-1);
+	}
+	return TS_SUCCESS;
+}
+
+ts_bool vertex_untaint(ts_vertex *vtx, ts_uint level){
+		(*(vtx->locked))=0;
+	if(level==0){
+		return TS_SUCCESS;	
+	}	
+	ts_uint i;
+	for(i=0; i<vtx->neigh_no; i++){
+		vertex_untaint(vtx->neigh[i], level-1);
+	}
+	return TS_SUCCESS;
+}
+
+inline ts_bool vertex_tainted(ts_vertex *vtx, ts_uint level, ts_uint amount){
+	if(*(vtx->locked)>amount) return 1;
+	if(level==0){
+		return 0;
+	}
+	ts_uint i;
+	for(i=0; i<vtx->neigh_no;i++){
+		if(vertex_tainted(vtx->neigh[i], level-1, amount)) return 1;
+	}
+	
+	return 0;
+}

--
Gitblit v1.9.3