/*
* Copyright (C) 2008 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.screenshot;
import com.android.ddmlib.AndroidDebugBridge;
import com.android.ddmlib.Device;
import com.android.ddmlib.Log;
import com.android.ddmlib.RawImage;
import com.android.ddmlib.Log.ILogOutput;
import com.android.ddmlib.Log.LogLevel;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import javax.imageio.ImageIO;
/**
* Connects to a device using ddmlib and dumps its event log as long as the device is connected.
*/
public class Screenshot {
public static void main(String[] args) {
boolean device = false;
boolean emulator = false;
String serial = null;
String filepath = null;
boolean landscape = false;
if (args.length == 0) {
printUsageAndQuit();
}
// parse command line parameters.
int index = 0;
do {
String argument = args[index++];
if ("-d".equals(argument)) {
if (emulator || serial != null) {
printAndExit("-d conflicts with -e and -s", false /* terminate */);
}
device = true;
} else if ("-e".equals(argument)) {
if (device || serial != null) {
printAndExit("-e conflicts with -d and -s", false /* terminate */);
}
emulator = true;
} else if ("-s".equals(argument)) {
// quick check on the next argument.
if (index == args.length) {
printAndExit("Missing serial number after -s", false /* terminate */);
}
if (device || emulator) {
printAndExit("-s conflicts with -d and -e", false /* terminate */);
}
serial = args[index++];
} else if ("-l".equals(argument)) {
landscape = true;
} else {
// get the filepath and break.
filepath = argument;
// should not be any other device.
if (index < args.length) {
printAndExit("Too many arguments!", false /* terminate */);
}
}
} while (index < args.length);
if (filepath == null) {
printUsageAndQuit();
}
Log.setLogOutput(new ILogOutput() {
public void printAndPromptLog(LogLevel logLevel, String tag, String message) {
System.err.println(logLevel.getStringValue() + ":" + tag + ":" + message);
}
public void printLog(LogLevel logLevel, String tag, String message) {
System.err.println(logLevel.getStringValue() + ":" + tag + ":" + message);
}
});
// init the lib
// [try to] ensure ADB is running
String adbLocation = System.getProperty("com.android.screenshot.bindir"); //$NON-NLS-1$
if (adbLocation != null && adbLocation.length() != 0) {
adbLocation += File.separator + "adb"; //$NON-NLS-1$
} else {
adbLocation = "adb"; //$NON-NLS-1$
}
AndroidDebugBridge.init(false /* debugger support */);
try {
AndroidDebugBridge bridge = AndroidDebugBridge.createBridge(
adbLocation, true /* forceNewBridge */);
// we can't just ask for the device list right away, as the internal thread getting
// them from ADB may not be done getting the first list.
// Since we don't really want getDevices() to be blocking, we wait here manually.
int count = 0;
while (bridge.hasInitialDeviceList() == false) {
try {
Thread.sleep(100);
count++;
} catch (InterruptedException e) {
// pass
}
// let's not wait > 10 sec.
if (count > 100) {
System.err.println("Timeout getting device list!");
return;
}
}
// now get the devices
Device[] devices = bridge.getDevices();
if (devices.length == 0) {
printAndExit("No devices found!", true /* terminate */);
}
Device target = null;
if (emulator || device) {
for (Device d : devices) {
// this test works because emulator and device can't both be true at the same
// time.
if (d.isEmulator() == emulator) {
// if we already found a valid target, we print an error and return.
if (target != null) {
if (emulator) {
printAndExit("Error: more than one emulator launched!",
true /* terminate */);
} else {
printAndExit("Error: more than one device connected!",true /* terminate */);
}
}
target = d;
}
}
} else if (serial != null) {
for (Device d : devices) {
if (serial.equals(d.getSerialNumber())) {
target = d;
break;
}
}
} else {
if (devices.length > 1) {
printAndExit("Error: more than one emulator or device available!",
true /* terminate */);
}
target = devices[0];
}
if (target != null) {
try {
System.out.println("Taking screenshot from: " + target.getSerialNumber());
getDeviceImage(target, filepath, landscape);
System.out.println("Success.");
} catch (IOException e) {
e.printStackTrace();
}
} else {
printAndExit("Could not find matching device/emulator.", true /* terminate */);
}
} finally {
AndroidDebugBridge.terminate();
}
}
/*
* Grab an image from an ADB-connected device.
*/
private static void getDeviceImage(Device device, String filepath, boolean landscape)
throws IOException {
RawImage rawImage;
try {
rawImage = device.getScreenshot();
}
catch (IOException ioe) {
printAndExit("Unable to get frame buffer: " + ioe.getMessage(), true /* terminate */);
return;
}
// device/adb not available?
if (rawImage == null)
return;
assert rawImage.bpp == 16;
BufferedImage image;
if (landscape) {
// convert raw data to an Image
image = new BufferedImage(rawImage.height, rawImage.width,
BufferedImage.TYPE_INT_ARGB);
byte[] buffer = rawImage.data;
int index = 0;
for (int y = 0 ; y < rawImage.height ; y++) {
for (int x = 0 ; x < rawImage.width ; x++) {
int value = buffer[index++] & 0x00FF;
value |= (buffer[index++] << 8) & 0x0FF00;
int r = ((value >> 11) & 0x01F) << 3;
int g = ((value >> 5) & 0x03F) << 2;
int b = ((value >> 0) & 0x01F) << 3;
value = 0xFF << 24 | r << 16 | g << 8 | b;
image.setRGB(y, rawImage.width - x - 1, value);
}
}
} else {
// convert raw data to an Image
image = new BufferedImage(rawImage.width, rawImage.height,
BufferedImage.TYPE_INT_ARGB);
byte[] buffer = rawImage.data;
int index = 0;
for (int y = 0 ; y < rawImage.height ; y++) {
for (int x = 0 ; x < rawImage.width ; x++) {
int value = buffer[index++] & 0x00FF;
value |= (buffer[index++] << 8) & 0x0FF00;
int r = ((value >> 11) & 0x01F) << 3;
int g = ((value >> 5) & 0x03F) << 2;
int b = ((value >> 0) & 0x01F) << 3;
value = 0xFF << 24 | r << 16 | g << 8 | b;
image.setRGB(x, y, value);
}
}
}
if (!ImageIO.write(image, "png", new File(filepath))) {
throw new IOException("Failed to find png writer");
}
}
private static void printUsageAndQuit() {
// 80 cols marker: 01234567890123456789012345678901234567890123456789012345678901234567890123456789
System.out.println("Usage: screenshot2 [-d | -e | -s SERIAL] [-l] OUT_FILE");
System.out.println("");
System.out.println(" -d Uses the first device found.");
System.out.println(" -e Uses the first emulator found.");
System.out.println(" -s Targets the device by serial number.");
System.out.println("");
System.out.println(" -l Rotate images for landscape mode.");
System.out.println("");
System.exit(1);
}
private static void printAndExit(String message, boolean terminate) {
System.out.println(message);
if (terminate) {
AndroidDebugBridge.terminate();
}
System.exit(1);
}
}