git.mutantstargoat.com Git - dosdemo/blob - src/polytest.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <math.h>
   5 #include "screen.h"
   6 #include "demo.h"
   7 #include "3dgfx.h"
   8 #include "gfxutil.h"
   9 #include "polyfill.h"   /* just for struct pimage */
  10 #include "cfgopt.h"
  11 #include "mesh.h"
  12 #include "bsptree.h"
  13
  14 static int init(void);
  15 static void destroy(void);
  16 static void start(long trans_time);
  17 static void draw(void);
  18 static void draw_lowres_raster(void);
  19 static void keypress(int key);
  20 static int gen_texture(struct pimage *img, int xsz, int ysz);
  21
  22 static struct screen scr = {
  23         "polytest",
  24         init,
  25         destroy,
  26         start, 0,
  27         draw,
  28         keypress
  29 };
  30
  31 static float cam_theta, cam_phi = 25;
  32 static float cam_dist = 3;
  33 static struct g3d_mesh cube, torus;
  34 static struct bsptree torus_bsp;
  35
  36 static struct pimage tex;
  37
  38 static int use_bsp = 1;
  39
  40 #define LOWRES_SCALE    10
  41 static uint16_t *lowres_pixels;
  42 static int lowres_width, lowres_height;
  43
  44 struct screen *polytest_screen(void)
  45 {
  46         return &scr;
  47 }
  48
  49 static int init(void)
  50 {
  51         int i;
  52
  53         gen_cube_mesh(&cube, 1.0, 0);
  54         gen_torus_mesh(&torus, 1.0, 0.25, 24, 12);
  55         /* scale texcoords */
  56         for(i=0; i<torus.vcount; i++) {
  57                 torus.varr[i].u *= 4.0;
  58                 torus.varr[i].v *= 2.0;
  59         }
  60
  61         init_bsp(&torus_bsp);
  62         if(bsp_add_mesh(&torus_bsp, &torus) == -1) {
  63                 fprintf(stderr, "failed to construct torus BSP tree\n");
  64                 return -1;
  65         }
  66
  67         gen_texture(&tex, 128, 128);
  68
  69 #ifdef DEBUG_POLYFILL
  70         lowres_width = fb_width / LOWRES_SCALE;
  71         lowres_height = fb_height / LOWRES_SCALE;
  72         lowres_pixels = malloc(lowres_width * lowres_height * 2);
  73         scr.draw = draw_debug;
  74 #endif
  75
  76         return 0;
  77 }
  78
  79 static void destroy(void)
  80 {
  81         free(lowres_pixels);
  82         free(cube.varr);
  83         free(torus.varr);
  84         free(torus.iarr);
  85         destroy_bsp(&torus_bsp);
  86 }
  87
  88 static void start(long trans_time)
  89 {
  90         g3d_matrix_mode(G3D_PROJECTION);
  91         g3d_load_identity();
  92         g3d_perspective(50.0, 1.3333333, 0.5, 100.0);
  93
  94         g3d_enable(G3D_CULL_FACE);
  95         g3d_enable(G3D_LIGHTING);
  96         g3d_enable(G3D_LIGHT0);
  97
  98         g3d_polygon_mode(G3D_TEX_GOURAUD);
  99 }
 100
 101 static void update(void)
 102 {
 103         mouse_orbit_update(&cam_theta, &cam_phi, &cam_dist);
 104 }
 105
 106
 107 static void draw(void)
 108 {
 109         float vdir[3];
 110         const float *mat;
 111
 112         update();
 113
 114         memset(fb_pixels, 0, fb_width * fb_height * 2);
 115
 116         g3d_matrix_mode(G3D_MODELVIEW);
 117         g3d_load_identity();
 118         g3d_translate(0, 0, -cam_dist);
 119         if(opt.sball) {
 120                 g3d_mult_matrix(sball_matrix);
 121         } else {
 122                 g3d_rotate(cam_phi, 1, 0, 0);
 123                 g3d_rotate(cam_theta, 0, 1, 0);
 124         }
 125
 126         g3d_light_pos(0, -10, 10, 20);
 127
 128         g3d_mtl_diffuse(0.4, 0.7, 1.0);
 129         g3d_set_texture(tex.width, tex.height, tex.pixels);
 130
 131         if(use_bsp) {
 132                 /* calc world-space view direction */
 133                 mat = g3d_get_matrix(G3D_MODELVIEW, 0);
 134                 /* transform (0, 0, -1) with transpose(mat3x3) */
 135                 vdir[0] = -mat[2];
 136                 vdir[1] = -mat[6];
 137                 vdir[2] = -mat[10];
 138
 139                 draw_bsp(&torus_bsp, vdir[0], vdir[1], vdir[2]);
 140         } else {
 141                 zsort_mesh(&torus);
 142                 draw_mesh(&torus);
 143         }
 144
 145         /*draw_mesh(&cube);*/
 146         swap_buffers(fb_pixels);
 147 }
 148
 149 static void draw_debug(void)
 150 {
 151         update();
 152
 153         memset(lowres_pixels, 0, lowres_width * lowres_height * 2);
 154
 155         g3d_matrix_mode(G3D_MODELVIEW);
 156         g3d_load_identity();
 157         g3d_translate(0, 0, -cam_dist);
 158         g3d_rotate(cam_phi, 1, 0, 0);
 159         g3d_rotate(cam_theta, 0, 1, 0);
 160
 161         g3d_framebuffer(lowres_width, lowres_height, lowres_pixels);
 162         /*zsort(&torus);*/
 163         draw_mesh(&cube);
 164
 165         draw_lowres_raster();
 166
 167
 168         g3d_framebuffer(fb_width, fb_height, fb_pixels);
 169
 170         g3d_polygon_mode(G3D_WIRE);
 171         draw_mesh(&cube);
 172         g3d_polygon_mode(G3D_FLAT);
 173
 174         swap_buffers(fb_pixels);
 175 }
 176
 177
 178 static void draw_huge_pixel(uint16_t *dest, int dest_width, uint16_t color)
 179 {
 180         int i, j;
 181         uint16_t grid_color = PACK_RGB16(127, 127, 127);
 182
 183         for(i=0; i<LOWRES_SCALE; i++) {
 184                 for(j=0; j<LOWRES_SCALE; j++) {
 185                         dest[j] = i == 0 || j == 0 ? grid_color : color;
 186                 }
 187                 dest += dest_width;
 188         }
 189 }
 190
 191 static void draw_lowres_raster(void)
 192 {
 193         int i, j;
 194         uint16_t *sptr = lowres_pixels;
 195         uint16_t *dptr = fb_pixels;
 196
 197         for(i=0; i<lowres_height; i++) {
 198                 for(j=0; j<lowres_width; j++) {
 199                         draw_huge_pixel(dptr, fb_width, *sptr++);
 200                         dptr += LOWRES_SCALE;
 201                 }
 202                 dptr += fb_width * LOWRES_SCALE - fb_width;
 203         }
 204 }
 205
 206 static void keypress(int key)
 207 {
 208         switch(key) {
 209         case 'b':
 210                 use_bsp = !use_bsp;
 211                 printf("drawing with %s\n", use_bsp ? "BSP tree" : "z-sorting");
 212                 break;
 213         }
 214 }
 215
 216 static int gen_texture(struct pimage *img, int xsz, int ysz)
 217 {
 218         int i, j;
 219         uint16_t *pix;
 220
 221         if(!(img->pixels = malloc(xsz * ysz * sizeof *pix))) {
 222                 return -1;
 223         }
 224         pix = img->pixels;
 225
 226         for(i=0; i<ysz; i++) {
 227                 for(j=0; j<xsz; j++) {
 228                         int val = i ^ j;
 229
 230                         *pix++ = PACK_RGB16(val, val, val);
 231                 }
 232         }
 233
 234         img->width = xsz;
 235         img->height = ysz;
 236         return 0;
 237 }