src/common/readoct.c

#ifndef lint
static const char       RCSid[] = "$Id: readoct.c,v 2.30 2016/03/04 00:21:21 greg Exp $";
#endif
/*
 *  readoct.c - routines to read octree information.
 */

#include "copyright.h"

#include  <stdio.h>
#include  <time.h>

#include  "platform.h"
#include  "paths.h"
#include  "standard.h"
#include  "octree.h"
#include  "object.h"
#include  "otypes.h"
#include  "resolu.h"

static double  ogetflt(void);
static long  ogetint(int);
static char  *ogetstr(char *);
static int  nonsurfintree(OCTREE ot);
static void  octerror(int  etyp, char  *msg);
static void  skiptree(void);
static OCTREE  getfullnode(void), gettree(void);

static char  *infn;                     /* input file specification */
static FILE  *infp;                     /* input file stream */
static int  objsize;                    /* size of stored OBJECT's */
static OBJECT  objorig;                 /* zeroeth object */
static OBJECT  fnobjects;               /* number of objects in this file */


int
readoct(                                /* read in octree file or stream */
        char  *inpspec,
        int  load,
        CUBE  *scene,
        char  *ofn[]
)
{
        char  sbuf[512];
        int  nf;
        int  i;
        long  m;
        
        if (inpspec == NULL) {
                infn = "standard input";
                infp = stdin;
        } else if (inpspec[0] == '!') {
                infn = inpspec;
                if ((infp = popen(inpspec+1, "r")) == NULL) {
                        sprintf(errmsg, "cannot execute \"%s\"", inpspec);
                        error(SYSTEM, errmsg);
                }
        } else {
                infn = inpspec;
                if ((infp = fopen(inpspec, "r")) == NULL) {
                        sprintf(errmsg, "cannot open octree file \"%s\"",
                                        inpspec);
                        error(SYSTEM, errmsg);
                }
        }
        SET_FILE_BINARY(infp);
                                        /* get header */
        if (checkheader(infp, OCTFMT, load&IO_INFO ? stdout : (FILE *)NULL) < 0)
                octerror(USER, "not an octree");
                                        /* check format */
        if ((objsize = ogetint(2)-OCTMAGIC) <= 0 ||
                        objsize > MAXOBJSIZ || objsize > sizeof(long))
                octerror(USER, "incompatible octree format");
                                        /* get boundaries */
        if (load & IO_BOUNDS) {
                for (i = 0; i < 3; i++)
                        scene->cuorg[i] = atof(ogetstr(sbuf));
                scene->cusize = atof(ogetstr(sbuf));
        } else {
                for (i = 0; i < 4; i++)
                        ogetstr(sbuf);
        }
        objorig = nobjects;             /* set object offset */
        nf = 0;                         /* get object files */
        while (*ogetstr(sbuf)) {
                if (load & IO_SCENE)
                        readobj(sbuf);
                if (load & IO_FILES)
                        ofn[nf] = savqstr(sbuf);
                nf++;
        }
        if (load & IO_FILES)
                ofn[nf] = NULL;
                                        /* get number of objects */
        fnobjects = m = ogetint(objsize);
        if (fnobjects != m)
                octerror(USER, "too many objects");

        if (load & IO_TREE)             /* get the octree */
                scene->cutree = gettree();
        else if (load & IO_SCENE && nf == 0)
                skiptree();
                
        if (load & IO_SCENE) {          /* get the scene */
            if (nf == 0) {
                                        /* load binary scene data */
                readscene(infp, objsize);

            } else {                    /* consistency checks */
                                /* check object count */
                if (nobjects != objorig+fnobjects)
                        octerror(USER, "bad object count; octree stale?");
                                /* check for non-surfaces */
                if (nonsurfintree(scene->cutree))
                        octerror(USER, "modifier in tree; octree stale?");
            }
        }
                                /* close the input */
        if (infn[0] == '!')
                pclose(infp);
        else if (infp != stdin)
                fclose(infp);
        return(nf);
}


static char *
ogetstr(char *s)                /* get null-terminated string */
{
        extern char  *getstr();

        if (getstr(s, infp) == NULL)
                octerror(USER, "truncated octree");
        return(s);
}


