trunk/src/jp/nyatla/nyartoolkit/core/squaredetect/NyARSquareDetector_ARToolKit.java

   1 package jp.nyatla.nyartoolkit.core.squaredetect;\r
   2 \r
   3 import jp.nyatla.nyartoolkit.NyARException;\r
   4 import jp.nyatla.nyartoolkit.core.labeling.LabelOverlapChecker;\r
   5 import jp.nyatla.nyartoolkit.core.labeling.artoolkit.NyARLabelingImage;\r
   6 import jp.nyatla.nyartoolkit.core.labeling.artoolkit.NyARLabelingLabel;\r
   7 import jp.nyatla.nyartoolkit.core.labeling.artoolkit.NyARLabelingLabelStack;\r
   8 import jp.nyatla.nyartoolkit.core.labeling.artoolkit.NyARLabeling_ARToolKit;\r
   9 import jp.nyatla.nyartoolkit.core.param.NyARCameraDistortionFactor;\r
  10 import jp.nyatla.nyartoolkit.core.raster.NyARBinRaster;\r
  11 import jp.nyatla.nyartoolkit.core.types.NyARIntSize;\r
  12 \r
  13 \r
  14 \r
  15 public class NyARSquareDetector_ARToolKit implements INyARSquareDetector\r
  16 {\r
  17         private static final int AR_AREA_MAX = 100000;// #define AR_AREA_MAX 100000\r
  18         private static final int AR_AREA_MIN = 70;// #define AR_AREA_MIN 70\r
  19         private final int _width;\r
  20         private final int _height;\r
  21 \r
  22         private final NyARLabeling_ARToolKit _labeling;\r
  23 \r
  24         private final NyARLabelingImage _limage;\r
  25 \r
  26         private final LabelOverlapChecker<NyARLabelingLabel> _overlap_checker = new LabelOverlapChecker<NyARLabelingLabel>(32,NyARLabelingLabel.class);\r
  27         private final SquareContourDetector _sqconvertor;\r
  28         private final ContourPickup _cpickup=new ContourPickup();\r
  29         \r
  30         private final int _max_coord;\r
  31         private final int[] _xcoord;\r
  32         private final int[] _ycoord;    \r
  33         /**\r
  34          * 最大i_squre_max個のマーカーを検出するクラスを作成する。\r
  35          * \r
  36          * @param i_param\r
  37          */\r
  38         public NyARSquareDetector_ARToolKit(NyARCameraDistortionFactor i_dist_factor_ref,NyARIntSize i_size) throws NyARException\r
  39         {\r
  40                 this._width = i_size.w;\r
  41                 this._height = i_size.h;\r
  42                 //歪み計算テーブルを作ると、8*width/height*2の領域を消費します。\r
  43                 //領域を取りたくない場合は、i_dist_factor_refの値をそのまま使ってください。\r
  44                 this._labeling = new NyARLabeling_ARToolKit();\r
  45                 this._sqconvertor=new SquareContourDetector(i_size,i_dist_factor_ref);\r
  46                 this._limage = new NyARLabelingImage(this._width, this._height);\r
  47 \r
  48                 // 輪郭の最大長は画面に映りうる最大の長方形サイズ。\r
  49                 int number_of_coord = (this._width + this._height) * 2;\r
  50 \r
  51                 // 輪郭バッファは頂点変換をするので、輪郭バッファの２倍取る。\r
  52                 this._max_coord = number_of_coord;\r
  53                 this._xcoord = new int[number_of_coord * 2];\r
  54                 this._ycoord = new int[number_of_coord * 2];\r
  55                 return;\r
  56         }\r
  57 \r
  58         /**\r
  59          * arDetectMarker2を基にした関数\r
  60          * この関数はNyARSquare要素のうち、directionを除くパラメータを取得して返します。\r
  61          * directionの確定は行いません。\r
  62          * @param i_raster\r
  63          * 解析する２値ラスタイメージを指定します。\r
  64          * @param o_square_stack\r
  65          * 抽出した正方形候補を格納するリスト\r
  66          * @throws NyARException\r
  67          */\r
  68         public final void detectMarker(NyARBinRaster i_raster, NyARSquareStack o_square_stack) throws NyARException\r
  69         {\r
  70                 final NyARLabelingImage limage = this._limage;\r
  71 \r
  72                 // 初期化\r
  73 \r
  74                 // マーカーホルダをリセット\r
  75                 o_square_stack.clear();\r
  76 \r
  77                 // ラベル数が0ならここまで(Labeling内部でソートするようにした。)\r
  78                 final int label_num = this._labeling.labeling(i_raster,this._limage);\r
  79                 if (label_num < 1) {\r
  80                         return;\r
  81                 }\r
  82 \r
  83                 final NyARLabelingLabelStack stack = limage.getLabelStack();\r
  84                 final NyARLabelingLabel[] labels = stack.getArray();\r
  85 \r
  86                 // デカいラベルを読み飛ばし\r
  87                 int i;\r
  88                 for (i = 0; i < label_num; i++) {\r
  89                         // 検査対象内のラベルサイズになるまで無視\r
  90                         if (labels[i].area <= AR_AREA_MAX) {\r
  91                                 break;\r
  92                         }\r
  93                 }\r
  94 \r
  95                 final int xsize = this._width;\r
  96                 final int ysize = this._height;\r
  97                 final int[] xcoord = this._xcoord;\r
  98                 final int[] ycoord = this._ycoord;\r
  99                 final int coord_max = this._max_coord;\r
 100                 final LabelOverlapChecker<NyARLabelingLabel> overlap = this._overlap_checker;\r
 101 \r
 102                 //重なりチェッカの最大数を設定\r
 103                 overlap.setMaxLabels(label_num);\r
 104 \r
 105                 for (; i < label_num; i++) {\r
 106                         final NyARLabelingLabel label_pt = labels[i];\r
 107                         final int label_area = label_pt.area;\r
 108                         // 検査対象サイズよりも小さくなったら終了\r
 109                         if (label_area < AR_AREA_MIN) {\r
 110                                 break;\r
 111                         }\r
 112                         // クリップ領域が画面の枠に接していれば除外\r
 113                         if (label_pt.clip_l == 1 || label_pt.clip_r == xsize - 2) {// if(wclip[i*4+0] == 1 || wclip[i*4+1] ==xsize-2){\r
 114                                 continue;\r
 115                         }\r
 116                         if (label_pt.clip_t == 1 || label_pt.clip_b == ysize - 2) {// if( wclip[i*4+2] == 1 || wclip[i*4+3] ==ysize-2){\r
 117                                 continue;\r
 118                         }\r
 119                         // 既に検出された矩形との重なりを確認\r
 120                         if (!overlap.check(label_pt)) {\r
 121                                 // 重なっているようだ。\r
 122                                 continue;\r
 123                         }\r
 124                         // 輪郭を取得\r
 125                         final int coord_num = _cpickup.getContour(limage,limage.getTopClipTangentX(label_pt),label_pt.clip_t, coord_max, xcoord, ycoord);\r
 126                         if (coord_num == coord_max) {\r
 127                                 // 輪郭が大きすぎる。\r
 128                                 continue;\r
 129                         }\r
 130                         //輪郭分析用に正規化する。\r
 131                         final int vertex1 = SquareContourDetector.normalizeCoord(xcoord, ycoord, coord_num);\r
 132 \r
 133                         //ここから先が輪郭分析\r
 134                         NyARSquare square_ptr = o_square_stack.prePush();\r
 135                         if(!this._sqconvertor.coordToSquare(xcoord,ycoord,vertex1,coord_num,label_area,square_ptr)){\r
 136                                 o_square_stack.pop();// 頂点の取得が出来なかったので破棄\r
 137                                 continue;                               \r
 138                         }\r
 139                         // 検出済の矩形の属したラベルを重なりチェックに追加する。\r
 140                         overlap.push(label_pt);\r
 141                 }\r
 142                 return;\r
 143         }\r
 144 \r
 145 }\r
 146 \r
 147 \r
 148 \r