src/util/eplus_idf.c

#ifndef lint
static const char RCSid[] = "$Id$";
#endif
/*
 *  eplus_idf.c
 *
 *  EnergyPlus Input Data File i/o routines
 *
 *  Created by Greg Ward on 1/31/14.
 */

#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <stdlib.h>
#include "eplus_idf.h"

#ifdef getc_unlocked            /* avoid horrendous overhead of flockfile */
#undef getc
#define getc    getc_unlocked
#endif

/* Create a new parameter with empty field list (comment optional) */
IDF_PARAMETER *
idf_newparam(IDF_LOADED *idf, const char *pname, const char *comm,
                        IDF_PARAMETER *prev)
{
        LUENT           *pent;
        IDF_PARAMETER   *pnew;

        if ((idf == NULL) | (pname == NULL))
                return(NULL);
        if (comm == NULL) comm = "";
        pent = lu_find(&idf->ptab, pname);
        if (pent == NULL)
                return(NULL);
        if (pent->key == NULL) {        /* new parameter name/type? */
                pent->key = (char *)malloc(strlen(pname)+1);
                if (pent->key == NULL)
                        return(NULL);
                strcpy(pent->key, pname);
        }
        pnew = (IDF_PARAMETER *)malloc(sizeof(IDF_PARAMETER)+strlen(comm));
        if (pnew == NULL)
                return(NULL);
        strcpy(pnew->rem, comm);
        pnew->flist = NULL;
        pnew->pname = pent->key;        /* add to table */
        pnew->pnext = (IDF_PARAMETER *)pent->data;
        pent->data = (char *)pnew;
        pnew->dnext = NULL;             /* add to file list */
        if (prev != NULL || (prev = idf->plast) != NULL) {
                pnew->dnext = prev->dnext;
                if (prev == idf->plast)
                        idf->plast = pnew;
        }
        if (idf->pfirst == NULL)
                idf->pfirst = idf->plast = pnew;
        else
                prev->dnext = pnew;
        return(pnew);
}

/* Add a field to the given parameter and follow with the given text */
int
idf_addfield(IDF_PARAMETER *param, const char *fval, const char *comm)
{
        int             fnum = 1;       /* returned argument number */
        IDF_FIELD       *fnew, *flast;
        char            *cp;

        if ((param == NULL) | (fval == NULL))
                return(0);
        if (comm == NULL) comm = "";
        fnew = (IDF_FIELD *)malloc(sizeof(IDF_FIELD)+strlen(fval)+strlen(comm));
        if (fnew == NULL)
                return(0);
        fnew->next = NULL;
        cp = fnew->arg;                 /* copy argument and comments */
        while ((*cp++ = *fval++))
                ;
        fnew->rem = cp;
        while ((*cp++ = *comm++))
                ;
                                        /* add to parameter's field list */
        if ((flast = param->flist) != NULL) {
                ++fnum;
                while (flast->next != NULL) {
                        flast = flast->next;
                        ++fnum;
                }
        }
        if (flast == NULL)
                param->flist = fnew;
        else
                flast->next = fnew;
        return(fnum);
}

/* Delete the specified parameter from loaded IDF */
int
idf_delparam(IDF_LOADED *idf, IDF_PARAMETER *param)
{
        LUENT           *pent;
        IDF_PARAMETER   *pptr, *plast;

        if ((idf == NULL) | (param == NULL))
                return(0);
                                        /* remove from parameter table */
        pent = lu_find(&idf->ptab, param->pname);
        for (plast = NULL, pptr = (IDF_PARAMETER *)pent->data;
                                pptr != NULL; plast = pptr, pptr = pptr->pnext)
                if (pptr == param)
                        break;
        if (pptr == NULL)
                return(0);
        if (plast == NULL)
                pent->data = (char *)param->pnext;
        else
                plast->pnext = param->pnext;
                                        /* remove from global list */
        for (plast = NULL, pptr = idf->pfirst;
                                pptr != NULL; plast = pptr, pptr = pptr->dnext)
                if (pptr == param)
                        break;
        if (pptr == NULL)
                return(0);
        if (plast == NULL)
                idf->pfirst = param->dnext;
        else
                plast->dnext = param->dnext;
        if (idf->plast == param)
                idf->plast = plast;
                                        /* free field list */
        while (param->flist != NULL) {
                IDF_FIELD       *fdel = param->flist;
                param->flist = fdel->next;
                free(fdel);
        }
        free(param);                    /* free parameter struct */
        return(1);
}

