diff options
Diffstat (limited to 'utils/wpseditor/screenshot/main.c')
| -rw-r--r-- | utils/wpseditor/screenshot/main.c | 361 |
1 files changed, 361 insertions, 0 deletions
diff --git a/utils/wpseditor/screenshot/main.c b/utils/wpseditor/screenshot/main.c new file mode 100644 index 0000000000..fb81e069b0 --- /dev/null +++ b/utils/wpseditor/screenshot/main.c @@ -0,0 +1,361 @@ +/*************************************************************************** + * __________ __ ___. + * Open \______ \ ____ ____ | | _\_ |__ _______ ___ + * Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ / + * Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < < + * Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \ + * \/ \/ \/ \/ \/ + * $Id$ + * + * Copyright (C) 2008 by Maurus Cuelenaere + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY + * KIND, either express or implied. + * + ****************************************************************************/ + +#include <stdio.h> +#include <stdarg.h> +#include <string.h> +#include <dlfcn.h> +#include <unistd.h> +#include "gd.h" +#include "gdfonts.h" +#include "api.h" + +#define DEBUGF1 _debug +#define DEBUGF2 if(verbose) _debug + +#define getFont() gdFontGetSmall() + +static struct trackstate mp3data = +{ +(char*)"Test title", +(char*)"Test artist", +(char*)"Test album", +(char*)"Test genre", +(char*)"Test disc", +(char*)"Test track", +(char*)"Test year", +(char*)"Test composer", +(char*)"Test comment", +(char*)"Test album artist", +(char*)"Test grouping", +1, /* int discnum */ +1, /* int tracknum */ +1, /* int version */ +1, /* int layer */ +2008, /* int year */ + +100, /* int length */ +70 /* int elapsed */ +}; + +static struct wpsstate wpsdata = {-20, -1, -1, 70, API_STATUS_FASTFORWARD}; + /* volume, fontheight, fontwidth, battery_level, audio_status */ + +static struct proxy_api api; +static bool verbose = false; +static int (*wps_init)(const char* buff, struct proxy_api *api, bool isfile); +static int (*wps_display)(); +static int (*wps_refresh)(); +static gdImagePtr framebuffer; +static gdImagePtr backdrop; + +extern gdImagePtr gdImageCreateFromBmp(FILE * inFile); +extern char *get_current_dir_name (void) __THROW; + +int _debug(const char* fmt,...) +{ +#if 0 /* Doesn't want to compile ?? */ + struct va_list ap; + + va_start(ap, fmt); + + fprintf(stdout, "[DBG] "); + vfprintf(stdout, fmt, ap); + fprintf(stdout, "\n"); + + va_end(ap); + + return 0; +#else + return -1; +#endif +} + +void _putsxy(int x, int y, const unsigned char *str) +{ + struct viewport_api avp; + int black = gdImageColorAllocate(framebuffer, 0, 0, 0); + + api.get_current_vp(&avp); + + gdImageString(framebuffer, getFont(), x + avp.x, y + avp.y - avp.fontheight, (unsigned char*)str, black); +} + +void _transparent_bitmap_part(const void *src, int src_x, int src_y, + int stride, int x, int y, int width, int height) +{ + FILE *_image; + gdImagePtr image; + int pink; + + DEBUGF2("transparent_bitmap_part(const void *src=%s, int src_x=%d, int src_y=%d, int stride=%d, int x=%d, int y=%d, int width=%d, int height=%d", (char*)src, src_x, src_y, stride, x, y, width, height); + + _image = fopen(src, "rb"); + if(_image == NULL) + return; + + image = gdImageCreateFromBmp(_image); + fclose(_image); + + pink = gdTrueColor(255, 0, 255); + gdImageColorTransparent(image, pink); + + gdImageCopy(framebuffer, image, x, y, src_x, src_y, width, height); + + gdImageDestroy(image); +} + +void _bitmap_part(const void *src, int src_x, int src_y, + int stride, int x, int y, int width, int height) +{ + FILE *_image; + gdImagePtr image; + + DEBUGF2("bitmap_part(const void *src=%s, int src_x=%d, int src_y=%d, int stride=%d, int x=%d, int y=%d, int width=%d, int height=%d", (char*)src, src_x, src_y, stride, x, y, width, height); + + _image = fopen(src, "rb"); + if(_image == NULL) + return; + + image = gdImageCreateFromBmp(_image); + fclose(_image); + + gdImageCopy(framebuffer, image, x, y, src_x, src_y, width, height); + + gdImageDestroy(image); +} + +void _drawpixel(int x, int y) +{ + int black = gdImageColorAllocate(framebuffer, 0, 0, 0); + gdImageSetPixel(framebuffer, x, y, black); +} + +void _fillrect(int x, int y, int width, int height) +{ + /* Don't draw this as backdrop is used */ +#if 0 + int black = gdImageColorAllocate(framebuffer, 0, 0, 0); + gdImageFilledRectangle(framebuffer, x, y, x+width, y+height, black); +#endif +} + +void _hline(int x1, int x2, int y) +{ + int black = gdImageColorAllocate(framebuffer, 0, 0, 0); + gdImageLine(framebuffer, x1, y, x2, y, black); +} + +void _vline(int x, int y1, int y2) +{ + int black = gdImageColorAllocate(framebuffer, 0, 0, 0); + gdImageLine(framebuffer, x, y1, x, y2, black); +} + +void _clear_viewport(int x, int y, int w, int h, int color) +{ + if(backdrop == NULL) + return; + + gdImageCopy(framebuffer, backdrop, x, y, x, y, w, h); +} + +static bool _load_wps_backdrop(char* filename) +{ + FILE *image; + if(backdrop != NULL) + gdImageDestroy(backdrop); + + DEBUGF2("load backdrop: %s", filename); + + image = fopen(filename, "rb"); + if(image == NULL) + return false; + + backdrop = gdImageCreateFromBmp(image); + fclose(image); + + return true; +} + +int _read_bmp_file(const char* filename, int *width, int *height) +{ + FILE *_image; + gdImagePtr image; + + DEBUGF2("load backdrop: %s", filename); + + _image = fopen(filename, "rb"); + if(_image == NULL) + return 0; + + image = gdImageCreateFromBmp(_image); + fclose(_image); + + *width = image->sx; + *height = image->sy; + + gdImageDestroy(image); + + return 1; +} + +static void _drawBackdrop() +{ + if(backdrop == NULL) + return; + + gdImageCopy(framebuffer, backdrop, 0, 0, 0, 0, backdrop->sx, backdrop->sy); +} + +static int screenshot(char *model, char *wps, char *png) +{ + char lib[255]; + void *handle; + FILE *out, *in; + + in = fopen(wps, "rb"); + if(in == NULL) + { + fprintf(stderr, "[ERR] Cannot open WPS: %s\n", wps); + return -1; + } + fclose(in); + + out = fopen(png, "wb"); + if(out == NULL) + { + fprintf(stderr, "[ERR] Cannot open PNG: %s\n", png); + return -2; + } + + snprintf(lib, 255, "%s/libwps_%s.so", (char*)get_current_dir_name(), (char*)model); + handle = dlopen(lib, RTLD_LAZY); + if (!handle) + { + fprintf(stderr, "[ERR] Cannot open library: %s\n", dlerror()); + fclose(out); + return -3; + } + + wps_init = dlsym(handle, "wps_init"); + wps_display = dlsym(handle, "wps_display"); + wps_refresh = dlsym(handle, "wps_refresh"); + + if (!wps_init || !wps_display || !wps_refresh) + { + fprintf(stderr, "[ERR] Failed to resolve funcs!"); + dlclose(handle); + fclose(out); + return -4; + } + + memset(&api, 0, sizeof(struct proxy_api)); + + if(verbose) + api.verbose = 3; + else + api.verbose = 0; + + api.putsxy = &_putsxy; + api.transparent_bitmap_part = &_transparent_bitmap_part; + api.bitmap_part = &_bitmap_part; + api.drawpixel = &_drawpixel; + api.fillrect = &_fillrect; + api.hline = &_hline; + api.vline = &_vline; + api.clear_viewport = &_clear_viewport; + api.load_wps_backdrop = &_load_wps_backdrop; + api.read_bmp_file = &_read_bmp_file; + api.debugf = &_debug; + + wps_init(wps, &api, true); + + framebuffer = gdImageCreateTrueColor(api.getwidth(), api.getheight()); + + _drawBackdrop(); + + if(strcmp(api.get_model_name(), model) != 0) + { + fprintf(stderr, "[ERR] Model name doesn't match the one supplied by the library\n"); + fprintf(stderr, " %s <-> %s\n", model, api.get_model_name()); + dlclose(handle); + fclose(out); + gdImageDestroy(framebuffer); + gdImageDestroy(backdrop); + wps_init = NULL; + wps_display = NULL; + wps_refresh = NULL; + return -5; + } + fprintf(stdout, "[INFO] Model: %s\n", api.get_model_name()); + + wpsdata.fontheight = getFont()->h; + wpsdata.fontwidth = getFont()->w; + api.set_wpsstate(wpsdata); + api.set_trackstate(mp3data); + api.set_next_trackstate(mp3data); + + _drawBackdrop(); + wps_refresh(); + gdImagePng(framebuffer, out); + + fprintf(stdout, "[INFO] Image written\n"); + + dlclose(handle); + fclose(out); + gdImageDestroy(framebuffer); + gdImageDestroy(backdrop); + + wps_init = NULL; + wps_display = NULL; + wps_refresh = NULL; + + return 0; +} + +static void usage(void) +{ + fprintf(stderr, "Usage: screenshot [-V] <MODEL> <WPS_FILE> <OUT_PNG>\n"); + fprintf(stderr, "Example: screenshot h10_5gb iCatcher.wps out.png\n"); +} + +int main(int argc, char ** argv) +{ + if(argc < 4) + { + usage(); + return -1; + } + + if(strcmp(argv[1], "-V") == 0) + { + verbose = true; + return screenshot(argv[2], argv[3], argv[4]); + } + else + { + verbose = false; + return screenshot(argv[1], argv[2], argv[3]); + } + + return 0; +} |