git.mutantstargoat.com Git - retroray/blob - src/scene.c

   1 /*
   2 RetroRay - integrated standalone vintage modeller/renderer
   3 Copyright (C) 2023  John Tsiombikas <nuclear@mutantstargoat.com>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <https://www.gnu.org/licenses/>.
  17 */
  18 #include <stdlib.h>
  19 #include <float.h>
  20 #include "scene.h"
  21 #include "rt.h"
  22 #include "darray.h"
  23 #include "logger.h"
  24
  25 struct scene *create_scene(void)
  26 {
  27         struct scene *scn;
  28
  29         if(!(scn = malloc(sizeof *scn))) {
  30                 errormsg("failed to allocate scene\n");
  31                 return 0;
  32         }
  33         scn->objects = darr_alloc(0, sizeof *scn->objects);
  34
  35         return scn;
  36 }
  37
  38 void free_scene(struct scene *scn)
  39 {
  40         int i;
  41
  42         if(!scn) return;
  43
  44         for(i=0; i<darr_size(scn->objects); i++) {
  45                 free_object(scn->objects[i]);
  46         }
  47         darr_free(scn->objects);
  48         free(scn);
  49 }
  50
  51 int scn_add_object(struct scene *scn, struct object *obj)
  52 {
  53         darr_push(scn->objects, &obj);
  54         return 0;
  55 }
  56
  57 int scn_rm_object(struct scene *scn, int idx)
  58 {
  59         int numobj = darr_size(scn->objects);
  60
  61         if(idx < 0 || idx >= numobj) {
  62                 return -1;
  63         }
  64
  65         free_object(scn->objects[idx]);
  66
  67         if(idx < numobj - 1) {
  68                 scn->objects[idx] = scn->objects[numobj - 1];
  69         }
  70         darr_pop(scn->objects);
  71         return 0;
  72 }
  73
  74 int scn_num_objects(const struct scene *scn)
  75 {
  76         return darr_size(scn->objects);
  77 }
  78
  79 int scn_object_index(const struct scene *scn, const struct object *obj)
  80 {
  81         int i, num = darr_size(scn->objects);
  82         for(i=0; i<num; i++) {
  83                 if(scn->objects[i] == obj) {
  84                         return i;
  85                 }
  86         }
  87         return -1;
  88 }
  89
  90 int scn_intersect(const struct scene *scn, const cgm_ray *ray, struct rayhit *hit)
  91 {
  92         int i, numobj;
  93         struct rayhit hit0;
  94         struct csghit chit;
  95
  96         hit0.t = FLT_MAX;
  97         hit0.obj = 0;
  98
  99         numobj = darr_size(scn->objects);
 100         for(i=0; i<numobj; i++) {
 101                 if(ray_object(ray, scn->objects[i], &chit) && chit.ivlist[0].a.t < hit0.t) {
 102                         hit0 = chit.ivlist[0].a;
 103                 }
 104         }
 105
 106         if(hit0.obj) {
 107                 if(hit) *hit = hit0;
 108                 return 1;
 109         }
 110         return 0;
 111 }
 112
 113 struct object *create_object(int type)
 114 {
 115         struct object *obj;
 116         struct sphere *sph;
 117         char buf[32];
 118         static int objid;
 119
 120         if(!(obj = calloc(1, sizeof *obj))) {
 121                 errormsg("failed to allocate object\n");
 122                 return 0;
 123         }
 124         obj->type = type;
 125
 126         cgm_vcons(&obj->pos, 0, 0, 0);
 127         cgm_qcons(&obj->rot, 0, 0, 0, 1);
 128         cgm_vcons(&obj->scale, 1, 1, 1);
 129         cgm_vcons(&obj->pivot, 0, 0, 0);
 130         cgm_midentity(obj->xform);
 131         cgm_midentity(obj->inv_xform);
 132
 133         switch(type) {
 134         case OBJ_SPHERE:
 135                 sph = (struct sphere*)obj;
 136                 sph->rad = 1.0f;
 137                 sprintf(buf, "sphere%03d", objid);
 138                 break;
 139
 140         default:
 141                 sprintf(buf, "object%03d", objid);
 142                 break;
 143         }
 144
 145         set_object_name(obj, buf);
 146         objid++;
 147         return obj;
 148 }
 149
 150 void free_object(struct object *obj)
 151 {
 152         if(!obj) return;
 153
 154         free(obj->name);
 155         free(obj);
 156 }
 157
 158 int set_object_name(struct object *obj, const char *name)
 159 {
 160         char *str = strdup(name);
 161         if(!str) return -1;
 162
 163         free(obj->name);
 164         obj->name = str;
 165         return 0;
 166 }
 167
 168 void calc_object_matrix(struct object *obj)
 169 {
 170         int i;
 171         float rmat[16];
 172         float *mat = obj->xform;
 173
 174         cgm_mtranslation(mat, obj->pivot.x, obj->pivot.y, obj->pivot.z);
 175         cgm_mrotation_quat(rmat, &obj->rot);
 176
 177         for(i=0; i<3; i++) {
 178                 mat[i] = rmat[i];
 179                 mat[4 + i] = rmat[4 + i];
 180                 mat[8 + i] = rmat[8 + i];
 181         }
 182
 183         mat[0] *= obj->scale.x; mat[4] *= obj->scale.y; mat[8] *= obj->scale.z; mat[12] += obj->pos.x;
 184         mat[1] *= obj->scale.x; mat[5] *= obj->scale.y; mat[9] *= obj->scale.z; mat[13] += obj->pos.y;
 185         mat[2] *= obj->scale.x; mat[6] *= obj->scale.y; mat[10] *= obj->scale.z; mat[14] += obj->pos.z;
 186
 187         cgm_mpretranslate(mat, -obj->pivot.x, -obj->pivot.y, -obj->pivot.z);
 188         /* that's basically: pivot * rotation * translation * scaling * -pivot */
 189
 190         cgm_mcopy(obj->inv_xform, mat);
 191         cgm_minverse(obj->inv_xform);
 192 }