gart_dma.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 <string.h>
  33
  34 #include <sys/types.h>
  35 #include <sys/stat.h>
  36 #include <sys/uio.h>
  37 #include <sys/ioctl.h>
  38 #include <sys/mman.h>
  39 #include <sys/fbio.h>
  40 #include <sys/consio.h>
  41 #include <fcntl.h>
  42 #include <unistd.h>
  43
  44 #include "ps3gpu_ctl.h"
  45 #include "ps3gpu_mth.h"
  46 #include "reset_gpu_state.h"
  47 #include "util.h"
  48
  49 #define PS3GPU_DEV_PATH "/dev/ps3gpu"
  50
  51 #define DISPLAY_WIDTH   1920
  52 #define DISPLAY_HEIGHT  1080
  53 #define DISPLAY_BPP     4
  54 #define DISPLAY_PITCH   (DISPLAY_WIDTH * DISPLAY_BPP)
  55
  56 int
  57 main(int argc, char **argv)
  58 {
  59         struct ps3gpu_ctl_context_allocate context_allocate;
  60         struct ps3gpu_ctl_context_free context_free;
  61         struct ps3gpu_ctl_memory_allocate memory_allocate;
  62         struct ps3gpu_ctl_setup_control setup_control;
  63         struct ps3gpu_ctl_flip flip;
  64         int context_id;
  65         volatile uint32_t *control;
  66         uint32_t *fifo, *reset_gpu, *vram, *gart;
  67         unsigned long fifo_handle, vram_handle, gart_handle;
  68         unsigned int fifo_gaddr, reset_gpu_gaddr, vram_gaddr, gart_gaddr;
  69         int fd = -1;
  70         int x, y;
  71         int w, h;
  72         int err;
  73
  74         /* Open GPU device */
  75
  76         fd = open(PS3GPU_DEV_PATH, O_RDWR);
  77         if (fd < 0) {
  78                 perror("open");
  79                 goto done;
  80         }
  81
  82         /* Create GPU context */
  83
  84         context_allocate.vram_size = 64; /* MB */
  85
  86         err = ioctl(fd, PS3GPU_CTL_CONTEXT_ALLOCATE, &context_allocate);
  87         if (err < 0) {
  88                 perror("ioctl");
  89                 goto done;
  90         }
  91
  92         context_id = context_allocate.context_id;
  93
  94         printf("context id %d\n", context_id);
  95         printf("control handle 0x%lx size %d\n",
  96             context_allocate.control_handle, context_allocate.control_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         /* Allocate FIFO */
 108
 109         memory_allocate.context_id = context_id;
 110         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_GART;
 111         memory_allocate.size = 64 * 1024;
 112         memory_allocate.align = 12;
 113
 114         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 115         if (err < 0) {
 116                 perror("ioctl");
 117                 goto done;
 118         }
 119
 120         fifo_handle = memory_allocate.handle;
 121         fifo_gaddr = memory_allocate.gpu_addr;
 122
 123         printf("fifo handle 0x%lx gpu addr 0x08%x\n",
 124             fifo_handle, fifo_gaddr);
 125
 126         /* Map FIFO */
 127
 128         fifo = mmap(NULL, memory_allocate.size,
 129             PROT_READ | PROT_WRITE, MAP_SHARED, fd, fifo_handle);
 130         if (fifo == (void *) MAP_FAILED) {
 131                 perror("mmap");
 132                 goto done;
 133         }
 134
 135         /* Setup FIFO */
 136
 137         setup_control.context_id = context_id;
 138         setup_control.put = fifo_handle;
 139         setup_control.get = fifo_handle;
 140         setup_control.ref = 0xdeadbabe;
 141
 142         err = ioctl(fd, PS3GPU_CTL_SETUP_CONTROL, &setup_control);
 143         if (err < 0) {
 144                 perror("ioctl");
 145                 goto done;
 146         }
 147
 148         printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 149             control[0x10], control[0x11], control[0x12]);
 150
 151         /* Allocate FIFO for resetting GPU state */
 152
 153         memory_allocate.context_id = context_id;
 154         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_GART;
 155         memory_allocate.size = 8 * 1024;
 156         memory_allocate.align = 12;
 157
 158         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 159         if (err < 0) {
 160                 perror("ioctl");
 161                 goto done;
 162         }
 163
 164         reset_gpu_gaddr = memory_allocate.gpu_addr;
 165
 166         printf("reset GPU state handle 0x%lx gpu addr 0x%08x\n",
 167             memory_allocate.handle, reset_gpu_gaddr);
 168
 169         /* Map FIFO for resetting GPU state */
 170
 171         reset_gpu = mmap(NULL, memory_allocate.size,
 172             PROT_READ | PROT_WRITE, MAP_SHARED, fd, memory_allocate.handle);
 173         if (reset_gpu == (void *) MAP_FAILED) {
 174                 perror("mmap");
 175                 goto done;
 176         }
 177
 178         memcpy(reset_gpu, reset_gpu_state_3d, reset_gpu_state_3d_size);
 179
 180         /* Kick FIFO */
 181
 182         fifo[0] = PS3GPU_MTH_HDR(0, 0, reset_gpu_gaddr | PS3GPU_MTH_ADDR_CALL);
 183         fifo[1] = PS3GPU_MTH_HDR(1, 0, PS3GPU_MTH_ADDR_REF);
 184         fifo[2] = 0xcafef00d;
 185
 186         control[0x10] = fifo_gaddr + 3 * sizeof(uint32_t);
 187
 188         while (control[0x10] != control[0x11])
 189                 usleep(1000);
 190
 191         printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 192             control[0x10], control[0x11], control[0x12]);
 193
 194         /* Allocate VRAM */
 195
 196         memory_allocate.context_id = context_id;
 197         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_VIDEO;
 198         memory_allocate.size = 9 * 1024 * 1024;
 199         memory_allocate.align = 12;
 200
 201         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 202         if (err < 0) {
 203                 perror("ioctl");
 204                 goto done;
 205         }
 206
 207         vram_handle = memory_allocate.handle;
 208         vram_gaddr = memory_allocate.gpu_addr;
 209
 210         printf("VRAM handle 0x%lx gpu addr 0x%08x\n",
 211             vram_handle, vram_gaddr);
 212
 213         /* Map VRAM */
 214
 215         vram = mmap(NULL, memory_allocate.size,
 216             PROT_READ | PROT_WRITE, MAP_SHARED, fd, vram_handle);
 217         if (vram == (void *) MAP_FAILED) {
 218                 perror("mmap");
 219                 goto done;
 220         }
 221
 222         memset(vram, 0x40, memory_allocate.size);
 223
 224         /* Flip */
 225
 226         flip.context_id = context_id;
 227         flip.head = PS3GPU_CTL_HEAD_A;
 228         flip.offset = vram_handle;
 229
 230         err = ioctl(fd, PS3GPU_CTL_FLIP, &flip);
 231         if (err < 0) {
 232                 perror("ioctl");
 233                 goto done;
 234         }
 235
 236         flip.context_id = context_id;
 237         flip.head = PS3GPU_CTL_HEAD_B;
 238         flip.offset = vram_handle;
 239
 240         err = ioctl(fd, PS3GPU_CTL_FLIP, &flip);
 241         if (err < 0) {
 242                 perror("ioctl");
 243                 goto done;
 244         }
 245
 246         usleep(2000000);
 247
 248         /* Allocate GART */
 249
 250         memory_allocate.context_id = context_id;
 251         memory_allocate.type = PS3GPU_CTL_MEMORY_TYPE_GART;
 252         memory_allocate.size = 1024 * 1024;
 253         memory_allocate.align = 12;
 254
 255         err = ioctl(fd, PS3GPU_CTL_MEMORY_ALLOCATE, &memory_allocate);
 256         if (err < 0) {
 257                 perror("ioctl");
 258                 goto done;
 259         }
 260
 261         gart_handle = memory_allocate.handle;
 262         gart_gaddr = memory_allocate.gpu_addr;
 263
 264         printf("GART handle 0x%lx gpu addr 0x%08x\n",
 265             gart_handle, gart_gaddr);
 266
 267         /* Map GART */
 268
 269         gart = mmap(NULL, memory_allocate.size,
 270             PROT_READ | PROT_WRITE, MAP_SHARED, fd, gart_handle);
 271         if (gart == (void *) MAP_FAILED) {
 272                 perror("mmap");
 273                 goto done;
 274         }
 275
 276         /* Test RSX DMA */
 277
 278         w = 200;
 279         h = 200;
 280
 281         for (y = 0; y < h; y++) {
 282                 for (x = 0; x < w; x++) {
 283                         if (y < (h / 2))
 284                                 gart[y * w + x] = 0x0000ff00;
 285                         else
 286                                 gart[y * w + x] = 0xffffff00;
 287                 }
 288         }
 289
 290         setup_control.context_id = context_id;
 291         setup_control.put = fifo_handle;
 292         setup_control.get = fifo_handle;
 293         setup_control.ref = 0xdeadbabe;
 294
 295         err = ioctl(fd, PS3GPU_CTL_SETUP_CONTROL, &setup_control);
 296         if (err < 0) {
 297                 perror("ioctl");
 298                 goto done;
 299         }
 300
 301         x = 200;
 302         y = 800;
 303
 304         err = transfer_data(fifo, 0xfeed0001, 0xfeed0000,
 305             vram_gaddr + y * DISPLAY_PITCH + x * DISPLAY_BPP, DISPLAY_PITCH,
 306             gart_gaddr + 0, w * DISPLAY_BPP,
 307             w * DISPLAY_BPP, h);
 308
 309         x = 800;
 310         y = 800;
 311
 312         err += transfer_data(fifo + err, 0xfeed0001, 0xfeed0000,
 313             vram_gaddr + y * DISPLAY_PITCH + x * DISPLAY_BPP, DISPLAY_PITCH,
 314             gart_gaddr + 0, w * DISPLAY_BPP,
 315             w * DISPLAY_BPP, h);
 316
 317         control[0x10] = fifo_gaddr + err * sizeof(uint32_t);
 318
 319         while (control[0x10] != control[0x11])
 320                 usleep(1000);
 321
 322         printf("FIFO put 0x%08x get 0x%08x ref 0x%08x\n",
 323             control[0x10], control[0x11], control[0x12]);
 324
 325         /*
 326          * convert -depth 8 -size 1920x1080 rgba:image.argb \
 327          *    -color-matrix "0 1 0 0, 0 0 1 0, 0 0 0 1, 1 0 0 0" \
 328          *    -alpha off -background black -flatten image.jpg
 329          */
 330
 331         save_image("image.argb", (const char *) vram, DISPLAY_PITCH * DISPLAY_HEIGHT);
 332
 333         usleep(5000000);
 334
 335         /* Destroy GPU context */
 336
 337         context_free.context_id = context_id;
 338
 339         err = ioctl(fd, PS3GPU_CTL_CONTEXT_FREE, &context_free);
 340         if (err < 0) {
 341                 perror("ioctl");
 342                 goto done;
 343         }
 344
 345 done:
 346
 347         if (fd >= 0)
 348                 close(fd);
 349
 350         /* Restore console */
 351
 352         ioctl(0, SW_TEXT_80x25, NULL);
 353
 354         exit(0);
 355 }