public static void vertical9(
      Kernel1D_I32 kernel, ImageSInt16 image, ImageInt16 dest, int divisor, boolean includeBorder) {
    final short[] dataSrc = image.data;
    final short[] dataDst = dest.data;

    final int k1 = kernel.data[0];
    final int k2 = kernel.data[1];
    final int k3 = kernel.data[2];
    final int k4 = kernel.data[3];
    final int k5 = kernel.data[4];
    final int k6 = kernel.data[5];
    final int k7 = kernel.data[6];
    final int k8 = kernel.data[7];
    final int k9 = kernel.data[8];

    final int radius = kernel.getRadius();

    final int imgWidth = dest.getWidth();
    final int imgHeight = dest.getHeight();
    final int halfDivisor = divisor / 2;

    final int yEnd = imgHeight - radius;

    final int xBorder = includeBorder ? 0 : radius;

    for (int y = radius; y < yEnd; y++) {
      int indexDst = dest.startIndex + y * dest.stride + xBorder;
      int i = image.startIndex + (y - radius) * image.stride;
      final int iEnd = i + imgWidth - xBorder;

      for (i += xBorder; i < iEnd; i++) {
        int indexSrc = i;

        int total = (dataSrc[indexSrc]) * k1;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k2;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k3;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k4;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k5;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k6;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k7;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k8;
        indexSrc += image.stride;
        total += (dataSrc[indexSrc]) * k9;

        dataDst[indexDst++] = (short) ((total + halfDivisor) / divisor);
      }
    }
  }
Exemplo n.º 2
0
  /** Sets a rectangle inside the image with the specified value. */
  public static void fillRectangle(
      ImageInt16 img, int value, int x0, int y0, int width, int height) {
    int x1 = x0 + width;
    int y1 = y0 + height;

    for (int y = y0; y < y1; y++) {
      for (int x = x0; x < x1; x++) {
        if (img.isInBounds(x, y)) img.set(x, y, value);
      }
    }
  }
Exemplo n.º 3
0
  /**
   * Fills the whole image with the specified pixel value
   *
   * @param img An image.
   * @param value The value that the image is being filled with.
   */
  public static void fill(ImageInt16 img, int value) {
    final int h = img.getHeight();
    final int w = img.getWidth();

    short[] data = img.data;

    for (int y = 0; y < h; y++) {
      int index = img.getStartIndex() + y * img.getStride();
      for (int x = 0; x < w; x++) {
        data[index++] = (short) value;
      }
    }
  }
Exemplo n.º 4
0
  /**
   * Computes the mean squared error (MSE) between the two images.
   *
   * @param imgA first image. Not modified.
   * @param imgB second image. Not modified.
   * @return error between the two images.
   */
  public static double computeMeanSquaredError(ImageInt16 imgA, ImageInt16 imgB) {
    final int h = imgA.getHeight();
    final int w = imgA.getWidth();

    double total = 0;

    for (int y = 0; y < h; y++) {
      for (int x = 0; x < w; x++) {
        double difference = imgA.get(x, y) - imgB.get(x, y);
        total += difference * difference;
      }
    }

    return total / (w * h);
  }
Exemplo n.º 5
0
  /**
   * Sets each value in the image to a value drawn from an uniform distribution that has a range of
   * min <= X < max.
   */
  public static void randomize(ImageInt16 img, Random rand, int min, int max) {
    final int h = img.getHeight();
    final int w = img.getWidth();

    int range = max - min;

    short[] data = img.data;

    for (int y = 0; y < h; y++) {
      int index = img.getStartIndex() + y * img.getStride();
      for (int x = 0; x < w; x++) {
        data[index++] = (short) (rand.nextInt(range) + min);
      }
    }
  }
Exemplo n.º 6
0
  /** Flips the image from top to bottom */
  public static void flipVertical(ImageInt16 img) {
    int h2 = img.height / 2;

    for (int y = 0; y < h2; y++) {
      int index1 = img.getStartIndex() + y * img.getStride();
      int index2 = img.getStartIndex() + (img.height - y - 1) * img.getStride();

      int end = index1 + img.width;

      while (index1 < end) {
        int tmp = img.data[index1];
        img.data[index1++] = img.data[index2];
        img.data[index2++] = (short) tmp;
      }
    }
  }
Exemplo n.º 7
0
  /**
   * Checks to see if the BufferedImage has the same intensity values as the ImageUInt8
   *
   * @param imgA BufferedImage
   * @param imgB ImageUInt8
   */
  public static void checkEquals(BufferedImage imgA, ImageInt16 imgB) {

    if (imgA.getRaster() instanceof ByteInterleavedRaster
        && imgA.getType() != BufferedImage.TYPE_BYTE_INDEXED) {
      ByteInterleavedRaster raster = (ByteInterleavedRaster) imgA.getRaster();

      if (raster.getNumBands() == 1) {
        int strideA = raster.getScanlineStride();
        int offsetA = raster.getDataOffset(0) - raster.getNumBands() + 1;

        // handle a special case where the RGB conversion is screwed
        for (int i = 0; i < imgA.getHeight(); i++) {
          for (int j = 0; j < imgA.getWidth(); j++) {
            int valB = imgB.get(j, i);
            int valA = raster.getDataStorage()[offsetA + i * strideA + j];
            if (!imgB.getTypeInfo().isSigned()) valA &= 0xFFFF;

            if (valA != valB)
              throw new RuntimeException("Images are not equal: " + valA + " " + valB);
          }
        }
        return;
      }
    } else if (imgA.getRaster() instanceof ShortInterleavedRaster) {
      ShortInterleavedRaster raster = (ShortInterleavedRaster) imgA.getRaster();

      if (raster.getNumBands() == 1) {
        int strideA = raster.getScanlineStride();
        int offsetA = raster.getDataOffset(0) - raster.getNumBands() + 1;

        // handle a special case where the RGB conversion is screwed
        for (int i = 0; i < imgA.getHeight(); i++) {
          for (int j = 0; j < imgA.getWidth(); j++) {
            int valB = imgB.get(j, i);
            int valA = raster.getDataStorage()[offsetA + i * strideA + j];
            if (!imgB.getTypeInfo().isSigned()) valA &= 0xFFFF;

            if (valA != valB)
              throw new RuntimeException("Images are not equal: " + valA + " " + valB);
          }
        }
      }
    } else {
      for (int y = 0; y < imgA.getHeight(); y++) {
        for (int x = 0; x < imgA.getWidth(); x++) {
          int rgb = imgA.getRGB(x, y);

          int gray = ((((rgb >>> 16) & 0xFF) + ((rgb >>> 8) & 0xFF) + (rgb & 0xFF)) / 3);
          int grayB = imgB.get(x, y);
          if (!imgB.getTypeInfo().isSigned()) gray &= 0xFFFF;

          if (Math.abs(gray - grayB) != 0) {
            throw new RuntimeException(
                "images are not equal: (" + x + " , " + y + ") A = " + gray + " B = " + grayB);
          }
        }
      }
    }
  }