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(int 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)
184 float sum = sharpen_getpixel(x, 1);
188 sum -= sharpen_getpixel(x - 1, 0);
189 sum -= sharpen_getpixel(x - 1, 1);
190 sum -= sharpen_getpixel(x - 1, 2);
191 sum -= sharpen_getpixel(x + 1, 0);
192 sum -= sharpen_getpixel(x + 1, 1);
193 sum -= sharpen_getpixel(x + 1, 2);
194 sum -= sharpen_getpixel(x, 0);
195 sum -= sharpen_getpixel(x, 2);
196 return bound(0, sum * a, 1);
198 void sharpen_shift(int w)
200 for(int i = 0; i < w; ++i)
202 sharpen_buffer[i] = sharpen_buffer[i + RADAR_WIDTH_MAX];
203 sharpen_buffer[i + RADAR_WIDTH_MAX] = sharpen_buffer[i + 2 * RADAR_WIDTH_MAX];
204 sharpen_buffer[i + 2 * RADAR_WIDTH_MAX] = 0;
207 void sharpen_init(int w)
209 for(int i = 0; i < w; ++i)
211 sharpen_buffer[i] = 0;
212 sharpen_buffer[i + RADAR_WIDTH_MAX] = 0;
213 sharpen_buffer[i + 2 * RADAR_WIDTH_MAX] = 0;
218 if(radarmapper.count & 4)
222 else if(radarmapper.count & 2)
224 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"));
230 void RadarMap_Think()
234 // size: pixel width/height
235 // maxs: cell width/height
244 get_mi_min_max_texcoords(1);
245 self.mins = mi_picmin;
246 self.maxs_x = (mi_picmax.x - mi_picmin.x) / self.size.x;
247 self.maxs_y = (mi_picmax.y - mi_picmin.y) / self.size.y;
248 self.maxs_z = mi_max.z - mi_min.z;
249 print("Picture mins/maxs: ", ftos(self.maxs.x), " and ", ftos(self.maxs.y), " should match\n");
250 self.netname = strzone(strcat("gfx/", mi_shortname, "_radar.xpm"));
251 if(!(self.count & 1))
253 self.cnt = fopen(self.netname, FILE_READ);
255 self.cnt = fopen(strcat("gfx/", mi_shortname, "_radar.tga"), FILE_READ);
257 self.cnt = fopen(strcat("gfx/", mi_shortname, "_radar.png"), FILE_READ);
259 self.cnt = fopen(strcat("gfx/", mi_shortname, "_radar.jpg"), FILE_READ);
261 self.cnt = fopen(strcat("gfx/", mi_shortname, "_mini.tga"), FILE_READ);
263 self.cnt = fopen(strcat("gfx/", mi_shortname, "_mini.png"), FILE_READ);
265 self.cnt = fopen(strcat("gfx/", mi_shortname, "_mini.jpg"), FILE_READ);
270 print(self.netname, " already exists, aborting (you may want to specify --force)\n");
275 self.cnt = fopen(self.netname, FILE_WRITE);
278 print("Error writing ", self.netname, "\n");
283 print("Writing to ", self.netname, "...\n");
284 fputs(self.cnt, "/* XPM */\n");
285 fputs(self.cnt, "static char *RadarMap[] = {\n");
286 fputs(self.cnt, "/* columns rows colors chars-per-pixel */\n");
287 fputs(self.cnt, strcat("\"", ftos(self.size.x), " ", ftos(self.size.y), " 256 2\",\n"));
288 for(i = 0; i < 256; ++i)
290 si = substring(doublehex, i*2, 2);
291 fputs(self.cnt, strcat("\"", si, " c #", si, si, si, "\",\n"));
294 self.nextthink = time;
295 sharpen_init(self.size.x);
297 else if(self.frame <= self.size.y)
299 // fill the sharpen buffer with this line
300 sharpen_shift(self.size.x);
307 for(x = 0; x < self.size.x; ++x)
309 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);
314 for(x = 0; x < self.size.x; ++x)
316 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);
321 for(x = 0; x < self.size.x; ++x)
323 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);
328 for(x = 0; x < self.size.x; ++x)
330 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);
336 // do we have enough lines?
339 // write a pixel line
340 fputs(self.cnt, "\"");
341 for(x = 0; x < self.size.x; ++x)
343 l = sharpen_get(x, self.ltime);
344 fputs(self.cnt, substring(doublehex, 2 * floor(l * 256.0), 2));
346 if(self.frame == self.size.y)
347 fputs(self.cnt, "\"\n");
350 fputs(self.cnt, "\",\n");
351 print(ftos(self.size.y - self.frame), " lines left\n");
355 // is this the last line? then write back the missing line
356 if(self.frame == self.size.y)
358 sharpen_shift(self.size.x);
359 // write a pixel line
360 fputs(self.cnt, "\"");
361 for(x = 0; x < self.size.x; ++x)
363 l = sharpen_get(x, self.ltime);
364 fputs(self.cnt, substring(doublehex, 2 * floor(l * 256.0), 2));
366 if(self.frame == self.size.y)
367 fputs(self.cnt, "\"\n");
370 fputs(self.cnt, "\",\n");
371 print(ftos(self.size.y - self.frame), " lines left\n");
376 self.nextthink = time;
381 fputs(self.cnt, "};\n");
383 print("Finished. Please edit data/", self.netname, " with an image editing application and place it in the TGA format in the gfx folder.\n");
388 float RadarMap_Make(float argc)
394 radarmapper = spawn();
395 radarmapper.classname = "radarmapper";
396 radarmapper.think = RadarMap_Think;
397 radarmapper.nextthink = time;
398 radarmapper.count = 8; // default to the --trace method, as it is faster now
399 radarmapper.ltime = 1;
400 radarmapper.size = '512 512 1';
401 for(i = 1; i < argc; ++i)
405 case "--force": { radarmapper.count |= 1; break; }
406 case "--loop": { radarmapper.count |= 2; break; }
407 case "--quit": { radarmapper.count |= 4; break; }
408 case "--block": { radarmapper.count &= ~24; break; }
409 case "--trace": { radarmapper.count &= ~24; radarmapper.count |= 8; break; }
410 case "--sample": { radarmapper.count &= ~24; radarmapper.count |= 16; break; }
411 case "--lineblock": { radarmapper.count |= 24; break; }
412 case "--flags": { ++i; radarmapper.count = stof(argv(i)); break; } // for the recursive call
413 case "--sharpen": { ++i; radarmapper.ltime = stof(argv(i)); break; } // for the recursive call
414 case "--res": // minor alias
415 case "--resolution": { ++i; radarmapper.size_x = stof(argv(i)); ++i; radarmapper.size_y = stof(argv(i)); break; }
416 case "--qual": // minor alias
417 case "--quality": { ++i; radarmapper.size_z = stof(argv(i)); break; }
427 if(radarmapper) // after doing the arguments, see if we successfully went forward.
429 print("Radarmap entity spawned.\n");
430 return true; // if so, don't print usage.