src/jp/nyatla/utils/j2se/LabelingBufferdImage.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.utils.j2se;\r
  33 \r
  34 import java.awt.Graphics;\r
  35 import java.awt.image.*;\r
  36 import java.awt.color.*;\r
  37 import java.awt.*;\r
  38 \r
  39 \r
  40 import jp.nyatla.nyartoolkit.NyARException;\r
  41 import jp.nyatla.nyartoolkit.core.types.*;\r
  42 import jp.nyatla.nyartoolkit.core.rasterreader.*;\r
  43 import jp.nyatla.nyartoolkit.core.raster.*;\r
  44 import jp.nyatla.nyartoolkit.core.types.stack.*;\r
  45 import jp.nyatla.nyartoolkit.core.labeling.*;\r
  46 import jp.nyatla.nyartoolkit.core.labeling.artoolkit.NyARLabelingImage;\r
  47 import jp.nyatla.nyartoolkit.core.labeling.artoolkit.NyARLabelingLabel;\r
  48 \r
  49 /**\r
  50  * bitmapとして利用可能なラベリングイメージです。\r
  51  * \r
  52  * \r
  53  */\r
  54 public class LabelingBufferdImage extends BufferedImage\r
  55 {\r
  56         public final static int COLOR_125_COLOR = 0;// 125色ラベルモード\r
  57 \r
  58         public final static int COLOR_256_MONO = 1;// 64階調モノクロモード\r
  59 \r
  60         public final static int COLOR_64_MONO = 2;// 64階調モノクロモード\r
  61 \r
  62         public final static int COLOR_32_MONO = 3;// 32階調モノクロモード\r
  63 \r
  64         public final static int COLOR_16_MONO = 4;// 16階調モノクロモード\r
  65 \r
  66         public final static int COLOR_8_MONO = 5;// 16階調モノクロモード\r
  67 \r
  68         private int[] _rgb_table_125;\r
  69 \r
  70 \r
  71         /**\r
  72          * i_width x i_heightの大きさのイメージを作成します。\r
  73          * \r
  74          * @param i_width\r
  75          * @param i_height\r
  76          */\r
  77         public LabelingBufferdImage(int i_width, int i_height)\r
  78         {\r
  79                 super(i_width, i_height, ColorSpace.TYPE_RGB);\r
  80                 // RGBテーブルを作成\r
  81                 this._rgb_table_125 = new int[125];\r
  82                 for (int i = 0; i < 5; i++) {\r
  83                         for (int i2 = 0; i2 < 5; i2++) {\r
  84                                 for (int i3 = 0; i3 < 5; i3++) {\r
  85                                         this._rgb_table_125[((i * 5) + i2) * 5 + i3] = ((((i * 63) << 8) | (i2 * 63)) << 8) | (i3 * 63);\r
  86                                 }\r
  87                         }\r
  88                 }\r
  89 /*              case COLOR_256_MONO:\r
  90                         this._rgb_table = new int[256];\r
  91                         this._number_of_color = 256;\r
  92                         for (int i = 0; i < 256; i++) {\r
  93                                 this._rgb_table[i] = (i << 16) | (i << 8) | i;\r
  94                         }\r
  95                         break;\r
  96                 case COLOR_64_MONO:\r
  97                         this._rgb_table = new int[64];\r
  98                         this._number_of_color = 64;\r
  99                         for (int i = 0; i < 64; i++) {\r
 100                                 int m = (i * 4);\r
 101                                 this._rgb_table[i] = (m << 16) | (m << 8) | m;\r
 102                         }\r
 103                         break;\r
 104                 case COLOR_32_MONO:\r
 105                         this._rgb_table = new int[32];\r
 106                         this._number_of_color = 32;\r
 107                         for (int i = 0; i < 32; i++) {\r
 108                                 int m = (i * 8);\r
 109                                 this._rgb_table[i] = (m << 16) | (m << 8) | m;\r
 110                         }\r
 111                         break;\r
 112                 case COLOR_16_MONO:\r
 113                         this._rgb_table = new int[32];\r
 114                         this._number_of_color = 16;\r
 115                         for (int i = 0; i < 16; i++) {\r
 116                                 int m = (i * 8);\r
 117                                 this._rgb_table[i] = (m << 16) | (m << 8) | m;\r
 118                         }\r
 119                         break;\r
 120                 }*/\r
 121         }\r
 122 \r
 123 \r
 124 \r
 125         public void drawImage(NyARGrayscaleRaster i_raster) throws NyARException\r
 126         {\r
 127                 assert (i_raster.getBufferReader().getBufferType() == INyARBufferReader.BUFFERFORMAT_INT1D_GRAY_8);\r
 128 \r
 129                 int w = this.getWidth();\r
 130                 int h = this.getHeight();\r
 131                 // サイズをチェック\r
 132                 NyARIntSize size = i_raster.getSize();\r
 133                 if (size.h > h || size.w > w) {\r
 134                         throw new NyARException();\r
 135                 }\r
 136 \r
 137                 int[] limg;\r
 138                 // イメージの描画\r
 139                 limg = (int[]) i_raster.getBufferReader().getBuffer();\r
 140                 for (int i = 0; i < h; i++) {\r
 141                         for (int i2 = 0; i2 < w; i2++) {\r
 142                                 this.setRGB(i2, i,limg[i*w+i2]);\r
 143                         }\r
 144                 }\r
 145                 return;\r
 146         }\r
 147         /**\r
 148          * バイナリラスタ\r
 149          * @param i_raster\r
 150          * @throws NyARException\r
 151          */\r
 152         public void drawImage(NyARBinRaster i_raster) throws NyARException\r
 153         {\r
 154                 assert (i_raster.getBufferReader().getBufferType() == INyARBufferReader.BUFFERFORMAT_INT1D_BIN_8);\r
 155 \r
 156                 int w = this.getWidth();\r
 157                 int h = this.getHeight();\r
 158                 // サイズをチェック\r
 159                 NyARIntSize size = i_raster.getSize();\r
 160                 if (size.h > h || size.w > w) {\r
 161                         throw new NyARException();\r
 162                 }\r
 163 \r
 164                 int[] limg;\r
 165                 // イメージの描画\r
 166                 limg = (int[]) i_raster.getBufferReader().getBuffer();\r
 167                 for (int i = 0; i < h; i++) {\r
 168                         for (int i2 = 0; i2 < w; i2++) {\r
 169                                 this.setRGB(i2, i, limg[i*w+i2] > 0 ? 255 : 0);\r
 170                         }\r
 171                 }\r
 172                 return;\r
 173         }\r
 174                 \r
 175         /**\r
 176          * ラベリングイメージを書く\r
 177          * @param i_raster\r
 178          * @throws NyARException\r
 179          */\r
 180         public void drawLabel(NyARLabelingImage i_image) throws NyARException\r
 181         {\r
 182                 int w = this.getWidth();\r
 183                 int h = this.getHeight();\r
 184                 // サイズをチェック\r
 185                 NyARIntSize size = i_image.getSize();\r
 186                 if (size.h > h || size.w > w) {\r
 187                         throw new NyARException();\r
 188                 }\r
 189                 int[] index_array=i_image.getIndexArray();\r
 190 \r
 191                 int[] limg;\r
 192                 // イメージの描画\r
 193                 limg = (int[]) i_image.getBufferReader().getBuffer();\r
 194                 for (int i = 0; i < h; i++) {\r
 195                         for (int i2 = 0; i2 < w; i2++) {\r
 196                                 int t=limg[i*w+i2]-1;\r
 197                                 if(t<0){\r
 198                                         t=0;\r
 199                                 }else{\r
 200                                         t=index_array[t];\r
 201                                 }\r
 202                                 this.setRGB(i2, i,_rgb_table_125[t% _rgb_table_125.length]);\r
 203                         }\r
 204                 }\r
 205                 return;\r
 206         }\r
 207         /**\r
 208          * \r
 209          * @param i_stack\r
 210          */\r
 211 \r
 212         public void overlayData(NyARLabelingLabel i_label)\r
 213         {\r
 214                 Graphics g = this.getGraphics();\r
 215                 g.setColor(Color.red);\r
 216                 g.drawRect(i_label.clip_l,i_label.clip_t,i_label.clip_r-i_label.clip_l,i_label.clip_b-i_label.clip_t);\r
 217                 return;\r
 218         }       \r
 219         /**\r
 220          * \r
 221          * @param i_stack\r
 222          */\r
 223 \r
 224         public void overlayData(NyARIntPointStack i_stack)\r
 225         {\r
 226                 int count = i_stack.getLength();\r
 227                 NyARIntPoint2d[] items = i_stack.getArray();\r
 228                 Graphics g = this.getGraphics();\r
 229                 for (int i = 0; i < count; i++) {\r
 230                         int x = items[i].x;\r
 231                         int y = items[i].y;\r
 232                         g.setColor(Color.red);\r
 233                         g.drawLine(x - 5, y, x + 5, y);\r
 234                         g.drawLine(x, y + 5, x, y - 5);\r
 235                 }\r
 236                 return;\r
 237         }\r
 238         public void overlayData(NyARIntRectStack i_stack)\r
 239         {\r
 240                 Color[] c=new Color[]{Color.cyan,Color.red,Color.green};\r
 241                 int count = i_stack.getLength();\r
 242                 NyARIntRect[] items = i_stack.getArray();\r
 243                 Graphics g = this.getGraphics();\r
 244                 for (int i = 0; i < count; i++) {\r
 245                         int x = items[i].x;\r
 246                         int y = items[i].y;\r
 247                         g.setColor(c[i%1]);\r
 248                         g.drawRect(x,y,items[i].w,items[i].h);\r
 249                 }\r
 250                 return;\r
 251         }\r
 252 }\r