VisClient/org/hfbk/vis/Partition.java

Go to the documentation of this file.

package org.hfbk.vis;

import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;

import org.hfbk.vis.visnode.VisNode;
import org.lwjgl.util.vector.Vector3f;

public class Partition extends HashMap<Partition.Triple, List<VisNode>>{
        
        public static class Triple{
                int x,y,z;
        
                public Triple(int x, int y, int z) {
                        this.x=x; this.y=y; this.z=z;  
                }
                public int hashCode() {
                        return x^y^z;
                }
                
                public boolean equals(Object other) {
                        if (!(other instanceof Triple)) return false;
                        Triple ot=(Triple)other;
                        return ot.x==x && ot.y==y && ot.z==z;
                }
                
                public Triple[] getHalfNeighbourhood(){
                        return new Triple[]{
                                        new Triple(x+1, y, z), //main axes
                                        new Triple(x, y+1, z),
                                        new Triple(x, y, z+1),
                                        new Triple(x+1, y+1, z), //2d diagonals
                                        new Triple(x-1, y+1, z),
                                        new Triple(x, y+1, z+1),
                                        new Triple(x, y-1, z+1),
                                        new Triple(x+1, y, z+1),
                                        new Triple(x-1, y, z+1),
                                        new Triple(x+1, y+1, z+1), //3d diagonals
                                        new Triple(x-1, y+1, z+1),
                                        new Triple(x+1, y-1, z+1),
                                        new Triple(x-1, y-1, z+1)
                        };
                }
        }

        float cellsize;
        
        public Partition(Collection<VisNode> nodes, float cellsize) {
                this.cellsize=cellsize;
                for(VisNode n : nodes)
                        add(n); 
        }
        
        void add(VisNode n){
                Vector3f p=n.position;
                Triple t=new Triple((int)(p.x/cellsize), (int)(p.y/cellsize), (int)(p.z/cellsize));                     
                
                List<VisNode> l=get(t);
                if (l==null){
                        l=new ArrayList<VisNode>();
                        put(t, l);
                }
                l.add(n);                       
        }
        
        public float getMeanCount(){
                int i=0; float count=0;
                for (List l: values()){
                        count+=l.size();
                        i++;
                }
                return count/i;
        }       
                        
        public List<List<VisNode>> getHalfNeighbourhood(Triple cell){
                Triple[] half=cell.getHalfNeighbourhood();
                List<List<VisNode>> nodes=new ArrayList<List<VisNode>>(half.length);
                
                for (Triple t: half) 
                        nodes.add(get(t));
                return nodes;
        }
}

---