cluster.c rewritten from memory

Samo Penic

2016-07-04 b4c13e874faf0b0919ca45722a8072ae36f26e90

cluster.c rewritten from memory

 src/cluster.c

@@ -32,6 +32,44 @@
    return TS_SUCCESS;
}

ts_bool cluster_list_compact(ts_cluster_list *cstlist){


    ts_uint i,n=cstlist->n;
	
    for(i=0;i<cstlist->n;i++){
        if(cstlist->cluster[i]==NULL){
            do{
                n--;
            } while(cstlist->cluster[n]==NULL && n>i);
            cstlist->cluster[i]=cstlist->cluster[n];
            cstlist->cluster[n]=NULL;
        }
    }
    cstlist->cluster=(ts_cluster **)realloc(cstlist->cluster,n*sizeof(ts_cluster *));

    return TS_SUCCESS;
}


ts_bool cluster_join(ts_cluster *cluster1, ts_cluster *cluster2){
    ts_cluster *master_cluster,*slave_cluster;
    ts_uint i;
    if(cluster1->idx<cluster2->idx){
        master_cluster=cluster1;
        slave_cluster=cluster2;
    } else {
        master_cluster=cluster2;
        slave_cluster=cluster1;
    }
    for(i=0;i<slave_cluster->nvtx;i++){
        cluster_add_vertex(master_cluster,slave_cluster->vtx[i]);
    }
    cluster_free(slave_cluster);
    slave_cluster=NULL;
    return TS_SUCCESS;
}

ts_bool cluster_free(ts_cluster *cluster){
    if(cluster!=NULL){
        if(cluster->vtx!=NULL)
@@ -47,8 +85,70 @@
        for(i=0;i<cstlist->n;i++){
            cluster_free(cstlist->cluster[i]);
        }
        free(cstlist->cluster);
        free(cstlist);
    }
    return TS_SUCCESS;
}


/* This is a stub function. User should check whether the vertex is clustering or not. */
ts_bool is_clusterable(ts_vertex *vtx){

return 1;
}


ts_cluster *cluster_vertex_neighbor(ts_vertex *vtx){
    int j;
    for(j=0;j<vtx->neigh_no;j++){
        if(vtx->neigh[j]->cluster!=NULL)
            return vtx->neigh[j]->cluster;
    }
    return NULL;
}

ts_bool cluster_vertex_neighbor_check(ts_vertex *vtx){

    int j;
    for(j=0;j<vtx->neigh_no;j++){
        if(vtx->neigh[j]->cluster!=NULL){
            if(vtx->neigh[j]->cluster!=vtx->cluster){
                cluster_join(vtx->cluster, vtx->neigh[j]->cluster);
            }
        }
    }
    return TS_SUCCESS;
}


ts_bool clusterize_vesicle(ts_vesicle *vesicle, ts_cluster_list *cstlist){

    int i;
    ts_vertex *vtx;
    ts_cluster *cst;
    for(i=0;i<vesicle->vlist->n;i++){
    //for each vertex
        vtx=vesicle->vlist->vtx[i];
        if(is_clusterable(vtx)){
            if(vtx->cluster!=NULL){
                //find first neigbor with cluster index
                cst=cluster_vertex_neighbor(vtx);
                if(cst==NULL){
                    //no clusters are around us, vo we are probably lonely vertex or no surronding vertex has been mapped yet.
                    cst=new_cluster(cstlist);
                    cluster_add_vertex(cst,vtx);
                } else {
                    //we are added to the first cluster found
                    cluster_add_vertex(cst,vtx);
                    cluster_vertex_neighbor_check(vtx);
                    cluster_list_compact(cstlist);
                }

            }
        }
    }


    return TS_SUCCESS;
}