vertex_array.c

   1 /*-
   2  * Copyright (C) 2011, 2012 glevand <geoffrey.levand@mail.ru>
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer,
  10  *    without modification, immediately at the beginning of the file.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  *
  26  * $FreeBSD$
  27  */
  28
  29 #include <stdio.h>
  30 #include <stdlib.h>
  31 #include <stdint.h>
  32 #include <stddef.h>
  33 #include <string.h>
  34
  35 #include <sys/types.h>
  36 #include <sys/stat.h>
  37 #include <sys/uio.h>
  38 #include <sys/ioctl.h>
  39 #include <sys/mman.h>
  40 #include <sys/fbio.h>
  41 #include <sys/consio.h>
  42 #include <fcntl.h>
  43 #include <unistd.h>
  44
  45 #include "ps3gpu_ctl.h"
  46 #include "ps3gpu_mth.h"
  47 #include "reset_gpu_state.h"
  48 #include "util.h"
  49
  50 struct vertex {
  51         float x, y, z;
  52         uint32_t rgba;
  53 };
  54
  55 int
  56 main(int argc, char **argv)
  57 {
  58         struct ps3gpu_ctl_context_allocate context_allocate;
  59         struct ps3gpu_ctl_context_free context_free;
  60         struct ps3gpu_ctl_memory_allocate memory_allocate;
  61         struct ps3gpu_ctl_setup_control setup_control;
  62         struct ps3gpu_ctl_display_buffer_set display_buffer_set;
  63         int context_id;
  64         volatile uint32_t *control;
  65         volatile uint8_t *driver_info;
  66         uint32_t *fifo, *reset_gpu, *db[2], *zb, *fp, *verts;
  67         unsigned long fifo_handle, db_handle[2], zb_handle, fp_handle, verts_handle;
  68         unsigned int fifo_gaddr, reset_gpu_gaddr, db_gaddr[2], zb_gaddr, fp_gaddr, verts_gaddr;
  69         int fd = -1;
  70         int i, err;
  71
  72         /* Open GPU device */
  73
  74         fd = open(PS3GPU_DEV_PATH, O_RDWR);
  75         if (fd < 0) {
  76                 perror("open");
  77                 goto done;
  78         }
  79
  80         /* Create GPU context */
  81
  82         context_allocate.vram_size = 64; /* MB */
  83
  84         err = ioctl(fd, PS3GPU_CTL_CONTEXT_ALLOCATE, &context_allocate);
  85         if (err < 0) {
  86                 perror("ioctl");
  87                 goto done;
  88         }
  89
  90         context_id = context_allocate.context_id;
  91
  92         printf("context id %d\n", context_id);
  93         printf("control handle 0x%lx size %d\n",
  94             context_allocate.control_handle, context_allocate.control_size);
  95         printf("driver_info handle 0x%lx size %d\n",
  96             context_allocate.driver_info_handle, context_allocate.driver_info_size);
  97
  98         /* Map control registers */
  99
 100         control = mmap(NULL, context_allocate.control_size,
 101             PROT_READ | PROT_WRITE, MAP_SHARED, fd, context_allocate.control_handle);
 102         if (control == (void *) MAP_FAILED) {
 103                 perror("mmap");
 104                 goto done;
 105         }
 106
 107         /* Map driver info */
 108
 109         driver_info = mmap(NULL, context_allocate.driver_info_size,
 110             PROT_READ | PROT_WRITE, MAP_SHARED, fd, context_allocate.driver_info_handle);
 111         if (driver_info == (void *) MAP_FAILED) {
 112                 perror("mmap");
 113                 goto done;
 114         }
 115
 116         printf("channel id %d\n", get_channel_id(driver_info));
 117
 118         /* Allocate FIFO */
 119
 120         memory_allocate.context_id = context_id;
 121         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_GART;
 122         memory_allocate.size = 64 * 1024;
 123         memory_allocate.align = 12;
 124
 125         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 126         if (err < 0) {
 127                 perror("ioctl");
 128                 goto done;
 129         }
 130
 131         fifo_handle = memory_allocate.handle;
 132         fifo_gaddr = memory_allocate.gpu_addr;
 133
 134         printf("fifo handle 0x%lx gpu addr 0x%08x\n",
 135             fifo_handle, fifo_gaddr);
 136
 137         /* Map FIFO */
 138
 139         fifo = mmap(NULL, memory_allocate.size,
 140             PROT_READ | PROT_WRITE, MAP_SHARED, fd, fifo_handle);
 141         if (fifo == (void *) MAP_FAILED) {
 142                 perror("mmap");
 143                 goto done;
 144         }
 145
 146         /* Setup FIFO */
 147
 148         setup_control.context_id = context_id;
 149         setup_control.put = fifo_handle;
 150         setup_control.get = fifo_handle;
 151         setup_control.ref = 0xdeadbabe;
 152
 153         err = ioctl(fd, PS3GPU_CTL_SETUP_CONTROL, &setup_control);
 154         if (err < 0) {
 155                 perror("ioctl");
 156                 goto done;
 157         }
 158
 159         printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 160             control[0x10], control[0x11], control[0x12]);
 161
 162         /* Allocate FIFO for resetting GPU state */
 163
 164         memory_allocate.context_id = context_id;
 165         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_GART;
 166         memory_allocate.size = 4 * 1024;
 167         memory_allocate.align = 12;
 168
 169         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 170         if (err < 0) {
 171                 perror("ioctl");
 172                 goto done;
 173         }
 174
 175         reset_gpu_gaddr = memory_allocate.gpu_addr;
 176
 177         printf("reset GPU state handle 0x%lx gpu addr 0x%08x\n",
 178             memory_allocate.handle, reset_gpu_gaddr);
 179
 180         /* Map FIFO for resetting GPU state */
 181
 182         reset_gpu = mmap(NULL, memory_allocate.size,
 183             PROT_READ | PROT_WRITE, MAP_SHARED, fd, memory_allocate.handle);
 184         if (reset_gpu == (void *) MAP_FAILED) {
 185                 perror("mmap");
 186                 goto done;
 187         }
 188
 189         memcpy(reset_gpu, reset_gpu_state_3d, reset_gpu_state_3d_size);
 190
 191         /* Kick FIFO */
 192
 193         fifo[0] = PS3GPU_MTH_HDR(0, 0, reset_gpu_gaddr | PS3GPU_MTH_ADDR_CALL);
 194         fifo[1] = PS3GPU_MTH_HDR(1, 0, PS3GPU_MTH_ADDR_REF);
 195         fifo[2] = 0xcafef00d;
 196
 197         control[0x10] = fifo_gaddr + 3 * sizeof(uint32_t);
 198
 199         err = wait_fifo_idle(control);
 200         if (err < 0) {
 201                 fprintf(stderr, "FIFO timeout: put 0x%08x get 0x%08x ref 0x%08x\n",
 202                     control[0x10], control[0x11], control[0x12]);
 203                 dump_fifo(stderr, fifo, 0x400);
 204                 goto done;
 205         }
 206
 207         printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 208             control[0x10], control[0x11], control[0x12]);
 209
 210         /* Allocate display buffers */
 211
 212         memory_allocate.context_id = context_id;
 213         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_VIDEO;
 214         memory_allocate.size = ROUNDUP(DISPLAY_HEIGHT * DISPLAY_PITCH, 4 * 1024);
 215         memory_allocate.align = 12;
 216
 217         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 218         if (err < 0) {
 219                 perror("ioctl");
 220                 goto done;
 221         }
 222
 223         db_handle[0] = memory_allocate.handle;
 224         db_gaddr[0] = memory_allocate.gpu_addr;
 225
 226         printf("DB0 handle 0x%lx gpu addr 0x%08x\n",
 227             db_handle[0], db_gaddr[0]);
 228
 229         db[0] = mmap(NULL, memory_allocate.size,
 230             PROT_READ | PROT_WRITE, MAP_SHARED, fd, db_handle[0]);
 231         if (db[0] == (void *) MAP_FAILED) {
 232                 perror("mmap");
 233                 goto done;
 234         }
 235
 236         memory_allocate.context_id = context_id;
 237         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_VIDEO;
 238         memory_allocate.size = ROUNDUP(DISPLAY_HEIGHT * DISPLAY_PITCH, 4 * 1024);
 239         memory_allocate.align = 12;
 240
 241         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 242         if (err < 0) {
 243                 perror("ioctl");
 244                 goto done;
 245         }
 246
 247         db_handle[1] = memory_allocate.handle;
 248         db_gaddr[1] = memory_allocate.gpu_addr;
 249
 250         printf("DB1 handle 0x%lx gpu addr 0x%08x\n",
 251             db_handle[1], db_gaddr[1]);
 252
 253         db[1] = mmap(NULL, memory_allocate.size,
 254             PROT_READ | PROT_WRITE, MAP_SHARED, fd, db_handle[1]);
 255         if (db[1] == (void *) MAP_FAILED) {
 256                 perror("mmap");
 257                 goto done;
 258         }
 259
 260         /* Allocate depth buffer */
 261
 262         memory_allocate.context_id = context_id;
 263         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_VIDEO;
 264         memory_allocate.size = ROUNDUP(DISPLAY_HEIGHT * DISPLAY_PITCH, 4 * 1024);
 265         memory_allocate.align = 12;
 266
 267         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 268         if (err < 0) {
 269                 perror("ioctl");
 270                 goto done;
 271         }
 272
 273         zb_handle = memory_allocate.handle;
 274         zb_gaddr = memory_allocate.gpu_addr;
 275
 276         printf("ZB handle 0x%lx gpu addr 0x%08x\n",
 277             zb_handle, zb_gaddr);
 278
 279         zb = mmap(NULL, memory_allocate.size,
 280             PROT_READ | PROT_WRITE, MAP_SHARED, fd, zb_handle);
 281         if (zb == (void *) MAP_FAILED) {
 282                 perror("mmap");
 283                 goto done;
 284         }
 285
 286         /* Allocate fragment program */
 287
 288         memory_allocate.context_id = context_id;
 289         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_VIDEO;
 290         memory_allocate.size = 4 * 1024;
 291         memory_allocate.align = 12;
 292
 293         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 294         if (err < 0) {
 295                 perror("ioctl");
 296                 goto done;
 297         }
 298
 299         fp_handle = memory_allocate.handle;
 300         fp_gaddr = memory_allocate.gpu_addr;
 301
 302         printf("FP handle 0x%lx gpu addr 0x%08x\n",
 303             fp_handle, fp_gaddr);
 304
 305         fp = mmap(NULL, memory_allocate.size,
 306             PROT_READ | PROT_WRITE, MAP_SHARED, fd, fp_handle);
 307         if (fp == (void *) MAP_FAILED) {
 308                 perror("mmap");
 309                 goto done;
 310         }
 311
 312         /* Allocate vertices */
 313
 314         memory_allocate.context_id = context_id;
 315         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_VIDEO;
 316         memory_allocate.size = 4 * 1024;
 317         memory_allocate.align = 12;
 318
 319         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 320         if (err < 0) {
 321                 perror("ioctl");
 322                 goto done;
 323         }
 324
 325         verts_handle = memory_allocate.handle;
 326         verts_gaddr = memory_allocate.gpu_addr;
 327
 328         printf("VERTS handle 0x%lx gpu addr 0x%08x\n",
 329             verts_handle, verts_gaddr);
 330
 331         verts = mmap(NULL, memory_allocate.size,
 332             PROT_READ | PROT_WRITE, MAP_SHARED, fd, verts_handle);
 333         if (verts == (void *) MAP_FAILED) {
 334                 perror("mmap");
 335                 goto done;
 336         }
 337
 338         /* Set display buffers */
 339
 340         display_buffer_set.context_id = context_id;
 341         display_buffer_set.buffer_id = 0;
 342         display_buffer_set.width = DISPLAY_WIDTH;
 343         display_buffer_set.height = DISPLAY_HEIGHT;
 344         display_buffer_set.pitch = DISPLAY_PITCH;
 345         display_buffer_set.offset = db_handle[0];
 346
 347         err = ioctl(fd, PS3GPU_CTL_DISPLAY_BUFFER_SET, &display_buffer_set);
 348         if (err < 0) {
 349                 perror("ioctl");
 350                 goto done;
 351         }
 352
 353         display_buffer_set.context_id = context_id;
 354         display_buffer_set.buffer_id = 1;
 355         display_buffer_set.width = DISPLAY_WIDTH;
 356         display_buffer_set.height = DISPLAY_HEIGHT;
 357         display_buffer_set.pitch = DISPLAY_PITCH;
 358         display_buffer_set.offset = db_handle[1];
 359
 360         err = ioctl(fd, PS3GPU_CTL_DISPLAY_BUFFER_SET, &display_buffer_set);
 361         if (err < 0) {
 362                 perror("ioctl");
 363                 goto done;
 364         }
 365
 366         const struct surface_desc surf_desc[] = {
 367                 /* display buffer 0 */
 368                 {
 369                         .sd_color_loc = { 0xfeed0000, 0xfeed0000, 0xfeed0000, 0xfeed0000 },
 370                         .sd_color_off = { db_gaddr[0], 0, 0, 0 },
 371                         .sd_color_pitch = { DISPLAY_PITCH, 64, 64, 64 },
 372                         .sd_color_fmt = 0x8,
 373                         .sd_color_target = 0x1,
 374                         .sd_depth_loc = 0xfeed0000,
 375                         .sd_depth_off = zb_gaddr,
 376                         .sd_depth_pitch = DISPLAY_PITCH,
 377                         .sd_depth_fmt = 0x2,
 378                         .sd_x = 0,
 379                         .sd_y = 0,
 380                         .sd_w = DISPLAY_WIDTH,
 381                         .sd_h = DISPLAY_HEIGHT,
 382                 },
 383                 /* display buffer 1 */
 384                 {
 385                         .sd_color_loc = { 0xfeed0000, 0xfeed0000, 0xfeed0000, 0xfeed0000 },
 386                         .sd_color_off = { db_gaddr[1], 0, 0, 0 },
 387                         .sd_color_pitch = { DISPLAY_PITCH, 64, 64, 64 },
 388                         .sd_color_fmt = 0x8,
 389                         .sd_color_target = 0x1,
 390                         .sd_depth_loc = 0xfeed0000,
 391                         .sd_depth_off = zb_gaddr,
 392                         .sd_depth_pitch = DISPLAY_PITCH,
 393                         .sd_depth_fmt = 0x2,
 394                         .sd_x = 0,
 395                         .sd_y = 0,
 396                         .sd_w = DISPLAY_WIDTH,
 397                         .sd_h = DISPLAY_HEIGHT,
 398                 },
 399         };
 400
 401         const uint32_t clear_color[] = {
 402                 0xff404040,
 403                 0xffffffff,
 404         };
 405
 406         const float vp_offset[] = { DISPLAY_WIDTH * 0.5f, DISPLAY_HEIGHT * 0.5f, 0.5f, 0.0f };
 407         const float vp_scale[] = { DISPLAY_WIDTH * 0.5f, DISPLAY_HEIGHT * 0.5f, 0.5f, 0.0f };
 408
 409         const uint32_t vertex_prg[] = {
 410                 /* MOV o[0], v[0] */
 411                 0x401f9c6c, 0x0040000d, 0x8106c083, 0x6041ff80,
 412                 /* MOV o[1], v[3] */
 413                 0x401f9c6c, 0x0040030d, 0x8106c083, 0x6041ff85,
 414         };
 415
 416         uint32_t frag_prg[] = {
 417                 /* MOVR R0, f[1] */
 418                 0x01003e01, 0x1c9dc801, 0x0001c800, 0x3fe1c800,
 419         };
 420
 421         /*
 422          *            (0.0, -0.5)
 423          *                 /\
 424          *                /  \
 425          *               /    \
 426          *              /      \
 427          * (-0.5, 0.5) /________\ (0.5, 0.5)
 428          *
 429          */
 430
 431         const struct vertex triangle_verts[] = {
 432                 {  0.0f, -0.5f, -1.0f, 0xff0000ff },
 433                 { -0.5f,  0.5f, -1.0f, 0x00ff00ff },
 434                 {  0.5f,  0.5f, -1.0f, 0x0000ffff },
 435         };
 436
 437         /* Swap half-words in fragment program */
 438
 439         for (i = 0; i < ARRAY_SIZE(frag_prg); i++)
 440                 frag_prg[i] = (frag_prg[i] << 16) | (frag_prg[i] >> 16);
 441
 442         for (i = 0; i < ARRAY_SIZE(surf_desc); i++) {
 443                 setup_control.context_id = context_id;
 444                 setup_control.put = fifo_handle;
 445                 setup_control.get = fifo_handle;
 446                 setup_control.ref = 0xdeadbabe;
 447
 448                 err = ioctl(fd, PS3GPU_CTL_SETUP_CONTROL, &setup_control);
 449                 if (err < 0) {
 450                         perror("ioctl");
 451                         goto done;
 452                 }
 453
 454                 printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 455                     control[0x10], control[0x11], control[0x12]);
 456
 457                 err += set_surface(fifo + err, &surf_desc[i]);
 458                 err += set_depth_mask(fifo + err, 0x00000000);
 459                 err += set_color_mask(fifo + err, 0x01010101);
 460                 err += set_color_mask_mrt(fifo + err, 0x00000000);
 461                 err += set_clear_color(fifo + err, clear_color[i]);
 462                 err += set_scissor(fifo + err, 0, 0, 4095, 4095);
 463                 err += clear_surface(fifo + err, 0x000000f1);
 464
 465                 err += set_viewport(fifo + err, 0, 0, DISPLAY_WIDTH, DISPLAY_HEIGHT,
 466                     0.0f, 1.0f, vp_offset, vp_scale);
 467
 468                 /* Set vertex shader */
 469
 470                 err += load_vertex_prg(fifo + err, 0, vertex_prg, ARRAY_SIZE(vertex_prg) / 4);
 471                 err += set_vertex_prg_start_slot(fifo + err, 0);
 472                 err += set_vertex_prg_reg_count(fifo + err, 2);
 473                 err += set_vertex_attr_inmask(fifo + err, (1 << 3) | (1 << 0));
 474                 err += set_vertex_attr_outmask(fifo + err, (1 << 2) | (1 << 0));
 475
 476                 /* Transfer fragment program to VRAM */
 477
 478                 err += transfer_inline(fifo + err, 0xfeed0000, fp_gaddr,
 479                     frag_prg, ARRAY_SIZE(frag_prg));
 480
 481                 /* Set fragment shader */
 482
 483                 err += set_frag_prg(fifo + err, 0x1, fp_gaddr);
 484                 err += frag_prg_ctrl(fifo + err, 2, 0, 0, 0, 0);
 485
 486                 err += set_front_poly_mode(fifo + err, 0x1b02);
 487                 err += set_shade_mode(fifo + err, 0x1d01);
 488
 489                 /* Transfer vertices to VRAM */
 490
 491                 err += transfer_inline(fifo + err, 0xfeed0000, verts_gaddr,
 492                     (uint32_t *) triangle_verts, sizeof(triangle_verts) / sizeof(uint32_t));
 493
 494                 /* register 0 - position */
 495                 err += set_vertex_data_arr(fifo + err, 0, 0, sizeof(struct vertex),
 496                     3, 2, verts_gaddr + offsetof(struct vertex, x), 0x0);
 497                 /* register 3 - color */
 498                 err += set_vertex_data_arr(fifo + err, 3, 0, sizeof(struct vertex),
 499                     4, 4, verts_gaddr + offsetof(struct vertex, rgba), 0x0);
 500
 501                 err += draw_arrays(fifo + err, 0x5, 0, ARRAY_SIZE(triangle_verts));
 502
 503                 err += flip_display_buffer(fifo + err, get_channel_id(driver_info), i, 0);
 504
 505                 /*
 506                  * Label with index 0 (head 0) is set by LV1 to 0x00000000 when flip is complete.
 507                  * Let GPU wait for it.
 508                  */
 509
 510                 err += wait_label(fifo + err, 0, 0x00000000);
 511
 512                 control[0x10] = fifo_gaddr + err * sizeof(uint32_t);
 513
 514                 err = wait_fifo_idle(control);
 515                 if (err < 0) {
 516                         fprintf(stderr, "FIFO timeout: put 0x%08x get 0x%08x ref 0x%08x\n",
 517                             control[0x10], control[0x11], control[0x12]);
 518                         dump_fifo(stderr, fifo, 0x400);
 519                         goto done;
 520                 }
 521
 522                 printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 523                     control[0x10], control[0x11], control[0x12]);
 524
 525                 usleep(1000000);
 526         }
 527
 528         save_image("image.argb", (const char *) db[0], DISPLAY_PITCH * DISPLAY_HEIGHT);
 529
 530         /* Destroy GPU context */
 531
 532         context_free.context_id = context_id;
 533
 534         err = ioctl(fd, PS3GPU_CTL_CONTEXT_FREE, &context_free);
 535         if (err < 0) {
 536                 perror("ioctl");
 537                 goto done;
 538         }
 539
 540 done:
 541
 542         if (fd >= 0)
 543                 close(fd);
 544
 545         /* Restore console */
 546
 547         ioctl(0, SW_TEXT_80x25, NULL);
 548
 549         exit(0);
 550 }