android/src/com/google/zxing/client/android/YUVLuminanceSource.java

   1 /*
   2  * Copyright 2009 ZXing authors
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package com.google.zxing.client.android;
  18
  19 import com.google.zxing.LuminanceSource;
  20
  21 import android.graphics.Bitmap;
  22
  23 /**
  24  * This object extends LuminanceSource around an array of YUV data returned from the camera driver,
  25  * with the option to crop to a rectangle within the full data. This can be used to exclude
  26  * superfluous pixels around the perimeter and speed up decoding.
  27  *
  28  * @author dswitkin@google.com (Daniel Switkin)
  29  */
  30 public final class YUVLuminanceSource extends LuminanceSource {
  31
  32   private final byte[] yuvData;
  33   private final int dataWidth;
  34   private final int dataHeight;
  35   private final int left;
  36   private final int top;
  37
  38   public YUVLuminanceSource(byte[] yuvData, int dataWidth, int dataHeight, int left, int top,
  39       int width, int height) {
  40     super(width, height);
  41
  42     if (left + width > dataWidth || top + height > dataHeight) {
  43       throw new IllegalArgumentException("Crop rectangle does not fit within image data.");
  44     }
  45
  46     this.yuvData = yuvData;
  47     this.dataWidth = dataWidth;
  48     this.dataHeight = dataHeight;
  49     this.left = left;
  50     this.top = top;
  51   }
  52
  53   public byte[] getRow(int y, byte[] row) {
  54     if (y < 0 || y >= getHeight()) {
  55       throw new IllegalArgumentException("Requested row is outside the image: " + y);
  56     }
  57     int width = getWidth();
  58     if (row == null || row.length < width) {
  59       row = new byte[width];
  60     }
  61     int offset = (y + top) * dataWidth + left;
  62     byte[] yuv = yuvData;
  63     for (int x = 0; x < width; x++) {
  64       row[x] = yuv[offset + x];
  65     }
  66     return row;
  67   }
  68
  69   public byte[] getMatrix() {
  70     int width = getWidth();
  71     int height = getHeight();
  72
  73     // If the caller asks for the entire underlying image, save the copy and give them the
  74     // original data. The docs specifically warn that result.length must be ignored.
  75     if (width == dataWidth && height == dataHeight) {
  76       return yuvData;
  77     }
  78
  79     int area = width * height;
  80     byte[] matrix = new byte[area];
  81     byte[] yuv = yuvData;
  82     int inputOffset = top * dataWidth + left;
  83     for (int y = 0; y < height; y++) {
  84       int outputOffset = y * width;
  85       for (int x = 0; x < width; x++) {
  86         // TODO: Compare performance with using System.arraycopy().
  87         matrix[outputOffset + x] = yuv[inputOffset + x];
  88       }
  89       inputOffset += dataWidth;
  90     }
  91     return matrix;
  92   }
  93
  94   public boolean isCropSupported() {
  95     return true;
  96   }
  97
  98   public LuminanceSource crop(int left, int top, int width, int height) {
  99     return new YUVLuminanceSource(yuvData, dataWidth, dataHeight, left, top, width, height);
 100   }
 101
 102   /**
 103    * Creates a greyscale Android Bitmap from the YUV data based on the crop rectangle.
 104    *
 105    * @return An 8888 bitmap.
 106    */
 107   public Bitmap renderToBitmap() {
 108     int width = getWidth();
 109     int height = getHeight();
 110     int[] pixels = new int[width * height];
 111     byte[] yuv = yuvData;
 112     int inputOffset = top * dataWidth + left;
 113
 114     for (int y = 0; y < height; y++) {
 115       int outputOffset = y * width;
 116       for (int x = 0; x < width; x++) {
 117         int grey = yuv[inputOffset + x] & 0xff;
 118         pixels[outputOffset + x] = (0xff << 24) | (grey << 16) | (grey << 8) | grey;
 119       }
 120       inputOffset += dataWidth;
 121     }
 122
 123     Bitmap bitmap = Bitmap.createBitmap(width, height, Bitmap.Config.ARGB_8888);
 124     bitmap.setPixels(pixels, 0, width, 0, 0, width, height);
 125     return bitmap;
 126   }
 127
 128 }