/* Get a named parameter list */
IDF_PARAMETER *
idf_getparam(IDF_LOADED *idf, const char *pname)
{
        if ((idf == NULL) | (pname == NULL))
                return(NULL);

        return((IDF_PARAMETER *)lu_find(&idf->ptab,pname)->data);
}

/* Read an argument including terminating ',' or ';' -- return which */
static int
idf_read_argument(char *buf, FILE *fp, int trim)
{
        int     skipwhite = trim;
        char    *cp = buf;
        int     c;

        while ((c = getc(fp)) != EOF && (c != ',') & (c != ';')) {
                if (skipwhite && isspace(c))
                        continue;
                skipwhite = 0;
                if (cp-buf < IDF_MAXLINE-1)
                        *cp++ = c;
        }
        if (trim)
                while (cp > buf && isspace(cp[-1]))
                        --cp;
        *cp = '\0';
        return(c);
}

/* Read a comment, including all white space up to next alpha character */
static void
idf_read_comment(char *buf, int len, FILE *fp)
{
        int     incomm = 0;
        char    *cp = buf;
        char    dummyc;
        int     c;

        if ((buf == NULL) | (len <= 0)) {
                buf = &dummyc;
                len = 1;
        }
        while ((c = getc(fp)) != EOF && isspace(c) | incomm) {
                if (c == '!')
                        ++incomm;
                else if (c == '\n')
                        incomm = 0;
                if (cp-buf < len-1)
                        *cp++ = c;
        }
        *cp = '\0';
        if (c != EOF)
                ungetc(c, fp);
}

/* Read a parameter and fields from an open file and add to end of list */
IDF_PARAMETER *
idf_readparam(IDF_LOADED *idf, FILE *fp)
{
        char            abuf[IDF_MAXLINE], cbuf[IDF_MAXLINE];
        int             delim;
        IDF_PARAMETER   *pnew;
        
        if ((delim = idf_read_argument(abuf, fp, 1)) == EOF)
                return(NULL);
        idf_read_comment(cbuf, IDF_MAXLINE, fp);
        pnew = idf_newparam(idf, abuf, cbuf, NULL);
        while (delim == ',')
                if ((delim = idf_read_argument(abuf, fp, 1)) != EOF) {
                        idf_read_comment(cbuf, IDF_MAXLINE, fp);
                        idf_addfield(pnew, abuf, cbuf);
                }
        if (delim != ';')
                fprintf(stderr, "Expected ';' at end of parameter list\n");
        return(pnew);
}

/* Initialize an IDF struct */
IDF_LOADED *
idf_create(const char *hdrcomm)
{
        IDF_LOADED      *idf = (IDF_LOADED *)calloc(1, sizeof(IDF_LOADED));

        if (idf == NULL)
                return(NULL);
        idf->ptab.hashf = &lu_shash;
        idf->ptab.keycmp = &strcmp;
        idf->ptab.freek = &free;
        lu_init(&idf->ptab, 200);
        if (hdrcomm != NULL && *hdrcomm) {
                idf->hrem = (char *)malloc(strlen(hdrcomm)+1);
                if (idf->hrem != NULL)
                        strcpy(idf->hrem, hdrcomm);
        }
        return(idf);
}

/* Load an Input Data File */
IDF_LOADED *
idf_load(const char *fname)
{
        char            *hdrcomm;
        FILE            *fp;
        IDF_LOADED      *idf;

        if (fname == NULL)
                fp = stdin;             /* open file if not stdin */
        else if ((fp = fopen(fname, "r")) == NULL)
                return(NULL);
                                        /* read header comments */
        hdrcomm = (char *)malloc(100*IDF_MAXLINE);
        idf_read_comment(hdrcomm, 100*IDF_MAXLINE, fp);
        idf = idf_create(hdrcomm);      /* create IDF struct */
        free(hdrcomm);
        if (idf == NULL)
                return(NULL);
                                        /* read each parameter */
        while (idf_readparam(idf, fp) != NULL)
                ;
        if (fp != stdin)                /* close file if not stdin */
                fclose(fp);
        return(idf);                    /* success! */
}

