src/hd/rholo2l.c

#ifndef lint
static const char       RCSid[] = "$Id: rholo2l.c,v 3.13 2003/06/26 00:58:10 schorsch Exp $";
#endif
/*
 * Routines for local rtrace execution
 */

#include <signal.h>
#include <sys/time.h>
#include <string.h>

#include "rholo.h"
#include "random.h"
#include "paths.h"
#include "selcall.h"
#include "rtprocess.h"

#ifndef MAXPROC
#define MAXPROC         64
#endif

int     nprocs = 0;                             /* running process count */

static char     pfile[] = TEMPLATE;             /* persist file name */

static SUBPROC  rtpd[MAXPROC];          /* process descriptors */
static float    *rtbuf = NULL;                  /* allocated i/o buffer */
static int      maxqlen = 0;                    /* maximum packets per queue */

static PACKET   *pqueue[MAXPROC];               /* packet queues */
static int      pqlen[MAXPROC];                 /* packet queue lengths */


int
start_rtrace()                  /* start rtrace process */
{
        static char     buf1[8];
        int     rmaxpack = 0;
        int     psiz, n;
                                        /* get number of processes */
        if (ncprocs <= 0 || nprocs > 0)
                return(0);
        if (ncprocs > MAXPROC) {
                sprintf(errmsg,
                        "number of rtrace processes reduced from %d to %d",
                                ncprocs, MAXPROC);
                error(WARNING, errmsg);
                ncprocs = MAXPROC;
        }
        if (rtargv[rtargc-1] != vval(OCTREE)) {
                                                /* add compulsory options */
                rtargv[rtargc++] = "-i-";
                rtargv[rtargc++] = "-I-";
                rtargv[rtargc++] = "-h-";
                rtargv[rtargc++] = "-ld-";
                sprintf(buf1, "%d", RPACKSIZ);
                rtargv[rtargc++] = "-x"; rtargv[rtargc++] = buf1;
                rtargv[rtargc++] = "-y"; rtargv[rtargc++] = "0";
                rtargv[rtargc++] = "-fff";
                rtargv[rtargc++] = vbool(VDIST) ? "-ovl" : "-ovL";
                if (nowarn)
                        rtargv[rtargc++] = "-w-";
                if (ncprocs > 1) {
                        mktemp(pfile);
                        rtargv[rtargc++] = "-PP"; rtargv[rtargc++] = pfile;
                }
                rtargv[rtargc++] = vval(OCTREE);
                rtargv[rtargc] = NULL;
        }
        maxqlen = 0;
        for (nprocs = 0; nprocs < ncprocs; nprocs++) {  /* spawn children */
                psiz = open_process(&rtpd[nprocs], rtargv);
                if (psiz <= 0)
                        error(SYSTEM, "cannot start rtrace process");
                n = psiz/(RPACKSIZ*6*sizeof(float));
                if (maxqlen == 0) {
                        if (!(maxqlen = n))
                                error(INTERNAL,
                                        "bad pipe buffer size assumption");
                        sleep(2);
                } else if (n != maxqlen)
                        error(INTERNAL, "varying pipe buffer size!");
                rmaxpack += n;
        }
        rtbuf = (float *)malloc(RPACKSIZ*6*sizeof(float)*maxqlen);
        if (rtbuf == NULL)
                error(SYSTEM, "malloc failure in start_rtrace");
        return(rmaxpack);
}


static int
bestout()                       /* get best process to process packet */
{
        int     cnt;
        register int    pn, i;

        pn = 0;                 /* find shortest queue */
        for (i = 1; i < nprocs; i++)
                if (pqlen[i] < pqlen[pn])
                        pn = i;
                                /* sanity check */
        if (pqlen[pn] == maxqlen)
                return(-1);
        cnt = 0;                /* count number of ties */
        for (i = pn; i < nprocs; i++)
                if (pqlen[i] == pqlen[pn])
                        cnt++;
                                /* break ties fairly */
        if ((cnt = random() % cnt))
                for (i = pn; i < nprocs; i++)
                        if (pqlen[i] == pqlen[pn] && !cnt--)
                                return(i);
        return(pn);
}