static OCTREE
getfullnode()                   /* get a set, return fullnode */
{
        OBJECT  set[MAXSET+1];
        register int  i;
        register long  m;

        if ((set[0] = ogetint(objsize)) > MAXSET)
                octerror(USER, "bad set in getfullnode");
        for (i = 1; i <= set[0]; i++) {
                m = ogetint(objsize) + objorig;
                if ((set[i] = m) != m)
                        octerror(USER, "too many objects");
        }
        return(fullnode(set));
}       


static long
ogetint(int siz)                /* get a siz-byte integer */
{
        extern long  getint();
        register long  r;

        r = getint(siz, infp);
        if (feof(infp))
                octerror(USER, "truncated octree");
        return(r);
}


static double
ogetflt()                       /* get a floating point number */
{
        extern double  getflt();
        double  r;

        r = getflt(infp);
        if (feof(infp))
                octerror(USER, "truncated octree");
        return(r);
}
        

static OCTREE
gettree()                       /* get a pre-ordered octree */
{
        register OCTREE  ot;
        register int  i;
        
        switch (getc(infp)) {
        case OT_EMPTY:
                return(EMPTY);
        case OT_FULL:
                return(getfullnode());
        case OT_TREE:
                if ((ot = octalloc()) == EMPTY)
                        octerror(SYSTEM, "out of tree space in gettree");
                for (i = 0; i < 8; i++)
                        octkid(ot, i) = gettree();
                return(ot);
        case EOF:
                octerror(USER, "truncated octree");
        default:
                octerror(USER, "damaged octree");
        }
        return EMPTY; /* pro forma return */
}


static int
nonsurfintree(OCTREE ot)                /* check tree for modifiers */
{
        OBJECT  set[MAXSET+1];
        register int  i;

        if (isempty(ot))
                return(0);
        if (istree(ot)) {
                for (i = 0; i < 8; i++)
                        if (nonsurfintree(octkid(ot, i)))
                                return(1);
                return(0);
        }
        objset(set, ot);
        for (i = set[0]; i > 0; i-- )
                if (ismodifier(objptr(set[i])->otype))
                        return(1);
        return(0);
}


static void
skiptree(void)                          /* skip octree on input */
{
        register int  i;
        
        switch (getc(infp)) {
        case OT_EMPTY:
                return;
        case OT_FULL:
                for (i = ogetint(objsize)*objsize; i-- > 0; )
                        if (getc(infp) == EOF)
                                octerror(USER, "truncated octree");
                return;
        case OT_TREE:
                for (i = 0; i < 8; i++)
                        skiptree();
                return;
        case EOF:
                octerror(USER, "truncated octree");
        default:
                octerror(USER, "damaged octree");
        }
}


