src/jp/nyatla/nyartoolkit/core/transmat/optimize/NyARRotTransOptimize_O2.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_O2 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_O2(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 trans\r
  83          * @param i_vertex3d\r
  84          * [m][3]\r
  85          * @param i_vertex2d\r
  86          * [n][2]\r
  87          * @return\r
  88          * @throws NyARException\r
  89          */\r
  90         private double modifyMatrix(NyARRotMatrix io_rot,NyARDoublePoint3d trans, NyARDoublePoint3d[] i_vertex3d, NyARDoublePoint2d[] i_vertex2d) throws NyARException\r
  91         {\r
  92                 double factor;\r
  93                 double a2, b2, c2;\r
  94                 double h, x, y;\r
  95                 double err, minerr = 0;\r
  96                 int t1, t2, t3;\r
  97                 int best_idx=0;\r
  98 \r
  99                 factor = 10.0 * Math.PI / 180.0;\r
 100                 double rot0, rot1, rot2;\r
 101                 double combo00, combo01, combo02, combo03, combo10, combo11, combo12, combo13, combo20, combo21, combo22, combo23;\r
 102                 double combo02_2, combo02_5, combo02_8, combo02_11;\r
 103                 double combo22_2, combo22_5, combo22_8, combo22_11;\r
 104                 double combo12_2, combo12_5, combo12_8, combo12_11;\r
 105                 // vertex展開\r
 106                 final double VX00, VX01, VX02, VX10, VX11, VX12, VX20, VX21, VX22, VX30, VX31, VX32;\r
 107                 VX00 = i_vertex3d[0].x;\r
 108                 VX01 = i_vertex3d[0].y;\r
 109                 VX02 = i_vertex3d[0].z;\r
 110                 VX10 = i_vertex3d[1].x;\r
 111                 VX11 = i_vertex3d[1].y;\r
 112                 VX12 = i_vertex3d[1].z;\r
 113                 VX20 = i_vertex3d[2].x;\r
 114                 VX21 = i_vertex3d[2].y;\r
 115                 VX22 = i_vertex3d[2].z;\r
 116                 VX30 = i_vertex3d[3].x;\r
 117                 VX31 = i_vertex3d[3].y;\r
 118                 VX32 = i_vertex3d[3].z;\r
 119                 final double P2D00, P2D01, P2D10, P2D11, P2D20, P2D21, P2D30, P2D31;\r
 120                 P2D00 = i_vertex2d[0].x;\r
 121                 P2D01 = i_vertex2d[0].y;\r
 122                 P2D10 = i_vertex2d[1].x;\r
 123                 P2D11 = i_vertex2d[1].y;\r
 124                 P2D20 = i_vertex2d[2].x;\r
 125                 P2D21 = i_vertex2d[2].y;\r
 126                 P2D30 = i_vertex2d[3].x;\r
 127                 P2D31 = i_vertex2d[3].y;\r
 128                 final NyARPerspectiveProjectionMatrix prjmat = this._projection_mat_ref;\r
 129                 final double CP0 = prjmat.m00,CP1 = prjmat.m01,CP2 = prjmat.m02,CP4 = prjmat.m10,CP5 = prjmat.m11,CP6 = prjmat.m12,CP8 = prjmat.m20,CP9 = prjmat.m21,CP10 = prjmat.m22;\r
 130                 combo03 = CP0 * trans.x + CP1 * trans.y + CP2 * trans.z + prjmat.m03;\r
 131                 combo13 = CP4 * trans.x + CP5 * trans.y + CP6 * trans.z + prjmat.m13;\r
 132                 combo23 = CP8 * trans.x + CP9 * trans.y + CP10 * trans.z + prjmat.m23;\r
 133                 double CACA, SASA, SACA, CA, SA;\r
 134                 double CACACB, SACACB, SASACB, CASB, SASB;\r
 135                 double SACASC, SACACBSC, SACACBCC, SACACC;\r
 136                 final double[][] double1D = this.__modifyMatrix_double1D;\r
 137 \r
 138                 final double[] a_factor = double1D[1];\r
 139                 final double[] sinb = double1D[2];\r
 140                 final double[] cosb = double1D[3];\r
 141                 final double[] b_factor = double1D[4];\r
 142                 final double[] sinc = double1D[5];\r
 143                 final double[] cosc = double1D[6];\r
 144                 final double[] c_factor = double1D[7];\r
 145                 double w, w2;\r
 146                 double wsin, wcos;\r
 147                 \r
 148                 final NyARDoublePoint3d angle = io_rot.refAngle();\r
 149                 a2 = angle.x;\r
 150                 b2 = angle.y;\r
 151                 c2 = angle.z;\r
 152 \r
 153                 // comboの3行目を先に計算\r
 154                 for (int i = 0; i < 10; i++) {\r
 155                         minerr = 1000000000.0;\r
 156                         // sin-cosテーブルを計算(これが外に出せるとは…。)\r
 157                         for (int j = 0; j < 3; j++) {\r
 158                                 w2 = factor * (j - 1);\r
 159                                 w = a2 + w2;\r
 160                                 a_factor[j] = w;\r
 161                                 w = b2 + w2;\r
 162                                 b_factor[j] = w;\r
 163                                 sinb[j] = Math.sin(w);\r
 164                                 cosb[j] = Math.cos(w);\r
 165                                 w = c2 + w2;\r
 166                                 c_factor[j] = w;\r
 167                                 sinc[j] = Math.sin(w);\r
 168                                 cosc[j] = Math.cos(w);\r
 169                         }\r
 170                         //\r
 171                         for (t1 = 0; t1 < 3; t1++) {\r
 172                                 SA = Math.sin(a_factor[t1]);\r
 173                                 CA = Math.cos(a_factor[t1]);\r
 174                                 // Optimize\r
 175                                 CACA = CA * CA;\r
 176                                 SASA = SA * SA;\r
 177                                 SACA = SA * CA;\r
 178                                 for (t2 = 0; t2 < 3; t2++) {\r
 179                                         wsin = sinb[t2];\r
 180                                         wcos = cosb[t2];\r
 181                                         CACACB = CACA * wcos;\r
 182                                         SACACB = SACA * wcos;\r
 183                                         SASACB = SASA * wcos;\r
 184                                         CASB = CA * wsin;\r
 185                                         SASB = SA * wsin;\r
 186                                         // comboの計算1\r
 187                                         combo02 = CP0 * CASB + CP1 * SASB + CP2 * wcos;\r
 188                                         combo12 = CP4 * CASB + CP5 * SASB + CP6 * wcos;\r
 189                                         combo22 = CP8 * CASB + CP9 * SASB + CP10 * wcos;\r
 190 \r
 191                                         combo02_2 = combo02 * VX02 + combo03;\r
 192                                         combo02_5 = combo02 * VX12 + combo03;\r
 193                                         combo02_8 = combo02 * VX22 + combo03;\r
 194                                         combo02_11 = combo02 * VX32 + combo03;\r
 195                                         combo12_2 = combo12 * VX02 + combo13;\r
 196                                         combo12_5 = combo12 * VX12 + combo13;\r
 197                                         combo12_8 = combo12 * VX22 + combo13;\r
 198                                         combo12_11 = combo12 * VX32 + combo13;\r
 199                                         combo22_2 = combo22 * VX02 + combo23;\r
 200                                         combo22_5 = combo22 * VX12 + combo23;\r
 201                                         combo22_8 = combo22 * VX22 + combo23;\r
 202                                         combo22_11 = combo22 * VX32 + combo23;\r
 203                                         for (t3 = 0; t3 < 3; t3++) {\r
 204                                                 wsin = sinc[t3];\r
 205                                                 wcos = cosc[t3];\r
 206                                                 SACASC = SACA * wsin;\r
 207                                                 SACACC = SACA * wcos;\r
 208                                                 SACACBSC = SACACB * wsin;\r
 209                                                 SACACBCC = SACACB * wcos;\r
 210 \r
 211                                                 rot0 = CACACB * wcos + SASA * wcos + SACACBSC - SACASC;\r
 212                                                 rot1 = SACACBCC - SACACC + SASACB * wsin + CACA * wsin;\r
 213                                                 rot2 = -CASB * wcos - SASB * wsin;\r
 214                                                 combo00 = CP0 * rot0 + CP1 * rot1 + CP2 * rot2;\r
 215                                                 combo10 = CP4 * rot0 + CP5 * rot1 + CP6 * rot2;\r
 216                                                 combo20 = CP8 * rot0 + CP9 * rot1 + CP10 * rot2;\r
 217 \r
 218                                                 rot0 = -CACACB * wsin - SASA * wsin + SACACBCC - SACACC;\r
 219                                                 rot1 = -SACACBSC + SACASC + SASACB * wcos + CACA * wcos;\r
 220                                                 rot2 = CASB * wsin - SASB * wcos;\r
 221                                                 combo01 = CP0 * rot0 + CP1 * rot1 + CP2 * rot2;\r
 222                                                 combo11 = CP4 * rot0 + CP5 * rot1 + CP6 * rot2;\r
 223                                                 combo21 = CP8 * rot0 + CP9 * rot1 + CP10 * rot2;\r
 224                                                 //\r
 225                                                 err = 0.0;\r
 226                                                 h = combo20 * VX00 + combo21 * VX01 + combo22_2;\r
 227                                                 x = P2D00 - (combo00 * VX00 + combo01 * VX01 + combo02_2) / h;\r
 228                                                 y = P2D01 - (combo10 * VX00 + combo11 * VX01 + combo12_2) / h;\r
 229                                                 err += x * x + y * y;\r
 230                                                 h = combo20 * VX10 + combo21 * VX11 + combo22_5;\r
 231                                                 x = P2D10 - (combo00 * VX10 + combo01 * VX11 + combo02_5) / h;\r
 232                                                 y = P2D11 - (combo10 * VX10 + combo11 * VX11 + combo12_5) / h;\r
 233                                                 err += x * x + y * y;\r
 234                                                 h = combo20 * VX20 + combo21 * VX21 + combo22_8;\r
 235                                                 x = P2D20 - (combo00 * VX20 + combo01 * VX21 + combo02_8) / h;\r
 236                                                 y = P2D21 - (combo10 * VX20 + combo11 * VX21 + combo12_8) / h;\r
 237                                                 err += x * x + y * y;\r
 238                                                 h = combo20 * VX30 + combo21 * VX31 + combo22_11;\r
 239                                                 x = P2D30 - (combo00 * VX30 + combo01 * VX31 + combo02_11) / h;\r
 240                                                 y = P2D31 - (combo10 * VX30 + combo11 * VX31 + combo12_11) / h;\r
 241                                                 err += x * x + y * y;\r
 242                                                 if (err < minerr) {\r
 243                                                         minerr = err;\r
 244                                                         a2 = a_factor[t1];\r
 245                                                         b2 = b_factor[t2];\r
 246                                                         c2 = c_factor[t3];\r
 247                                                         best_idx=t1+t2*3+t3*9;\r
 248                                                 }\r
 249                                         }\r
 250                                 }\r
 251                         }\r
 252                         if (best_idx==(1+3+9)) {\r
 253                                 factor *= 0.5;\r
 254                         }\r
 255                 }\r
 256                 io_rot.setAngle(a2, b2, c2);\r
 257                 /* printf("factor = %10.5f\n", factor*180.0/MD_PI); */\r
 258                 return minerr /4;\r
 259         }       \r
 260         \r
 261         \r
 262 }\r