int
slots_avail()                   /* count packet slots available */
{
        register int    nslots = 0;
        register int    i;

        for (i = nprocs; i--; )
                nslots += maxqlen - pqlen[i];
        return(nslots);
}


queue_packet(p)                 /* queue up a beam packet */
register PACKET *p;
{
        int     pn, n;
                                /* determine process to write to */
        if ((pn = bestout()) < 0)
                error(INTERNAL, "rtrace input queues are full!");
                                /* write out the packet */
        packrays(rtbuf, p);
        if ((n = p->nr) < RPACKSIZ)     /* add flush block? */
                memset((char *)(rtbuf+6*n++), '\0', 6*sizeof(float));
        if (writebuf(rtpd[pn].w, (char *)rtbuf, 6*sizeof(float)*n) < 0)
                error(SYSTEM, "write error in queue_packet");
        p->next = NULL;
        if (!pqlen[pn]++)       /* add it to the end of the queue */
                pqueue[pn] = p;
        else {
                register PACKET *rpl = pqueue[pn];
                while (rpl->next != NULL)
                        rpl = rpl->next;
                rpl->next = p;
        }
}


PACKET *
get_packets(poll)               /* read packets from rtrace processes */
int     poll;
{
        static struct timeval   tpoll;  /* zero timeval struct */
        fd_set  readset, errset;
        PACKET  *pldone = NULL, *plend;
        register PACKET *p;
        int     n, nr;
        register int    pn;
        float   *bp;
                                        /* prepare select call */
        FD_ZERO(&readset); FD_ZERO(&errset); n = 0;
        for (pn = nprocs; pn--; ) {
                if (pqlen[pn])
                        FD_SET(rtpd[pn].r, &readset);
                FD_SET(rtpd[pn].r, &errset);
                if (rtpd[pn].r >= n)
                        n = rtpd[pn].r + 1;
        }
                                        /* make the call */
        n = select(n, &readset, (fd_set *)NULL, &errset,
                        poll ? &tpoll : (struct timeval *)NULL);
        if (n < 0) {
                if (errno == EINTR)     /* interrupted select call */
                        return(NULL);
                error(SYSTEM, "select call failure in get_packets");
        }
        if (n == 0)                     /* is nothing ready? */
                return(NULL);
                                        /* make read call(s) */
        for (pn = 0; pn < nprocs; pn++) {
                if (!FD_ISSET(rtpd[pn].r, &readset) &&
                                !FD_ISSET(rtpd[pn].r, &errset))
                        continue;
        reread:
                n = read(rtpd[pn].r, (char *)rtbuf,
                                4*sizeof(float)*RPACKSIZ*pqlen[pn]);
                if (n < 0) {
                        if (errno == EINTR | errno == EAGAIN)
                                goto reread;
                        error(SYSTEM, "read error in get_packets");
                }
                if (n == 0)
                        goto eoferr;
                bp = rtbuf;                             /* finish processing */
                for (p = pqueue[pn]; n && p != NULL; p = p->next) {
                        if ((nr = p->nr) < RPACKSIZ)
                                nr++;                   /* add flush block */
                        n -= 4*sizeof(float)*nr;
                        if (n < 0) {                    /* get remainder */
                                n += readbuf(rtpd[pn].r,
                                                (char *)(bp+4*nr)+n, -n);
                                if (n)
                                        goto eoferr;
                        }
                        donerays(p, bp);
                        bp += 4*nr;
                        pqlen[pn]--;
                }
                if (n)                                  /* read past end? */
                        error(INTERNAL, "packet sync error in get_packets");
                                                        /* take from queue */
                if (pldone == NULL)
                        pldone = plend = pqueue[pn];
                else
                        plend->next = pqueue[pn];
                while (plend->next != p)
                        plend = plend->next;
                plend->next = NULL;
                pqueue[pn] = p;
        }
        return(pldone);                         /* return finished packets */
eoferr:
        error(USER, "rtrace process died");
}


