1 files changed, 85 insertions, 0 deletions
diff --git a/src/io/appium/droiddriver/instrumentation/InstrumentationUiDevice.java b/src/io/appium/droiddriver/instrumentation/InstrumentationUiDevice.java
new file mode 100644
index 0000000..3e3b35c
--- /dev/null
+++ b/src/io/appium/droiddriver/instrumentation/InstrumentationUiDevice.java
@@ -0,0 +1,85 @@
+/*
+ * Copyright (C) 2013 DroidDriver committers
+ *
+ * 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 io.appium.droiddriver.instrumentation;
+
+import android.graphics.Bitmap;
+import android.graphics.Bitmap.Config;
+import android.graphics.Canvas;
+import android.graphics.Rect;
+import android.graphics.RectF;
+import android.util.Log;
+import android.view.View;
+
+import io.appium.droiddriver.base.BaseUiDevice;
+import io.appium.droiddriver.base.DroidDriverContext;
+import io.appium.droiddriver.util.Logs;
+
+class InstrumentationUiDevice extends BaseUiDevice {
+  private final DroidDriverContext<View, ViewElement> context;
+
+  InstrumentationUiDevice(DroidDriverContext<View, ViewElement> context) {
+    this.context = context;
+  }
+
+  @Override
+  protected Bitmap takeScreenshot() {
+    ScreenshotRunnable screenshotRunnable =
+        new ScreenshotRunnable(context.getDriver().getRootElement().getRawElement());
+    context.runOnMainSync(screenshotRunnable);
+    return screenshotRunnable.screenshot;
+  }
+
+  @Override
+  protected DroidDriverContext<View, ViewElement> getContext() {
+    return context;
+  }
+
+  private static class ScreenshotRunnable implements Runnable {
+    private final View rootView;
+    Bitmap screenshot;
+
+    private ScreenshotRunnable(View rootView) {
+      this.rootView = rootView;
+    }
+
+    @Override
+    public void run() {
+      try {
+        rootView.destroyDrawingCache();
+        rootView.buildDrawingCache(false);
+        Bitmap drawingCache = rootView.getDrawingCache();
+        int[] xy = new int[2];
+        rootView.getLocationOnScreen(xy);
+        if (xy[0] == 0 && xy[1] == 0) {
+          screenshot = Bitmap.createBitmap(drawingCache);
+        } else {
+          Canvas canvas = new Canvas();
+          Rect rect = new Rect(0, 0, drawingCache.getWidth(), drawingCache.getHeight());
+          rect.offset(xy[0], xy[1]);
+          screenshot =
+              Bitmap.createBitmap(rect.width() + xy[0], rect.height() + xy[1], Config.ARGB_8888);
+          canvas.setBitmap(screenshot);
+          canvas.drawBitmap(drawingCache, null, new RectF(rect), null);
+          canvas.setBitmap(null);
+        }
+        rootView.destroyDrawingCache();
+      } catch (Throwable e) {
+        Logs.log(Log.ERROR, e);
+      }
+    }
+  }
+}