content: /*
   * Copyright (C) 2011 The Android Open Source Project
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  
  package android.filterfw.core;
  
  import android.filterfw.core.Frame;
  import android.filterfw.core.FrameFormat;
  import android.filterfw.core.FrameManager;
  import android.filterfw.core.GLFrame;
  import android.filterfw.core.NativeBuffer;
  import android.graphics.Bitmap;
  
  import java.nio.ByteBuffer;
  
  /**
   * @hide
   */
  public class NativeFrame extends Frame {
  
      private int nativeFrameId = -1;
  
      NativeFrame(FrameFormat format, FrameManager frameManager) {
          super(format, frameManager);
          int capacity = format.getSize();
          nativeAllocate(capacity);
          setReusable(capacity != 0);
      }
  
      @Override
      protected synchronized void releaseNativeAllocation() {
          nativeDeallocate();
          nativeFrameId = -1;
      }
  
      @Override
      protected synchronized boolean hasNativeAllocation() {
          return nativeFrameId != -1;
      }
  
      @Override
      public int getCapacity() {
          return getNativeCapacity();
      }
  
      /**
       * Returns the native frame's Object value.
       *
       * If the frame's base-type is not TYPE_OBJECT, this returns a data buffer containing the native
       * data (this is equivalent to calling getData().
       * If the frame is based on an object type, this type is expected to be a subclass of
       * NativeBuffer. The NativeBuffer returned is only valid for as long as the frame is alive. If
       * you need to hold on to the returned value, you must retain it.
       */
      @Override
      public Object getObjectValue() {
          // If this is not a structured frame, return our data
          if (getFormat().getBaseType() != FrameFormat.TYPE_OBJECT) {
              return getData();
          }
  
          // Get the structure class
          Class structClass = getFormat().getObjectClass();
          if (structClass == null) {
              throw new RuntimeException("Attempting to get object data from frame that does " +
                                         "not specify a structure object class!");
          }
  
          // Make sure it is a NativeBuffer subclass
          if (!NativeBuffer.class.isAssignableFrom(structClass)) {
              throw new RuntimeException("NativeFrame object class must be a subclass of " +
                                         "NativeBuffer!");
          }
  
          // Instantiate a new empty instance of this class
          NativeBuffer structData = null;
          try {
            structData = (NativeBuffer)structClass.newInstance();
          } catch (Exception e) {
            throw new RuntimeException("Could not instantiate new structure instance of type '" +
                                       structClass + "'!");
          }
  
          // Wrap it around our data
          if (!getNativeBuffer(structData)) {
              throw new RuntimeException("Could not get the native structured data for frame!");
          }
  
          // Attach this frame to it
          structData.attachToFrame(this);
  
          return structData;
      }
  
      @Override
      public void setInts(int[] ints) {
          assertFrameMutable();
          if (ints.length * nativeIntSize() > getFormat().getSize()) {
              throw new RuntimeException(
                  "NativeFrame cannot hold " + ints.length + " integers. (Can only hold " +
                  (getFormat().getSize() / nativeIntSize()) + " integers).");
          } else if (!setNativeInts(ints)) {
              throw new RuntimeException("Could not set int values for native frame!");
          }
      }
  
      @Override
      public int[] getInts() {
          return getNativeInts(getFormat().getSize());
      }
  
      @Override
      public void setFloats(float[] floats) {
          assertFrameMutable();
          if (floats.length * nativeFloatSize() > getFormat().getSize()) {
              throw new RuntimeException(
                  "NativeFrame cannot hold " + floats.length + " floats. (Can only hold " +
                  (getFormat().getSize() / nativeFloatSize()) + " floats).");
          } else if (!setNativeFloats(floats)) {
              throw new RuntimeException("Could not set int values for native frame!");
          }
      }
  
      @Override
      public float[] getFloats() {
          return getNativeFloats(getFormat().getSize());
      }
  
      // TODO: This function may be a bit confusing: Is the offset the target or source offset? Maybe
      // we should allow specifying both? (May be difficult for other frame types).
      @Override
      public void setData(ByteBuffer buffer, int offset, int length) {
          assertFrameMutable();
          byte[] bytes = buffer.array();
          if ((length + offset) > buffer.limit()) {
              throw new RuntimeException("Offset and length exceed buffer size in native setData: " +
                                         (length + offset) + " bytes given, but only " + buffer.limit() +
                                         " bytes available!");
          } else if (getFormat().getSize() != length) {
              throw new RuntimeException("Data size in setData does not match native frame size: " +
                                         "Frame size is " + getFormat().getSize() + " bytes, but " +
                                         length + " bytes given!");
          } else if (!setNativeData(bytes, offset, length)) {
              throw new RuntimeException("Could not set native frame data!");
          }
      }
  
      @Override
      public ByteBuffer getData() {
          byte[] data = getNativeData(getFormat().getSize());
          return data == null ? null : ByteBuffer.wrap(data);
      }
  
      @Override
      public void setBitmap(Bitmap bitmap) {
          assertFrameMutable();
          if (getFormat().getNumberOfDimensions() != 2) {
              throw new RuntimeException("Attempting to set Bitmap for non 2-dimensional native frame!");
          } else if (getFormat().getWidth()  != bitmap.getWidth() ||
                     getFormat().getHeight() != bitmap.getHeight()) {
              throw new RuntimeException("Bitmap dimensions do not match native frame dimensions!");
          } else {
              Bitmap rgbaBitmap = convertBitmapToRGBA(bitmap);
              int byteCount = rgbaBitmap.getByteCount();
              int bps = getFormat().getBytesPerSample();
              if (!setNativeBitmap(rgbaBitmap, byteCount, bps)) {
                  throw new RuntimeException("Could not set native frame bitmap data!");
              }
          }
      }
  
      @Override
      public Bitmap getBitmap() {
          if (getFormat().getNumberOfDimensions() != 2) {
              throw new RuntimeException("Attempting to get Bitmap for non 2-dimensional native frame!");
          }
          Bitmap result = Bitmap.createBitmap(getFormat().getWidth(),
                                              getFormat().getHeight(),
                                              Bitmap.Config.ARGB_8888);
          int byteCount = result.getByteCount();
          int bps = getFormat().getBytesPerSample();
          if (!getNativeBitmap(result, byteCount, bps)) {
              throw new RuntimeException("Could not get bitmap data from native frame!");
          }
          return result;
      }
  
      @Override
      public void setDataFromFrame(Frame frame) {
          // Make sure frame fits
          if (getFormat().getSize() < frame.getFormat().getSize()) {
              throw new RuntimeException(
                  "Attempting to assign frame of size " + frame.getFormat().getSize() + " to " +
                  "smaller native frame of size " + getFormat().getSize() + "!");
          }
  
          // Invoke optimized implementations if possible
          if (frame instanceof NativeFrame) {
              nativeCopyFromNative((NativeFrame)frame);
          } else if (frame instanceof GLFrame) {
              nativeCopyFromGL((GLFrame)frame);
          } else if (frame instanceof SimpleFrame) {
              setObjectValue(frame.getObjectValue());
          } else {
              super.setDataFromFrame(frame);
          }
      }
  
      @Override
      public String toString() {
          return "NativeFrame id: " + nativeFrameId + " (" + getFormat() + ") of size "
              + getCapacity();
      }
  
      static {
          System.loadLibrary("filterfw");
      }
  
      private native boolean nativeAllocate(int capacity);
  
      private native boolean nativeDeallocate();
  
      private native int getNativeCapacity();
  
      private static native int nativeIntSize();
  
      private static native int nativeFloatSize();
  
      private native boolean setNativeData(byte[] data, int offset, int length);
  
      private native byte[] getNativeData(int byteCount);
  
      private native boolean getNativeBuffer(NativeBuffer buffer);
  
      private native boolean setNativeInts(int[] ints);
  
      private native boolean setNativeFloats(float[] floats);
  
      private native int[] getNativeInts(int byteCount);
  
      private native float[] getNativeFloats(int byteCount);
  
      private native boolean setNativeBitmap(Bitmap bitmap, int size, int bytesPerSample);
  
      private native boolean getNativeBitmap(Bitmap bitmap, int size, int bytesPerSample);
  
      private native boolean nativeCopyFromNative(NativeFrame frame);
  
      private native boolean nativeCopyFromGL(GLFrame frame);
  }
  