1 files changed, 83 insertions, 0 deletions
diff --git a/app/src/main/java/com/javispedro/vndroid/ScreenGrabber.java b/app/src/main/java/com/javispedro/vndroid/ScreenGrabber.java
new file mode 100644
index 0000000..54d11bc
--- /dev/null
+++ b/app/src/main/java/com/javispedro/vndroid/ScreenGrabber.java
@@ -0,0 +1,83 @@
+package com.javispedro.vndroid;
+
+import android.content.Context;
+import android.content.ContextWrapper;
+import android.hardware.display.VirtualDisplay;
+import android.media.Image;
+import android.media.ImageReader;
+import android.support.annotation.Nullable;
+import android.util.Log;
+
+public abstract class ScreenGrabber extends ContextWrapper {
+    private static final String TAG = ScreenGrabber.class.getSimpleName();
+
+    protected static final int MAX_IMAGES = 3;
+
+    @Nullable protected ImageReader reader;
+    @Nullable protected VirtualDisplay display;
+    @Nullable protected Callback callback;
+
+    public interface Callback {
+        void onImage(Image image);
+    }
+
+    protected class VirtualDisplayCallback extends VirtualDisplay.Callback {
+        @Override
+        public void onPaused() {
+            Log.d(TAG, "onVirtualDisplayPaused");
+        }
+
+        @Override
+        public void onResumed() {
+            Log.d(TAG, "onVirtualDisplayResumed");
+        }
+
+        @Override
+        public void onStopped() {
+            Log.d(TAG, "onVirtualDisplayStopped");
+        }
+    }
+
+    protected class ImageReaderCallback implements ImageReader.OnImageAvailableListener {
+        @Override
+        public void onImageAvailable(ImageReader reader) {
+            if (callback != null) {
+                Image image = reader.acquireLatestImage();
+                callback.onImage(image);
+            }
+        }
+    }
+
+    protected ScreenGrabber(Context context) {
+        super(context);
+    }
+
+    public abstract void start();
+
+    public void stop() {
+        if (display != null) {
+            display.release();
+            display = null;
+        }
+        if (reader != null) {
+            reader.close();
+            reader = null;
+        }
+    }
+
+    public void setCallback(Callback c) {
+        callback = c;
+    }
+
+    public int scaleInputX(int x) {
+        return x;
+    }
+
+    public int scaleInputY(int y) {
+        return y;
+    }
+
+    public boolean hasSizeChanged() { return false; }
+
+    public void updateScreenSize() { }
+}