Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 9afc7b02 authored by Mathias Agopian's avatar Mathias Agopian
Browse files

fix [3374821] Cannot take screenshot of HC preview 

the screencap tool will now fallback to /dev/graphics/fb0 is the
screenshot API fails.

Change-Id: I3f593847c72f4ca5429d4f64ca22bd5436dd2051
parent 5cf98da8

cmds/screencap/screencap.cpp

+88 −26
Original line number Diff line number Diff line
@@ -19,6 +19,10 @@ 
#include <stdio.h>

#include <fcntl.h>



#include <linux/fb.h>

#include <sys/ioctl.h>

#include <sys/mman.h>



#include <binder/IMemory.h>

#include <surfaceflinger/SurfaceComposerClient.h>
@@ -28,14 +32,15 @@ 


using namespace android;



static void usage()

static void usage(const char* pname)

{

    fprintf(stderr,

            "usage: screenshot [-hp] [FILENAME]\n"

            "usage: %s [-hp] [FILENAME]\n"

            "   -h: this message\n"

            "   -p: save the file as a png.\n"

            "If FILENAME ends with .png it will be saved as a png.\n"

            "If FILENAME is not given, the results will be printed to stdout.\n"

            "If FILENAME is not given, the results will be printed to stdout.\n",

            pname

    );

}
@@ -54,8 +59,33 @@ static SkBitmap::Config flinger2skia(PixelFormat f) 
    }

}



static status_t vinfoToPixelFormat(const fb_var_screeninfo& vinfo,

        uint32_t* bytespp, uint32_t* f)

{



    switch (vinfo.bits_per_pixel) {

        case 16:

            *f = PIXEL_FORMAT_RGB_565;

            *bytespp = 2;

            break;

        case 24:

            *f = PIXEL_FORMAT_RGB_888;

            *bytespp = 3;

            break;

        case 32:

            // TODO: do better decoding of vinfo here

            *f = PIXEL_FORMAT_RGBX_8888;

            *bytespp = 4;

            break;

        default:

            return BAD_VALUE;

    }

    return NO_ERROR;

}



int main(int argc, char** argv)

{

    const char* pname = argv[0];

    bool png = false;

    int c;

    while ((c = getopt(argc, argv, "ph")) != -1) {
@@ -65,7 +95,7 @@ int main(int argc, char** argv) 
                break;

            case '?':

            case 'h':

                usage();

                usage(pname);

                return 1;

        }

    }
@@ -79,7 +109,7 @@ int main(int argc, char** argv) 
        const char* fn = argv[0];

        fd = open(fn, O_WRONLY | O_CREAT | O_TRUNC, 0664);

        if (fd == -1) {

            fprintf(stderr, "Error opening file: (%d) %s\n", errno, strerror(errno));

            fprintf(stderr, "Error opening file: %s (%s)\n", fn, strerror(errno));

            return 1;

        }

        const int len = strlen(fn);
@@ -89,20 +119,48 @@ int main(int argc, char** argv) 
    }

    

    if (fd == -1) {

        usage();

        usage(pname);

        return 1;

    }



    void const* mapbase = MAP_FAILED;

    ssize_t mapsize = -1;



    void const* base = 0;

    uint32_t w, h, f;

    size_t size = 0;



    ScreenshotClient screenshot;

    if (screenshot.update() != NO_ERROR) {

        return 0;

    if (screenshot.update() == NO_ERROR) {

        base = screenshot.getPixels();

        w = screenshot.getWidth();

        h = screenshot.getHeight();

        f = screenshot.getFormat();

        size = screenshot.getSize();

    } else {

        const char* fbpath = "/dev/graphics/fb0";

        int fb = open(fbpath, O_RDONLY);

        if (fb >= 0) {

            struct fb_var_screeninfo vinfo;

            if (ioctl(fb, FBIOGET_VSCREENINFO, &vinfo) == 0) {

                uint32_t bytespp;

                if (vinfoToPixelFormat(vinfo, &bytespp, &f) == NO_ERROR) {

                    size_t offset = (vinfo.xoffset + vinfo.yoffset*vinfo.xres) * bytespp;

                    w = vinfo.xres;

                    h = vinfo.yres;

                    size = w*h*bytespp;

                    mapsize = offset + size;

                    mapbase = mmap(0, mapsize, PROT_READ, MAP_PRIVATE, fb, 0);

                    if (mapbase != MAP_FAILED) {

                        base = (void const *)((char const *)mapbase + offset);

                    }

                }

            }

            close(fb);

        }

    }



    void const* base = screenshot.getPixels();

    uint32_t w = screenshot.getWidth();

    uint32_t h = screenshot.getHeight();

    uint32_t f = screenshot.getFormat();



    if (base) {

        if (png) {

            SkBitmap b;

            b.setConfig(flinger2skia(f), w, h);
@@ -115,8 +173,12 @@ int main(int argc, char** argv) 
            write(fd, &w, 4);

            write(fd, &h, 4);

            write(fd, &f, 4);

        write(fd, base, w*h*4);

            write(fd, base, size);

        }

    }

    close(fd);

    if (mapbase != MAP_FAILED) {

        munmap((void *)mapbase, mapsize);

    }

    return 0;

}