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                 this._labeling = new NyARLabeling_ARToolKit();\r
  43                 this._sqconvertor=new SquareContourDetector(i_size,i_dist_factor_ref);\r
  44                 this._limage = new NyARLabelingImage(this._width, this._height);\r
  45 \r
  46                 // 輪郭の最大長は画面に映りうる最大の長方形サイズ。\r
  47                 int number_of_coord = (this._width + this._height) * 2;\r
  48 \r
  49                 // 輪郭バッファは頂点変換をするので、輪郭バッファの２倍取る。\r
  50                 this._max_coord = number_of_coord;\r
  51                 this._xcoord = new int[number_of_coord * 2];\r
  52                 this._ycoord = new int[number_of_coord * 2];\r
  53                 return;\r
  54         }\r
  55 \r
  56         /**\r
  57          * arDetectMarker2を基にした関数\r
  58          * この関数はNyARSquare要素のうち、directionを除くパラメータを取得して返します。\r
  59          * directionの確定は行いません。\r
  60          * @param i_raster\r
  61          * 解析する２値ラスタイメージを指定します。\r
  62          * @param o_square_stack\r
  63          * 抽出した正方形候補を格納するリスト\r
  64          * @throws NyARException\r
  65          */\r
  66         public final void detectMarker(NyARBinRaster i_raster, NyARSquareStack o_square_stack) throws NyARException\r
  67         {\r
  68                 final NyARLabelingImage limage = this._limage;\r
  69 \r
  70                 // 初期化\r
  71 \r
  72                 // マーカーホルダをリセット\r
  73                 o_square_stack.clear();\r
  74 \r
  75                 // ラベル数が0ならここまで(Labeling内部でソートするようにした。)\r
  76                 final int label_num = this._labeling.labeling(i_raster,this._limage);\r
  77                 if (label_num < 1) {\r
  78                         return;\r
  79                 }\r
  80 \r
  81                 final NyARLabelingLabelStack stack = limage.getLabelStack();\r
  82                 final NyARLabelingLabel[] labels = stack.getArray();\r
  83 \r
  84                 // デカいラベルを読み飛ばし\r
  85                 int i;\r
  86                 for (i = 0; i < label_num; i++) {\r
  87                         // 検査対象内のラベルサイズになるまで無視\r
  88                         if (labels[i].area <= AR_AREA_MAX) {\r
  89                                 break;\r
  90                         }\r
  91                 }\r
  92 \r
  93                 final int xsize = this._width;\r
  94                 final int ysize = this._height;\r
  95                 final int[] xcoord = this._xcoord;\r
  96                 final int[] ycoord = this._ycoord;\r
  97                 final int coord_max = this._max_coord;\r
  98                 final LabelOverlapChecker<NyARLabelingLabel> overlap = this._overlap_checker;\r
  99 \r
 100                 //重なりチェッカの最大数を設定\r
 101                 overlap.setMaxLabels(label_num);\r
 102 \r
 103                 for (; i < label_num; i++) {\r
 104                         final NyARLabelingLabel label_pt = labels[i];\r
 105                         final int label_area = label_pt.area;\r
 106                         // 検査対象サイズよりも小さくなったら終了\r
 107                         if (label_area < AR_AREA_MIN) {\r
 108                                 break;\r
 109                         }\r
 110                         // クリップ領域が画面の枠に接していれば除外\r
 111                         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
 112                                 continue;\r
 113                         }\r
 114                         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
 115                                 continue;\r
 116                         }\r
 117                         // 既に検出された矩形との重なりを確認\r
 118                         if (!overlap.check(label_pt)) {\r
 119                                 // 重なっているようだ。\r
 120                                 continue;\r
 121                         }\r
 122                         // 輪郭を取得\r
 123                         final int coord_num = _cpickup.getContour(limage,limage.getTopClipTangentX(label_pt),label_pt.clip_t, coord_max, xcoord, ycoord);\r
 124                         if (coord_num == coord_max) {\r
 125                                 // 輪郭が大きすぎる。\r
 126                                 continue;\r
 127                         }\r
 128                         //輪郭分析用に正規化する。\r
 129                         final int vertex1 = SquareContourDetector.normalizeCoord(xcoord, ycoord, coord_num);\r
 130 \r
 131                         //ここから先が輪郭分析\r
 132                         NyARSquare square_ptr = o_square_stack.prePush();\r
 133                         if(!this._sqconvertor.coordToSquare(xcoord,ycoord,vertex1,coord_num,label_area,square_ptr)){\r
 134                                 o_square_stack.pop();// 頂点の取得が出来なかったので破棄\r
 135                                 continue;                               \r
 136                         }\r
 137                         // 検出済の矩形の属したラベルを重なりチェックに追加する。\r
 138                         overlap.push(label_pt);\r
 139                 }\r
 140                 return;\r
 141         }\r
 142 \r
 143 }\r
 144 \r
 145 \r
 146 \r