ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/radiance/ray/src/rt/RtraceSimulManager.cpp
Revision: 2.19
Committed: Wed Nov 6 18:28:52 2024 UTC (5 months, 3 weeks ago) by greg
Branch: MAIN
Changes since 2.18: +3 -1 lines
Log Message:
perf(rxtrace,rxpict,rxpiece,rxcontrib): Improved exit strategy in children

File Contents

# User Rev Content
1 greg 2.1 #ifndef lint
2 greg 2.19 static const char RCSid[] = "$Id: RtraceSimulManager.cpp,v 2.18 2024/10/10 21:02:52 greg Exp $";
3 greg 2.1 #endif
4     /*
5     * RtraceSimulManager.cpp
6     *
7     * Rtrace simulation manager class implementation
8     *
9     * Created by Greg Ward on 2/2/2023.
10     */
11    
12 greg 2.5 #include <unistd.h>
13 greg 2.12 #include <ctype.h>
14 greg 2.1 #include "RtraceSimulManager.h"
15     #include "source.h"
16    
17     // Load octree and prepare renderer
18     bool
19     RadSimulManager::LoadOctree(const char *octn)
20     {
21     if (octname) { // already running?
22     if (octn && !strcmp(octn, octname))
23     return true;
24 greg 2.6 Cleanup();
25 greg 2.1 }
26 greg 2.12 if (!octn) // don't support stdin octree
27 greg 2.1 return false;
28    
29 greg 2.13 NewHeader(octn); // get octree header
30 greg 2.1 ray_init((char *)octn);
31     return true;
32     }
33    
34 greg 2.12 // callback function for loading header
35     static int
36     add2header(char *str, void *p)
37     {
38     if (isheadid(str))
39     return 0;
40     if (isformat(str))
41     return 0;
42    
43     return (*(RadSimulManager *)p).AddHeader(str);
44     }
45    
46     // Prepare header from previous input (or clear)
47     // Normally called during octree load
48     bool
49 greg 2.13 RadSimulManager::NewHeader(const char *inspec)
50 greg 2.12 {
51 greg 2.13 if (hlen) {
52     free(header); header = NULL; hlen = 0;
53 greg 2.12 }
54 greg 2.13 if (!inspec || !*inspec)
55 greg 2.12 return false;
56 greg 2.13 if (inspec[0] == '!') // save command as header?
57     return AddHeader(inspec+1);
58     // attempt to read from file
59     FILE *fp = fopen(inspec, "rb");
60     if (!fp) return false;
61 greg 2.12 bool ok = (getheader(fp, add2header, this) >= 0);
62     fclose(fp);
63     return ok;
64     }
65    
66 greg 2.13 // Add a string to header (adds newline if missing)
67 greg 2.12 bool
68     RadSimulManager::AddHeader(const char *str)
69     {
70     if (!str) return false;
71     int len = strlen(str);
72 greg 2.14 while (len && (str[len-1] == '\n') | (str[len-1] == '\r'))
73 greg 2.13 --len; // don't copy EOL(s)
74     if (!len)
75     return false;
76     if (hlen)
77     header = (char *)realloc(header, hlen+len+2);
78     else
79 greg 2.12 header = (char *)malloc(len+2);
80 greg 2.13 if (!header) {
81     hlen = 0; // XXX should report?
82     return false;
83 greg 2.12 }
84 greg 2.13 memcpy(header+hlen, str, len);
85     hlen += len;
86     header[hlen++] = '\n'; // add single EOL
87     header[hlen] = '\0';
88 greg 2.12 return true;
89     }
90    
91     // helper function to check for white-space and quotations
92     static int
93     check_special(const char *s)
94     {
95     int space_found = 0;
96    
97     while (*s) {
98     if ((*s == '"') | (*s == '\''))
99     return *s; // quotes have priority
100     if (isspace(*s))
101     space_found = *s;
102     s++;
103     }
104     return space_found;
105     }
106    
107     // Append program line to header
108     bool
109 greg 2.15 RadSimulManager::AddHeader(int ac, char *av[])
110 greg 2.12 {
111     if ((ac <= 0) | !av) return false;
112     int len = 0;
113     int n;
114     for (n = 0; n < ac; n++) {
115     if (!av[n]) return false;
116     len += strlen(av[n]) + 3;
117     }
118 greg 2.13 if (hlen) // add to header
119 greg 2.12 header = (char *)realloc(header, hlen+len+1);
120 greg 2.13 else
121 greg 2.12 header = (char *)malloc(len+1);
122 greg 2.13
123 greg 2.12 for (n = 0; n < ac; n++) {
124     int c = check_special(av[n]);
125 greg 2.13 len = strlen(av[n]);
126     if (c | !len) { // need to quote argument?
127     if ((c == '"') | (c == '\n')) c = '\'';
128 greg 2.12 else c = '"';
129     header[hlen++] = c;
130     strcpy(header+hlen, av[n]);
131 greg 2.13 hlen += len;
132 greg 2.12 header[hlen++] = c;
133     } else {
134     strcpy(header+hlen, av[n]);
135 greg 2.13 hlen += len;
136 greg 2.12 }
137     header[hlen++] = ' ';
138     }
139     header[hlen-1] = '\n'; // terminate line
140     header[hlen] = '\0';
141     return true;
142     }
143    
144 greg 2.17 // Look for specific header keyword, return value
145     const char *
146     RadSimulManager::GetHeadStr(const char *key, bool inOK) const
147     {
148     if (!key | !hlen || strchr(key, '\n'))
149     return NULL;
150     if (inOK) // skip leading spaces?
151     while (isspace(*key)) key++;
152    
153     const int klen = strlen(key);
154     if (!klen)
155     return NULL;
156     const char * cp = header;
157     while (*cp) {
158     if (inOK) { // skip leading spaces?
159     while (isspace(*cp) && *cp++ != '\n')
160     ;
161     if (cp[-1] == '\n')
162     continue;
163     }
164     if (!strncmp(cp, key, klen))
165     return cp+klen; // found it!
166    
167     while (*cp && *cp++ != '\n')
168     ;
169     }
170     return NULL;
171     }
172    
173 greg 2.5 // How many processors are available?
174 greg 2.2 int
175     RadSimulManager::GetNCores()
176     {
177 greg 2.6 return sysconf(_SC_NPROCESSORS_ONLN);
178 greg 2.2 }
179    
180 greg 2.1 // Set number of computation threads (0 => #cores)
181     int
182     RadSimulManager::SetThreadCount(int nt)
183     {
184 greg 2.6 if (!Ready())
185     return 0;
186    
187     if (nt <= 0) nt = castonly ? 1 : GetNCores();
188    
189     if (nt == 1)
190     ray_pclose(ray_pnprocs);
191     else if (nt < ray_pnprocs)
192     ray_pclose(ray_pnprocs - nt);
193     else if (nt > ray_pnprocs)
194     ray_popen(nt - ray_pnprocs);
195 greg 2.5
196 greg 2.6 return NThreads();
197 greg 2.1 }
198    
199 greg 2.4 // Assign ray to subthread (fails if NThreads()<2)
200     bool
201     RadSimulManager::SplitRay(RAY *r)
202     {
203 greg 2.6 if (!ray_pnprocs || ThreadsAvailable() < 1)
204 greg 2.4 return false;
205    
206 greg 2.6 return (ray_psend(r) > 0);
207 greg 2.4 }
208    
209     // Process a ray (in subthread), optional result
210     bool
211     RadSimulManager::ProcessRay(RAY *r)
212     {
213 greg 2.6 if (!Ready()) return false;
214    
215     if (!ray_pnprocs) { // single-threaded mode?
216 greg 2.4 samplendx++;
217     rayvalue(r);
218     return true;
219     }
220 greg 2.6 int rv = ray_pqueue(r);
221     if (rv < 0) {
222     error(WARNING, "ray tracing process(es) died");
223 greg 2.4 return false;
224     }
225 greg 2.6 return (rv > 0);
226 greg 2.4 }
227    
228     // Wait for next result (or fail)
229     bool
230     RadSimulManager::WaitResult(RAY *r)
231     {
232 greg 2.6 if (!ray_pnprocs)
233     return false;
234    
235     return (ray_presult(r, 0) > 0);
236 greg 2.4 }
237    
238 greg 2.1 // Close octree, free data, return status
239     int
240 greg 2.3 RadSimulManager::Cleanup(bool everything)
241 greg 2.1 {
242 greg 2.19 if (ray_pnprocs < 0)
243     return 0; // skip in child process
244 greg 2.13 NewHeader();
245 greg 2.6 if (!ray_pnprocs)
246 greg 2.11 ray_done(everything);
247     else
248 greg 2.5 ray_pdone(everything);
249 greg 2.1 return 0;
250     }
251    
252     // How many threads are currently unoccupied?
253     int
254     RadSimulManager::ThreadsAvailable() const
255     {
256 greg 2.6 if (!ray_pnprocs) return 1;
257    
258 greg 2.5 return ray_pnidle;
259 greg 2.1 }
260    
261     // Global pointer to simulation manager for trace call-back (only one)
262     static const RtraceSimulManager * ourRTsimMan = NULL;
263    
264 greg 2.7 // Call-back for trace output
265     void
266 greg 2.1 RtraceSimulManager::RTracer(RAY *r)
267     {
268     (*ourRTsimMan->traceCall)(r, ourRTsimMan->tcData);
269     }
270    
271 greg 2.6 // Call-back for FIFO output
272     int
273     RtraceSimulManager::Rfifout(RAY *r)
274     {
275     return (*ourRTsimMan->cookedCall)(r, ourRTsimMan->ccData);
276     }
277    
278 greg 2.1 // Check for changes to render flags & adjust accordingly
279     bool
280     RtraceSimulManager::UpdateMode()
281     {
282     rtFlags &= RTmask;
283     if (!cookedCall)
284     rtFlags &= ~RTdoFIFO;
285     if (!traceCall)
286     rtFlags &= ~RTtraceSources;
287     if (rtFlags & RTimmIrrad)
288     rtFlags &= ~RTlimDist;
289    
290     int misMatch = rtFlags ^ curFlags;
291     // updates based on toggled flags
292     if (misMatch & RTtraceSources) {
293 greg 2.18 int sn = nsources;
294 greg 2.1 if (rtFlags & RTtraceSources) {
295 greg 2.18 srcFollowed.NewBitMap(nsources);
296     while (sn--) {
297     if (source[sn].sflags & SFOLLOW)
298     continue;
299 greg 2.1 source[sn].sflags |= SFOLLOW;
300 greg 2.18 srcFollowed.Set(sn);
301     }
302     } else {
303     while (sn--)
304     if (srcFollowed.Check(sn))
305     source[sn].sflags &= ~SFOLLOW;
306     srcFollowed.NewBitMap(0);
307     }
308 greg 2.1 }
309 greg 2.4 if (misMatch & RTdoFIFO && FlushQueue() < 0)
310     return false;
311 greg 2.1 curFlags = rtFlags;
312 greg 2.6 // update callbacks
313     if (traceCall)
314 greg 2.1 trace = RTracer;
315 greg 2.6 else if (trace == RTracer)
316 greg 2.1 trace = NULL;
317 greg 2.6 if (rtFlags & RTdoFIFO)
318     ray_fifo_out = Rfifout;
319     else if (ray_fifo_out == Rfifout)
320     ray_fifo_out = NULL;
321     if ((trace != RTracer) & (ray_fifo_out != Rfifout)) {
322 greg 2.1 ourRTsimMan = NULL;
323 greg 2.6 } else if (ourRTsimMan != this) {
324     if (ourRTsimMan)
325     error(WARNING, "Competing top-level simulation managers?");
326     ourRTsimMan = this;
327 greg 2.1 }
328     return true;
329     }
330    
331     extern "C" int m_normal(OBJREC *m, RAY *r);
332    
333 greg 2.6 // compute irradiance rather than radiance
334 greg 2.1 static void
335     rayirrad(RAY *r)
336     {
337     /* pretend we hit surface */
338     r->rxt = r->rot = 1e-5;
339     VSUM(r->rop, r->rorg, r->rdir, r->rot);
340     r->ron[0] = -r->rdir[0];
341     r->ron[1] = -r->rdir[1];
342     r->ron[2] = -r->rdir[2];
343     r->rod = 1.0;
344     /* compute result */
345     r->revf = raytrace;
346     m_normal(&Lamb, r);
347     r->revf = rayirrad;
348     }
349    
350 greg 2.6 // compute first ray intersection only
351 greg 2.1 static void
352     raycast(RAY *r)
353     {
354     if (!localhit(r, &thescene)) {
355     if (r->ro == &Aftplane) { /* clipped */
356     r->ro = NULL;
357     r->rot = FHUGE;
358     } else
359     sourcehit(r);
360     }
361     }
362    
363     // Add ray bundle to queue w/ optional 1st ray ID
364     int
365     RtraceSimulManager::EnqueueBundle(const FVECT orig_direc[], int n, RNUMBER rID0)
366     {
367     int nqueued = 0;
368     RAY res;
369    
370     if (!Ready())
371     return -1;
372    
373     if (castonly && !cookedCall)
374 greg 2.8 error(INTERNAL, "EnqueueBundle() called in castonly mode without cookedCall");
375 greg 2.1
376     if (!UpdateMode()) // update rendering mode if requested
377     return -1;
378    
379 greg 2.9 if (rID0 && curFlags&RTdoFIFO)
380     error(INTERNAL, "Ray number assignment unsupported with FIFO");
381    
382 greg 2.1 while (n-- > 0) { // queue each ray
383     VCOPY(res.rorg, orig_direc[0]);
384     VCOPY(res.rdir, orig_direc[1]);
385 greg 2.16 res.rmax = .0;
386 greg 2.1 orig_direc += 2;
387     rayorigin(&res, PRIMARY, NULL, NULL);
388 greg 2.9 res.rno = rID0 ? (lastRayID = rID0++) : ++lastRayID;
389 greg 2.1 if (curFlags & RTimmIrrad)
390     res.revf = rayirrad;
391     else if (castonly)
392     res.revf = raycast;
393 greg 2.4 double d = normalize(res.rdir);
394     bool sendRes = (cookedCall != NULL);
395 greg 2.10 if (d > .0) { // direction vector is valid?
396 greg 2.1 if (curFlags & RTlimDist)
397     res.rmax = d;
398 greg 2.7 if (((curFlags&RTdoFIFO) != 0) & (ray_pnprocs > 0)) {
399     if (ray_fifo_in(&res) < 0)
400     return -1;
401 greg 2.4 sendRes = false;
402 greg 2.6 } else
403     sendRes &= ProcessRay(&res);
404 greg 2.1 } else if (ThreadsAvailable() < NThreads() &&
405 greg 2.4 FlushQueue() < 0)
406 greg 2.1 return -1;
407 greg 2.6 // may be dummy ray
408     if (sendRes && (*cookedCall)(&res, ccData) < 0)
409     return -1;
410 greg 2.3 nqueued++;
411 greg 2.1 }
412     return nqueued;
413     }
414    
415     // Finish pending rays and complete callbacks
416 greg 2.4 int
417 greg 2.1 RtraceSimulManager::FlushQueue()
418     {
419 greg 2.7 if (curFlags & RTdoFIFO) {
420     if (ray_pnprocs)
421     return ray_fifo_flush();
422     return 0;
423     }
424 greg 2.4 int nsent = 0;
425     RAY res;
426    
427     while (WaitResult(&res)) {
428     if (!cookedCall) continue;
429 greg 2.6 int rv = (*cookedCall)(&res, ccData);
430     if (rv < 0) return -1;
431     nsent += rv;
432 greg 2.4 }
433     return nsent;
434 greg 2.1 }