-/* このソースは実験用のソースです。\r
- * 動いたり動かなかったりします。\r
+/* \r
+ * PROJECT: NyARToolkit(Extension)\r
+ * --------------------------------------------------------------------------------\r
+ * The NyARToolkit is Java edition ARToolKit class library.\r
+ * Copyright (C)2008-2009 Ryo Iizuka\r
+ *\r
+ * This program is free software: you can redistribute it and/or modify\r
+ * it under the terms of the GNU General Public License as published by\r
+ * the Free Software Foundation, either version 3 of the License, or\r
+ * (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 program. If not, see <http://www.gnu.org/licenses/>.\r
+ * \r
+ * For further information please contact.\r
+ * http://nyatla.jp/nyatoolkit/\r
+ * <airmail(at)ebony.plala.or.jp> or <nyatla(at)nyatla.jp>\r
* \r
*/\r
package jp.nyatla.nyartoolkit.core.labeling.rlelabeling;\r
\r
import jp.nyatla.nyartoolkit.NyARException;\r
import jp.nyatla.nyartoolkit.core.raster.*;\r
+import jp.nyatla.nyartoolkit.core.types.stack.NyARObjectStack;\r
\r
+class RleInfoStack extends NyARObjectStack<RleInfoStack.RleInfo>\r
+{\r
+ public class RleInfo\r
+ {\r
+ //継承メンバ\r
+ public int entry_x; // フラグメントラベルの位置\r
+ public int area;\r
+ public int clip_r;\r
+ public int clip_l;\r
+ public int clip_b;\r
+ public int clip_t;\r
+ public long pos_x;\r
+ public long pos_y; \r
+ } \r
+ public RleInfoStack(int i_length) throws NyARException\r
+ {\r
+ super(i_length, RleInfoStack.RleInfo.class);\r
+ return;\r
+ }\r
\r
-\r
+ protected RleInfoStack.RleInfo createElement()\r
+ {\r
+ return new RleInfoStack.RleInfo();\r
+ }\r
+}\r
/**\r
* [strage class]\r
*/\r
}\r
}\r
\r
+\r
// RleImageをラベリングする。\r
public class NyARLabeling_Rle\r
{\r
+ private static final int AR_AREA_MAX = 100000;// #define AR_AREA_MAX 100000\r
+ private static final int AR_AREA_MIN = 70;// #define AR_AREA_MIN 70\r
+ \r
+ private RleInfoStack _rlestack;\r
private RleElement[] _rle1;\r
-\r
private RleElement[] _rle2;\r
+ private int _max_area;\r
+ private int _min_area;\r
\r
- public NyARLabeling_Rle(int i_width)\r
+ public NyARLabeling_Rle(int i_width,int i_height) throws NyARException\r
{\r
+ this._rlestack=new RleInfoStack(i_width*i_height*2048/(320*240)+32);\r
this._rle1 = RleElement.createArray(i_width/2+1);\r
this._rle2 = RleElement.createArray(i_width/2+1);\r
+ setAreaRange(AR_AREA_MAX,AR_AREA_MIN);\r
+\r
+ return;\r
+ }\r
+ /**\r
+ * 対象サイズ\r
+ * @param i_max\r
+ * @param i_min\r
+ */\r
+ public void setAreaRange(int i_max,int i_min)\r
+ {\r
+ this._max_area=i_max;\r
+ this._min_area=i_min;\r
return;\r
}\r
\r
/**\r
- * i_bin_bufのbinイメージをREL圧縮する。\r
- * \r
- * @param i_bin_raster\r
+ * i_bin_bufのgsイメージをREL圧縮する。\r
+ * @param i_bin_buf\r
+ * @param i_st\r
+ * @param i_len\r
+ * @param i_out\r
+ * @param i_th\r
+ * BINラスタのときは0,GSラスタの時は閾値を指定する。\r
+ * この関数は、閾値を暗点と認識します。\r
+ * 暗点<=th<明点\r
+ * @return\r
*/\r
- private int toRel(int[] i_bin_buf, int i_st, int i_len, RleElement[] i_out)\r
+ private int toRel(int[] i_bin_buf, int i_st, int i_len, RleElement[] i_out,int i_th)\r
{\r
int current = 0;\r
int r = -1;\r
final int right_edge = i_st + i_len - 1;\r
while (x < right_edge) {\r
// 暗点(0)スキャン\r
- if (i_bin_buf[x] != 0) {\r
- x++;\r
+ if (i_bin_buf[x] > i_th) {\r
+ x++;//明点\r
continue;\r
}\r
// 暗点発見→暗点長を調べる\r
r++;// 暗点+1\r
x++;\r
while (x < right_edge) {\r
- if (i_bin_buf[x] != 0) {\r
+ if (i_bin_buf[x] > i_th) {\r
// 明点(1)→暗点(0)配列終了>登録\r
i_out[current].r = r;\r
current++;\r
}\r
}\r
// 最後の1点だけ判定方法が少し違うの。\r
- if (i_bin_buf[x] != 0) {\r
+ if (i_bin_buf[x] > i_th) {\r
// 明点→rカウント中なら暗点配列終了>登録\r
if (r >= 0) {\r
i_out[current].r = r;\r
return current;\r
}\r
\r
- private void addFragment(RleElement i_rel_img, int i_nof, int i_row_index, int i_rel_index,RleLabelFragmentInfoStack o_stack) throws NyARException\r
+ private void addFragment(RleElement i_rel_img, int i_nof, int i_row_index,RleInfoStack o_stack) throws NyARException\r
{\r
int l=i_rel_img.l;\r
final int len=i_rel_img.r - l;\r
i_rel_img.fid = i_nof;// REL毎の固有ID\r
- RleLabelFragmentInfoStack.RleLabelFragmentInfo v = o_stack.prePush();\r
+ RleInfoStack.RleInfo v = o_stack.prePush();\r
v.entry_x = l;\r
v.area =len;\r
v.clip_l=l;\r
v.clip_r=i_rel_img.r-1;\r
v.clip_t=i_row_index;\r
v.clip_b=i_row_index;\r
- v.pos_x+=(len*(2*l+(len-1)))/2;\r
- v.pos_y+=i_row_index*len;\r
+ v.pos_x=(len*(2*l+(len-1)))/2;\r
+ v.pos_y=i_row_index*len;\r
\r
return;\r
}\r
-\r
- //\r
- public int labeling(NyARBinRaster i_bin_raster, int i_top, int i_bottom,RleLabelFragmentInfoStack o_stack) throws NyARException\r
+ //所望のラスタからBIN-RLEに変換しながらの低速系も準備しようかな\r
+ \r
+ /**\r
+ * 単一閾値を使ってGSラスタをBINラスタに変換しながらラベリングします。\r
+ * @param i_gs_raster\r
+ * @param i_top\r
+ * @param i_bottom\r
+ * @param o_stack\r
+ * @return\r
+ * @throws NyARException\r
+ */\r
+ public int labeling(NyARBinRaster i_bin_raster, int i_top, int i_bottom,NyARRleLabelFragmentInfoStack o_stack) throws NyARException\r
+ {\r
+ assert(i_bin_raster.isEqualBufferType(NyARBufferType.INT1D_BIN_8));\r
+ return this.imple_labeling(i_bin_raster,0,i_top,i_bottom,o_stack);\r
+ }\r
+ /**\r
+ * BINラスタをラベリングします。\r
+ * @param i_gs_raster\r
+ * @param i_th\r
+ * 画像を2値化するための閾値。暗点<=th<明点となります。\r
+ * @param i_top\r
+ * @param i_bottom\r
+ * @param o_stack\r
+ * @return\r
+ * @throws NyARException\r
+ */\r
+ public int labeling(NyARGrayscaleRaster i_gs_raster,int i_th, int i_top, int i_bottom,NyARRleLabelFragmentInfoStack o_stack) throws NyARException\r
+ {\r
+ assert(i_gs_raster.isEqualBufferType(NyARBufferType.INT1D_GRAY_8));\r
+ return this.imple_labeling(i_gs_raster,i_th,i_top,i_bottom,o_stack);\r
+ }\r
+ private int imple_labeling(INyARRaster i_raster,int i_th,int i_top, int i_bottom,NyARRleLabelFragmentInfoStack o_stack) throws NyARException\r
{\r
// リセット処理\r
- o_stack.clear();\r
+ final RleInfoStack rlestack=this._rlestack;\r
+ rlestack.clear();\r
+\r
//\r
RleElement[] rle_prev = this._rle1;\r
RleElement[] rle_current = this._rle2;\r
int len_prev = 0;\r
int len_current = 0;\r
- final int width = i_bin_raster.getWidth();\r
- int[] in_buf = (int[]) i_bin_raster.getBufferReader().getBuffer();\r
+ final int width = i_raster.getWidth();\r
+ int[] in_buf = (int[]) i_raster.getBuffer();\r
\r
int id_max = 0;\r
int label_count=0;\r
// 初段登録\r
\r
- len_prev = toRel(in_buf, i_top, width, rle_prev);\r
+ len_prev = toRel(in_buf, i_top, width, rle_prev,i_th);\r
for (int i = 0; i < len_prev; i++) {\r
// フラグメントID=フラグメント初期値、POS=Y値、RELインデクス=行\r
- addFragment(rle_prev[i], id_max, i_top, i,o_stack);\r
+ addFragment(rle_prev[i], id_max, i_top,rlestack);\r
id_max++;\r
// nofの最大値チェック\r
label_count++;\r
}\r
- RleLabelFragmentInfoStack.RleLabelFragmentInfo[] f_array = o_stack.getArray();\r
+ RleInfoStack.RleInfo[] f_array = rlestack.getArray();\r
// 次段結合\r
for (int y = i_top + 1; y < i_bottom; y++) {\r
// カレント行の読込\r
- len_current = toRel(in_buf, y * width, width, rle_current);\r
+ len_current = toRel(in_buf, y * width, width, rle_current,i_th);\r
int index_prev = 0;\r
\r
SCAN_CUR: for (int i = 0; i < len_current; i++) {\r
continue;\r
} else if (rle_prev[index_prev].l - rle_current[i].r > 0) {// 0なら8方位ラベリングになる\r
// prevがcur右方にある→独立フラグメント\r
- addFragment(rle_current[i], id_max, y, i,o_stack);\r
+ addFragment(rle_current[i], id_max, y,rlestack);\r
id_max++;\r
label_count++;\r
// 次のindexをしらべる\r
continue SCAN_CUR;\r
}\r
id=rle_prev[index_prev].fid;//ルートフラグメントid\r
- RleLabelFragmentInfoStack.RleLabelFragmentInfo id_ptr = f_array[id];\r
+ RleInfoStack.RleInfo id_ptr = f_array[id];\r
//結合対象(初回)->prevのIDをコピーして、ルートフラグメントの情報を更新\r
rle_current[i].fid = id;//フラグメントIDを保存\r
//\r
\r
//結合するルートフラグメントを取得\r
final int prev_id =rle_prev[index_prev].fid;\r
- RleLabelFragmentInfoStack.RleLabelFragmentInfo prev_ptr = f_array[prev_id];\r
+ RleInfoStack.RleInfo prev_ptr = f_array[prev_id];\r
if (id != prev_id){\r
- if(prev_ptr.area==0){\r
- System.out.println("ERRRR");\r
- }\r
label_count--;\r
//prevとcurrentのフラグメントidを書き換える。\r
for(int i2=index_prev;i2<len_prev;i2++){\r
// curにidが割り当てられたかを確認\r
// 右端独立フラグメントを追加\r
if (id < 0){\r
- addFragment(rle_current[i], id_max, y, i,o_stack);\r
+ addFragment(rle_current[i], id_max, y,rlestack);\r
id_max++;\r
label_count++;\r
}\r
len_prev = len_current;\r
rle_current = tmp;\r
}\r
- //ソートする。\r
- o_stack.sortByArea();\r
- //ラベル数を再設定\r
- o_stack.reserv(label_count);\r
- //posを計算\r
- for(int i=0;i<label_count;i++){\r
- final RleLabelFragmentInfoStack.RleLabelFragmentInfo tmp=f_array[i];\r
- tmp.pos_x/=tmp.area;\r
- tmp.pos_y/=tmp.area;\r
+ //対象のラベルだけ転写\r
+ o_stack.init(label_count);\r
+ NyARRleLabelFragmentInfoStack.RleLabelFragmentInfo[] o_dest_array=o_stack.getArray();\r
+ final int max=this._max_area;\r
+ final int min=this._min_area;\r
+ int active_labels=0;\r
+ for(int i=id_max-1;i>=0;i--){\r
+ final int area=f_array[i].area;\r
+ if(area<min || area>max){//対象外のエリア0のもminではじく\r
+ continue;\r
+ }\r
+ //\r
+ final RleInfoStack.RleInfo src_info=f_array[i];\r
+ final NyARRleLabelFragmentInfoStack.RleLabelFragmentInfo dest_info=o_dest_array[active_labels];\r
+ dest_info.area=area;\r
+ dest_info.clip_b=src_info.clip_b;\r
+ dest_info.clip_r=src_info.clip_r;\r
+ dest_info.clip_t=src_info.clip_t;\r
+ dest_info.clip_l=src_info.clip_l;\r
+ dest_info.entry_x=src_info.entry_x;\r
+ dest_info.pos_x=src_info.pos_x/src_info.area;\r
+ dest_info.pos_y=src_info.pos_y/src_info.area;\r
+ active_labels++;\r
}\r
+ //ラベル数を再設定\r
+ o_stack.pops(label_count-active_labels);\r
//ラベル数を返却\r
- return label_count;\r
+ return active_labels;\r
} \r
}\r
\r