c8ff41f4eb8107dab90ba66459505ef257d23106
[divverent/darkplaces.git] / r_lightning.c
1
2 #include "quakedef.h"
3 #include "image.h"
4
5 cvar_t r_lightningbeam_thickness = {CVAR_SAVE, "r_lightningbeam_thickness", "4", "thickness of the lightning beam effect"};
6 cvar_t r_lightningbeam_scroll = {CVAR_SAVE, "r_lightningbeam_scroll", "5", "speed of texture scrolling on the lightning beam effect"};
7 cvar_t r_lightningbeam_repeatdistance = {CVAR_SAVE, "r_lightningbeam_repeatdistance", "128", "how far to stretch the texture along the lightning beam effect"};
8 cvar_t r_lightningbeam_color_red = {CVAR_SAVE, "r_lightningbeam_color_red", "1", "color of the lightning beam effect"};
9 cvar_t r_lightningbeam_color_green = {CVAR_SAVE, "r_lightningbeam_color_green", "1", "color of the lightning beam effect"};
10 cvar_t r_lightningbeam_color_blue = {CVAR_SAVE, "r_lightningbeam_color_blue", "1", "color of the lightning beam effect"};
11 cvar_t r_lightningbeam_qmbtexture = {CVAR_SAVE, "r_lightningbeam_qmbtexture", "0", "load the qmb textures/particles/lightning.pcx texture instead of generating one, can look better"};
12
13 rtexture_t *r_lightningbeamtexture;
14 rtexture_t *r_lightningbeamqmbtexture;
15 rtexturepool_t *r_lightningbeamtexturepool;
16
17 int r_lightningbeamelements[18] = {0, 1, 2, 0, 2, 3, 4, 5, 6, 4, 6, 7, 8, 9, 10, 8, 10, 11};
18
19 void r_lightningbeams_start(void)
20 {
21         r_lightningbeamtexturepool = R_AllocTexturePool();
22         r_lightningbeamtexture = NULL;
23         r_lightningbeamqmbtexture = NULL;
24 }
25
26 void r_lightningbeams_setupqmbtexture(void)
27 {
28         r_lightningbeamqmbtexture = loadtextureimage(r_lightningbeamtexturepool, "textures/particles/lightning.pcx", 0, 0, false, TEXF_ALPHA | TEXF_PRECACHE);
29         if (r_lightningbeamqmbtexture == NULL)
30                 Cvar_SetValueQuick(&r_lightningbeam_qmbtexture, false);
31 }
32
33 void r_lightningbeams_setuptexture(void)
34 {
35 #if 0
36 #define BEAMWIDTH 128
37 #define BEAMHEIGHT 64
38 #define PATHPOINTS 8
39         int i, j, px, py, nearestpathindex, imagenumber;
40         float particlex, particley, particlexv, particleyv, dx, dy, s, maxpathstrength;
41         unsigned char *pixels;
42         int *image;
43         struct lightningpathnode_s
44         {
45                 float x, y, strength;
46         }
47         path[PATHPOINTS], temppath;
48
49         image = Mem_Alloc(tempmempool, BEAMWIDTH * BEAMHEIGHT * sizeof(int));
50         pixels = Mem_Alloc(tempmempool, BEAMWIDTH * BEAMHEIGHT * sizeof(unsigned char[4]));
51
52         for (imagenumber = 0, maxpathstrength = 0.0339476;maxpathstrength < 0.5;imagenumber++, maxpathstrength += 0.01)
53         {
54         for (i = 0;i < PATHPOINTS;i++)
55         {
56                 path[i].x = lhrandom(0, 1);
57                 path[i].y = lhrandom(0.2, 0.8);
58                 path[i].strength = lhrandom(0, 1);
59         }
60         for (i = 0;i < PATHPOINTS;i++)
61         {
62                 for (j = i + 1;j < PATHPOINTS;j++)
63                 {
64                         if (path[j].x < path[i].x)
65                         {
66                                 temppath = path[j];
67                                 path[j] = path[i];
68                                 path[i] = temppath;
69                         }
70                 }
71         }
72         particlex = path[0].x;
73         particley = path[0].y;
74         particlexv = lhrandom(0, 0.02);
75         particlexv = lhrandom(-0.02, 0.02);
76         memset(image, 0, BEAMWIDTH * BEAMHEIGHT * sizeof(int));
77         for (i = 0;i < 65536;i++)
78         {
79                 for (nearestpathindex = 0;nearestpathindex < PATHPOINTS;nearestpathindex++)
80                         if (path[nearestpathindex].x > particlex)
81                                 break;
82                 nearestpathindex %= PATHPOINTS;
83                 dx = path[nearestpathindex].x + lhrandom(-0.01, 0.01);dx = bound(0, dx, 1) - particlex;if (dx < 0) dx += 1;
84                 dy = path[nearestpathindex].y + lhrandom(-0.01, 0.01);dy = bound(0, dy, 1) - particley;
85                 s = path[nearestpathindex].strength / sqrt(dx*dx+dy*dy);
86                 particlexv = particlexv /* (1 - lhrandom(0.08, 0.12))*/ + dx * s;
87                 particleyv = particleyv /* (1 - lhrandom(0.08, 0.12))*/ + dy * s;
88                 particlex += particlexv * maxpathstrength;particlex -= (int) particlex;
89                 particley += particleyv * maxpathstrength;particley = bound(0, particley, 1);
90                 px = particlex * BEAMWIDTH;
91                 py = particley * BEAMHEIGHT;
92                 if (px >= 0 && py >= 0 && px < BEAMWIDTH && py < BEAMHEIGHT)
93                         image[py*BEAMWIDTH+px] += 16;
94         }
95
96         for (py = 0;py < BEAMHEIGHT;py++)
97         {
98                 for (px = 0;px < BEAMWIDTH;px++)
99                 {
100                         pixels[(py*BEAMWIDTH+px)*4+0] = bound(0, image[py*BEAMWIDTH+px] * 1.0f, 255.0f);
101                         pixels[(py*BEAMWIDTH+px)*4+1] = bound(0, image[py*BEAMWIDTH+px] * 1.0f, 255.0f);
102                         pixels[(py*BEAMWIDTH+px)*4+2] = bound(0, image[py*BEAMWIDTH+px] * 1.0f, 255.0f);
103                         pixels[(py*BEAMWIDTH+px)*4+3] = 255;
104                 }
105         }
106
107         Image_WriteTGARGBA(va("lightningbeam%i.tga", imagenumber), BEAMWIDTH, BEAMHEIGHT, pixels);
108         }
109
110         r_lightningbeamtexture = R_LoadTexture2D(r_lightningbeamtexturepool, "lightningbeam", BEAMWIDTH, BEAMHEIGHT, pixels, TEXTYPE_RGBA, TEXF_PRECACHE, NULL);
111
112         Mem_Free(pixels);
113         Mem_Free(image);
114 #else
115 #define BEAMWIDTH 64
116 #define BEAMHEIGHT 128
117         float r, g, b, intensity, fx, width, center;
118         int x, y;
119         unsigned char *data, *noise1, *noise2;
120
121         data = (unsigned char *)Mem_Alloc(tempmempool, BEAMWIDTH * BEAMHEIGHT * 4);
122         noise1 = (unsigned char *)Mem_Alloc(tempmempool, BEAMHEIGHT * BEAMHEIGHT);
123         noise2 = (unsigned char *)Mem_Alloc(tempmempool, BEAMHEIGHT * BEAMHEIGHT);
124         fractalnoise(noise1, BEAMHEIGHT, BEAMHEIGHT / 8);
125         fractalnoise(noise2, BEAMHEIGHT, BEAMHEIGHT / 16);
126
127         for (y = 0;y < BEAMHEIGHT;y++)
128         {
129                 width = 0.15;//((noise1[y * BEAMHEIGHT] * (1.0f / 256.0f)) * 0.1f + 0.1f);
130                 center = (noise1[y * BEAMHEIGHT + (BEAMHEIGHT / 2)] / 256.0f) * (1.0f - width * 2.0f) + width;
131                 for (x = 0;x < BEAMWIDTH;x++, fx++)
132                 {
133                         fx = (((float) x / BEAMWIDTH) - center) / width;
134                         intensity = 1.0f - sqrt(fx * fx);
135                         if (intensity > 0)
136                                 intensity = pow(intensity, 2) * ((noise2[y * BEAMHEIGHT + x] * (1.0f / 256.0f)) * 0.33f + 0.66f);
137                         intensity = bound(0, intensity, 1);
138                         r = intensity * 1.0f;
139                         g = intensity * 1.0f;
140                         b = intensity * 1.0f;
141                         data[(y * BEAMWIDTH + x) * 4 + 0] = (unsigned char)(bound(0, r, 1) * 255.0f);
142                         data[(y * BEAMWIDTH + x) * 4 + 1] = (unsigned char)(bound(0, g, 1) * 255.0f);
143                         data[(y * BEAMWIDTH + x) * 4 + 2] = (unsigned char)(bound(0, b, 1) * 255.0f);
144                         data[(y * BEAMWIDTH + x) * 4 + 3] = (unsigned char)255;
145                 }
146         }
147
148         r_lightningbeamtexture = R_LoadTexture2D(r_lightningbeamtexturepool, "lightningbeam", BEAMWIDTH, BEAMHEIGHT, data, TEXTYPE_RGBA, TEXF_PRECACHE, NULL);
149         Mem_Free(noise1);
150         Mem_Free(noise2);
151         Mem_Free(data);
152 #endif
153 }
154
155 void r_lightningbeams_shutdown(void)
156 {
157         r_lightningbeamtexture = NULL;
158         r_lightningbeamqmbtexture = NULL;
159         R_FreeTexturePool(&r_lightningbeamtexturepool);
160 }
161
162 void r_lightningbeams_newmap(void)
163 {
164 }
165
166 void R_LightningBeams_Init(void)
167 {
168         Cvar_RegisterVariable(&r_lightningbeam_thickness);
169         Cvar_RegisterVariable(&r_lightningbeam_scroll);
170         Cvar_RegisterVariable(&r_lightningbeam_repeatdistance);
171         Cvar_RegisterVariable(&r_lightningbeam_color_red);
172         Cvar_RegisterVariable(&r_lightningbeam_color_green);
173         Cvar_RegisterVariable(&r_lightningbeam_color_blue);
174         Cvar_RegisterVariable(&r_lightningbeam_qmbtexture);
175         R_RegisterModule("R_LightningBeams", r_lightningbeams_start, r_lightningbeams_shutdown, r_lightningbeams_newmap);
176 }
177
178 void R_CalcLightningBeamPolygonVertex3f(float *v, const float *start, const float *end, const float *offset)
179 {
180         // near right corner
181         VectorAdd     (start, offset, (v + 0));
182         // near left corner
183         VectorSubtract(start, offset, (v + 3));
184         // far left corner
185         VectorSubtract(end  , offset, (v + 6));
186         // far right corner
187         VectorAdd     (end  , offset, (v + 9));
188 }
189
190 void R_CalcLightningBeamPolygonTexCoord2f(float *tc, float t1, float t2)
191 {
192         if (r_lightningbeam_qmbtexture.integer)
193         {
194                 // near right corner
195                 tc[0] = t1;tc[1] = 0;
196                 // near left corner
197                 tc[2] = t1;tc[3] = 1;
198                 // far left corner
199                 tc[4] = t2;tc[5] = 1;
200                 // far right corner
201                 tc[6] = t2;tc[7] = 0;
202         }
203         else
204         {
205                 // near right corner
206                 tc[0] = 0;tc[1] = t1;
207                 // near left corner
208                 tc[2] = 1;tc[3] = t1;
209                 // far left corner
210                 tc[4] = 1;tc[5] = t2;
211                 // far right corner
212                 tc[6] = 0;tc[7] = t2;
213         }
214 }
215
216 void R_FogLightningBeam_Vertex3f_Color4f(const float *v, float *c, int numverts, float r, float g, float b, float a)
217 {
218         int i;
219         float fog;
220         for (i = 0;i < numverts;i++, v += 3, c += 4)
221         {
222                 fog = FogPoint_World(v);
223                 c[0] = r * fog;
224                 c[1] = g * fog;
225                 c[2] = b * fog;
226                 c[3] = a;
227         }
228 }
229
230 float beamrepeatscale;
231
232 void R_DrawLightningBeam_TransparentCallback(const entity_render_t *ent, const rtlight_t *rtlight, int numsurfaces, int *surfacelist)
233 {
234         int surfacelistindex;
235         rmeshstate_t m;
236         float vertex3f[12*3];
237         float texcoord2f[12*2];
238         float color4f[12*4];
239         R_Mesh_Matrix(&identitymatrix);
240         GL_BlendFunc(GL_SRC_ALPHA, GL_ONE);
241         GL_DepthMask(false);
242         GL_DepthRange(0, 1);
243         GL_PolygonOffset(r_refdef.polygonfactor, r_refdef.polygonoffset);
244         GL_DepthTest(true);
245         if (r_lightningbeam_qmbtexture.integer && r_lightningbeamqmbtexture == NULL)
246                 r_lightningbeams_setupqmbtexture();
247         if (!r_lightningbeam_qmbtexture.integer && r_lightningbeamtexture == NULL)
248                 r_lightningbeams_setuptexture();
249
250         R_Mesh_VertexPointer(vertex3f, 0, 0);
251         // FIXME: fixed function path can't properly handle r_view.colorscale > 1
252         if (r_refdef.fogenabled)
253         {
254                 // per vertex colors if fog is used
255                 R_Mesh_ColorPointer(color4f, 0, 0);
256                 R_FogLightningBeam_Vertex3f_Color4f(vertex3f, color4f, 12, r_lightningbeam_color_red.value * r_view.colorscale, r_lightningbeam_color_green.value * r_view.colorscale, r_lightningbeam_color_blue.value * r_view.colorscale, 1);
257         }
258         else
259         {
260                 // solid color if fog is not used
261                 R_Mesh_ColorPointer(NULL, 0, 0);
262                 GL_Color(r_lightningbeam_color_red.value * r_view.colorscale, r_lightningbeam_color_green.value * r_view.colorscale, r_lightningbeam_color_blue.value * r_view.colorscale, 1);
263         }
264         memset(&m, 0, sizeof(m));
265         if (r_lightningbeam_qmbtexture.integer)
266                 m.tex[0] = R_GetTexture(r_lightningbeamqmbtexture);
267         else
268                 m.tex[0] = R_GetTexture(r_lightningbeamtexture);
269         m.pointer_texcoord[0] = texcoord2f;
270         R_Mesh_TextureState(&m);
271
272         for (surfacelistindex = 0;surfacelistindex < numsurfaces;surfacelistindex++)
273         {
274                 const beam_t *b = cl.beams + surfacelist[surfacelistindex];
275                 vec3_t beamdir, right, up, offset, start, end;
276                 float length, t1, t2;
277
278                 CL_Beam_CalculatePositions(b, start, end);
279
280                 // calculate beam direction (beamdir) vector and beam length
281                 // get difference vector
282                 VectorSubtract(end, start, beamdir);
283                 // find length of difference vector
284                 length = sqrt(DotProduct(beamdir, beamdir));
285                 // calculate scale to make beamdir a unit vector (normalized)
286                 t1 = 1.0f / length;
287                 // scale beamdir so it is now normalized
288                 VectorScale(beamdir, t1, beamdir);
289
290                 // calculate up vector such that it points toward viewer, and rotates around the beamdir
291                 // get direction from start of beam to viewer
292                 VectorSubtract(r_view.origin, start, up);
293                 // remove the portion of the vector that moves along the beam
294                 // (this leaves only a vector pointing directly away from the beam)
295                 t1 = -DotProduct(up, beamdir);
296                 VectorMA(up, t1, beamdir, up);
297                 // generate right vector from forward and up, the result is unnormalized
298                 CrossProduct(beamdir, up, right);
299                 // now normalize the right vector and up vector
300                 VectorNormalize(right);
301                 VectorNormalize(up);
302
303                 // calculate T coordinate scrolling (start and end texcoord along the beam)
304                 t1 = r_refdef.time * -r_lightningbeam_scroll.value;// + beamrepeatscale * DotProduct(start, beamdir);
305                 t1 = t1 - (int) t1;
306                 t2 = t1 + beamrepeatscale * length;
307
308                 // the beam is 3 polygons in this configuration:
309                 //  *   2
310                 //   * *
311                 // 1******
312                 //   * *
313                 //  *   3
314                 // they are showing different portions of the beam texture, creating an
315                 // illusion of a beam that appears to curl around in 3D space
316                 // (and realize that the whole polygon assembly orients itself to face
317                 //  the viewer)
318
319                 // polygon 1, verts 0-3
320                 VectorScale(right, r_lightningbeam_thickness.value, offset);
321                 R_CalcLightningBeamPolygonVertex3f(vertex3f + 0, start, end, offset);
322                 // polygon 2, verts 4-7
323                 VectorAdd(right, up, offset);
324                 VectorScale(offset, r_lightningbeam_thickness.value * 0.70710681f, offset);
325                 R_CalcLightningBeamPolygonVertex3f(vertex3f + 12, start, end, offset);
326                 // polygon 3, verts 8-11
327                 VectorSubtract(right, up, offset);
328                 VectorScale(offset, r_lightningbeam_thickness.value * 0.70710681f, offset);
329                 R_CalcLightningBeamPolygonVertex3f(vertex3f + 24, start, end, offset);
330                 R_CalcLightningBeamPolygonTexCoord2f(texcoord2f + 0, t1, t2);
331                 R_CalcLightningBeamPolygonTexCoord2f(texcoord2f + 8, t1 + 0.33, t2 + 0.33);
332                 R_CalcLightningBeamPolygonTexCoord2f(texcoord2f + 16, t1 + 0.66, t2 + 0.66);
333                 if (r_refdef.fogenabled)
334                 {
335                         // per vertex colors if fog is used
336                         R_FogLightningBeam_Vertex3f_Color4f(vertex3f, color4f, 12, r_lightningbeam_color_red.value, r_lightningbeam_color_green.value, r_lightningbeam_color_blue.value, 1);
337                 }
338
339                 // draw the 3 polygons as one batch of 6 triangles using the 12 vertices
340                 GL_LockArrays(0, 12);
341                 R_Mesh_Draw(0, 12, 6, r_lightningbeamelements, 0, 0);
342                 GL_LockArrays(0, 0);
343         }
344 }
345
346 extern cvar_t cl_beams_polygons;
347 void R_DrawLightningBeams(void)
348 {
349         int i;
350         beam_t *b;
351
352         if (!cl_beams_polygons.integer)
353                 return;
354
355         beamrepeatscale = 1.0f / r_lightningbeam_repeatdistance.value;
356         for (i = 0, b = cl.beams;i < cl.num_beams;i++, b++)
357         {
358                 if (b->model && b->lightning)
359                 {
360                         vec3_t org, start, end, dir;
361                         vec_t dist;
362                         CL_Beam_CalculatePositions(b, start, end);
363                         // calculate the nearest point on the line (beam) for depth sorting
364                         VectorSubtract(end, start, dir);
365                         dist = (DotProduct(r_view.origin, dir) - DotProduct(start, dir)) / (DotProduct(end, dir) - DotProduct(start, dir));
366                         dist = bound(0, dist, 1);
367                         VectorLerp(start, dist, end, org);
368                         // now we have the nearest point on the line, so sort with it
369                         R_MeshQueue_AddTransparent(org, R_DrawLightningBeam_TransparentCallback, NULL, i, NULL);
370                 }
371         }
372 }
373