mpi/fruitfly/registration/PointMatch.java

   1 package mpi.fruitfly.registration;
   2
   3 import java.util.ArrayList;
   4 import java.util.Collection;
   5
   6 public class PointMatch
   7 {
   8         final private Point p1;
   9         final public Point getP1() { return p1; }
  10
  11         final private Point p2;
  12         final public Point getP2() { return p2; }
  13
  14         private float weight;
  15         final public float getWeight(){ return weight; }
  16         final public void setWeight( float weight ){ this.weight = weight; }
  17
  18         private float distance;
  19         final public float getDistance(){ return distance; }
  20
  21         public PointMatch(
  22                         Point p1,
  23                         Point p2,
  24                         float weight )
  25         {
  26                 this.p1 = p1;
  27                 this.p2 = p2;
  28
  29                 this.weight = weight;
  30
  31                 distance = Point.distance( p1, p2 );
  32         }
  33
  34         public PointMatch(
  35                         Point p1,
  36                         Point p2 )
  37         {
  38                 this.p1 = p1;
  39                 this.p2 = p2;
  40
  41                 weight = 1.0f;
  42
  43                 distance = Point.distance( p1, p2 );
  44         }
  45
  46         /**
  47          * apply a model to p1, update distance
  48          *
  49          * @param model
  50          */
  51         final public void apply( Model model )
  52         {
  53                 p1.apply( model );
  54                 distance = Point.distance( p1, p2 );
  55         }
  56
  57         /**
  58          * convert from Matches, weight is 1.0
  59          *
  60          * @param matches
  61          * @return
  62          */
  63         final public static ArrayList< PointMatch > fromMatches( Collection< Match > matches )
  64         {
  65                 ArrayList< PointMatch > list = new ArrayList< PointMatch >();
  66                 for ( Match match : matches )
  67                 {
  68                         list.add(
  69                                         new PointMatch(
  70                                                         new Point( match.p1 ),
  71                                                         new Point( match.p2 ) ) );
  72                 }
  73                 return list;
  74         }
  75
  76         /**
  77          * convert to Matches using the world coordinates, weight gets lost
  78          *
  79          * @param matches
  80          * @return
  81          */
  82         final public static ArrayList< Match > toMatches( Collection< PointMatch > matches )
  83         {
  84                 ArrayList< Match > list = new ArrayList< Match >();
  85                 for ( PointMatch match : matches )
  86                 {
  87                         list.add(
  88                                         new Match(
  89                                                         match.p1.getW(),
  90                                                         match.p2.getW() ) );
  91                 }
  92                 return list;
  93         }
  94
  95         final public static ArrayList< Match > toLocalMatches( Collection< PointMatch > matches )
  96         {
  97                 ArrayList< Match > list = new ArrayList< Match >();
  98                 for ( PointMatch match : matches )
  99                 {
 100                         list.add(
 101                                         new Match(
 102                                                         match.p2.getW(),
 103                                                         match.p1.getL() ) );
 104                 }
 105                 return list;
 106         }
 107
 108         /**
 109          * flip symmetrically, weight remains unchanged
 110          *
 111          * @param matches
 112          * @return
 113          */
 114         final public static ArrayList< PointMatch > flip( Collection< PointMatch > matches )
 115         {
 116                 ArrayList< PointMatch > list = new ArrayList< PointMatch >();
 117                 for ( PointMatch match : matches )
 118                 {
 119                         list.add(
 120                                         new PointMatch(
 121                                                         match.p2,
 122                                                         match.p1,
 123                                                         match.weight ) );
 124                 }
 125                 return list;
 126         }
 127
 128 }