1227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks/*
2227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Copyright 2013 The Android Open Source Project
3227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks *
4227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Licensed under the Apache License, Version 2.0 (the "License");
5227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * you may not use this file except in compliance with the License.
6227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * You may obtain a copy of the License at
7227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks *
8227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks *      http://www.apache.org/licenses/LICENSE-2.0
9227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks *
10227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * Unless required by applicable law or agreed to in writing, software
11227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * distributed under the License is distributed on an "AS IS" BASIS,
12227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * See the License for the specific language governing permissions and
14227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks * limitations under the License.
15227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks */
16227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
17227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspackage androidx.media.filterfw.samples.simplecamera;
18227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
19227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
20227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.content.res.AssetManager;
21227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.graphics.Bitmap;
22227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.graphics.BitmapFactory;
23227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.net.Uri;
24227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport android.provider.MediaStore;
25227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.Filter;
26227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameImage2D;
27227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameType;
28227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.FrameValue;
29227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.MffContext;
30227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport androidx.media.filterfw.MffFilterTestCase;
31227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
32227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.io.FileNotFoundException;
33227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.io.IOException;
34227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.util.concurrent.ExecutionException;
35227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricksimport java.util.concurrent.TimeoutException;
36227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
37227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendrickspublic class IfElseFilterTest extends MffFilterTestCase {
38227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    private final static int BIG_INPUT_WIDTH = 1536;
39227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    private final static int BIG_INPUT_HEIGHT = 2048;
40227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    private final static int SMALL_INPUT_WIDTH = 480;
41227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    private final static int SMALL_INPUT_HEIGHT = 640;
42227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
43227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    private AssetManager assetMgr = null;
44227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    @Override
45227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    protected Filter createFilter(MffContext mffContext) {
46227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        assetMgr = mffContext.getApplicationContext().getAssets();
47227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        return new IfElseFilter(mffContext, "ifElseFilter");
48227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    }
49227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
50227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    public void testIfElseFilterTrue() throws Exception {
51227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameImage2D image =
52227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                createFrame(FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_CPU),
53227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                        new int[] {BIG_INPUT_WIDTH,BIG_INPUT_HEIGHT}).asFrameImage2D();
54227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameImage2D video =
55227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                createFrame(FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_CPU),
56227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                        new int[] {SMALL_INPUT_WIDTH,SMALL_INPUT_HEIGHT}).asFrameImage2D();
57227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
58227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        // Image of legs
59227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        Bitmap videoBitmap = BitmapFactory.decodeStream(assetMgr.open("0002_000390.jpg"));
60227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        // Image of a face
61227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        Bitmap imageBitmap = BitmapFactory.decodeStream(assetMgr.open("XZZ019.jpg"));
62227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
63227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        image.setBitmap(imageBitmap);
64227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        injectInputFrame("falseResult", image);
65227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        video.setBitmap(videoBitmap);
66227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        injectInputFrame("trueResult", video);
67227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
68227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameValue conditionFrame = createFrame(FrameType.single(boolean.class), new int[] {1}).
69227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                asFrameValue();
70227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        conditionFrame.setValue(true);
71227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        injectInputFrame("condition", conditionFrame);
72227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
73227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        process();
74227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
75227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        // Ensure that for true, we use the video input
76227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameImage2D outputImage = getOutputFrame("output").asFrameImage2D();
77227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        assertEquals(outputImage, video);
78227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    }
79227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
80227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    public void testIfElseFilterFalse() throws Exception {
81227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
82227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameImage2D image =
83227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                createFrame(FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_CPU),
84227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                        new int[] {BIG_INPUT_WIDTH,BIG_INPUT_HEIGHT}).asFrameImage2D();
85227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameImage2D video =
86227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                createFrame(FrameType.image2D(FrameType.ELEMENT_RGBA8888, FrameType.READ_CPU),
87227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                        new int[] {SMALL_INPUT_WIDTH,SMALL_INPUT_HEIGHT}).asFrameImage2D();
88227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
89227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        // Image of legs
90227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        Bitmap videoBitmap = BitmapFactory.decodeStream(assetMgr.open("0002_000390.jpg"));
91227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        // Image of a face
92227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        Bitmap imageBitmap = BitmapFactory.decodeStream(assetMgr.open("XZZ019.jpg"));
93227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
94227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        image.setBitmap(imageBitmap);
95227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        injectInputFrame("falseResult", image);
96227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        video.setBitmap(videoBitmap);
97227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        injectInputFrame("trueResult", video);
98227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
99227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
100227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameValue conditionFrame = createFrame(FrameType.single(boolean.class), new int[] {1}).
101227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks                asFrameValue();
102227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        conditionFrame.setValue(false);
103227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        injectInputFrame("condition", conditionFrame);
104227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
105227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        process();
106227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks
107227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        // Ensure that for true, we use the video input
108227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        FrameImage2D outputImage = getOutputFrame("output").asFrameImage2D();
109227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks        assertEquals(outputImage, image);
110227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks    }
111227b47625d7482b5b47ad0e4c70ce0a246236adeBenjamin Hendricks}
112