PACKET *
do_packets(pl)                  /* queue a packet list, return finished */
register PACKET *pl;
{
        register PACKET *p;
                                        /* consistency check */
        if (nprocs < 1)
                error(CONSISTENCY, "do_packets called with no active process");
                                        /* queue each new packet */
        while (pl != NULL) {
                p = pl; pl = p->next; p->next = NULL;
                queue_packet(p);
        }
        return(get_packets(slots_avail()));     /* return processed packets */
}


PACKET *
flush_queue()                   /* empty all rtrace queues */
{
        PACKET  *rpdone = NULL;
        register PACKET *rpl;
        float   *bp;
        register PACKET *p;
        int     i, n, nr;

        for (i = 0; i < nprocs; i++)
                if (pqlen[i]) {
                        if (rpdone == NULL) {           /* tack on queue */
                                rpdone = rpl = pqueue[i];
                                if ((nr = rpl->nr) < RPACKSIZ) nr++;
                        } else {
                                rpl->next = pqueue[i];
                                nr = 0;
                        }
                        while (rpl->next != NULL) {
                                nr += (rpl = rpl->next)->nr;
                                if (rpl->nr < RPACKSIZ)
                                        nr++;           /* add flush block */
                        }
                        n = readbuf(rtpd[i].r, (char *)rtbuf,
                                        4*sizeof(float)*nr);
                        if (n < 0)
                                error(SYSTEM, "read failure in flush_queue");
                        bp = rtbuf;                     /* process packets */
                        for (p = pqueue[i]; p != NULL; p = p->next) {
                                if ((nr = p->nr) < RPACKSIZ)
                                        nr++;           /* add flush block */
                                n -= 4*sizeof(float)*nr;
                                if (n >= 0)
                                        donerays(p, bp);
                                else
                                        p->nr = 0;      /* short data error */
                                bp += 4*nr;
                        }
                        pqueue[i] = NULL;               /* zero this queue */
                        pqlen[i] = 0;
                }
        return(rpdone);         /* return all packets completed */
}


static
killpersist()                   /* kill persistent process */
{
        FILE    *fp;
        int     pid;

        if ((fp = fopen(pfile, "r")) == NULL)
                return;
        if (fscanf(fp, "%*s %d", &pid) != 1 || kill(pid, SIGALRM) < 0)
                unlink(pfile);
        fclose(fp);
}


