/*
 * Copyright (C) 2020 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.test.mediahelper;

import static com.google.common.truth.Truth.assertThat;

import android.graphics.Bitmap;
import android.graphics.BitmapFactory;
import android.media.MediaExtractor;
import android.media.MediaFormat;
import android.util.Log;

import java.io.FileDescriptor;
import java.io.IOException;
import java.util.concurrent.TimeUnit;

/** Helper functions for validating media files. */
public class MediaValidationHelper {
    private static final String TAG = MediaValidationHelper.class.getSimpleName();
    private static final String VIDEO_MIME_TYPE = "video";
    private static final String AUDIO_MIME_TYPE = "audio";

    /**
     * Validate video file.
     *
     * <p>This function validates that video contains at least 1 video track and 1 audio track. Also
     * validate that the video track has expected height and width. It also checks the video track
     * duration is bigger than the specified length.
     *
     * @param fileDescriptor The {@link FileDescriptor} of the video file.
     * @param expHeight Expected video track height.
     * @param expWidth Expected video track width.
     * @param minDurationMillis Minimum duration in milli seconds. The video track duration should
     *     be longer than it.
     * @throws IOException
     */
    public static void validateVideo(
            FileDescriptor fileDescriptor, int expHeight, int expWidth, long minDurationMillis)
            throws IOException {
        MediaExtractor extractor = new MediaExtractor();
        extractor.setDataSource(fileDescriptor);
        int numTracks = extractor.getTrackCount();
        assertThat(numTracks).isGreaterThan(1);
        Log.d(TAG, String.format("Number of tracks: %d", numTracks));
        boolean hasVideoTrack = false;
        boolean hasAudioTrack = false;
        for (int i = 0; i < numTracks; i++) {
            MediaFormat format = extractor.getTrackFormat(i);
            String mime = format.getString(MediaFormat.KEY_MIME);
            Log.d(TAG, String.format("Mime type: %s; format: %s", mime, format.toString()));
            if (mime.contains(VIDEO_MIME_TYPE)) {
                hasVideoTrack = true;
                validateVideoTrackMediaFormat(format, expHeight, expWidth, minDurationMillis);
                // TODO(b/164484222): Validate video track frame drop.
            } else if (mime.contains(AUDIO_MIME_TYPE)) {
                hasAudioTrack = true;
            }
        }
        assertThat(hasVideoTrack).isTrue();
        assertThat(hasAudioTrack).isTrue();
    }

    /**
     * Validates that video data {@link MediaFormat} is equal to the expected width and height. And
     * the duration is longer than certain value.
     *
     * @param format The {@link MediaFormat} of the video track.
     * @param expHeight Expected video track height.
     * @param expWidth Expected video track width.
     * @param minDurationMillis Minimum duration in milli seconds. The video track duration should
     *     be longer than it.
     */
    private static void validateVideoTrackMediaFormat(
            MediaFormat format, int expHeight, int expWidth, long minDurationMillis) {
        long durationMillis =
                TimeUnit.MICROSECONDS.toMillis(format.getLong(MediaFormat.KEY_DURATION));
        int width = format.getInteger(MediaFormat.KEY_WIDTH);
        int height = format.getInteger(MediaFormat.KEY_HEIGHT);
        Log.d(
                TAG,
                String.format(
                        "Duration: %d; Width: %d; Height: %d", durationMillis, width, height));
        assertThat(durationMillis).isGreaterThan(minDurationMillis);
        assertThat(width).isEqualTo(expWidth);
        assertThat(height).isEqualTo(expHeight);
    }

    /**
     * Validates that the image file height and weight is greater than certain value.
     *
     * @param fileDescriptor The {@link FileDescriptor} of the image.
     * @param minHeight Minimum height. The image height should be greater than this value.
     * @param minWidth Minimum width. The image width should be greater than this value.
     */
    public static void validateImage(FileDescriptor fileDescriptor, int minHeight, int minWidth) {
        Bitmap bitmap = BitmapFactory.decodeFileDescriptor(fileDescriptor);
        int height = bitmap.getHeight();
        int width = bitmap.getWidth();
        Log.d(TAG, String.format("Height: %d; Width: %d", height, width));
        assertThat(height).isGreaterThan(minHeight);
        assertThat(width).isGreaterThan(minWidth);
    }
}