165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn/*
265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * Copyright (C) 2011 The Android Open Source Project
365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn *
465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * Licensed under the Apache License, Version 2.0 (the "License");
565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * you may not use this file except in compliance with the License.
665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * You may obtain a copy of the License at
765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn *
865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn *      http://www.apache.org/licenses/LICENSE-2.0
965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn *
1065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * Unless required by applicable law or agreed to in writing, software
1165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * distributed under the License is distributed on an "AS IS" BASIS,
1265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * See the License for the specific language governing permissions and
1465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * limitations under the License.
1565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn */
1665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
1765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
1865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennpackage android.filterpacks.imageproc;
1965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
2065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.content.Context;
2165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.Filter;
2265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.FilterContext;
2365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.Frame;
2465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.FrameFormat;
2565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.FrameManager;
2665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.GenerateFinalPort;
2765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.GenerateFieldPort;
2865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.KeyValueMap;
2965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.MutableFrameFormat;
3065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.core.NativeFrame;
3165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.filterfw.format.ImageFormat;
3265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennimport android.graphics.Bitmap;
3365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
3465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn/**
3565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn * @hide
3665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn */
3765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Rennpublic class BitmapSource extends Filter {
3865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
3965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @GenerateFieldPort(name = "target")
4065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    String mTargetString;
4165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
4265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @GenerateFieldPort(name = "bitmap")
4365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    private Bitmap mBitmap;
4465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
4565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @GenerateFieldPort(name = "recycleBitmap", hasDefault = true)
4665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    private boolean mRecycleBitmap = true;
4765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
4865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @GenerateFieldPort(name = "repeatFrame", hasDefault = true)
4965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    boolean mRepeatFrame = false;
5065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
5165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    private int mTarget;
5265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    private Frame mImageFrame;
5365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
5465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    public BitmapSource(String name) {
5565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        super(name);
5665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    }
5765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
5865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
5965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @Override
6065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    public void setupPorts() {
6165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        // Setup output format
6265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        FrameFormat outputFormat = ImageFormat.create(ImageFormat.COLORSPACE_RGBA,
6365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn                                                      FrameFormat.TARGET_UNSPECIFIED);
6465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
6565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        // Add output port
6665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        addOutputPort("image", outputFormat);
6765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    }
6865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
6965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    public void loadImage(FilterContext filterContext) {
7065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        // Create frame with bitmap
7165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        mTarget = FrameFormat.readTargetString(mTargetString);
7265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        FrameFormat outputFormat = ImageFormat.create(mBitmap.getWidth(),
7365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn                                                      mBitmap.getHeight(),
7465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn                                                      ImageFormat.COLORSPACE_RGBA,
7565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn                                                      mTarget);
7665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        mImageFrame = filterContext.getFrameManager().newFrame(outputFormat);
7765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        mImageFrame.setBitmap(mBitmap);
7865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        mImageFrame.setTimestamp(Frame.TIMESTAMP_UNKNOWN);
7965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
8065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        // Free up space used by bitmap
8165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        if (mRecycleBitmap) {
8265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            mBitmap.recycle();
8365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        }
8465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        mBitmap = null;
8565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    }
8665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
8765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @Override
8865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    public void fieldPortValueUpdated(String name, FilterContext context) {
8965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        // Clear image (to trigger reload) in case parameters have been changed
9065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        if (name.equals("bitmap") || name.equals("target")) {
9165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            if (mImageFrame != null) {
9265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn                mImageFrame.release();
9365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn                mImageFrame = null;
9465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            }
9565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        }
9665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    }
9765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
9865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @Override
9965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    public void process(FilterContext context) {
10065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        if (mImageFrame == null) {
10165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            loadImage(context);
10265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        }
10365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
10465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        pushOutput("image", mImageFrame);
10565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
10665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        if (!mRepeatFrame) {
10765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            closeOutputPort("image");
10865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        }
10965953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    }
11065953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn
11165953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    @Override
11265953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    public void tearDown(FilterContext env) {
11365953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        if (mImageFrame != null) {
11465953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            mImageFrame.release();
11565953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn            mImageFrame = null;
11665953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn        }
11765953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn    }
11865953da4636fbf5f0a24b8f5f2b5fa7d76ff13d9Marius Renn}
119