tags/2.3.1/src/jp/nyatla/nyartoolkit/core/transmat/optimize/NyARRotTransOptimize.java

   1 /* \r
   2  * PROJECT: NyARToolkit\r
   3  * --------------------------------------------------------------------------------\r
   4  * This work is based on the original ARToolKit developed by\r
   5  *   Hirokazu Kato\r
   6  *   Mark Billinghurst\r
   7  *   HITLab, University of Washington, Seattle\r
   8  * http://www.hitl.washington.edu/artoolkit/\r
   9  *\r
  10  * The NyARToolkit is Java version ARToolkit class library.\r
  11  * Copyright (C)2008 R.Iizuka\r
  12  *\r
  13  * This program is free software; you can redistribute it and/or\r
  14  * modify it under the terms of the GNU General Public License\r
  15  * as published by the Free Software Foundation; either version 2\r
  16  * of the License, or (at your option) any later version.\r
  17  * \r
  18  * This program is distributed in the hope that it will be useful,\r
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  21  * GNU General Public License for more details.\r
  22  * \r
  23  * You should have received a copy of the GNU General Public License\r
  24  * along with this framework; if not, write to the Free Software\r
  25  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\r
  26  * \r
  27  * For further information please contact.\r
  28  *      http://nyatla.jp/nyatoolkit/\r
  29  *      <airmail(at)ebony.plala.or.jp>\r
  30  * \r
  31  */\r
  32 package jp.nyatla.nyartoolkit.core.transmat.optimize;\r
  33 \r
  34 \r
  35 import jp.nyatla.nyartoolkit.NyARException;\r
  36 import jp.nyatla.nyartoolkit.core.param.*;\r
  37 import jp.nyatla.nyartoolkit.core.transmat.fitveccalc.NyARFitVecCalculator;\r
  38 import jp.nyatla.nyartoolkit.core.transmat.rotmatrix.NyARRotMatrix;\r
  39 import jp.nyatla.nyartoolkit.core.types.NyARDoublePoint2d;\r
  40 import jp.nyatla.nyartoolkit.core.types.NyARDoublePoint3d;\r
  41 /**\r
  42  * 基本姿勢と実画像を一致するように、角度を微調整→平行移動量を再計算\r
  43  * を繰り返して、変換行列を最適化する。\r
  44  *\r
  45  */\r
  46 public class NyARRotTransOptimize implements INyARRotTransOptimize\r
  47 {\r
  48         private final static int AR_GET_TRANS_MAT_MAX_LOOP_COUNT = 5;// #define AR_GET_TRANS_MAT_MAX_LOOP_COUNT 5\r
  49         private final static double AR_GET_TRANS_MAT_MAX_FIT_ERROR = 1.0;// #define AR_GET_TRANS_MAT_MAX_FIT_ERROR 1.0\r
  50         private final NyARPerspectiveProjectionMatrix _projection_mat_ref;\r
  51         public NyARRotTransOptimize(NyARPerspectiveProjectionMatrix i_projection_mat_ref)\r
  52         {\r
  53                 this._projection_mat_ref=i_projection_mat_ref;\r
  54                 return;\r
  55         }\r
  56         \r
  57         final public double optimize(NyARRotMatrix io_rotmat,NyARDoublePoint3d io_transvec,NyARFitVecCalculator i_calculator) throws NyARException\r
  58         {\r
  59                 final NyARDoublePoint2d[] fit_vertex=i_calculator.getFitSquare();\r
  60                 final NyARDoublePoint3d[] offset_square=i_calculator.getOffsetVertex().vertex;\r
  61                 \r
  62                 double err = -1;\r
  63                 /*ループを抜けるタイミングをARToolKitと合わせるために変なことしてます。*/\r
  64                 for (int i = 0;; i++) {\r
  65                         // <arGetTransMat3>\r
  66                         err = modifyMatrix(io_rotmat,io_transvec,offset_square,fit_vertex);\r
  67                         i_calculator.calculateTransfer(io_rotmat, io_transvec);\r
  68                         err = modifyMatrix(io_rotmat,io_transvec,offset_square,fit_vertex);                     \r
  69                         // //</arGetTransMat3>\r
  70                         if (err < AR_GET_TRANS_MAT_MAX_FIT_ERROR || i == AR_GET_TRANS_MAT_MAX_LOOP_COUNT-1) {\r
  71                                 break;\r
  72                         }\r
  73                         i_calculator.calculateTransfer(io_rotmat, io_transvec);\r
  74                 }               \r
  75                 return err;\r
  76         }\r
  77         \r
  78         private final double[][] __modifyMatrix_double1D = new double[8][3];\r
  79         /**\r
  80          * arGetRot計算を階層化したModifyMatrix 896\r
  81          * \r
  82          * @param nyrot\r
  83          * @param trans\r
  84          * @param i_vertex3d\r
  85          * [m][3]\r
  86          * @param i_vertex2d\r
  87          * [n][2]\r
  88          * @return\r
  89          * @throws NyARException\r
  90          */\r
  91         private double modifyMatrix(NyARRotMatrix io_rot,NyARDoublePoint3d trans, NyARDoublePoint3d[] i_vertex3d, NyARDoublePoint2d[] i_vertex2d) throws NyARException\r
  92         {\r
  93                 double factor;\r
  94                 double a2, b2, c2;\r
  95                 double ma = 0.0, mb = 0.0, mc = 0.0;\r
  96                 double h, x, y;\r
  97                 double err, minerr = 0;\r
  98                 int t1, t2, t3;\r
  99                 int s1 = 0, s2 = 0, s3 = 0;\r
 100 \r
 101                 factor = 10.0 * Math.PI / 180.0;\r
 102                 double rot0, rot1, rot3, rot4, rot6, rot7;\r
 103                 double combo00, combo01, combo02, combo03, combo10, combo11, combo12, combo13, combo20, combo21, combo22, combo23;\r
 104                 double combo02_2, combo02_5, combo02_8, combo02_11;\r
 105                 double combo22_2, combo22_5, combo22_8, combo22_11;\r
 106                 double combo12_2, combo12_5, combo12_8, combo12_11;\r
 107                 // vertex展開\r
 108                 final double VX00, VX01, VX02, VX10, VX11, VX12, VX20, VX21, VX22, VX30, VX31, VX32;\r
 109                 NyARDoublePoint3d d_pt;\r
 110                 d_pt = i_vertex3d[0];\r
 111                 VX00 = d_pt.x;\r
 112                 VX01 = d_pt.y;\r
 113                 VX02 = d_pt.z;\r
 114                 d_pt = i_vertex3d[1];\r
 115                 VX10 = d_pt.x;\r
 116                 VX11 = d_pt.y;\r
 117                 VX12 = d_pt.z;\r
 118                 d_pt = i_vertex3d[2];\r
 119                 VX20 = d_pt.x;\r
 120                 VX21 = d_pt.y;\r
 121                 VX22 = d_pt.z;\r
 122                 d_pt = i_vertex3d[3];\r
 123                 VX30 = d_pt.x;\r
 124                 VX31 = d_pt.y;\r
 125                 VX32 = d_pt.z;\r
 126                 final double P2D00, P2D01, P2D10, P2D11, P2D20, P2D21, P2D30, P2D31;\r
 127                 NyARDoublePoint2d d_pt2;\r
 128                 d_pt2 = i_vertex2d[0];\r
 129                 P2D00 = d_pt2.x;\r
 130                 P2D01 = d_pt2.y;\r
 131                 d_pt2 = i_vertex2d[1];\r
 132                 P2D10 = d_pt2.x;\r
 133                 P2D11 = d_pt2.y;\r
 134                 d_pt2 = i_vertex2d[2];\r
 135                 P2D20 = d_pt2.x;\r
 136                 P2D21 = d_pt2.y;\r
 137                 d_pt2 = i_vertex2d[3];\r
 138                 P2D30 = d_pt2.x;\r
 139                 P2D31 = d_pt2.y;\r
 140                 final NyARPerspectiveProjectionMatrix prjmat = this._projection_mat_ref;\r
 141                 final double CP0, CP1, CP2, CP4, CP5, CP6, CP8, CP9, CP10;\r
 142                 CP0 = prjmat.m00;\r
 143                 CP1 = prjmat.m01;\r
 144                 CP2 = prjmat.m02;\r
 145                 CP4 = prjmat.m10;\r
 146                 CP5 = prjmat.m11;\r
 147                 CP6 = prjmat.m12;\r
 148                 CP8 = prjmat.m20;\r
 149                 CP9 = prjmat.m21;\r
 150                 CP10 = prjmat.m22;\r
 151                 combo03 = CP0 * trans.x + CP1 * trans.y + CP2 * trans.z + prjmat.m03;\r
 152                 combo13 = CP4 * trans.x + CP5 * trans.y + CP6 * trans.z + prjmat.m13;\r
 153                 combo23 = CP8 * trans.x + CP9 * trans.y + CP10 * trans.z + prjmat.m23;\r
 154                 double CACA, SASA, SACA, CA, SA;\r
 155                 double CACACB, SACACB, SASACB, CASB, SASB;\r
 156                 double SACASC, SACACBSC, SACACBCC, SACACC;\r
 157                 final double[][] double1D = this.__modifyMatrix_double1D;\r
 158 \r
 159 \r
 160                 final double[] a_factor = double1D[1];\r
 161                 final double[] sinb = double1D[2];\r
 162                 final double[] cosb = double1D[3];\r
 163                 final double[] b_factor = double1D[4];\r
 164                 final double[] sinc = double1D[5];\r
 165                 final double[] cosc = double1D[6];\r
 166                 final double[] c_factor = double1D[7];\r
 167                 double w, w2;\r
 168                 double wsin, wcos;\r
 169                 \r
 170                 //現在の角度を確保\r
 171                 final NyARDoublePoint3d angle = io_rot.refAngle();\r
 172                 a2 = angle.x;\r
 173                 b2 = angle.y;\r
 174                 c2 = angle.z;\r
 175 \r
 176                 // comboの3行目を先に計算\r
 177                 for (int i = 0; i < 10; i++) {\r
 178                         minerr = 1000000000.0;\r
 179                         // sin-cosテーブルを計算(これが外に出せるとは…。)\r
 180                         for (int j = 0; j < 3; j++) {\r
 181                                 w2 = factor * (j - 1);\r
 182                                 w = a2 + w2;\r
 183                                 a_factor[j] = w;\r
 184                                 w = b2 + w2;\r
 185                                 b_factor[j] = w;\r
 186                                 sinb[j] = Math.sin(w);\r
 187                                 cosb[j] = Math.cos(w);\r
 188                                 w = c2 + w2;\r
 189                                 c_factor[j] = w;\r
 190                                 sinc[j] = Math.sin(w);\r
 191                                 cosc[j] = Math.cos(w);\r
 192                         }\r
 193                         //\r
 194                         for (t1 = 0; t1 < 3; t1++) {\r
 195                                 SA = Math.sin(a_factor[t1]);\r
 196                                 CA = Math.cos(a_factor[t1]);\r
 197                                 // Optimize\r
 198                                 CACA = CA * CA;\r
 199                                 SASA = SA * SA;\r
 200                                 SACA = SA * CA;\r
 201                                 for (t2 = 0; t2 < 3; t2++) {\r
 202                                         wsin = sinb[t2];\r
 203                                         wcos = cosb[t2];\r
 204                                         CACACB = CACA * wcos;\r
 205                                         SACACB = SACA * wcos;\r
 206                                         SASACB = SASA * wcos;\r
 207                                         CASB = CA * wsin;\r
 208                                         SASB = SA * wsin;\r
 209                                         // comboの計算1\r
 210                                         combo02 = CP0 * CASB + CP1 * SASB + CP2 * wcos;\r
 211                                         combo12 = CP4 * CASB + CP5 * SASB + CP6 * wcos;\r
 212                                         combo22 = CP8 * CASB + CP9 * SASB + CP10 * wcos;\r
 213 \r
 214                                         combo02_2 = combo02 * VX02 + combo03;\r
 215                                         combo02_5 = combo02 * VX12 + combo03;\r
 216                                         combo02_8 = combo02 * VX22 + combo03;\r
 217                                         combo02_11 = combo02 * VX32 + combo03;\r
 218                                         combo12_2 = combo12 * VX02 + combo13;\r
 219                                         combo12_5 = combo12 * VX12 + combo13;\r
 220                                         combo12_8 = combo12 * VX22 + combo13;\r
 221                                         combo12_11 = combo12 * VX32 + combo13;\r
 222                                         combo22_2 = combo22 * VX02 + combo23;\r
 223                                         combo22_5 = combo22 * VX12 + combo23;\r
 224                                         combo22_8 = combo22 * VX22 + combo23;\r
 225                                         combo22_11 = combo22 * VX32 + combo23;\r
 226                                         for (t3 = 0; t3 < 3; t3++) {\r
 227                                                 wsin = sinc[t3];\r
 228                                                 wcos = cosc[t3];\r
 229                                                 SACASC = SACA * wsin;\r
 230                                                 SACACC = SACA * wcos;\r
 231                                                 SACACBSC = SACACB * wsin;\r
 232                                                 SACACBCC = SACACB * wcos;\r
 233 \r
 234                                                 rot0 = CACACB * wcos + SASA * wcos + SACACBSC - SACASC;\r
 235                                                 rot3 = SACACBCC - SACACC + SASACB * wsin + CACA * wsin;\r
 236                                                 rot6 = -CASB * wcos - SASB * wsin;\r
 237 \r
 238                                                 combo00 = CP0 * rot0 + CP1 * rot3 + CP2 * rot6;\r
 239                                                 combo10 = CP4 * rot0 + CP5 * rot3 + CP6 * rot6;\r
 240                                                 combo20 = CP8 * rot0 + CP9 * rot3 + CP10 * rot6;\r
 241 \r
 242                                                 rot1 = -CACACB * wsin - SASA * wsin + SACACBCC - SACACC;\r
 243                                                 rot4 = -SACACBSC + SACASC + SASACB * wcos + CACA * wcos;\r
 244                                                 rot7 = CASB * wsin - SASB * wcos;\r
 245                                                 combo01 = CP0 * rot1 + CP1 * rot4 + CP2 * rot7;\r
 246                                                 combo11 = CP4 * rot1 + CP5 * rot4 + CP6 * rot7;\r
 247                                                 combo21 = CP8 * rot1 + CP9 * rot4 + CP10 * rot7;\r
 248                                                 //\r
 249                                                 err = 0.0;\r
 250                                                 h = combo20 * VX00 + combo21 * VX01 + combo22_2;\r
 251                                                 x = P2D00 - (combo00 * VX00 + combo01 * VX01 + combo02_2) / h;\r
 252                                                 y = P2D01 - (combo10 * VX00 + combo11 * VX01 + combo12_2) / h;\r
 253                                                 err += x * x + y * y;\r
 254                                                 h = combo20 * VX10 + combo21 * VX11 + combo22_5;\r
 255                                                 x = P2D10 - (combo00 * VX10 + combo01 * VX11 + combo02_5) / h;\r
 256                                                 y = P2D11 - (combo10 * VX10 + combo11 * VX11 + combo12_5) / h;\r
 257                                                 err += x * x + y * y;\r
 258                                                 h = combo20 * VX20 + combo21 * VX21 + combo22_8;\r
 259                                                 x = P2D20 - (combo00 * VX20 + combo01 * VX21 + combo02_8) / h;\r
 260                                                 y = P2D21 - (combo10 * VX20 + combo11 * VX21 + combo12_8) / h;\r
 261                                                 err += x * x + y * y;\r
 262                                                 h = combo20 * VX30 + combo21 * VX31 + combo22_11;\r
 263                                                 x = P2D30 - (combo00 * VX30 + combo01 * VX31 + combo02_11) / h;\r
 264                                                 y = P2D31 - (combo10 * VX30 + combo11 * VX31 + combo12_11) / h;\r
 265                                                 err += x * x + y * y;\r
 266                                                 if (err < minerr) {\r
 267                                                         minerr = err;\r
 268                                                         ma = a_factor[t1];\r
 269                                                         mb = b_factor[t2];\r
 270                                                         mc = c_factor[t3];\r
 271                                                         s1 = t1 - 1;\r
 272                                                         s2 = t2 - 1;\r
 273                                                         s3 = t3 - 1;\r
 274                                                 }\r
 275                                         }\r
 276                                 }\r
 277                         }\r
 278                         if (s1 == 0 && s2 == 0 && s3 == 0) {\r
 279                                 factor *= 0.5;\r
 280                         }\r
 281                         a2 = ma;\r
 282                         b2 = mb;\r
 283                         c2 = mc;\r
 284                 }\r
 285                 io_rot.setAngle(ma, mb, mc);\r
 286                 /* printf("factor = %10.5f\n", factor*180.0/MD_PI); */\r
 287                 return minerr / 4;\r
 288         }       \r
 289         \r
 290         \r
 291 }\r