xref: /packages/apps/Gallery2/src/com/android/gallery3d/filtershow/filters/ImageFilterRedEye.java

/*
 * Copyright (C) 2012 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 com.android.gallery3d.filtershow.filters;

import android.graphics.Bitmap;
import android.graphics.Matrix;
import android.graphics.RectF;

import com.android.gallery3d.filtershow.imageshow.GeometryMetadata;

import java.util.Vector;

public class ImageFilterRedEye extends ImageFilter {
    private static final String LOGTAG = "ImageFilterRedEye";
    private Vector<RedEyeCandidate> mCandidates = null;

    public ImageFilterRedEye() {
        mName = "Red Eye";
    }

    @Override
    public ImageFilter clone() throws CloneNotSupportedException {
        ImageFilterRedEye filter = (ImageFilterRedEye) super.clone();
        if (mCandidates != null) {
            int size = mCandidates.size();
            filter.mCandidates = new Vector<RedEyeCandidate>();
            for (int i = 0; i < size; i++) {
                filter.mCandidates.add(new RedEyeCandidate(mCandidates.elementAt(i)));
            }
        }
        return filter;
    }

    @Override
    public boolean isNil() {
        if (mCandidates != null && mCandidates.size() > 0) {
            return false;
        }
        return true;
    }

    @Override
    public boolean same(ImageFilter filter) {
        boolean isRedEyeFilter = super.same(filter);
        if (!isRedEyeFilter) {
            return false;
        }
        ImageFilterRedEye redEyeFilter = (ImageFilterRedEye) filter;
        if (redEyeFilter.mCandidates == null && mCandidates == null) {
            return true;
        }
        if (redEyeFilter.mCandidates == null || mCandidates == null) {
            return false;
        }
        if (redEyeFilter.mCandidates.size() != mCandidates.size()) {
            return false;
        }
        int size = mCandidates.size();
        for (int i = 0; i < size; i++) {
            RedEyeCandidate c1 = mCandidates.elementAt(i);
            RedEyeCandidate c2 = redEyeFilter.mCandidates.elementAt(i);
            if (!c1.equals(c2)) {
                return false;
            }
        }
        return true;
    }

    public Vector<RedEyeCandidate> getCandidates() {
        if (mCandidates == null) {
            mCandidates = new Vector<RedEyeCandidate>();
        }
        return mCandidates;
    }

    public void addRect(RectF rect, RectF bounds) {
        if (mCandidates == null) {
            mCandidates = new Vector<RedEyeCandidate>();
        }
        Vector<RedEyeCandidate> intersects = new Vector<RedEyeCandidate>();
        for (int i = 0; i < mCandidates.size(); i++) {
            RedEyeCandidate r = mCandidates.elementAt(i);
            if (r.intersect(rect)) {
                intersects.add(r);
            }
        }
        for (int i = 0; i < intersects.size(); i++) {
            RedEyeCandidate r = intersects.elementAt(i);
            rect.union(r.mRect);
            bounds.union(r.mBounds);
            mCandidates.remove(r);
        }
        mCandidates.add(new RedEyeCandidate(rect, bounds));
    }

    public void clear() {
        if (mCandidates == null) {
            mCandidates = new Vector<RedEyeCandidate>();
        }
        mCandidates.clear();
    }

    native protected void nativeApplyFilter(Bitmap bitmap, int w, int h, short[] matrix);

    @Override
    public Bitmap apply(Bitmap bitmap, float scaleFactor, boolean highQuality) {
        int w = bitmap.getWidth();
        int h = bitmap.getHeight();
        short[] rect = new short[4];

        if (mCandidates != null && mCandidates.size() > 0) {
            for (int i = 0; i < mCandidates.size(); i++) {
                RectF r = new RectF(mCandidates.elementAt(i).mRect);
                GeometryMetadata geo = getImagePreset().mGeoData;
                Matrix originalToScreen = geo.getOriginalToScreen(true,
                        getImagePreset().getImageLoader().getOriginalBounds().width(),
                        getImagePreset().getImageLoader().getOriginalBounds().height(),
                        w, h);
                originalToScreen.mapRect(r);
                if (r.left < 0) {
                    r.left = 0;
                }
                if (r.left > w) {
                    r.left = w;
                }
                if (r.top < 0) {
                    r.top = 0;
                }
                if (r.top > h) {
                    r.top = h;
                }
                if (r.right < 0) {
                    r.right = 0;
                }
                if (r.right > w) {
                    r.right = w;
                }
                if (r.bottom < 0) {
                    r.bottom = 0;
                }
                if (r.bottom > h) {
                    r.bottom = h;
                }
                rect[0] = (short) r.left;
                rect[1] = (short) r.top;
                rect[2] = (short) r.width();
                rect[3] = (short) r.height();
                nativeApplyFilter(bitmap, w, h, rect);
            }
        }
        return bitmap;
    }
}