RetroArch/tasks/task_file_transfer.c
2015-07-09 00:24:10 +02:00

684 lines
16 KiB
C

/* RetroArch - A frontend for libretro.
* Copyright (C) 2011-2015 - Daniel De Matteis
*
* RetroArch 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 Found-
* ation, either version 3 of the License, or (at your option) any later version.
*
* RetroArch is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
* without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
* PURPOSE. See the GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License along with RetroArch.
* If not, see <http://www.gnu.org/licenses/>.
*/
#include <string.h>
#include <file/nbio.h>
#include <formats/image.h>
#include <formats/rpng.h>
#include <compat/strl.h>
#include <retro_miscellaneous.h>
#include <queues/message_queue.h>
#include <string/string_list.h>
#include <rhash.h>
#include "tasks.h"
#define CB_MENU_WALLPAPER 0xb476e505U
#define CB_MENU_BOXART 0x68b307cdU
enum nbio_image_status_enum
{
NBIO_IMAGE_STATUS_POLL = 0,
NBIO_IMAGE_STATUS_TRANSFER,
NBIO_IMAGE_STATUS_TRANSFER_PARSE,
NBIO_IMAGE_STATUS_PROCESS_TRANSFER,
NBIO_IMAGE_STATUS_PROCESS_TRANSFER_PARSE,
NBIO_IMAGE_STATUS_TRANSFER_PARSE_FREE
};
enum nbio_status_enum
{
NBIO_STATUS_POLL = 0,
NBIO_STATUS_TRANSFER,
NBIO_STATUS_TRANSFER_PARSE,
NBIO_STATUS_TRANSFER_PARSE_FREE
};
typedef struct nbio_image_handle
{
struct texture_image ti;
bool is_blocking;
bool is_blocking_on_processing;
bool is_finished;
transfer_cb_t cb;
#ifdef HAVE_RPNG
struct rpng_t *handle;
#endif
unsigned processing_pos_increment;
unsigned pos_increment;
uint64_t frame_count;
uint64_t processing_frame_count;
int processing_final_state;
msg_queue_t *msg_queue;
unsigned status;
} nbio_image_handle_t;
typedef struct nbio_handle
{
nbio_image_handle_t image;
bool is_finished;
transfer_cb_t cb;
struct nbio_t *handle;
unsigned pos_increment;
uint64_t frame_count;
msg_queue_t *msg_queue;
unsigned status;
} nbio_handle_t;
static nbio_handle_t *nbio_ptr;
void *rarch_main_data_nbio_get_ptr(void)
{
return nbio_ptr;
}
msg_queue_t *rarch_main_data_nbio_get_msg_queue_ptr(void)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
if (!nbio)
return NULL;
return nbio->msg_queue;
}
void *rarch_main_data_nbio_get_handle(void)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
if (!nbio)
return NULL;
return nbio->handle;
}
msg_queue_t *rarch_main_data_nbio_image_get_msg_queue_ptr(void)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
if (!nbio)
return NULL;
#ifdef HAVE_RPNG
return nbio->image.msg_queue;
#else
return NULL;
#endif
}
void *rarch_main_data_nbio_image_get_handle(void)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
if (!nbio)
return NULL;
#ifdef HAVE_RPNG
return nbio->image.handle;
#else
return NULL;
#endif
}
#ifdef HAVE_MENU
#include "../menu/menu_driver.h"
#ifdef HAVE_RPNG
static int cb_image_menu_upload_generic(nbio_handle_t *nbio)
{
unsigned r_shift, g_shift, b_shift, a_shift;
if (!nbio)
return -1;
if (nbio->image.processing_final_state == IMAGE_PROCESS_ERROR ||
nbio->image.processing_final_state == IMAGE_PROCESS_ERROR_END)
return -1;
texture_image_set_color_shifts(&r_shift, &g_shift, &b_shift,
&a_shift);
texture_image_color_convert(r_shift, g_shift, b_shift,
a_shift, &nbio->image.ti);
nbio->image.is_blocking_on_processing = false;
nbio->image.is_blocking = true;
nbio->image.is_finished = true;
nbio->is_finished = true;
return 0;
}
static int cb_image_menu_wallpaper_upload(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (cb_image_menu_upload_generic(nbio) != 0)
return -1;
menu_driver_load_image(&nbio->image.ti, MENU_IMAGE_WALLPAPER);
texture_image_free(&nbio->image.ti);
return 0;
}
static int cb_image_menu_boxart_upload(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (cb_image_menu_upload_generic(nbio) != 0)
return -1;
menu_driver_load_image(&nbio->image.ti, MENU_IMAGE_BOXART);
texture_image_free(&nbio->image.ti);
return 0;
}
static int cb_image_menu_generic(nbio_handle_t *nbio)
{
unsigned width, height;
int retval;
if (!nbio)
return -1;
if ( !nbio->image.handle->has_ihdr ||
!nbio->image.handle->has_idat ||
!nbio->image.handle->has_iend)
return -1;
retval = rpng_nbio_load_image_argb_process(nbio->image.handle,
&nbio->image.ti.pixels, &width, &height);
nbio->image.ti.width = width;
nbio->image.ti.height = height;
if (retval == IMAGE_PROCESS_ERROR || retval == IMAGE_PROCESS_ERROR_END)
return -1;
nbio->image.is_blocking_on_processing = true;
nbio->image.is_finished = false;
return 0;
}
static int cb_image_menu_wallpaper(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (cb_image_menu_generic(nbio) != 0)
return -1;
nbio->image.cb = &cb_image_menu_wallpaper_upload;
return 0;
}
static int cb_image_menu_boxart(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (cb_image_menu_generic(nbio) != 0)
return -1;
nbio->image.cb = &cb_image_menu_boxart_upload;
return 0;
}
static int rarch_main_data_image_iterate_poll(nbio_handle_t *nbio)
{
const char *path = NULL;
if (!nbio)
return -1;
path = msg_queue_pull(nbio->image.msg_queue);
if (!path)
return -1;
/* Can only deal with one image transfer at a time for now */
if (nbio->image.handle)
return -1;
/* We need to load the image file first. */
msg_queue_clear(nbio->msg_queue);
msg_queue_push(nbio->msg_queue, path, 0, 1);
return 0;
}
static int rarch_main_data_image_iterate_transfer(nbio_handle_t *nbio)
{
unsigned i;
if (!nbio)
return -1;
if (nbio->image.is_finished)
return 0;
for (i = 0; i < nbio->image.pos_increment; i++)
{
unsigned ret;
if (!rpng_nbio_load_image_argb_iterate(
nbio->image.handle->buff_data,
nbio->image.handle, &ret))
goto error;
nbio->image.handle->buff_data += ret;
}
nbio->image.frame_count++;
return 0;
error:
return -1;
}
static int rarch_main_data_image_iterate_process_transfer(nbio_handle_t *nbio)
{
unsigned i, width, height;
int retval = 0;
if (!nbio)
return -1;
for (i = 0; i < nbio->image.processing_pos_increment; i++)
{
retval = rpng_nbio_load_image_argb_process(nbio->image.handle,
&nbio->image.ti.pixels, &width, &height);
nbio->image.ti.width = width;
nbio->image.ti.height = height;
if (retval != IMAGE_PROCESS_NEXT)
break;
}
nbio->image.processing_frame_count++;
if (retval == IMAGE_PROCESS_NEXT)
return 0;
nbio->image.processing_final_state = retval;
return -1;
}
static int rarch_main_data_image_iterate_parse_free(nbio_handle_t *nbio)
{
if (!nbio)
return -1;
rpng_nbio_load_image_free(nbio->image.handle);
nbio->image.handle = NULL;
nbio->image.frame_count = 0;
nbio->image.processing_frame_count = 0;
msg_queue_clear(nbio->image.msg_queue);
return 0;
}
static int rarch_main_data_image_iterate_process_transfer_parse(nbio_handle_t *nbio)
{
size_t len = 0;
if (nbio->image.handle && nbio->image.cb)
nbio->image.cb(nbio, len);
return 0;
}
static int rarch_main_data_image_iterate_transfer_parse(nbio_handle_t *nbio)
{
size_t len = 0;
if (nbio->image.handle && nbio->image.cb)
nbio->image.cb(nbio, len);
return 0;
}
void rarch_main_data_nbio_image_iterate(bool is_thread)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
nbio_image_handle_t *image = nbio ? &nbio->image : NULL;
if (!image || !nbio)
return;
(void)is_thread;
switch (image->status)
{
case NBIO_IMAGE_STATUS_PROCESS_TRANSFER:
if (rarch_main_data_image_iterate_process_transfer(nbio) == -1)
image->status = NBIO_IMAGE_STATUS_PROCESS_TRANSFER_PARSE;
break;
case NBIO_IMAGE_STATUS_TRANSFER_PARSE:
rarch_main_data_image_iterate_transfer_parse(nbio);
if (image->is_blocking_on_processing)
image->status = NBIO_IMAGE_STATUS_PROCESS_TRANSFER;
break;
case NBIO_IMAGE_STATUS_TRANSFER:
if (!image->is_blocking)
if (rarch_main_data_image_iterate_transfer(nbio) == -1)
image->status = NBIO_IMAGE_STATUS_TRANSFER_PARSE;
break;
default:
case NBIO_IMAGE_STATUS_POLL:
if (rarch_main_data_image_iterate_poll(nbio) == 0)
image->status = NBIO_IMAGE_STATUS_TRANSFER;
break;
}
}
void rarch_main_data_nbio_image_upload_iterate(bool is_thread)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
nbio_image_handle_t *image = nbio ? &nbio->image : NULL;
if (!image || !nbio)
return;
(void)is_thread;
switch (image->status)
{
case NBIO_IMAGE_STATUS_PROCESS_TRANSFER_PARSE:
rarch_main_data_image_iterate_process_transfer_parse(nbio);
if (image->is_finished)
image->status = NBIO_IMAGE_STATUS_TRANSFER_PARSE_FREE;
break;
case NBIO_IMAGE_STATUS_TRANSFER_PARSE_FREE:
rarch_main_data_image_iterate_parse_free(nbio);
image->status = NBIO_IMAGE_STATUS_POLL;
break;
}
}
#endif
#endif
static int cb_nbio_default(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (!data)
return -1;
(void)len;
nbio->is_finished = true;
return 0;
}
#ifdef HAVE_RPNG
static int cb_nbio_generic(nbio_handle_t *nbio, size_t *len)
{
void *ptr = NULL;
if (!nbio->image.handle)
{
nbio->image.cb = NULL;
return -1;
}
ptr = nbio_get_ptr(nbio->handle, len);
if (!ptr)
{
rpng_nbio_load_image_free(nbio->image.handle);
nbio->image.handle = NULL;
nbio->image.cb = NULL;
return -1;
}
nbio->image.handle->buff_data = (uint8_t*)ptr;
nbio->image.pos_increment = (*len / 2) ? (*len / 2) : 1;
nbio->image.processing_pos_increment = (*len / 4) ? (*len / 4) : 1;
if (!rpng_nbio_load_image_argb_start(nbio->image.handle))
{
rpng_nbio_load_image_free(nbio->image.handle);
return -1;
}
nbio->image.is_blocking = false;
nbio->image.is_finished = false;
nbio->is_finished = true;
return 0;
}
static int cb_nbio_image_menu_wallpaper(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (!nbio || !data)
return -1;
nbio->image.handle = (struct rpng_t*)calloc(1, sizeof(struct rpng_t));
nbio->image.cb = &cb_image_menu_wallpaper;
return cb_nbio_generic(nbio, &len);
}
static int cb_nbio_image_menu_boxart(void *data, size_t len)
{
nbio_handle_t *nbio = (nbio_handle_t*)data;
if (!nbio || !data)
return -1;
nbio->image.handle = (struct rpng_t*)calloc(1, sizeof(struct rpng_t));
nbio->image.cb = &cb_image_menu_boxart;
return cb_nbio_generic(nbio, &len);
}
#endif
static int rarch_main_data_nbio_iterate_poll(nbio_handle_t *nbio)
{
char elem0[PATH_MAX_LENGTH] = {0};
uint32_t cb_type_hash = 0;
struct nbio_t* handle = NULL;
struct string_list *str_list = NULL;
const char *path = NULL;
if (!nbio)
return -1;
path = msg_queue_pull(nbio->msg_queue);
if (!path)
return -1;
/* Can only deal with one NBIO transfer at a time for now */
if (nbio->handle)
return -1;
str_list = string_split(path, "|");
if (!str_list)
goto error;
if (str_list->size > 0)
{
unsigned elem0_hash = 0;
strlcpy(elem0, str_list->elems[0].data, sizeof(elem0));
elem0_hash = djb2_calculate(elem0);
/* TODO/FIXME - should be able to deal with this
* in a better way. */
switch(elem0_hash)
{
case CB_MENU_WALLPAPER:
case CB_MENU_BOXART:
goto error;
default:
break;
}
}
if (str_list->size > 1)
cb_type_hash = djb2_calculate(str_list->elems[1].data);
handle = nbio_open(elem0, NBIO_READ);
if (!handle)
{
RARCH_ERR("Could not create new file loading handle.\n");
goto error;
}
nbio->handle = handle;
nbio->is_finished = false;
nbio->cb = &cb_nbio_default;
switch (cb_type_hash)
{
#if defined(HAVE_MENU) && defined(HAVE_RPNG)
case CB_MENU_WALLPAPER:
nbio->cb = &cb_nbio_image_menu_wallpaper;
break;
case CB_MENU_BOXART:
nbio->cb = &cb_nbio_image_menu_boxart;
break;
#endif
}
nbio_begin_read(handle);
string_list_free(str_list);
return 0;
error:
if (str_list)
string_list_free(str_list);
return -1;
}
static int rarch_main_data_nbio_iterate_transfer(nbio_handle_t *nbio)
{
size_t i;
if (!nbio)
return -1;
nbio->pos_increment = 5;
if (nbio->is_finished)
return 0;
for (i = 0; i < nbio->pos_increment; i++)
{
if (nbio_iterate(nbio->handle))
return -1;
}
nbio->frame_count++;
return 0;
}
static int rarch_main_data_nbio_iterate_parse_free(nbio_handle_t *nbio)
{
if (!nbio || !nbio->is_finished)
return -1;
nbio_free(nbio->handle);
nbio->handle = NULL;
nbio->is_finished = false;
nbio->frame_count = 0;
msg_queue_clear(nbio->msg_queue);
return 0;
}
static int rarch_main_data_nbio_iterate_parse(nbio_handle_t *nbio)
{
int len = 0;
if (!nbio)
return -1;
if (nbio->cb)
nbio->cb(nbio, len);
return 0;
}
void rarch_main_data_nbio_iterate(bool is_thread)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
if (!nbio)
return;
switch (nbio->status)
{
case NBIO_STATUS_TRANSFER_PARSE:
rarch_main_data_nbio_iterate_parse(nbio);
nbio->status = NBIO_STATUS_TRANSFER_PARSE_FREE;
break;
case NBIO_STATUS_TRANSFER:
if (rarch_main_data_nbio_iterate_transfer(nbio) == -1)
nbio->status = NBIO_STATUS_TRANSFER_PARSE;
break;
case NBIO_STATUS_TRANSFER_PARSE_FREE:
rarch_main_data_nbio_iterate_parse_free(nbio);
nbio->status = NBIO_STATUS_POLL;
break;
case NBIO_STATUS_POLL:
default:
if (rarch_main_data_nbio_iterate_poll(nbio) == 0)
nbio->status = NBIO_STATUS_TRANSFER;
break;
}
}
void rarch_main_data_nbio_init_msg_queue(void)
{
nbio_handle_t *nbio = (nbio_handle_t*)
rarch_main_data_nbio_get_ptr();
if (!nbio)
return;
if (!nbio->msg_queue)
rarch_assert(nbio->msg_queue = msg_queue_new(8));
if (!nbio->image.msg_queue)
rarch_assert(nbio->image.msg_queue = msg_queue_new(8));
}
void rarch_main_data_nbio_uninit(void)
{
if (nbio_ptr)
free(nbio_ptr);
nbio_ptr = NULL;
}
void rarch_main_data_nbio_init(void)
{
nbio_ptr = (nbio_handle_t*)calloc(1, sizeof(*nbio_ptr));
}