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

import android.app.Activity;
import android.graphics.Bitmap;
import android.graphics.Canvas;
import android.graphics.Color;
import android.graphics.Paint;
import android.graphics.Picture;
import android.os.Bundle;
import android.view.SurfaceHolder;
import android.view.SurfaceView;
import android.view.View;
import android.view.ViewDebug;
import android.webkit.WebChromeClient;
import android.webkit.WebView;
import android.webkit.WebViewClient;
import android.widget.ImageView;
import android.widget.LinearLayout;
import android.widget.LinearLayout.LayoutParams;
import android.widget.ProgressBar;

import java.io.PipedInputStream;
import java.io.PipedOutputStream;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.Future;

public class PictureCaptureDemo extends Activity {
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);

        final LinearLayout layout = new LinearLayout(this);
        layout.setOrientation(LinearLayout.VERTICAL);

        final LinearLayout inner = new LinearLayout(this);
        inner.setOrientation(LinearLayout.HORIZONTAL);
        ProgressBar spinner = new ProgressBar(this, null, android.R.attr.progressBarStyleLarge);
        inner.addView(spinner,
                new LayoutParams(LayoutParams.WRAP_CONTENT, LayoutParams.WRAP_CONTENT));

        inner.addView(new View(this), new LayoutParams(50, 1));

        Picture picture = new Picture();
        Canvas canvas = picture.beginRecording(100, 100);
        canvas.drawColor(Color.RED);
        Paint paint = new Paint();
        paint.setTextSize(32);
        paint.setColor(Color.BLACK);
        canvas.drawText("Hello", 0, 50, paint);
        picture.endRecording();

        ImageView iv1 = new ImageView(this);
        iv1.setImageBitmap(Bitmap.createBitmap(picture, 100, 100, Bitmap.Config.ARGB_8888));
        inner.addView(iv1, new LayoutParams(LayoutParams.WRAP_CONTENT, LayoutParams.WRAP_CONTENT));

        inner.addView(new View(this), new LayoutParams(50, 1));

        ImageView iv2 = new ImageView(this);
        iv2.setImageBitmap(Bitmap.createBitmap(picture, 100, 100, Bitmap.Config.HARDWARE));
        inner.addView(iv2, new LayoutParams(LayoutParams.WRAP_CONTENT, LayoutParams.WRAP_CONTENT));

        layout.addView(inner,
                new LayoutParams(LayoutParams.MATCH_PARENT, LayoutParams.WRAP_CONTENT));
        // For testing with a functor in the tree
        WebView wv = new WebView(this);
        wv.setWebViewClient(new WebViewClient());
        wv.setWebChromeClient(new WebChromeClient());
        wv.loadUrl("https://google.com");
        layout.addView(wv, new LayoutParams(LayoutParams.MATCH_PARENT, 400));

        SurfaceView mySurfaceView = new SurfaceView(this);
        layout.addView(mySurfaceView,
                new LayoutParams(LayoutParams.MATCH_PARENT, 600));

        setContentView(layout);

        mySurfaceView.getHolder().addCallback(new SurfaceHolder.Callback() {
            private AutoCloseable mStopCapture;

            @Override
            public void surfaceCreated(SurfaceHolder holder) {
                final Random rand = new Random();
                mStopCapture = ViewDebug.startRenderingCommandsCapture(mySurfaceView,
                        mCaptureThread, (picture) -> {
                            if (rand.nextInt(20) == 0) {
                                try {
                                    Thread.sleep(100);
                                } catch (InterruptedException e) {
                                }
                            }
                            Canvas canvas = holder.lockCanvas();
                            if (canvas == null) {
                                return false;
                            }
                            canvas.drawPicture(picture);
                            holder.unlockCanvasAndPost(canvas);
                            picture.close();
                            return true;
                        });
            }

            @Override
            public void surfaceChanged(SurfaceHolder holder, int format, int width, int height) {

            }

            @Override
            public void surfaceDestroyed(SurfaceHolder holder) {
                if (mStopCapture != null) {
                    try {
                        mStopCapture.close();
                    } catch (Exception e) {
                    }
                    mStopCapture = null;
                }
            }
        });
    }

    ExecutorService mCaptureThread = Executors.newSingleThreadExecutor();
    ExecutorService mExecutor = Executors.newSingleThreadExecutor();

    Picture deepCopy(Picture src) {
        try {
            PipedInputStream inputStream = new PipedInputStream();
            PipedOutputStream outputStream = new PipedOutputStream(inputStream);
            Future<Picture> future = mExecutor.submit(() -> Picture.createFromStream(inputStream));
            src.writeToStream(outputStream);
            outputStream.close();
            return future.get();
        } catch (Exception ex) {
            throw new RuntimeException(ex);
        }
    }
}