src/edu/berkeley/qfat/geom/PointSet.java

   1 package edu.berkeley.qfat.geom;
   2 import java.util.*;
   3
   4 public class PointSet<V extends HasPoint> implements Iterable<V> {
   5
   6     private RTree<V> rtree = new RTree<V>();
   7
   8     private HashMap<Point,V> exact = new HashMap<Point,V>();
   9
  10     public int size() { return exact.size(); }
  11
  12     public Iterator<V> iterator() {
  13         return exact.values().iterator();
  14     }
  15     public void clear() {
  16         exact = new HashMap<Point,V>();
  17         rtree = new RTree<V>();
  18     }
  19
  20     public V get(Point p) {
  21         return exact.get(p);
  22     }
  23
  24     public void rebuild() {
  25     }
  26
  27     public void add(V v) {
  28         V x = get(v.getPoint());
  29         if (x != null && x.equals(v)) return;
  30         if (x != null) throw new Error("duplicates!");
  31         Point p = v.getPoint();
  32         rtree.insert(v);
  33         exact.put(p, v);
  34     }
  35
  36     public void remove(V v) {
  37         Point p = v.getPoint();
  38         rtree.remove(v);
  39         exact.remove(p);
  40     }
  41
  42     public V nearest(Point p) {
  43         if (exact.size()==0) return null;
  44         return rtree.nearest(p);
  45     }
  46
  47     public Vec diagonal() {
  48         float min_x = Float.MAX_VALUE;
  49         float min_y = Float.MAX_VALUE;
  50         float min_z = Float.MAX_VALUE;
  51         float max_x = Float.MIN_VALUE;
  52         float max_y = Float.MIN_VALUE;
  53         float max_z = Float.MIN_VALUE;
  54         for(Point p : exact.keySet()) {
  55             if (p.x < min_x) min_x = p.x;
  56             if (p.y < min_y) min_y = p.y;
  57             if (p.z < min_z) min_z = p.z;
  58             if (p.x > max_x) max_x = p.x;
  59             if (p.y > max_y) max_y = p.y;
  60             if (p.z > max_z) max_z = p.z;
  61         }
  62         return new Vec(max_x - min_x, max_y - min_y, max_z - min_z);
  63     }
  64
  65     public Point centroid() {
  66         float min_x = Float.MAX_VALUE;
  67         float min_y = Float.MAX_VALUE;
  68         float min_z = Float.MAX_VALUE;
  69         float max_x = Float.MIN_VALUE;
  70         float max_y = Float.MIN_VALUE;
  71         float max_z = Float.MIN_VALUE;
  72         for(Point p : exact.keySet()) {
  73             if (p.x < min_x) min_x = p.x;
  74             if (p.y < min_y) min_y = p.y;
  75             if (p.z < min_z) min_z = p.z;
  76             if (p.x > max_x) max_x = p.x;
  77             if (p.y > max_y) max_y = p.y;
  78             if (p.z > max_z) max_z = p.z;
  79         }
  80         return new Point((float)(max_x + min_x)/2,
  81                      (float)(max_y + min_y)/2,
  82                      (float)(max_z + min_z)/2);
  83     }
  84 }