4 #include "../../dpdefs/progsdefs.qh"
5 #include "../../dpdefs/dpextensions.qh"
6 #include "../../common/util.qh"
9 #include "../../csqcmodellib/sv_model.qh"
12 // ===============================================
13 // Generates radar map images for use in the HUD
14 // ===============================================
16 float FullTraceFraction(vector a, vector mi, vector ma, vector b)
29 while(vlen(c - b) > 1)
33 tracebox(c, mi, ma, b, MOVE_WORLDONLY, world);
38 black += vlen(trace_endpos - c);
42 n += tracebox_inverted(c, mi, ma, b, MOVE_WORLDONLY, world, false, world);
44 white += vlen(trace_endpos - c);
49 dprint("HOLY SHIT! FullTraceFraction: ", ftos(n), " total traces, ", ftos(m), " iterations\n");
51 return white / (black + white);
53 float RadarMapAtPoint_Trace(float x, float y, float w, float h, float zmin, float zsize, float q)
58 ma = '1 0 0' * w + '0 1 0' * h;
59 a = '1 0 0' * x + '0 1 0' * y + '0 0 1' * zmin;
60 b = '1 0 0' * x + '0 1 0' * y + '0 0 1' * (zsize + zmin);
62 return FullTraceFraction(a, mi, ma, b);
64 float RadarMapAtPoint_LineBlock(float x, float y, float w, float h, float zmin, float zsize, float q)
71 // 256q-1 is the ideal sample count to map equal amount of sample values to one pixel value
74 dz = (zsize / q) * '0 0 1';
75 ma = '1 0 0' * w + '0 1 0' * h + dz;
76 o = '1 0 0' * x + '0 1 0' * y + '0 0 1' * zmin;
78 if(x < world.absmin.x - w)
80 if(y < world.absmin.y - h)
82 if(x > world.absmax.x)
84 if(y > world.absmax.y)
88 for(i = 0; i < q; ++i)
91 v1 = v2 = o + dz * i + mi;
92 v1_x += random() * (ma.x - mi.x);
93 v1_y += random() * (ma.y - mi.y);
94 v1_z += random() * (ma.z - mi.z);
95 v2_x += random() * (ma.x - mi.x);
96 v2_y += random() * (ma.y - mi.y);
97 v2_z += random() * (ma.z - mi.z);
98 traceline(v1, v2, MOVE_WORLDONLY, world);
99 if(trace_startsolid || trace_fraction < 1)
104 float RadarMapAtPoint_Block(float x, float y, float w, float h, float zmin, float zsize, float q)
111 // 256q-1 is the ideal sample count to map equal amount of sample values to one pixel value
114 dz = (zsize / q) * '0 0 1';
115 ma = '1 0 0' * w + '0 1 0' * h + dz;
116 o = '1 0 0' * x + '0 1 0' * y + '0 0 1' * zmin;
118 if(x < world.absmin.x - w)
120 if(y < world.absmin.y - h)
122 if(x > world.absmax.x)
124 if(y > world.absmax.y)
128 for(i = 0; i < q; ++i)
130 tracebox(o + dz * i, mi, ma, o + dz * i, MOVE_WORLDONLY, world);
136 float RadarMapAtPoint_Sample(float x, float y, float w, float h, float zmin, float zsize, float q)
140 q *= 4; // choose q so it matches the regular algorithm in speed
143 // 256q-1 is the ideal sample count to map equal amount of sample values to one pixel value
146 ma = '1 0 0' * w + '0 1 0' * h;
147 a = '1 0 0' * x + '0 1 0' * y + '0 0 1' * zmin;
148 b = '1 0 0' * w + '0 1 0' * h + '0 0 1' * zsize;
153 for(i = 0; i < q; ++i)
156 v_x = a.x + random() * b.x;
157 v_y = a.y + random() * b.y;
158 v_z = a.z + random() * b.z;
159 traceline(v, v, MOVE_WORLDONLY, world);
166 void sharpen_set(float x, float v)
168 sharpen_buffer[x + 2 * RADAR_WIDTH_MAX] = v;
170 float sharpen_getpixel(float x, float y)
174 if(x >= RADAR_WIDTH_MAX)
180 return sharpen_buffer[x + y * RADAR_WIDTH_MAX];
182 float sharpen_get(float x, float a)
185 sum = sharpen_getpixel(x, 1);
189 sum -= sharpen_getpixel(x - 1, 0);
190 sum -= sharpen_getpixel(x - 1, 1);
191 sum -= sharpen_getpixel(x - 1, 2);
192 sum -= sharpen_getpixel(x + 1, 0);
193 sum -= sharpen_getpixel(x + 1, 1);
194 sum -= sharpen_getpixel(x + 1, 2);
195 sum -= sharpen_getpixel(x, 0);
196 sum -= sharpen_getpixel(x, 2);
197 return bound(0, sum * a, 1);
199 void sharpen_shift(float w)
202 for(i = 0; i < w; ++i)
204 sharpen_buffer[i] = sharpen_buffer[i + RADAR_WIDTH_MAX];
205 sharpen_buffer[i + RADAR_WIDTH_MAX] = sharpen_buffer[i + 2 * RADAR_WIDTH_MAX];
206 sharpen_buffer[i + 2 * RADAR_WIDTH_MAX] = 0;
209 void sharpen_init(float w)
212 for(i = 0; i < w; ++i)
214 sharpen_buffer[i] = 0;
215 sharpen_buffer[i + RADAR_WIDTH_MAX] = 0;
216 sharpen_buffer[i + 2 * RADAR_WIDTH_MAX] = 0;
221 if(radarmapper.count & 4)
225 else if(radarmapper.count & 2)
227 localcmd(strcat("defer 1 \"sv_cmd radarmap --flags ", ftos(radarmapper.count), strcat(" --res ", ftos(radarmapper.size.x), " ", ftos(radarmapper.size.y), " --sharpen ", ftos(radarmapper.ltime), " --qual ", ftos(radarmapper.size.z)), "\"\n"));
233 void RadarMap_Think()
237 // size: pixel width/height
238 // maxs: cell width/height
247 get_mi_min_max_texcoords(1);
248 self.mins = mi_picmin;
249 self.maxs_x = (mi_picmax.x - mi_picmin.x) / self.size.x;
250 self.maxs_y = (mi_picmax.y - mi_picmin.y) / self.size.y;
251 self.maxs_z = mi_max.z - mi_min.z;
252 print("Picture mins/maxs: ", ftos(self.maxs.x), " and ", ftos(self.maxs.y), " should match\n");
253 self.netname = strzone(strcat("gfx/", mi_shortname, "_radar.xpm"));
254 if(!(self.count & 1))
256 self.cnt = fopen(self.netname, FILE_READ);
258 self.cnt = fopen(strcat("gfx/", mi_shortname, "_radar.tga"), FILE_READ);
260 self.cnt = fopen(strcat("gfx/", mi_shortname, "_radar.png"), FILE_READ);
262 self.cnt = fopen(strcat("gfx/", mi_shortname, "_radar.jpg"), FILE_READ);
264 self.cnt = fopen(strcat("gfx/", mi_shortname, "_mini.tga"), FILE_READ);
266 self.cnt = fopen(strcat("gfx/", mi_shortname, "_mini.png"), FILE_READ);
268 self.cnt = fopen(strcat("gfx/", mi_shortname, "_mini.jpg"), FILE_READ);
273 print(self.netname, " already exists, aborting (you may want to specify --force)\n");
278 self.cnt = fopen(self.netname, FILE_WRITE);
281 print("Error writing ", self.netname, "\n");
286 print("Writing to ", self.netname, "...\n");
287 fputs(self.cnt, "/* XPM */\n");
288 fputs(self.cnt, "static char *RadarMap[] = {\n");
289 fputs(self.cnt, "/* columns rows colors chars-per-pixel */\n");
290 fputs(self.cnt, strcat("\"", ftos(self.size.x), " ", ftos(self.size.y), " 256 2\",\n"));
291 for(i = 0; i < 256; ++i)
293 si = substring(doublehex, i*2, 2);
294 fputs(self.cnt, strcat("\"", si, " c #", si, si, si, "\",\n"));
297 self.nextthink = time;
298 sharpen_init(self.size.x);
300 else if(self.frame <= self.size.y)
302 // fill the sharpen buffer with this line
303 sharpen_shift(self.size.x);
310 for(x = 0; x < self.size.x; ++x)
312 l = RadarMapAtPoint_Block(self.mins.x + x * self.maxs.x, self.mins.y + (self.size.y - self.frame) * self.maxs.y, self.maxs.x, self.maxs.y, self.mins.z, self.maxs.z, self.size.z);
317 for(x = 0; x < self.size.x; ++x)
319 l = RadarMapAtPoint_Trace(self.mins.x + x * self.maxs.x, self.mins.y + (self.size.y - self.frame) * self.maxs.y, self.maxs.x, self.maxs.y, self.mins.z, self.maxs.z, self.size.z);
324 for(x = 0; x < self.size.x; ++x)
326 l = RadarMapAtPoint_Sample(self.mins.x + x * self.maxs.x, self.mins.y + (self.size.y - self.frame) * self.maxs.y, self.maxs.x, self.maxs.y, self.mins.z, self.maxs.z, self.size.z);
331 for(x = 0; x < self.size.x; ++x)
333 l = RadarMapAtPoint_LineBlock(self.mins.x + x * self.maxs.x, self.mins.y + (self.size.y - self.frame) * self.maxs.y, self.maxs.x, self.maxs.y, self.mins.z, self.maxs.z, self.size.z);
339 // do we have enough lines?
342 // write a pixel line
343 fputs(self.cnt, "\"");
344 for(x = 0; x < self.size.x; ++x)
346 l = sharpen_get(x, self.ltime);
347 fputs(self.cnt, substring(doublehex, 2 * floor(l * 256.0), 2));
349 if(self.frame == self.size.y)
350 fputs(self.cnt, "\"\n");
353 fputs(self.cnt, "\",\n");
354 print(ftos(self.size.y - self.frame), " lines left\n");
358 // is this the last line? then write back the missing line
359 if(self.frame == self.size.y)
361 sharpen_shift(self.size.x);
362 // write a pixel line
363 fputs(self.cnt, "\"");
364 for(x = 0; x < self.size.x; ++x)
366 l = sharpen_get(x, self.ltime);
367 fputs(self.cnt, substring(doublehex, 2 * floor(l * 256.0), 2));
369 if(self.frame == self.size.y)
370 fputs(self.cnt, "\"\n");
373 fputs(self.cnt, "\",\n");
374 print(ftos(self.size.y - self.frame), " lines left\n");
379 self.nextthink = time;
384 fputs(self.cnt, "};\n");
386 print("Finished. Please edit data/", self.netname, " with an image editing application and place it in the TGA format in the gfx folder.\n");
391 float RadarMap_Make(float argc)
397 radarmapper = spawn();
398 radarmapper.classname = "radarmapper";
399 radarmapper.think = RadarMap_Think;
400 radarmapper.nextthink = time;
401 radarmapper.count = 8; // default to the --trace method, as it is faster now
402 radarmapper.ltime = 1;
403 radarmapper.size = '512 512 1';
404 for(i = 1; i < argc; ++i)
408 case "--force": { radarmapper.count |= 1; break; }
409 case "--loop": { radarmapper.count |= 2; break; }
410 case "--quit": { radarmapper.count |= 4; break; }
411 case "--block": { radarmapper.count &= ~24; break; }
412 case "--trace": { radarmapper.count &= ~24; radarmapper.count |= 8; break; }
413 case "--sample": { radarmapper.count &= ~24; radarmapper.count |= 16; break; }
414 case "--lineblock": { radarmapper.count |= 24; break; }
415 case "--flags": { ++i; radarmapper.count = stof(argv(i)); break; } // for the recursive call
416 case "--sharpen": { ++i; radarmapper.ltime = stof(argv(i)); break; } // for the recursive call
417 case "--res": // minor alias
418 case "--resolution": { ++i; radarmapper.size_x = stof(argv(i)); ++i; radarmapper.size_y = stof(argv(i)); break; }
419 case "--qual": // minor alias
420 case "--quality": { ++i; radarmapper.size_z = stof(argv(i)); break; }
430 if(radarmapper) // after doing the arguments, see if we successfully went forward.
432 print("Radarmap entity spawned.\n");
433 return true; // if so, don't print usage.