OSDN Git Service

初期インポート。
[nyartoolkit-and/nyartoolkit-android-0.9.git] / src / jp / nyatla / nyartoolkit / core / NyARColorPatt_O3.java
diff --git a/src/jp/nyatla/nyartoolkit/core/NyARColorPatt_O3.java b/src/jp/nyatla/nyartoolkit/core/NyARColorPatt_O3.java
new file mode 100644 (file)
index 0000000..19c569f
--- /dev/null
@@ -0,0 +1,373 @@
+/* \r
+ * PROJECT: NyARToolkit\r
+ * --------------------------------------------------------------------------------\r
+ * This work is based on the original ARToolKit developed by\r
+ *   Hirokazu Kato\r
+ *   Mark Billinghurst\r
+ *   HITLab, University of Washington, Seattle\r
+ * http://www.hitl.washington.edu/artoolkit/\r
+ *\r
+ * The NyARToolkit is Java version ARToolkit class library.\r
+ * Copyright (C)2008 R.Iizuka\r
+ *\r
+ * This program is free software; you can redistribute it and/or\r
+ * modify it under the terms of the GNU General Public License\r
+ * as published by the Free Software Foundation; either version 2\r
+ * of the License, or (at your option) any later version.\r
+ * \r
+ * This program is distributed in the hope that it will be useful,\r
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
+ * GNU General Public License for more details.\r
+ * \r
+ * You should have received a copy of the GNU General Public License\r
+ * along with this framework; if not, write to the Free Software\r
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA\r
+ * \r
+ * For further information please contact.\r
+ *     http://nyatla.jp/nyatoolkit/\r
+ *     <airmail(at)ebony.plala.or.jp>\r
+ * \r
+ */\r
+package jp.nyatla.nyartoolkit.core;\r
+\r
+\r
+\r
+import jp.nyatla.nyartoolkit.NyARException;\r
+import jp.nyatla.nyartoolkit.core.raster.NyARRaster;\r
+\r
+/**\r
+ * 24ビットカラーのマーカーを保持するために使うクラスです。\r
+ * このクラスは、ARToolkitのパターンと、ラスタから取得したパターンを保持します。\r
+ * 演算順序を含む最適化をしたもの\r
+ *\r
+ */\r
+public class NyARColorPatt_O3 implements NyARColorPatt\r
+{\r
+    private static final int AR_PATT_SAMPLE_NUM=64;//#define   AR_PATT_SAMPLE_NUM   64\r
+    private int extpat[][][];\r
+    private int width;\r
+    private int height;\r
+    public NyARColorPatt_O3(int i_width,int i_height)\r
+    {\r
+       this.width=i_width;\r
+       this.height=i_height;\r
+       this.extpat=new int[i_height][i_width][3];\r
+    }\r
+//    public void setSize(int i_new_width,int i_new_height)\r
+//    {\r
+//     int array_w=this.extpat[0].length;\r
+//     int array_h=this.extpat.length;\r
+//     //十分なサイズのバッファがあるか確認\r
+//     if(array_w>=i_new_width && array_h>=i_new_height){\r
+//         //OK 十分だ→サイズ調整のみ\r
+//     }else{\r
+//         //足りないよ→取り直し\r
+//         this.extpat=new int[i_new_height][i_new_width][3];\r
+//     }\r
+//        this.width =i_new_width;\r
+//        this.height=i_new_height;\r
+//        return;\r
+//    }    \r
+    public int[][][] getPatArray()\r
+    {\r
+       return extpat;\r
+    }\r
+    public int getWidth()\r
+    {\r
+       return width;\r
+    }\r
+    public int getHeight()\r
+    {\r
+       return height;\r
+    }\r
+    private final NyARMat wk_get_cpara_a=new NyARMat(8,8);\r
+    private final NyARMat wk_get_cpara_b=new NyARMat(8,1);\r
+\r
+    /**\r
+     * @param world\r
+     * @param vertex\r
+     * @param o_para\r
+     * @throws NyARException\r
+     */\r
+    private boolean get_cpara(double vertex_0[], double vertex_1[],NyARMat o_para) throws NyARException\r
+    {\r
+       double world[][]=this.wk_pickFromRaster_world;\r
+       NyARMat a =wk_get_cpara_a;//次処理で値を設定するので、初期化不要// new NyARMat( 8, 8 );\r
+       double[][] a_array=a.getArray();\r
+       NyARMat b =wk_get_cpara_b;//次処理で値を設定するので、初期化不要// new NyARMat( 8, 1 );\r
+       double[][] b_array=b.getArray();\r
+       double[] a_pt0,a_pt1,world_pti;\r
+\r
+       for(int i = 0; i < 4; i++ ) {\r
+           a_pt0=a_array[i*2];\r
+           a_pt1=a_array[i*2+1];\r
+           world_pti=world[i];\r
+\r
+           a_pt0[0]=world_pti[0];//a->m[i*16+0]  = world[i][0];\r
+           a_pt0[1]=world_pti[1];//a->m[i*16+1]  = world[i][1];\r
+           a_pt0[2]=1.0;//a->m[i*16+2]  = 1.0;\r
+           a_pt0[3]=0.0;//a->m[i*16+3]  = 0.0;\r
+           a_pt0[4]=0.0;//a->m[i*16+4]  = 0.0;\r
+           a_pt0[5]=0.0;//a->m[i*16+5]  = 0.0;\r
+           a_pt0[6]=-world_pti[0] * vertex_0[i];//a->m[i*16+6]  = -world[i][0] * vertex[i][0];\r
+           a_pt0[7]=-world_pti[1] * vertex_0[i];//a->m[i*16+7]  = -world[i][1] * vertex[i][0];\r
+           a_pt1[0]=0.0;//a->m[i*16+8]  = 0.0;\r
+           a_pt1[1]=0.0;//a->m[i*16+9]  = 0.0;\r
+           a_pt1[2]=0.0;//a->m[i*16+10] = 0.0;\r
+           a_pt1[3]=world_pti[0];//a->m[i*16+11] = world[i][0];\r
+           a_pt1[4]=world_pti[1];//a->m[i*16+12] = world[i][1];\r
+           a_pt1[5]=1.0;//a->m[i*16+13] = 1.0;\r
+           a_pt1[6]=-world_pti[0] * vertex_1[i];//a->m[i*16+14] = -world[i][0] * vertex[i][1];\r
+           a_pt1[7]=-world_pti[1] * vertex_1[i];//a->m[i*16+15] = -world[i][1] * vertex[i][1];\r
+           b_array[i*2+0][0]=vertex_0[i];//b->m[i*2+0] = vertex[i][0];\r
+           b_array[i*2+1][0]=vertex_1[i];//b->m[i*2+1] = vertex[i][1];\r
+       }\r
+       if(!a.matrixSelfInv()){\r
+           return false;\r
+       }           \r
+\r
+       o_para.matrixMul(a, b);\r
+       return true;\r
+    }\r
+\r
+  //   private final double[] wk_pickFromRaster_para=new double[9];//[3][3];\r
+    private final double[][] wk_pickFromRaster_world={//double    world[4][2];\r
+           {100.0,     100.0},\r
+           {100.0+10.0,100.0},\r
+           {100.0+10.0,100.0 + 10.0},\r
+           {100.0,     100.0 + 10.0}\r
+    };\r
+    /**\r
+     * pickFromRaster関数から使う変数です。\r
+     *\r
+     */\r
+    private static void initValue_wk_pickFromRaster_ext_pat2(int[][][] i_ext_pat2,int i_width,int i_height)\r
+    {\r
+       int i,i2;\r
+       int[][] pt2;\r
+       int[]   pt1;\r
+       for(i=i_height-1;i>=0;i--){\r
+           pt2=i_ext_pat2[i];\r
+           for(i2=i_width-1;i2>=0;i2--){\r
+               pt1=pt2[i2];\r
+               pt1[0]=0;\r
+               pt1[1]=0;\r
+               pt1[2]=0;\r
+           }\r
+       }\r
+    }\r
+    private final double[][] wk_pickFromRaster_local=new double[2][4];\r
+    private final NyARMat wk_pickFromRaster_cpara=new NyARMat(8,1);\r
+    /**\r
+     * imageから、i_markerの位置にあるパターンを切り出して、保持します。\r
+     * Optimize:STEP[769->750]\r
+     * @param image\r
+     * @param i_marker\r
+     * @throws Exception\r
+     */\r
+    public boolean pickFromRaster(NyARRaster image, NyARMarker i_marker) throws NyARException\r
+    {\r
+       NyARMat cpara=this.wk_pickFromRaster_cpara;\r
+       //localの計算\r
+       int[] x_coord=i_marker.x_coord;\r
+       int[] y_coord=i_marker.y_coord;\r
+       int[] vertex=i_marker.mkvertex;\r
+       double[] local_0=wk_pickFromRaster_local[0];//double    local[4][2];    \r
+       double[] local_1=wk_pickFromRaster_local[1];//double    local[4][2];    \r
+       for(int i = 0; i < 4; i++ ) {\r
+           local_0[i] = x_coord[vertex[i]];\r
+           local_1[i] = y_coord[vertex[i]];\r
+       }\r
+       //xdiv2,ydiv2の計算\r
+       int xdiv2, ydiv2;\r
+       int l1,l2;\r
+       double w1,w2;\r
+\r
+       //x計算\r
+       w1=local_0[0] - local_0[1];\r
+       w2=local_1[0] - local_1[1];\r
+       l1 = (int)(w1*w1+w2*w2);\r
+       w1=local_0[2] - local_0[3];\r
+       w2=local_1[2] - local_1[3];\r
+       l2 = (int)(w1*w1+w2*w2);\r
+       if( l2 > l1 ){\r
+           l1 = l2;\r
+       }\r
+       l1=l1/4;\r
+       xdiv2 =this.width;\r
+       while( xdiv2*xdiv2 < l1 ){\r
+           xdiv2*=2;\r
+       }\r
+       if( xdiv2 > AR_PATT_SAMPLE_NUM)\r
+       {\r
+           xdiv2 =AR_PATT_SAMPLE_NUM;\r
+       }\r
+       \r
+       //y計算\r
+       w1=local_0[1] - local_0[2];\r
+       w2=local_1[1] - local_1[2];\r
+       l1 = (int)(w1*w1+ w2*w2);\r
+       w1=local_0[3] - local_0[0];\r
+       w2=local_1[3] - local_1[0];\r
+       l2 = (int)(w1*w1+ w2*w2);\r
+       if( l2 > l1 ){\r
+           l1 = l2;\r
+       }\r
+       ydiv2 =this.height;\r
+       l1=l1/4;\r
+       while( ydiv2*ydiv2 < l1 ){\r
+           ydiv2*=2;\r
+       }\r
+       if( ydiv2 >AR_PATT_SAMPLE_NUM)\r
+       {\r
+           ydiv2 = AR_PATT_SAMPLE_NUM;\r
+       }       \r
+       \r
+       //cparaの計算\r
+       if(!get_cpara(local_0,local_1,cpara)){\r
+           return false;\r
+       }\r
+       updateExtpat(image,cpara,xdiv2,ydiv2);\r
+\r
+       return true;\r
+    }\r
+    //かなり大きいワークバッファを取るな…。\r
+    private double[] wk_updateExtpat_para00_xw;\r
+    private double[] wk_updateExtpat_para10_xw;\r
+    private double[] wk_updateExtpat_para20_xw;\r
+    private int[] wk_updateExtpat_rgb_buf;\r
+    private int[] wk_updateExtpat_x_rgb_index;\r
+    private int[] wk_updateExtpat_y_rgb_index;\r
+    private int[] wk_updateExtpat_i_rgb_index;\r
+    private int wk_updateExtpat_buffer_size=0;\r
+\r
+    /**\r
+     * ワークバッファを予約する\r
+     * @param i_xdiv2\r
+     */\r
+    private void reservWorkBuffers(int i_xdiv2)\r
+    {\r
+        if(this.wk_updateExtpat_buffer_size<i_xdiv2){\r
+            wk_updateExtpat_para00_xw=new double[i_xdiv2];\r
+            wk_updateExtpat_para10_xw=new double[i_xdiv2];\r
+            wk_updateExtpat_para20_xw=new double[i_xdiv2];\r
+            wk_updateExtpat_rgb_buf=new int[i_xdiv2*3];\r
+            wk_updateExtpat_x_rgb_index=new int[i_xdiv2];\r
+            wk_updateExtpat_y_rgb_index=new int[i_xdiv2];\r
+            wk_updateExtpat_i_rgb_index=new int[i_xdiv2];\r
+            this.wk_updateExtpat_buffer_size=i_xdiv2;\r
+        }\r
+       //十分なら何もしない。\r
+       return;\r
+    }\r
+    \r
+    private void updateExtpat(NyARRaster image,NyARMat i_cpara,int i_xdiv2,int i_ydiv2) throws NyARException\r
+    {\r
+       int img_x=image.getWidth();\r
+       int img_y=image.getHeight();\r
+       final int[][][] L_extpat=this.extpat;\r
+       final int L_WIDTH=this.width;\r
+       final int L_HEIGHT=this.height;\r
+       /*wk_pickFromRaster_ext_pat2ワーク変数を初期化する。*/\r
+       //int[][][] ext_pat2=wk_pickFromRaster_ext_pat2;//ARUint32  ext_pat2[AR_PATT_SIZE_Y][AR_PATT_SIZE_X][3];\r
+       int extpat_j[][],extpat_j_i[];\r
+       //int ext_pat2_j[][],ext_pat2_j_i[];\r
+\r
+       initValue_wk_pickFromRaster_ext_pat2(L_extpat,L_WIDTH,L_HEIGHT);\r
+\r
+       double[][] cpara_array=i_cpara.getArray();\r
+       double para21_x_yw,para01_x_yw,para11_x_yw;\r
+       double para00,para01,para02,para10,para11,para12,para20,para21;\r
+        para00 = cpara_array[0*3+0][0];//para[i][0] = c->m[i*3+0];\r
+        para01 = cpara_array[0*3+1][0];//para[i][1] = c->m[i*3+1];\r
+        para02 = cpara_array[0*3+2][0];//para[i][2] = c->m[i*3+2];\r
+        para10 = cpara_array[1*3+0][0];//para[i][0] = c->m[i*3+0];\r
+        para11 = cpara_array[1*3+1][0];//para[i][1] = c->m[i*3+1];\r
+        para12 = cpara_array[1*3+2][0];//para[i][2] = c->m[i*3+2];\r
+        para20 = cpara_array[2*3+0][0];//para[2][0] = c->m[2*3+0];\r
+        para21 = cpara_array[2*3+1][0];//para[2][1] = c->m[2*3+1];\r
+\r
+       \r
+       double          d,yw;\r
+       int             xc, yc;\r
+       int i,j;\r
+       //      arGetCode_put_zero(ext_pat2);//put_zero( (ARUint8 *)ext_pat2, AR_PATT_SIZE_Y*AR_PATT_SIZE_X*3*sizeof(ARUint32) );\r
+       int xdiv = i_xdiv2/L_WIDTH;//xdiv = xdiv2/Config.AR_PATT_SIZE_X;\r
+       int ydiv = i_ydiv2/L_HEIGHT;//ydiv = ydiv2/Config.AR_PATT_SIZE_Y;\r
+       \r
+       //計算バッファを予約する\r
+       this.reservWorkBuffers(i_xdiv2);        \r
+       double[] para00_xw=this.wk_updateExtpat_para00_xw;\r
+       double[] para10_xw=this.wk_updateExtpat_para10_xw;\r
+       double[] para20_xw=this.wk_updateExtpat_para20_xw;\r
+       int[] x_rgb_index=this.wk_updateExtpat_x_rgb_index;\r
+       int[] y_rgb_index=this.wk_updateExtpat_y_rgb_index;\r
+       int[] i_rgb_index=this.wk_updateExtpat_i_rgb_index;\r
+       int[] rgb_buf=this.wk_updateExtpat_rgb_buf;\r
+       double xw;\r
+       for(i=0;i<i_xdiv2;i++){\r
+           xw= 102.5 + 5.0 * ((double)i+0.5) /i_xdiv2;\r
+           para20_xw[i]=para20*xw;\r
+           para00_xw[i]=para00*xw;\r
+           para10_xw[i]=para10*xw;\r
+       }\r
+\r
+       int index_num;\r
+\r
+       \r
+       for(j = 0; j < i_ydiv2; j++ ) {\r
+           yw = 102.5 + 5.0 * ((double)j+0.5) /i_ydiv2;\r
+           para21_x_yw=para21*yw+1.0;\r
+           para11_x_yw=para11*yw+para12;\r
+           para01_x_yw=para01*yw+para02;\r
+           extpat_j=L_extpat[j/ydiv];\r
+           index_num=0;\r
+           //ステップ1.RGB取得用のマップを作成\r
+           for(i = 0; i < i_xdiv2; i++ ) {\r
+               d = para20_xw[i] + para21_x_yw;\r
+               if( d == 0 ){\r
+                   throw new NyARException();\r
+               }\r
+               xc = (int)((para00_xw[i] + para01_x_yw)/d);\r
+               yc = (int)((para10_xw[i] + para11_x_yw)/d);\r
+               //範囲外は無視\r
+               if(xc<0 || xc >=img_x || yc<0 || yc >=img_y){\r
+                   continue;\r
+               }\r
+//             ピクセル値の計算\r
+//             image.getPixel(xc,yc,rgb_buf);\r
+//                ext_pat2_j_i=ext_pat2_j[i/xdiv];\r
+//                ext_pat2_j_i[0] += rgb_buf[0];//R\r
+//                ext_pat2_j_i[1] += rgb_buf[1];//G\r
+//                ext_pat2_j_i[2] += rgb_buf[2];//B\r
+\r
+               x_rgb_index[index_num]=xc;\r
+               y_rgb_index[index_num]=yc;\r
+               i_rgb_index[index_num]=i/xdiv;\r
+               index_num++;\r
+           }\r
+//         //ステップ2.ピクセル配列を取得\r
+           image.getPixelSet(x_rgb_index,y_rgb_index,index_num,rgb_buf);\r
+//         //ピクセル値の計算\r
+           for(i=index_num-1;i>=0;i--){\r
+                extpat_j_i=extpat_j[i_rgb_index[i]];\r
+                extpat_j_i[0] += rgb_buf[i*3+0];//R\r
+                extpat_j_i[1] += rgb_buf[i*3+1];//G\r
+                extpat_j_i[2] += rgb_buf[i*3+2];//B\r
+           }\r
+       }\r
+       /*<Optimize>*/\r
+       int xdiv_x_ydiv=xdiv*ydiv;\r
+       for(j =L_HEIGHT-1; j>=0; j--){\r
+           extpat_j=L_extpat[j];\r
+           for(i = L_WIDTH-1; i>=0; i--){                              // PRL 2006-06-08.\r
+               extpat_j_i=extpat_j[i];\r
+               extpat_j_i[0]/=(xdiv_x_ydiv);//ext_pat[j][i][0] = (byte)(ext_pat2[j][i][0] / (xdiv*ydiv));\r
+               extpat_j_i[1]/=(xdiv_x_ydiv);//ext_pat[j][i][1] = (byte)(ext_pat2[j][i][1] / (xdiv*ydiv));\r
+               extpat_j_i[2]/=(xdiv_x_ydiv);//ext_pat[j][i][2] = (byte)(ext_pat2[j][i][2] / (xdiv*ydiv));\r
+           }\r
+       }\r
+       return; \r
+    }\r
+}
\ No newline at end of file