TrakEM2_/src/main/java/lenscorrection/PolynomialModel2D.java

   1 /**
   2  * License: GPL
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License 2
   6  * as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public License
  14  * along with this program; if not, write to the Free Software
  15  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  16  */
  17 package lenscorrection;
  18
  19 import java.util.Collection;
  20
  21 import mpicbg.models.AbstractAffineModel2D;
  22 import mpicbg.models.AbstractModel;
  23 import mpicbg.models.IllDefinedDataPointsException;
  24 import mpicbg.models.NotEnoughDataPointsException;
  25 import mpicbg.models.PointMatch;
  26
  27 /**
  28  * A wrpper for {@link NonLinearTransform} and the {@link AbstractAffineModel2D}
  29  * to which it is regularized.
  30  *
  31  * @author Stephan Saalfeld <saalfeld@mpi-cbg.de>
  32  * @version 0.1a
  33  */
  34 public class PolynomialModel2D extends AbstractModel< PolynomialModel2D >
  35 {
  36         private NonLinearTransform nlt = new NonLinearTransform();
  37         private AbstractAffineModel2D< ? > affine = null;
  38         private float lambda = 0f;
  39
  40
  41         public AbstractAffineModel2D< ? > getAffine(){ return affine; }
  42         public void setAffine( final AbstractAffineModel2D< ? > affine ){ this.affine = affine; }
  43         public void setAffine( final Class< ? extends AbstractAffineModel2D< ? > > affineClass ) throws Exception
  44         {
  45                 affine = affineClass.newInstance();
  46         }
  47
  48         public int getOrder(){ return nlt.getDimension(); }
  49         public void setOrder( final int order ){ nlt.setDimension( order ); }
  50
  51         public float getLambda(){ return lambda; }
  52         public void setLambda( final float lambda ){ this.lambda = lambda; }
  53
  54
  55         @Override
  56         public PolynomialModel2D copy()
  57         {
  58                 final PolynomialModel2D clone = new PolynomialModel2D();
  59                 clone.nlt = nlt.copy();
  60                 clone.affine = affine.copy();
  61                 clone.lambda = lambda;
  62                 return clone;
  63         }
  64
  65         @Override
  66         public < P extends PointMatch >void fit( Collection< P > pointMatches ) throws NotEnoughDataPointsException, IllDefinedDataPointsException
  67         {
  68                 if ( pointMatches.size() < getMinNumMatches() )
  69                         throw new NotEnoughDataPointsException( pointMatches.size() + " data points are not enough to estimate a 2d polynomial of order " + nlt.getDimension() + ", at least " + getMinNumMatches() + " data points required." );
  70
  71                 affine.fit( pointMatches );
  72
  73                 final double h1[][] = new double[ pointMatches.size() ][ 2 ];
  74             final double h2[][] = new double[ pointMatches.size() ][ 2 ];
  75
  76             int i = 0;
  77                 for ( final P match : pointMatches )
  78             {
  79                 final float[] tmp1 = match.getP1().getL().clone();
  80                 affine.applyInPlace( tmp1 );
  81
  82                 final float[] tmp2 = match.getP2().getW();
  83
  84                 h1[ i ] = new double[]{ tmp1[ 0 ], tmp1[ 1 ] };
  85                         h2[ i ] = new double[]{ tmp2[ 0 ], tmp2[ 1 ] };
  86
  87                         ++i;
  88             }
  89
  90                 nlt.fit( h1, h2, lambda );
  91         }
  92
  93         @Override
  94         public int getMinNumMatches()
  95         {
  96                 return Math.max( affine.getMinNumMatches(), nlt.getMinNumMatches() );
  97         }
  98
  99         @Override
 100         public void set( PolynomialModel2D m )
 101         {
 102                 nlt.set( m.nlt );
 103                 affine = m.affine.copy();
 104                 lambda = m.lambda;
 105         }
 106
 107         @Override
 108         public String toString()
 109         {
 110                 return "affine: " + affine.toString() + "; polynomial: " + nlt.toString();
 111         }
 112
 113         @Override
 114         public float[] apply( float[] location )
 115         {
 116                 final float[] l = location.clone();
 117                 applyInPlace( l );
 118                 return l;
 119         }
 120
 121         @Override
 122         public void applyInPlace( float[] location )
 123         {
 124                 affine.applyInPlace( location );
 125                 nlt.applyInPlace( location );
 126         }
 127 }