org.libimobiledevice.ios.driver.binding.services.ScreenshotService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of reactor-x-ios-libimobiledevice Show documentation
libimoviledevice
The newest version!
/*
 * Copyright 2012-2013 eBay Software Foundation and ios-driver 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 org.libimobiledevice.ios.driver.binding.services;

import com.sun.jna.ptr.PointerByReference;

import org.apache.commons.imaging.ImageReadException;
import org.apache.commons.imaging.formats.tiff.TiffImageParser;
import org.libimobiledevice.ios.driver.binding.exceptions.SDKException;
import org.libimobiledevice.ios.driver.binding.raw.ImobiledeviceSdkLibrary;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.nio.LongBuffer;
import java.util.HashMap;
import javax.imageio.ImageIO;
import org.apache.commons.imaging.formats.png.PngImageParser;
import static org.libimobiledevice.ios.driver.binding.exceptions.SDKErrorCode.throwIfNeeded;
import static org.libimobiledevice.ios.driver.binding.raw.ImobiledeviceSdkLibrary.screenshot_service_free;
import static org.libimobiledevice.ios.driver.binding.raw.ImobiledeviceSdkLibrary.screenshot_service_new;
import static org.libimobiledevice.ios.driver.binding.raw.ImobiledeviceSdkLibrary.screenshot_service_take_screenshot;
import static org.libimobiledevice.ios.driver.binding.raw.ImobiledeviceSdkLibrary.sdk_idevice_screenshot_service_t;

public class ScreenshotService {
    private final ImobiledeviceSdkLibrary.sdk_idevice_screenshot_service_t service;

    public ScreenshotService(IOSDevice d) throws SDKException {
        InformationService info = new InformationService(d);
        boolean dev = info.isDevModeEnabled();
        info.free();
        if (!dev) {
            throw new SDKException(
                "to use screenshots on a device, you need the device to be in dev mode.");
        }
        PointerByReference ptr = new PointerByReference();
        throwIfNeeded(screenshot_service_new(d.getSDKHandle(), ptr));
        service = new sdk_idevice_screenshot_service_t(ptr.getValue());
    }

    public void free() throws SDKException {
        throwIfNeeded(screenshot_service_free(service));
    }

    private byte[] takeScreenshotAsTiff() throws SDKException {
        PointerByReference ptr = new PointerByReference();
        LongBuffer sizeptr = LongBuffer.allocate(1);

        int res = screenshot_service_take_screenshot(service, ptr, sizeptr);

        if (ptr == null) {
            throw new SDKException("Bug ? pointer should have been assigned by the screenshot service");
        } else if (ptr.getValue() == null) {
            throw new SDKException("Didn't get a value back. Something wrong in screenshot_service");
        } else {
            long size = sizeptr.get(0);
            byte[] b = ptr.getValue().getByteArray(0, (int) size);
            return b;
        }
    }

    private byte[] getPNGAsString() throws SDKException, ImageReadException, IOException {
        byte[] raw = takeScreenshotAsTiff();

        TiffImageParser parser = new TiffImageParser();
        BufferedImage image = parser.getBufferedImage(raw, new HashMap<>());
        File f = new File("screen.png");
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        ImageIO.write(image, "png", out);
        out.flush();
        return out.toByteArray();
    }

    /**
     * Gets a screenshot in PNG format.
     *
     * @return byte array
     *
     * @throws SDKException any issue
     */
    public byte[] takeScreenshot() throws SDKException {
        try {
            return getPNGAsString();
        } catch (SDKException | ImageReadException | IOException e) {
            throw new SDKException("Cannot take a screenshot : " + e.getMessage());
        }
    }

    /**
     * Gets a screenshot in PNG format.
     *
     * @param png file path
     *
     * @throws SDKException       any issue
     * @throws ImageReadException any issue
     * @throws IOException        any issue
     */
    public void takeScreenshot(File png) throws SDKException, ImageReadException, IOException {
        byte[] raw = takeScreenshotAsTiff();
        PngImageParser parser = new PngImageParser();
        BufferedImage image = parser.getBufferedImage(raw, new HashMap<>());
        ImageIO.write(image, "png", png);
    }
}