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() { return rtree.iterator(); }
  13
  14     public void clear() {
  15         exact.clear();
  16         rtree = new RTree<V>();
  17     }
  18
  19     public V get(Point p) {
  20         return exact.get(p);
  21     }
  22
  23     public void rebuild() {
  24     }
  25
  26     public void add(V v) {
  27         V x = get(v.getPoint());
  28         if (x != null && x.equals(v)) return;
  29         if (x != null) throw new Error("duplicates!");
  30         Point p = v.getPoint();
  31         rtree.insert(v);
  32         exact.put(p, v);
  33     }
  34
  35     public void remove(V v) {
  36         Point p = v.getPoint();
  37         rtree.remove(v);
  38         exact.remove(p);
  39     }
  40
  41     public V nearest(Point p) {
  42         if (exact.size()==0) return null;
  43         return rtree.nearest(p);
  44     }
  45
  46     public Vec diagonal() {
  47         float min_x = Float.MAX_VALUE;
  48         float min_y = Float.MAX_VALUE;
  49         float min_z = Float.MAX_VALUE;
  50         float max_x = Float.MIN_VALUE;
  51         float max_y = Float.MIN_VALUE;
  52         float max_z = Float.MIN_VALUE;
  53         for(V v : this) {
  54             Point p = v.getPoint();
  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(V v : this) {
  73             Point p = v.getPoint();
  74             if (p.x < min_x) min_x = p.x;
  75             if (p.y < min_y) min_y = p.y;
  76             if (p.z < min_z) min_z = p.z;
  77             if (p.x > max_x) max_x = p.x;
  78             if (p.y > max_y) max_y = p.y;
  79             if (p.z > max_z) max_z = p.z;
  80         }
  81         return new Point((float)(max_x + min_x)/2,
  82                      (float)(max_y + min_y)/2,
  83                      (float)(max_z + min_z)/2);
  84     }
  85 }