Home | History | Annotate | Download | only in ps3
      1 /*
      2  * SDL - Simple DirectMedia Layer
      3  * CELL BE Support for PS3 Framebuffer
      4  * Copyright (C) 2008, 2009 International Business Machines Corporation
      5  *
      6  * This library is free software; you can redistribute it and/or modify it
      7  * under the terms of the GNU Lesser General Public License as published
      8  * by the Free Software Foundation; either version 2.1 of the License, or
      9  * (at your option) any later version.
     10  *
     11  * This library is distributed in the hope that it will be useful, but
     12  * WITHOUT ANY WARRANTY; without even the implied warranty of
     13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
     14  * Lesser General Public License for more details.
     15  *
     16  * You should have received a copy of the GNU Lesser General Public
     17  * License along with this library; if not, write to the Free Software
     18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301
     19  * USA
     20  *
     21  *  Martin Lowinski  <lowinski [at] de [dot] ibm [ibm] com>
     22  *  Dirk Herrendoerfer <d.herrendoerfer [at] de [dot] ibm [dot] com>
     23  *  SPE code based on research by:
     24  *  Rene Becker
     25  *  Thimo Emmerich
     26  */
     27 
     28 #include "SDL_config.h"
     29 
     30 #include "SDL_video.h"
     31 #include "SDL_ps3video.h"
     32 #include "SDL_ps3yuv_c.h"
     33 #include "../SDL_yuvfuncs.h"
     34 #include "spulibs/spu_common.h"
     35 
     36 /* Stores the executable name */
     37 extern spe_program_handle_t yuv2rgb_spu;
     38 extern spe_program_handle_t bilin_scaler_spu;
     39 
     40 int SPE_Start(_THIS, spu_data_t * spe_data);
     41 int SPE_Stop(_THIS, spu_data_t * spe_data);
     42 int SPE_Boot(_THIS, spu_data_t * spe_data);
     43 int SPE_Shutdown(_THIS, spu_data_t * spe_data);
     44 int SPE_SendMsg(_THIS, spu_data_t * spe_data, unsigned int msg);
     45 int SPE_WaitForMsg(_THIS, spu_data_t * spe_data, unsigned int msg);
     46 void SPE_RunContext(void *thread_argp);
     47 
     48 
     49 /* The functions used to manipulate software video overlays */
     50 static struct private_yuvhwfuncs ps3_yuvfuncs = {
     51   PS3_LockYUVOverlay,
     52   PS3_UnlockYUVOverlay,
     53   PS3_DisplayYUVOverlay,
     54   PS3_FreeYUVOverlay
     55 };
     56 
     57 
     58 struct private_yuvhwdata {
     59 	SDL_Surface *display;
     60 	SDL_Surface *stretch;
     61     volatile void * pixels __attribute__((aligned(128)));
     62 
     63 	/* These are just so we don't have to allocate them separately */
     64 	Uint16 pitches[3];
     65 	Uint8 * planes[3];
     66 
     67 	unsigned int scale;
     68 
     69 	/* Scaled YUV picture */
     70 	Uint8 * scaler_out __attribute__((aligned(128)));
     71 
     72 	/* YUV2RGB converter data */
     73     volatile struct yuv2rgb_parms_t * converter_parms __attribute__((aligned(128)));
     74 
     75 	/* Scaler data */
     76     volatile struct scale_parms_t * scaler_parms __attribute__((aligned(128)));
     77 
     78 	Uint8 locked;
     79 };
     80 
     81 
     82 SDL_Overlay *PS3_CreateYUVOverlay(_THIS, int width, int height, Uint32 format, SDL_Surface *display) {
     83 	/* Only RGB packed pixel conversion supported */
     84 	if ((display->format->BytesPerPixel != 2) &&
     85 			(display->format->BytesPerPixel != 3) &&
     86 			(display->format->BytesPerPixel != 4))
     87 	{
     88 		SDL_SetError ("Can't use YUV data on non 16/24/32 bit surfaces");
     89 		return NULL;
     90 	}
     91 
     92 	/* Double-check the requested format. We'll only support YV12 */
     93 	switch (format) {
     94 	    case SDL_IYUV_OVERLAY:
     95 		case SDL_YV12_OVERLAY:
     96 			/* Supported YUV format */
     97 			break;
     98 		default:
     99 			SDL_SetError("Unsupported YUV format");
    100 			return NULL;
    101 	}
    102 
    103 	SDL_Overlay* overlay;
    104 	struct private_yuvhwdata* hwdata;
    105 
    106 	/* Create the overlay structure */
    107 	overlay = (SDL_Overlay *) SDL_calloc(1, sizeof(SDL_Overlay));
    108 	if (overlay == NULL) {
    109 		SDL_OutOfMemory();
    110 		return NULL;
    111 	}
    112 	SDL_memset(overlay, 0, (sizeof *overlay));
    113 
    114 	/* Set the basic attributes */
    115 	overlay->format = format;
    116 	overlay->w = width;
    117 	overlay->h = height;
    118 	overlay->hwdata = NULL;
    119 
    120 	/* Set up the PS3 YUV surface function structure */
    121 	overlay->hwfuncs = &ps3_yuvfuncs;
    122 
    123 	/* Create the pixel data and lookup tables */
    124 	hwdata = (struct private_yuvhwdata *) SDL_calloc(1, sizeof(struct private_yuvhwdata));
    125 	if (hwdata == NULL) {
    126 		SDL_OutOfMemory();
    127 		SDL_FreeYUVOverlay(overlay);
    128 		return NULL;
    129 	}
    130 	overlay->hwdata = hwdata;
    131 
    132 	hwdata->stretch = NULL;
    133 	hwdata->display = display;
    134 
    135 	/* Create SPU parms structure */
    136 	hwdata->converter_parms = (struct yuv2rgb_parms_t *) memalign(16, sizeof(struct yuv2rgb_parms_t));
    137 	hwdata->scaler_parms = (struct scale_parms_t *) memalign(16, sizeof(struct scale_parms_t));
    138 	if (hwdata->converter_parms == NULL || hwdata->scaler_parms == NULL) {
    139 		SDL_FreeYUVOverlay(overlay);
    140 		SDL_OutOfMemory();
    141 		return(NULL);
    142 	}
    143 
    144 	/* Set up the SPEs */
    145 	scaler_thread_data = (spu_data_t *) malloc(sizeof(spu_data_t));
    146 	converter_thread_data = (spu_data_t *) malloc(sizeof(spu_data_t));
    147 	if (converter_thread_data == NULL || scaler_thread_data == NULL) {
    148 		SDL_FreeYUVOverlay(overlay);
    149 		SDL_OutOfMemory();
    150 		return(NULL);
    151 	}
    152 
    153 	scaler_thread_data->program = bilin_scaler_spu;
    154 	scaler_thread_data->program_name = "bilin_scaler_spu";
    155 	scaler_thread_data->keepalive = 0;
    156 	scaler_thread_data->booted = 0;
    157 
    158 	converter_thread_data->program = yuv2rgb_spu;
    159 	converter_thread_data->program_name = "yuv2rgb_spu";
    160 	converter_thread_data->keepalive = 1;
    161 	converter_thread_data->booted = 0;
    162 
    163 	SPE_Start(this, converter_thread_data);
    164 
    165 	hwdata->pixels = (Uint8 *) memalign(16, width * height + ((width * height) >> 1));
    166 	if (hwdata->pixels == NULL) {
    167 		SDL_FreeYUVOverlay(overlay);
    168 		SDL_OutOfMemory();
    169 		return(NULL);
    170 	}
    171 
    172 	/* Find the pitch and offset values for the overlay */
    173 	overlay->pitches = hwdata->pitches;
    174 	overlay->pixels = hwdata->planes;
    175 	switch (format) {
    176 	    case SDL_YV12_OVERLAY:
    177 	    case SDL_IYUV_OVERLAY:
    178 			overlay->pitches[0] = overlay->w;
    179 			overlay->pitches[1] = overlay->pitches[0] / 2;
    180 			overlay->pitches[2] = overlay->pitches[0] / 2;
    181 			overlay->pixels[0] = (Uint8 *)hwdata->pixels;
    182 			overlay->pixels[1] = overlay->pixels[0] +
    183 				overlay->pitches[0] * overlay->h;
    184 			overlay->pixels[2] = overlay->pixels[1] +
    185 				overlay->pitches[1] * overlay->h / 2;
    186 			overlay->planes = 3;
    187 		break;
    188 	    default:
    189 		/* We should never get here (caught above) */
    190 		break;
    191 	}
    192 
    193 	/* We're all done.. */
    194 	return overlay;
    195 }
    196 
    197 
    198 int PS3_LockYUVOverlay(_THIS, SDL_Overlay *overlay) {
    199 	if (overlay == NULL) {
    200 		return -1;
    201 	}
    202 	overlay->hwdata->locked = 1;
    203 
    204 	return 0;
    205 }
    206 
    207 
    208 void PS3_UnlockYUVOverlay(_THIS, SDL_Overlay *overlay) {
    209 	if (overlay == NULL) {
    210 		return;
    211 	}
    212 	overlay->hwdata->locked = 0;
    213 
    214 	return;
    215 }
    216 
    217 
    218 int PS3_DisplayYUVOverlay(_THIS, SDL_Overlay *overlay, SDL_Rect *src, SDL_Rect *dst) {
    219 	if ((overlay == NULL) || (overlay->hwdata == NULL)) {
    220 		return -1;
    221 	}
    222 
    223 	Uint8 *lum, *Cr, *Cb;
    224 	struct private_yuvhwdata *hwdata;
    225 	SDL_Surface *display;
    226 
    227 	hwdata = overlay->hwdata;
    228 	display = hwdata->display;
    229 
    230 	/* Do we have to scale? */
    231 	if ((src->w != dst->w) || (src->h != dst->h) ) {
    232 		hwdata->scale = 1;
    233 		deprintf(1, "[PS3] We need to scale\n");
    234 	} else {
    235 		hwdata->scale = 0;
    236 		deprintf(1, "[PS3] No scaling\n");
    237 	}
    238 
    239 	/* Find out where the various portions of the image are */
    240 	switch (overlay->format) {
    241 		case SDL_YV12_OVERLAY:
    242 			lum = (Uint8 *)overlay->pixels[0];
    243 			Cr =  (Uint8 *)overlay->pixels[1];
    244 			Cb =  (Uint8 *)overlay->pixels[2];
    245 			break;
    246 		case SDL_IYUV_OVERLAY:
    247 			lum = (Uint8 *)overlay->pixels[0];
    248 			Cr =  (Uint8 *)overlay->pixels[2];
    249 			Cb =  (Uint8 *)overlay->pixels[1];
    250 			break;
    251 		default:
    252 			SDL_SetError("Unsupported YUV format in blit");
    253 			return -1;
    254 	}
    255 
    256 	if (hwdata->scale) {
    257 		/* Alloc mem for scaled YUV picture */
    258 		hwdata->scaler_out = (Uint8 *) memalign(16, dst->w * dst->h + ((dst->w * dst->h) >> 1));
    259 		if (hwdata->scaler_out == NULL) {
    260 			SDL_FreeYUVOverlay(overlay);
    261 			SDL_OutOfMemory();
    262 			return -1;
    263 		}
    264 
    265 		/* Set parms for scaling */
    266 		hwdata->scaler_parms->src_pixel_width = src->w;
    267 		hwdata->scaler_parms->src_pixel_height = src->h;
    268 		hwdata->scaler_parms->dst_pixel_width = dst->w;
    269 		hwdata->scaler_parms->dst_pixel_height = dst->h;
    270 		hwdata->scaler_parms->y_plane = lum;
    271 		hwdata->scaler_parms->v_plane = Cr;
    272 		hwdata->scaler_parms->u_plane = Cb;
    273 		hwdata->scaler_parms->dstBuffer = hwdata->scaler_out;
    274 		scaler_thread_data->argp = (void *)hwdata->scaler_parms;
    275 
    276 		/* Scale the YUV overlay to given size */
    277 		SPE_Start(this, scaler_thread_data);
    278 		SPE_Stop(this, scaler_thread_data);
    279 
    280 		/* Set parms for converting after scaling */
    281 		hwdata->converter_parms->y_plane = hwdata->scaler_out;
    282 		hwdata->converter_parms->v_plane = hwdata->scaler_out + dst->w * dst->h;
    283 		hwdata->converter_parms->u_plane = hwdata->scaler_out + dst->w * dst->h + ((dst->w * dst->h) >> 2);
    284 	} else {
    285 		/* Set parms for converting */
    286 		hwdata->converter_parms->y_plane = lum;
    287 		hwdata->converter_parms->v_plane = Cr;
    288 		hwdata->converter_parms->u_plane = Cb;
    289 	}
    290 
    291 	hwdata->converter_parms->src_pixel_width = dst->w;
    292 	hwdata->converter_parms->src_pixel_height = dst->h;
    293 	hwdata->converter_parms->dstBuffer = (Uint8 *) s_pixels;
    294 	converter_thread_data->argp = (void *)hwdata->converter_parms;
    295 
    296 	/* Convert YUV overlay to RGB */
    297 	SPE_SendMsg(this, converter_thread_data, SPU_START);
    298 	SPE_SendMsg(this, converter_thread_data, (unsigned int)converter_thread_data->argp);
    299 
    300 	/* Centering */
    301 	s_bounded_input_width = dst->w;
    302 	s_bounded_input_height = dst->h;
    303 
    304 	/* UpdateRects() will do the rest.. */
    305 	SDL_UpdateRects(display, 1, dst);
    306 
    307 	if (hwdata->scale)
    308 		SDL_free((void *)hwdata->scaler_out);
    309 
    310 	return 0;
    311 }
    312 
    313 
    314 void PS3_FreeYUVOverlay(_THIS, SDL_Overlay *overlay) {
    315 	if (overlay == NULL) {
    316 		return;
    317 	}
    318 
    319 	if (overlay->hwdata == NULL) {
    320 		return;
    321 	}
    322 
    323 	struct private_yuvhwdata * hwdata;
    324 	hwdata = overlay->hwdata;
    325 
    326 	if (scaler_thread_data)
    327 		SDL_free(scaler_thread_data);
    328 	if (converter_thread_data) {
    329 		SPE_Shutdown(this, converter_thread_data);
    330 		SDL_free(converter_thread_data);
    331 	}
    332 
    333 	if (hwdata) {
    334 		if (hwdata->pixels)
    335 			SDL_free((void *)hwdata->pixels);
    336 		SDL_free(hwdata);
    337 	}
    338 	return;
    339 }
    340 
    341