/* Write a parameter and fields to an open file */
int
idf_writeparam(IDF_PARAMETER *param, FILE *fp)
{
        IDF_FIELD       *fptr;

        if ((param == NULL) | (fp == NULL))
                return(0);
        fputs(param->pname, fp);
        fputc(',', fp);
        fputs(param->rem, fp);
        for (fptr = param->flist; fptr != NULL; fptr = fptr->next) {
                fputs(fptr->arg, fp);
                fputc((fptr->next==NULL ? ';' : ','), fp);
                fputs(fptr->rem, fp);
        }
        return(!ferror(fp));
}

/* Write out an Input Data File */
int
idf_write(IDF_LOADED *idf, const char *fname)
{
        FILE            *fp;
        IDF_PARAMETER   *pptr;

        if (idf == NULL)
                return(0);
        if (fname == NULL)
                fp = stdout;            /* open file if not stdout */
        else if ((fp = fopen(fname, "w")) == NULL)
                return(0);
        fputs(idf->hrem, fp);           /* write header then parameters */
        for (pptr = idf->pfirst; pptr != NULL; pptr = pptr->dnext)
                if (!idf_writeparam(pptr, fp)) {
                        fclose(fp);
                        return(0);
                }
        if (fp == stdout)               /* flush/close file & check status */
                return(fflush(fp) == 0);
        return(fclose(fp) == 0);
}

