Merge branch 'master' of mutantstargoat.com:/home/nuclear/git/dosdemo
[dosdemo] / src / grise.c
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <math.h>
5 #include <assert.h>
6 #include "imago2.h"
7 #include "demo.h"
8 #include "screen.h"
9
10 /* APPROX. 170 FPS Minimum */
11
12 #define BG_FILENAME "data/grise.png"
13 #define GROBJ_01_FILENAME "data/grobj_01.png"
14
15 #define BB_SIZE 512     /* Let's use a power of 2. Maybe we'll zoom/rotate the effect */
16
17 /* Every backBuffer scanline is guaranteed to have that many dummy pixels before and after */
18 #define PIXEL_PADDING 32
19
20 #define MIN_SCROLL PIXEL_PADDING
21 #define MAX_SCROLL (backgroundW - fb_width - MIN_SCROLL)
22
23 #define FAR_SCROLL_SPEED 50.0f
24 #define NEAR_SCROLL_SPEED 400.0f
25
26 #define HORIZON_HEIGHT 100
27 #define REFLECTION_HEIGHT (240 - HORIZON_HEIGHT)
28
29 #define NORMALMAP_SCANLINE 372
30
31 static int init(void);
32 static void destroy(void);
33 static void start(long trans_time);
34 static void stop(long trans_time);
35 static void draw(void);
36
37 static void convert32To16(unsigned int *src32, unsigned short *dst16, unsigned int pixelCount);
38 static void processNormal();
39 static void initScrollTables();
40 static void updateScrollTables(float dt);
41
42 static void rleEncode(unsigned char *pixels, unsigned int w, unsigned int h);
43
44 static unsigned short *background = 0;
45 static unsigned int backgroundW = 0;
46 static unsigned int backgroundH = 0;
47
48 static unsigned int lastFrameTime = 0;
49 static float lastFrameDuration = 0.0f;
50
51 static short *displacementMap;
52
53 static unsigned short *backBuffer;
54
55 static float scrollScaleTable[REFLECTION_HEIGHT];
56 static float scrollTable[REFLECTION_HEIGHT];
57 static int scrollTableRounded[REFLECTION_HEIGHT];
58 static int scrollModTable[REFLECTION_HEIGHT];
59 static float nearScrollAmount = 0.0f;
60
61 static struct screen scr = {
62         "galaxyrise",
63         init,
64         destroy,
65         start,
66         stop,
67         draw
68 };
69
70 struct screen *grise_screen(void)
71 {
72         return &scr;
73 }
74
75
76 static int init(void)
77 {
78         unsigned char *reflectedObject;
79         int reflectedObjectW, reflectedObjectH;
80
81         /* Allocate back buffer */
82         backBuffer = (unsigned short*) malloc(BB_SIZE * BB_SIZE * sizeof(unsigned short));
83
84         /* grise.png contains the background (horizon), baked reflection and normalmap for displacement */
85         if (!(background = img_load_pixels(BG_FILENAME, &backgroundW, &backgroundH, IMG_FMT_RGBA32))) {
86                 fprintf(stderr, "failed to load image " BG_FILENAME "\n");
87                 return -1;
88         }
89
90         /* Convert to 16bpp */
91         convert32To16((unsigned int*)background, background, backgroundW * NORMALMAP_SCANLINE); /* Normalmap will keep its 32 bit color */
92
93         /* Load reflected objects */
94         if (!(reflectedObject = img_load_pixels(GROBJ_01_FILENAME, &reflectedObjectW, &reflectedObjectH, IMG_FMT_GREY8))) {
95                 fprintf(stderr, "failed to load image " GROBJ_01_FILENAME "\n");
96                 return -1;
97         }
98
99         rleEncode(reflectedObject, reflectedObjectW, reflectedObjectH);
100
101         img_free_pixels(reflectedObject);
102
103         initScrollTables();
104
105         processNormal();
106
107 #ifdef MIKE_PC
108         return 0xCAFE;
109 #else
110         return 0;
111 #endif
112 }
113
114 static void destroy(void)
115 {
116         free(backBuffer);
117         backBuffer = 0;
118
119         img_free_pixels(background);
120 }
121
122 static void start(long trans_time)
123 {
124         lastFrameTime = time_msec;
125 }
126
127 static void stop(long trans_time)
128 {
129 }
130
131 static void draw(void)
132 {       
133         int scroll = MIN_SCROLL + (MAX_SCROLL - MIN_SCROLL) * mouse_x / fb_width;
134         unsigned short *dst = backBuffer + PIXEL_PADDING;
135         unsigned short *src = background + scroll;
136         int scanline = 0;
137         int i = 0;
138         short *dispScanline;
139         int d;
140
141         lastFrameDuration = (time_msec - lastFrameTime) / 1000.0f;
142         lastFrameTime = time_msec;
143
144         /* First, render the horizon */
145         for (scanline = 0; scanline < HORIZON_HEIGHT; scanline++) {
146                 memcpy(dst, src, fb_width * 2);
147                 src += backgroundW;
148                 dst += BB_SIZE;
149         }
150         
151         /* Create scroll opffsets for all scanlines of the normalmap */
152         updateScrollTables(lastFrameDuration);
153
154         /* Then, render the reflection under the horizon */
155         /* dst is already in place */
156         src = background + HORIZON_HEIGHT * backgroundW;
157         dispScanline = displacementMap;
158         for (scanline = 0; scanline < REFLECTION_HEIGHT; scanline++) {
159                 for (i = 0; i < fb_width; i++) {
160                         d = dispScanline[(i + scrollTableRounded[scanline]) % scrollModTable[scanline]];
161                         *dst++ = src[i + scroll + d];
162                 }
163                 src += backgroundW;
164                 dst += BB_SIZE - fb_width;
165                 dispScanline += backgroundW;
166         }
167
168         /* Blit effect to framebuffer */
169         src = backBuffer + PIXEL_PADDING;
170         dst = fb_pixels;
171         for (scanline = 0; scanline < fb_height; scanline++) {
172                 memcpy(dst, src, fb_width * 2);
173                 src += BB_SIZE;
174                 dst += fb_width;
175         }
176 }
177
178 /* src and dst can be the same */
179 static void convert32To16(unsigned int *src32, unsigned short *dst16, unsigned int pixelCount) {
180         unsigned int p;
181         while (pixelCount) {
182                 p = *src32++;
183                 *dst16++ =      ((p << 8) & 0xF800)             /* R */
184                         |               ((p >> 5) & 0x07E0)             /* G */
185                         |               ((p >> 19) & 0x001F);   /* B */
186                 pixelCount--;
187         }
188 }
189
190 /* Normal map preprocessing */
191 /* Scale normal with depth and unpack R component (horizontal component) */
192 static void processNormal() {
193         int scanline;
194         int i;
195         int x;
196         short maxDisplacement = 0;
197         short minDisplacement = 256;
198         unsigned short *dst;
199         short *dst2;
200         unsigned int *normalmap = (unsigned int*)background;
201         normalmap += NORMALMAP_SCANLINE * backgroundW;
202         dst = (unsigned short*)normalmap;
203         displacementMap = (short*)dst;
204         dst2 = displacementMap;
205
206         for (scanline = 0; scanline < REFLECTION_HEIGHT; scanline++) {
207                 scrollModTable[scanline] = (int) (backgroundW / scrollScaleTable[scanline] + 0.5f);
208                 for (i = 0; i < backgroundW; i++) {
209                         x = (int)(i * scrollScaleTable[scanline] + 0.5f);
210                         if (x < backgroundW) {
211                                 *dst = (unsigned short)(normalmap[x] >> 8) & 0xFF;
212                                 if ((short)*dst > maxDisplacement) maxDisplacement = (short)(*dst);
213                                 if ((short)*dst < minDisplacement) minDisplacement = (short)(*dst);
214                         } else {
215                                 *dst = 0;
216                         }
217                         dst++;
218                 }
219                 normalmap += backgroundW;
220         }
221
222         if (maxDisplacement == minDisplacement) {
223                 printf("Warning: grise normalmap fucked up\n");
224                 return;
225         }
226
227         /* Second pass - subtract half maximum displacement to displace in both directions */
228         for (scanline = 0; scanline < REFLECTION_HEIGHT; scanline++) {
229                 for (i = 0; i < backgroundW; i++) {
230                         /* Remember that MIN_SCROLL is the padding around the screen, so ti's the maximum displacement we can get (positive & negative) */
231                         *dst2 = 2 * MIN_SCROLL * (*dst2 - minDisplacement) / (maxDisplacement - minDisplacement) - MIN_SCROLL;
232                         *dst2 = (short)((float)*dst2 / scrollScaleTable[scanline] + 0.5f); /* Displacements must also scale with distance*/
233                         dst2++;
234                 }
235         }
236 }
237
238 static float distanceScale(int scanline) {
239         float farScale, t;
240         farScale = (float)NEAR_SCROLL_SPEED / (float)FAR_SCROLL_SPEED;
241         t = (float)scanline / ((float)REFLECTION_HEIGHT - 1);
242         return 1.0f / (1.0f / farScale + (1.0f - 1.0f / farScale) * t);
243 }
244
245 static void initScrollTables() {
246         int i = 0;
247         for (i = 0; i < REFLECTION_HEIGHT; i++) {
248                 scrollScaleTable[i] = distanceScale(i);
249                 scrollTable[i] = 0.0f;
250                 scrollTableRounded[i] = 0;
251         }
252 }
253
254
255 static void updateScrollTables(float dt) {
256         int i = 0;
257         
258         nearScrollAmount += dt * NEAR_SCROLL_SPEED;
259         nearScrollAmount = (float) fmod(nearScrollAmount, 512.0f);
260
261         for (i = 0; i < REFLECTION_HEIGHT; i++) {
262                 scrollTable[i] = nearScrollAmount / scrollScaleTable[i];
263                 scrollTableRounded[i] = (int)(scrollTable[i] + 0.5f) % scrollModTable[i];
264         }
265 }
266
267 static void rleEncode(unsigned char *pixels, unsigned int w, unsigned int h) {
268         int scanline;
269         int i;
270         int skipping = 1;
271
272         for (scanline = 0; scanline < h; scanline++) {
273                 for (i = 0; i < w; i++) {
274                         if (*pixels++) {
275                                 
276                         }
277                 }
278         }
279 }