package com.frogermcs.activityframemetrics;

import android.annotation.TargetApi;
import android.app.Activity;
import android.app.Application;
import android.os.Build;
import android.os.Bundle;
import android.os.Handler;
import android.util.Log;
import android.view.FrameMetrics;
import android.view.Window;

import java.util.HashMap;
import java.util.Map;

/**
 * Created by froger_mcs on 14/11/2016.
 */

public class ActivityFrameMetrics implements Application.ActivityLifecycleCallbacks {
    private static final float DEFAULT_WARNING_LEVEL_MS = 17.f;
    private static final float DEFAULT_ERROR_LEVEL_MS = 34.f;

    @Override
    public void onActivityCreated(Activity activity, Bundle savedInstanceState) {

    }

    @Override
    public void onActivityStarted(Activity activity) {

    }

    @Override
    public void onActivityResumed(Activity activity) {
        startFrameMetrics(activity);
    }

    @Override
    public void onActivityPaused(Activity activity) {
        stopFrameMetrics(activity);
    }

    @Override
    public void onActivityStopped(Activity activity) {

    }

    @Override
    public void onActivitySaveInstanceState(Activity activity, Bundle outState) {

    }

    @Override
    public void onActivityDestroyed(Activity activity) {

    }

    private float warningLevelMs;
    private float errorLevelMs;
    private boolean showWarning;
    private boolean showError;

    private Map<String, Window.OnFrameMetricsAvailableListener> frameMetricsAvailableListenerMap = new HashMap<>();

    private ActivityFrameMetrics() {
    }

    @TargetApi(Build.VERSION_CODES.N)
    public void startFrameMetrics(Activity activity) {
        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
            final String activityName = activity.getClass().getSimpleName();
            Window.OnFrameMetricsAvailableListener listener = new Window.OnFrameMetricsAvailableListener() {

                private int allFrames = 0;
                private int jankyFrames = 0;

                @Override
                public void onFrameMetricsAvailable(Window window, FrameMetrics frameMetrics, int dropCountSinceLastInvocation) {
                    FrameMetrics frameMetricsCopy = new FrameMetrics(frameMetrics);
                    allFrames++;
                    float totalDurationMs = (float) (0.000001 * frameMetricsCopy.getMetric(FrameMetrics.TOTAL_DURATION));
                    if (totalDurationMs > warningLevelMs) {
                        jankyFrames++;
                        String msg = String.format("Janky frame detected on %s with total duration: %.2fms\n", activityName, totalDurationMs);
                        float layoutMeasureDurationMs = (float) (0.000001 * frameMetricsCopy.getMetric(FrameMetrics.LAYOUT_MEASURE_DURATION));
                        float drawDurationMs = (float) (0.000001 * frameMetricsCopy.getMetric(FrameMetrics.DRAW_DURATION));
                        float gpuCommandMs = (float) (0.000001 * frameMetricsCopy.getMetric(FrameMetrics.COMMAND_ISSUE_DURATION));
                        float othersMs = totalDurationMs - layoutMeasureDurationMs - drawDurationMs - gpuCommandMs;
                        float jankyPercent = (float) jankyFrames / allFrames * 100;
                        msg += String.format("Layout/measure: %.2fms, draw:%.2fms, gpuCommand:%.2fms others:%.2fms\n",
                                layoutMeasureDurationMs, drawDurationMs, gpuCommandMs, othersMs);
                        msg += "Janky frames: " + jankyFrames + "/" + allFrames + "(" + jankyPercent + "%)";
                        if (showWarning && totalDurationMs > errorLevelMs) {
                            Log.e("FrameMetrics", msg);
                        } else if (showError) {
                            Log.w("FrameMetrics", msg);
                        }
                    }
                }
            };
            activity.getWindow().addOnFrameMetricsAvailableListener(listener, new Handler());
            frameMetricsAvailableListenerMap.put(activityName, listener);
        } else {
            Log.w("FrameMetrics", "FrameMetrics can work only with Android SDK 24 (Nougat) and higher");
        }
    }

    @TargetApi(Build.VERSION_CODES.N)
    public void stopFrameMetrics(Activity activity) {
        if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.N) {
            String activityName = activity.getClass().getName();
            Window.OnFrameMetricsAvailableListener onFrameMetricsAvailableListener = frameMetricsAvailableListenerMap.get(activityName);
            if (onFrameMetricsAvailableListener != null) {
                activity.getWindow().removeOnFrameMetricsAvailableListener(onFrameMetricsAvailableListener);
                frameMetricsAvailableListenerMap.remove(activityName);
            }
        }
    }

    public static class Builder {
        private float warningLevelMs = DEFAULT_WARNING_LEVEL_MS;
        private float errorLevelMs = DEFAULT_ERROR_LEVEL_MS;
        private boolean showWarnings = true;
        private boolean showErrors = true;

        public Builder() {
        }

        public Builder warningLevelMs(float warningLevelMs) {
            this.warningLevelMs = warningLevelMs;
            return this;
        }

        public Builder errorLevelMs(float errorLevelMs) {
            this.errorLevelMs = errorLevelMs;
            return this;
        }

        public Builder showWarnings(boolean show) {
            this.showWarnings = show;
            return this;
        }

        public Builder showErrors(boolean show) {
            this.showErrors = show;
            return this;
        }

        public ActivityFrameMetrics build() {
            ActivityFrameMetrics activityFrameMetrics = new ActivityFrameMetrics();
            activityFrameMetrics.warningLevelMs = this.warningLevelMs;
            activityFrameMetrics.errorLevelMs = this.errorLevelMs;
            activityFrameMetrics.showError = this.showErrors;
            activityFrameMetrics.showWarning = this.showWarnings;
            return activityFrameMetrics;
        }
    }
}