/* Free a loaded IDF */
void
idf_free(IDF_LOADED *idf)
{
        if (idf == NULL)
                return;
        if (idf->hrem != NULL)
                free(idf->hrem);
        while (idf->pfirst != NULL)
                idf_delparam(idf, idf->pfirst);
        lu_done(&idf->ptab);
}
Revision:	2.1
Committed:	Sat Feb 1 01:28:43 2014 UTC (10 years, 2 months ago) by greg
Content type:	text/plain
Branch:	MAIN
Log Message:	Added (thus far) untested routines for reading and writing EnergyPlus IDFs
#	User	Rev	Content
1	greg	2.1	#ifndef lint
2			static const char RCSid[] = "$Id$";
3			#endif
4			/*
5			* eplus_idf.c
6			*
7			* EnergyPlus Input Data File i/o routines
8			*
9			* Created by Greg Ward on 1/31/14.
10			*/
11
12			#include <stdio.h>
13			#include <string.h>
14			#include <ctype.h>
15			#include <stdlib.h>
16			#include "eplus_idf.h"
17
18			#ifdef getc_unlocked /* avoid horrendous overhead of flockfile */
19			#undef getc
20			#define getc getc_unlocked
21			#endif
22
23			/* Create a new parameter with empty field list (comment optional) */
24			IDF_PARAMETER *
25			idf_newparam(IDF_LOADED idf, const char pname, const char *comm,
26			IDF_PARAMETER *prev)
27			{
28			LUENT *pent;
29			IDF_PARAMETER *pnew;
30
31			if ((idf == NULL) \| (pname == NULL))
32			return(NULL);
33			if (comm == NULL) comm = "";
34			pent = lu_find(&idf->ptab, pname);
35			if (pent == NULL)
36			return(NULL);
37			if (pent->key == NULL) { /* new parameter name/type? */
38			pent->key = (char *)malloc(strlen(pname)+1);
39			if (pent->key == NULL)
40			return(NULL);
41			strcpy(pent->key, pname);
42			}
43			pnew = (IDF_PARAMETER *)malloc(sizeof(IDF_PARAMETER)+strlen(comm));
44			if (pnew == NULL)
45			return(NULL);
46			strcpy(pnew->rem, comm);
47			pnew->flist = NULL;
48			pnew->pname = pent->key; /* add to table */
49			pnew->pnext = (IDF_PARAMETER *)pent->data;
50			pent->data = (char *)pnew;
51			pnew->dnext = NULL; /* add to file list */
52			if (prev != NULL \|\| (prev = idf->plast) != NULL) {
53			pnew->dnext = prev->dnext;
54			if (prev == idf->plast)
55			idf->plast = pnew;
56			}
57			if (idf->pfirst == NULL)
58			idf->pfirst = idf->plast = pnew;
59			else
60			prev->dnext = pnew;
61			return(pnew);
62			}
63
64			/* Add a field to the given parameter and follow with the given text */
65			int
66			idf_addfield(IDF_PARAMETER param, const char fval, const char *comm)
67			{
68			int fnum = 1; /* returned argument number */
69			IDF_FIELD fnew, flast;
70			char *cp;
71
72			if ((param == NULL) \| (fval == NULL))
73			return(0);
74			if (comm == NULL) comm = "";
75			fnew = (IDF_FIELD *)malloc(sizeof(IDF_FIELD)+strlen(fval)+strlen(comm));
76			if (fnew == NULL)
77			return(0);
78			fnew->next = NULL;
79			cp = fnew->arg; /* copy argument and comments */
80			while ((cp++ = fval++))
81			;
82			fnew->rem = cp;
83			while ((cp++ = comm++))
84			;
85			/* add to parameter's field list */
86			if ((flast = param->flist) != NULL) {
87			++fnum;
88			while (flast->next != NULL) {
89			flast = flast->next;
90			++fnum;
91			}
92			}
93			if (flast == NULL)
94			param->flist = fnew;
95			else
96			flast->next = fnew;
97			return(fnum);
98			}
99
100			/* Delete the specified parameter from loaded IDF */
101			int
102			idf_delparam(IDF_LOADED idf, IDF_PARAMETER param)
103			{
104			LUENT *pent;
105			IDF_PARAMETER pptr, plast;
106
107			if ((idf == NULL) \| (param == NULL))
108			return(0);
109			/* remove from parameter table */
110			pent = lu_find(&idf->ptab, param->pname);
111			for (plast = NULL, pptr = (IDF_PARAMETER *)pent->data;
112			pptr != NULL; plast = pptr, pptr = pptr->pnext)
113			if (pptr == param)
114			break;
115			if (pptr == NULL)
116			return(0);
117			if (plast == NULL)
118			pent->data = (char *)param->pnext;
119			else
120			plast->pnext = param->pnext;
121			/* remove from global list */
122			for (plast = NULL, pptr = idf->pfirst;
123			pptr != NULL; plast = pptr, pptr = pptr->dnext)
124			if (pptr == param)
125			break;
126			if (pptr == NULL)
127			return(0);
128			if (plast == NULL)
129			idf->pfirst = param->dnext;
130			else
131			plast->dnext = param->dnext;
132			if (idf->plast == param)
133			idf->plast = plast;
134			/* free field list */
135			while (param->flist != NULL) {
136			IDF_FIELD *fdel = param->flist;
137			param->flist = fdel->next;
138			free(fdel);
139			}
140			free(param); /* free parameter struct */
141			return(1);
142			}
143
144			/* Get a named parameter list */
145			IDF_PARAMETER *
146			idf_getparam(IDF_LOADED idf, const char pname)
147			{
148			if ((idf == NULL) \| (pname == NULL))
149			return(NULL);
150
151			return((IDF_PARAMETER *)lu_find(&idf->ptab,pname)->data);
152			}
153
154			/* Read an argument including terminating ',' or ';' -- return which */
155			static int
156			idf_read_argument(char buf, FILE fp, int trim)
157			{
158			int skipwhite = trim;
159			char *cp = buf;
160			int c;
161
162			while ((c = getc(fp)) != EOF && (c != ',') & (c != ';')) {
163			if (skipwhite && isspace(c))
164			continue;
165			skipwhite = 0;
166			if (cp-buf < IDF_MAXLINE-1)
167			*cp++ = c;
168			}
169			if (trim)
170			while (cp > buf && isspace(cp[-1]))
171			--cp;
172			*cp = '\0';
173			return(c);
174			}
175
176			/* Read a comment, including all white space up to next alpha character */
177			static void
178			idf_read_comment(char buf, int len, FILE fp)
179			{
180			int incomm = 0;
181			char *cp = buf;
182			char dummyc;
183			int c;
184
185			if ((buf == NULL) \| (len <= 0)) {
186			buf = &dummyc;
187			len = 1;
188			}
189			while ((c = getc(fp)) != EOF && isspace(c) \| incomm) {
190			if (c == '!')
191			++incomm;
192			else if (c == '\n')
193			incomm = 0;
194			if (cp-buf < len-1)
195			*cp++ = c;
196			}
197			*cp = '\0';
198			if (c != EOF)
199			ungetc(c, fp);
200			}
201
202			/* Read a parameter and fields from an open file and add to end of list */
203			IDF_PARAMETER *
204			idf_readparam(IDF_LOADED idf, FILE fp)
205			{
206			char abuf[IDF_MAXLINE], cbuf[IDF_MAXLINE];
207			int delim;
208			IDF_PARAMETER *pnew;
209
210			if ((delim = idf_read_argument(abuf, fp, 1)) == EOF)
211			return(NULL);
212			idf_read_comment(cbuf, IDF_MAXLINE, fp);
213			pnew = idf_newparam(idf, abuf, cbuf, NULL);
214			while (delim == ',')
215			if ((delim = idf_read_argument(abuf, fp, 1)) != EOF) {
216			idf_read_comment(cbuf, IDF_MAXLINE, fp);
217			idf_addfield(pnew, abuf, cbuf);
218			}
219			if (delim != ';')
220			fprintf(stderr, "Expected ';' at end of parameter list\n");
221			return(pnew);
222			}
223
224			/* Initialize an IDF struct */
225			IDF_LOADED *
226			idf_create(const char *hdrcomm)
227			{
228			IDF_LOADED idf = (IDF_LOADED )calloc(1, sizeof(IDF_LOADED));
229
230			if (idf == NULL)
231			return(NULL);
232			idf->ptab.hashf = &lu_shash;
233			idf->ptab.keycmp = &strcmp;
234			idf->ptab.freek = &free;
235			lu_init(&idf->ptab, 200);
236			if (hdrcomm != NULL && *hdrcomm) {
237			idf->hrem = (char *)malloc(strlen(hdrcomm)+1);
238			if (idf->hrem != NULL)
239			strcpy(idf->hrem, hdrcomm);
240			}
241			return(idf);
242			}
243
244			/* Load an Input Data File */
245			IDF_LOADED *
246			idf_load(const char *fname)
247			{
248			char *hdrcomm;
249			FILE *fp;
250			IDF_LOADED *idf;
251
252			if (fname == NULL)
253			fp = stdin; /* open file if not stdin */
254			else if ((fp = fopen(fname, "r")) == NULL)
255			return(NULL);
256			/* read header comments */
257			hdrcomm = (char )malloc(100IDF_MAXLINE);
258			idf_read_comment(hdrcomm, 100*IDF_MAXLINE, fp);
259			idf = idf_create(hdrcomm); /* create IDF struct */
260			free(hdrcomm);
261			if (idf == NULL)
262			return(NULL);
263			/* read each parameter */
264			while (idf_readparam(idf, fp) != NULL)
265			;
266			if (fp != stdin) /* close file if not stdin */
267			fclose(fp);
268			return(idf); /* success! */
269			}
270
271			/* Write a parameter and fields to an open file */
272			int
273			idf_writeparam(IDF_PARAMETER param, FILE fp)
274			{
275			IDF_FIELD *fptr;
276
277			if ((param == NULL) \| (fp == NULL))
278			return(0);
279			fputs(param->pname, fp);
280			fputc(',', fp);
281			fputs(param->rem, fp);
282			for (fptr = param->flist; fptr != NULL; fptr = fptr->next) {
283			fputs(fptr->arg, fp);
284			fputc((fptr->next==NULL ? ';' : ','), fp);
285			fputs(fptr->rem, fp);
286			}
287			return(!ferror(fp));
288			}
289
290			/* Write out an Input Data File */
291			int
292			idf_write(IDF_LOADED idf, const char fname)
293			{
294			FILE *fp;
295			IDF_PARAMETER *pptr;
296
297			if (idf == NULL)
298			return(0);
299			if (fname == NULL)
300			fp = stdout; /* open file if not stdout */
301			else if ((fp = fopen(fname, "w")) == NULL)
302			return(0);
303			fputs(idf->hrem, fp); /* write header then parameters */
304			for (pptr = idf->pfirst; pptr != NULL; pptr = pptr->dnext)
305			if (!idf_writeparam(pptr, fp)) {
306			fclose(fp);
307			return(0);
308			}
309			if (fp == stdout) /* flush/close file & check status */
310			return(fflush(fp) == 0);
311			return(fclose(fp) == 0);
312			}
313
314			/* Free a loaded IDF */
315			void
316			idf_free(IDF_LOADED *idf)
317			{
318			if (idf == NULL)
319			return;
320			if (idf->hrem != NULL)
321			free(idf->hrem);
322			while (idf->pfirst != NULL)
323			idf_delparam(idf, idf->pfirst);
324			lu_done(&idf->ptab);
325			}