static void
octerror(int etyp, char *msg)           /* octree error */
{
        char  msgbuf[128];

        sprintf(msgbuf, "(%s): %s", infn, msg);
        error(etyp, msgbuf);
}
Revision:	2.31
Committed:	Sat May 4 00:32:47 2019 UTC (5 years ago) by greg
Content type:	text/plain
Branch:	MAIN
CVS Tags:	rad5R3
Changes since 2.30:	+2 -2 lines
Log Message:	Fixed issue with recovering after octree read from stdin
#	User	Rev	Content
1	greg	1.1	#ifndef lint
2	greg	2.31	static const char RCSid[] = "$Id: readoct.c,v 2.30 2016/03/04 00:21:21 greg Exp $";
3	greg	1.1	#endif
4			/*
5			* readoct.c - routines to read octree information.
6	greg	2.13	*/
7
8	greg	2.14	#include "copyright.h"
9	greg	1.1
10	schorsch	2.17	#include <stdio.h>
11	greg	2.19	#include <time.h>
12	schorsch	2.17
13	greg	2.23	#include "platform.h"
14	greg	2.29	#include "paths.h"
15	greg	1.1	#include "standard.h"
16			#include "octree.h"
17			#include "object.h"
18			#include "otypes.h"
19	schorsch	2.17	#include "resolu.h"
20	greg	1.1
21	schorsch	2.20	static double ogetflt(void);
22			static long ogetint(int);
23			static char ogetstr(char );
24	greg	2.27	static int nonsurfintree(OCTREE ot);
25	schorsch	2.17	static void octerror(int etyp, char *msg);
26			static void skiptree(void);
27	schorsch	2.20	static OCTREE getfullnode(void), gettree(void);
28	greg	1.1
29	greg	2.13	static char infn; / input file specification */
30	greg	1.1	static FILE infp; / input file stream */
31	greg	1.14	static int objsize; /* size of stored OBJECT's */
32	greg	1.1	static OBJECT objorig; /* zeroeth object */
33	greg	2.12	static OBJECT fnobjects; /* number of objects in this file */
34	greg	1.1
35
36			int
37	greg	2.27	readoct( /* read in octree file or stream */
38			char *inpspec,
39			int load,
40			CUBE *scene,
41			char *ofn[]
42			)
43	greg	1.1	{
44	greg	1.11	char sbuf[512];
45	greg	1.1	int nf;
46	greg	2.15	int i;
47	greg	1.15	long m;
48	greg	1.1
49	greg	2.13	if (inpspec == NULL) {
50	greg	1.1	infn = "standard input";
51			infp = stdin;
52	greg	2.13	} else if (inpspec[0] == '!') {
53			infn = inpspec;
54			if ((infp = popen(inpspec+1, "r")) == NULL) {
55			sprintf(errmsg, "cannot execute \"%s\"", inpspec);
56			error(SYSTEM, errmsg);
57			}
58	greg	1.1	} else {
59	greg	2.13	infn = inpspec;
60			if ((infp = fopen(inpspec, "r")) == NULL) {
61	greg	1.1	sprintf(errmsg, "cannot open octree file \"%s\"",
62	greg	2.13	inpspec);
63	greg	1.1	error(SYSTEM, errmsg);
64			}
65			}
66	schorsch	2.16	SET_FILE_BINARY(infp);
67	greg	1.1	/* get header */
68	greg	2.11	if (checkheader(infp, OCTFMT, load&IO_INFO ? stdout : (FILE *)NULL) < 0)
69	greg	1.12	octerror(USER, "not an octree");
70	greg	1.1	/* check format */
71	greg	2.4	if ((objsize = ogetint(2)-OCTMAGIC) <= 0 \|\|
72	greg	1.14	objsize > MAXOBJSIZ \|\| objsize > sizeof(long))
73	greg	1.12	octerror(USER, "incompatible octree format");
74	greg	1.1	/* get boundaries */
75			if (load & IO_BOUNDS) {
76			for (i = 0; i < 3; i++)
77	greg	2.4	scene->cuorg[i] = atof(ogetstr(sbuf));
78			scene->cusize = atof(ogetstr(sbuf));
79	greg	1.1	} else {
80			for (i = 0; i < 4; i++)
81	greg	2.4	ogetstr(sbuf);
82	greg	1.1	}
83			objorig = nobjects; /* set object offset */
84			nf = 0; /* get object files */
85	greg	2.4	while (*ogetstr(sbuf)) {
86	greg	1.1	if (load & IO_SCENE)
87			readobj(sbuf);
88			if (load & IO_FILES)
89			ofn[nf] = savqstr(sbuf);
90			nf++;
91			}
92			if (load & IO_FILES)
93			ofn[nf] = NULL;
94	greg	1.8	/* get number of objects */
95	greg	2.4	fnobjects = m = ogetint(objsize);
96	greg	1.15	if (fnobjects != m)
97			octerror(USER, "too many objects");
98	greg	1.1
99	greg	2.13	if (load & IO_TREE) /* get the octree */
100	greg	1.1	scene->cutree = gettree();
101	greg	2.5	else if (load & IO_SCENE && nf == 0)
102			skiptree();
103
104	schorsch	2.21	if (load & IO_SCENE) { /* get the scene */
105	greg	2.5	if (nf == 0) {
106	greg	2.15	/* load binary scene data */
107			readscene(infp, objsize);
108
109	greg	2.5	} else { /* consistency checks */
110			/* check object count */
111			if (nobjects != objorig+fnobjects)
112			octerror(USER, "bad object count; octree stale?");
113			/* check for non-surfaces */
114	greg	2.27	if (nonsurfintree(scene->cutree))
115	greg	2.5	octerror(USER, "modifier in tree; octree stale?");
116			}
117	schorsch	2.21	}
118	greg	2.13	/* close the input */
119			if (infn[0] == '!')
120			pclose(infp);
121	greg	2.31	else if (infp != stdin)
122	greg	2.13	fclose(infp);
123	greg	1.1	return(nf);
124			}
125
126
127			static char *
128	greg	2.27	ogetstr(char s) / get null-terminated string */
129	greg	1.1	{
130	greg	2.4	extern char *getstr();
131	greg	1.1
132	greg	2.4	if (getstr(s, infp) == NULL)
133			octerror(USER, "truncated octree");
134			return(s);
135	greg	1.1	}
136
137
138			static OCTREE
139			getfullnode() /* get a set, return fullnode */
140			{
141	greg	2.7	OBJECT set[MAXSET+1];
142	greg	1.1	register int i;
143	greg	1.14	register long m;
144	greg	1.1
145	greg	2.4	if ((set[0] = ogetint(objsize)) > MAXSET)
146	greg	1.1	octerror(USER, "bad set in getfullnode");
147	greg	1.14	for (i = 1; i <= set[0]; i++) {
148	greg	2.4	m = ogetint(objsize) + objorig;
149	greg	1.15	if ((set[i] = m) != m)
150	greg	1.14	octerror(USER, "too many objects");
151			}
152	greg	1.1	return(fullnode(set));
153			}
154
155
156			static long
157	greg	2.27	ogetint(int siz) /* get a siz-byte integer */
158	greg	1.1	{
159	greg	2.4	extern long getint();
160	greg	1.3	register long r;
161	greg	1.1
162	greg	2.4	r = getint(siz, infp);
163			if (feof(infp))
164			octerror(USER, "truncated octree");
165	greg	1.1	return(r);
166			}
167
168
169			static double
170	greg	2.4	ogetflt() /* get a floating point number */
171	greg	1.1	{
172	greg	2.4	extern double getflt();
173	greg	2.7	double r;
174	greg	1.1
175	greg	2.4	r = getflt(infp);
176			if (feof(infp))
177			octerror(USER, "truncated octree");
178			return(r);
179	greg	1.1	}
180
181
182			static OCTREE
183			gettree() /* get a pre-ordered octree */
184			{
185	greg	2.7	register OCTREE ot;
186	greg	1.1	register int i;
187
188			switch (getc(infp)) {
189			case OT_EMPTY:
190			return(EMPTY);
191			case OT_FULL:
192			return(getfullnode());
193			case OT_TREE:
194			if ((ot = octalloc()) == EMPTY)
195			octerror(SYSTEM, "out of tree space in gettree");
196			for (i = 0; i < 8; i++)
197			octkid(ot, i) = gettree();
198			return(ot);
199	greg	2.5	case EOF:
200			octerror(USER, "truncated octree");
201			default:
202			octerror(USER, "damaged octree");
203			}
204	greg	2.22	return EMPTY; /* pro forma return */
205	greg	2.12	}
206
207
208			static int
209	greg	2.27	nonsurfintree(OCTREE ot) /* check tree for modifiers */
210	greg	2.12	{
211	greg	2.27	OBJECT set[MAXSET+1];
212	greg	2.12	register int i;
213
214	greg	2.27	if (isempty(ot))
215			return(0);
216			if (istree(ot)) {
217			for (i = 0; i < 8; i++)
218			if (nonsurfintree(octkid(ot, i)))
219			return(1);
220			return(0);
221			}
222			objset(set, ot);
223	greg	2.28	for (i = set[0]; i > 0; i-- )
224	greg	2.27	if (ismodifier(objptr(set[i])->otype))
225	greg	2.12	return(1);
226			return(0);
227	greg	2.5	}
228
229
230	schorsch	2.17	static void
231			skiptree(void) /* skip octree on input */
232	greg	2.5	{
233			register int i;
234
235			switch (getc(infp)) {
236			case OT_EMPTY:
237			return;
238			case OT_FULL:
239	greg	2.6	for (i = ogetint(objsize)*objsize; i-- > 0; )
240			if (getc(infp) == EOF)
241			octerror(USER, "truncated octree");
242	greg	2.5	return;
243			case OT_TREE:
244			for (i = 0; i < 8; i++)
245			skiptree();
246			return;
247	greg	1.1	case EOF:
248			octerror(USER, "truncated octree");
249			default:
250			octerror(USER, "damaged octree");
251			}
252			}
253
254
255	schorsch	2.17	static void
256	greg	2.27	octerror(int etyp, char msg) / octree error */
257	greg	1.1	{
258			char msgbuf[128];
259
260			sprintf(msgbuf, "(%s): %s", infn, msg);
261			error(etyp, msgbuf);
262			}