int
end_rtrace()                    /* close rtrace process(es) */
{
        int     status = 0, rv;

        if (nprocs > 1)
                killpersist();
        while (nprocs > 0) {
                rv = close_process(&rtpd[--nprocs]);
                if (rv > 0)
                        status = rv;
        }
        free((void *)rtbuf);
        rtbuf = NULL;
        maxqlen = 0;
        return(status);
}
Revision:	3.14
Committed:	Mon Jun 30 14:59:12 2003 UTC (22 years, 4 months ago) by schorsch
Content type:	text/plain
Branch:	MAIN
Changes since 3.13:	+3 -2 lines
Log Message:	Replaced most outdated BSD function calls with their posix equivalents, and cleaned up a few other platform dependencies.
#	User	Rev	Content
1	gregl	3.1	#ifndef lint
2	schorsch	3.14	static const char RCSid[] = "$Id: rholo2l.c,v 3.13 2003/06/26 00:58:10 schorsch Exp $";
3	gregl	3.1	#endif
4			/*
5			* Routines for local rtrace execution
6			*/
7
8	schorsch	3.13	#include <signal.h>
9			#include <sys/time.h>
10	schorsch	3.14	#include <string.h>
11	schorsch	3.13
12	gregl	3.1	#include "rholo.h"
13			#include "random.h"
14	gregl	3.6	#include "paths.h"
15	gregl	3.3	#include "selcall.h"
16	schorsch	3.13	#include "rtprocess.h"
17	gregl	3.1
18			#ifndef MAXPROC
19	gregl	3.8	#define MAXPROC 64
20	gregl	3.1	#endif
21
22	gregl	3.8	int nprocs = 0; /* running process count */
23
24	gregl	3.6	static char pfile[] = TEMPLATE; /* persist file name */
25	gregl	3.1
26	schorsch	3.13	static SUBPROC rtpd[MAXPROC]; /* process descriptors */
27	gregl	3.1	static float rtbuf = NULL; / allocated i/o buffer */
28			static int maxqlen = 0; /* maximum packets per queue */
29
30			static PACKET pqueue[MAXPROC]; / packet queues */
31			static int pqlen[MAXPROC]; /* packet queue lengths */
32
33
34			int
35			start_rtrace() /* start rtrace process */
36			{
37			static char buf1[8];
38			int rmaxpack = 0;
39	gregl	3.9	int psiz, n;
40	gregl	3.1	/* get number of processes */
41	gregl	3.10	if (ncprocs <= 0 \|\| nprocs > 0)
42	gregl	3.1	return(0);
43	gregl	3.9	if (ncprocs > MAXPROC) {
44	gregl	3.1	sprintf(errmsg,
45			"number of rtrace processes reduced from %d to %d",
46	gregl	3.9	ncprocs, MAXPROC);
47	gregl	3.1	error(WARNING, errmsg);
48	gregl	3.9	ncprocs = MAXPROC;
49	gregl	3.1	}
50	gregl	3.9	if (rtargv[rtargc-1] != vval(OCTREE)) {
51			/* add compulsory options */
52			rtargv[rtargc++] = "-i-";
53			rtargv[rtargc++] = "-I-";
54			rtargv[rtargc++] = "-h-";
55			rtargv[rtargc++] = "-ld-";
56			sprintf(buf1, "%d", RPACKSIZ);
57			rtargv[rtargc++] = "-x"; rtargv[rtargc++] = buf1;
58			rtargv[rtargc++] = "-y"; rtargv[rtargc++] = "0";
59			rtargv[rtargc++] = "-fff";
60			rtargv[rtargc++] = vbool(VDIST) ? "-ovl" : "-ovL";
61			if (nowarn)
62			rtargv[rtargc++] = "-w-";
63			if (ncprocs > 1) {
64			mktemp(pfile);
65			rtargv[rtargc++] = "-PP"; rtargv[rtargc++] = pfile;
66			}
67			rtargv[rtargc++] = vval(OCTREE);
68			rtargv[rtargc] = NULL;
69	gregl	3.1	}
70			maxqlen = 0;
71	gregl	3.9	for (nprocs = 0; nprocs < ncprocs; nprocs++) { /* spawn children */
72	schorsch	3.13	psiz = open_process(&rtpd[nprocs], rtargv);
73	gregl	3.1	if (psiz <= 0)
74			error(SYSTEM, "cannot start rtrace process");
75			n = psiz/(RPACKSIZ6sizeof(float));
76			if (maxqlen == 0) {
77			if (!(maxqlen = n))
78			error(INTERNAL,
79			"bad pipe buffer size assumption");
80	gregl	3.11	sleep(2);
81	gregl	3.1	} else if (n != maxqlen)
82			error(INTERNAL, "varying pipe buffer size!");
83			rmaxpack += n;
84			}
85			rtbuf = (float )malloc(RPACKSIZ6sizeof(float)maxqlen);
86			if (rtbuf == NULL)
87			error(SYSTEM, "malloc failure in start_rtrace");
88			return(rmaxpack);
89			}
90
91
92			static int
93			bestout() /* get best process to process packet */
94			{
95			int cnt;
96			register int pn, i;
97
98			pn = 0; /* find shortest queue */
99			for (i = 1; i < nprocs; i++)
100			if (pqlen[i] < pqlen[pn])
101			pn = i;
102			/* sanity check */
103			if (pqlen[pn] == maxqlen)
104			return(-1);
105			cnt = 0; /* count number of ties */
106			for (i = pn; i < nprocs; i++)
107			if (pqlen[i] == pqlen[pn])
108			cnt++;
109			/* break ties fairly */
110			if ((cnt = random() % cnt))
111			for (i = pn; i < nprocs; i++)
112			if (pqlen[i] == pqlen[pn] && !cnt--)
113			return(i);
114			return(pn);
115			}
116
117
118			int
119			slots_avail() /* count packet slots available */
120			{
121			register int nslots = 0;
122			register int i;
123
124			for (i = nprocs; i--; )
125			nslots += maxqlen - pqlen[i];
126			return(nslots);
127			}
128
129
130			queue_packet(p) /* queue up a beam packet */
131			register PACKET *p;
132			{
133			int pn, n;
134			/* determine process to write to */
135			if ((pn = bestout()) < 0)
136			error(INTERNAL, "rtrace input queues are full!");
137			/* write out the packet */
138			packrays(rtbuf, p);
139			if ((n = p->nr) < RPACKSIZ) /* add flush block? */
140	schorsch	3.14	memset((char )(rtbuf+6n++), '\0', 6*sizeof(float));
141	schorsch	3.13	if (writebuf(rtpd[pn].w, (char )rtbuf, 6sizeof(float)*n) < 0)
142	gregl	3.1	error(SYSTEM, "write error in queue_packet");
143			p->next = NULL;
144			if (!pqlen[pn]++) /* add it to the end of the queue */
145			pqueue[pn] = p;
146			else {
147			register PACKET *rpl = pqueue[pn];
148			while (rpl->next != NULL)
149			rpl = rpl->next;
150			rpl->next = p;
151			}
152			}
153
154
155			PACKET *
156			get_packets(poll) /* read packets from rtrace processes */
157			int poll;
158			{
159			static struct timeval tpoll; /* zero timeval struct */
160			fd_set readset, errset;
161			PACKET pldone = NULL, plend;
162			register PACKET *p;
163			int n, nr;
164			register int pn;
165			float *bp;
166			/* prepare select call */
167			FD_ZERO(&readset); FD_ZERO(&errset); n = 0;
168			for (pn = nprocs; pn--; ) {
169			if (pqlen[pn])
170	schorsch	3.13	FD_SET(rtpd[pn].r, &readset);
171			FD_SET(rtpd[pn].r, &errset);
172			if (rtpd[pn].r >= n)
173			n = rtpd[pn].r + 1;
174	gregl	3.1	}
175			/* make the call */
176			n = select(n, &readset, (fd_set *)NULL, &errset,
177			poll ? &tpoll : (struct timeval *)NULL);
178			if (n < 0) {
179			if (errno == EINTR) /* interrupted select call */
180			return(NULL);
181			error(SYSTEM, "select call failure in get_packets");
182			}
183			if (n == 0) /* is nothing ready? */
184			return(NULL);
185			/* make read call(s) */
186			for (pn = 0; pn < nprocs; pn++) {
187	schorsch	3.13	if (!FD_ISSET(rtpd[pn].r, &readset) &&
188			!FD_ISSET(rtpd[pn].r, &errset))
189	gregl	3.1	continue;
190			reread:
191	schorsch	3.13	n = read(rtpd[pn].r, (char *)rtbuf,
192	gregl	3.1	4sizeof(float)RPACKSIZ*pqlen[pn]);
193			if (n < 0) {
194			if (errno == EINTR \| errno == EAGAIN)
195			goto reread;
196			error(SYSTEM, "read error in get_packets");
197			}
198			if (n == 0)
199			goto eoferr;
200			bp = rtbuf; /* finish processing */
201			for (p = pqueue[pn]; n && p != NULL; p = p->next) {
202			if ((nr = p->nr) < RPACKSIZ)
203			nr++; /* add flush block */
204			n -= 4sizeof(float)nr;
205			if (n < 0) { /* get remainder */
206	schorsch	3.13	n += readbuf(rtpd[pn].r,
207	gregl	3.1	(char )(bp+4nr)+n, -n);
208			if (n)
209			goto eoferr;
210			}
211			donerays(p, bp);
212			bp += 4*nr;
213			pqlen[pn]--;
214			}
215			if (n) /* read past end? */
216			error(INTERNAL, "packet sync error in get_packets");
217			/* take from queue */
218	gregl	3.2	if (pldone == NULL)
219	gregl	3.1	pldone = plend = pqueue[pn];
220			else
221			plend->next = pqueue[pn];
222			while (plend->next != p)
223			plend = plend->next;
224			plend->next = NULL;
225			pqueue[pn] = p;
226			}
227			return(pldone); /* return finished packets */
228			eoferr:
229	gregl	3.3	error(USER, "rtrace process died");
230	gregl	3.1	}
231
232
233			PACKET *
234			do_packets(pl) /* queue a packet list, return finished */
235			register PACKET *pl;
236			{
237			register PACKET *p;
238			/* consistency check */
239			if (nprocs < 1)
240			error(CONSISTENCY, "do_packets called with no active process");
241			/* queue each new packet */
242			while (pl != NULL) {
243			p = pl; pl = p->next; p->next = NULL;
244			queue_packet(p);
245			}
246			return(get_packets(slots_avail())); /* return processed packets */
247			}
248
249
250			PACKET *
251			flush_queue() /* empty all rtrace queues */
252			{
253			PACKET *rpdone = NULL;
254			register PACKET *rpl;
255			float *bp;
256			register PACKET *p;
257			int i, n, nr;
258
259			for (i = 0; i < nprocs; i++)
260			if (pqlen[i]) {
261			if (rpdone == NULL) { /* tack on queue */
262			rpdone = rpl = pqueue[i];
263	gregl	3.4	if ((nr = rpl->nr) < RPACKSIZ) nr++;
264	gregl	3.1	} else {
265			rpl->next = pqueue[i];
266	gregl	3.2	nr = 0;
267	gregl	3.1	}
268	gregl	3.2	while (rpl->next != NULL) {
269			nr += (rpl = rpl->next)->nr;
270			if (rpl->nr < RPACKSIZ)
271			nr++; /* add flush block */
272			}
273	schorsch	3.13	n = readbuf(rtpd[i].r, (char *)rtbuf,
274	gregl	3.2	4sizeof(float)nr);
275	gregl	3.1	if (n < 0)
276			error(SYSTEM, "read failure in flush_queue");
277			bp = rtbuf; /* process packets */
278			for (p = pqueue[i]; p != NULL; p = p->next) {
279			if ((nr = p->nr) < RPACKSIZ)
280			nr++; /* add flush block */
281			n -= 4sizeof(float)nr;
282			if (n >= 0)
283			donerays(p, bp);
284			else
285			p->nr = 0; /* short data error */
286			bp += 4*nr;
287			}
288			pqueue[i] = NULL; /* zero this queue */
289			pqlen[i] = 0;
290			}
291			return(rpdone); /* return all packets completed */
292			}
293
294
295			static
296			killpersist() /* kill persistent process */
297			{
298			FILE *fp;
299			int pid;
300
301	gregl	3.6	if ((fp = fopen(pfile, "r")) == NULL)
302	gregl	3.1	return;
303			if (fscanf(fp, "%*s %d", &pid) != 1 \|\| kill(pid, SIGALRM) < 0)
304	gregl	3.6	unlink(pfile);
305	gregl	3.1	fclose(fp);
306			}
307
308
309			int
310			end_rtrace() /* close rtrace process(es) */
311			{
312			int status = 0, rv;
313
314			if (nprocs > 1)
315			killpersist();
316			while (nprocs > 0) {
317	schorsch	3.13	rv = close_process(&rtpd[--nprocs]);
318	gregl	3.1	if (rv > 0)
319			status = rv;
320			}
321	greg	3.12	free((void *)rtbuf);
322	gregl	3.1	rtbuf = NULL;
323			maxqlen = 0;
324			return(status);
325			}