1/*
2 * Copyright © 2010 Intel Corporation
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
21 * DEALINGS IN THE SOFTWARE.
22 */
23
24#include <stdio.h>
25#include <string.h>
26#include <errno.h>
27#include <getopt.h>
28
29#include "glcpp.h"
30#include "main/mtypes.h"
31#include "main/shaderobj.h"
32#include "util/strtod.h"
33
34extern int glcpp_parser_debug;
35
36void
37_mesa_reference_shader(struct gl_context *ctx, struct gl_shader **ptr,
38                       struct gl_shader *sh)
39{
40   (void) ctx;
41   *ptr = sh;
42}
43
44/* Read from fp until EOF and return a string of everything read.
45 */
46static char *
47load_text_fp (void *ctx, FILE *fp)
48{
49#define CHUNK 4096
50	char *text = NULL;
51	size_t text_size = 0;
52	size_t total_read = 0;
53	size_t bytes;
54
55	while (1) {
56		if (total_read + CHUNK + 1 > text_size) {
57			text_size = text_size ? text_size * 2 : CHUNK + 1;
58			text = reralloc_size (ctx, text, text_size);
59			if (text == NULL) {
60				fprintf (stderr, "Out of memory\n");
61				return NULL;
62			}
63		}
64		bytes = fread (text + total_read, 1, CHUNK, fp);
65		total_read += bytes;
66
67		if (bytes < CHUNK) {
68			break;
69		}
70	}
71
72	text[total_read] = '\0';
73
74	return text;
75}
76
77static char *
78load_text_file(void *ctx, const char *filename)
79{
80	char *text;
81	FILE *fp;
82
83	if (filename == NULL || strcmp (filename, "-") == 0)
84		return load_text_fp (ctx, stdin);
85
86	fp = fopen (filename, "r");
87	if (fp == NULL) {
88		fprintf (stderr, "Failed to open file %s: %s\n",
89			 filename, strerror (errno));
90		return NULL;
91	}
92
93	text = load_text_fp (ctx, fp);
94
95	fclose(fp);
96
97	return text;
98}
99
100/* Initialize only those things that glcpp cares about.
101 */
102static void
103init_fake_gl_context (struct gl_context *gl_ctx)
104{
105	memset(gl_ctx, 0, sizeof(*gl_ctx));
106	gl_ctx->API = API_OPENGL_COMPAT;
107	gl_ctx->Const.DisableGLSLLineContinuations = false;
108}
109
110static void
111usage (void)
112{
113	fprintf (stderr,
114		 "Usage: glcpp [OPTIONS] [--] [<filename>]\n"
115		 "\n"
116		 "Pre-process the given filename (stdin if no filename given).\n"
117		 "The following options are supported:\n"
118		 "    --disable-line-continuations      Do not interpret lines ending with a\n"
119		 "                                      backslash ('\\') as a line continuation.\n");
120}
121
122enum {
123	DISABLE_LINE_CONTINUATIONS_OPT = CHAR_MAX + 1
124};
125
126static const struct option
127long_options[] = {
128	{"disable-line-continuations", no_argument, 0, DISABLE_LINE_CONTINUATIONS_OPT },
129        {"debug",                      no_argument, 0, 'd'},
130	{0,                            0,           0, 0 }
131};
132
133int
134main (int argc, char *argv[])
135{
136	char *filename = NULL;
137	void *ctx = ralloc(NULL, void*);
138	char *info_log = ralloc_strdup(ctx, "");
139	const char *shader;
140	int ret;
141	struct gl_context gl_ctx;
142	int c;
143
144	init_fake_gl_context (&gl_ctx);
145
146	while ((c = getopt_long(argc, argv, "d", long_options, NULL)) != -1) {
147		switch (c) {
148		case DISABLE_LINE_CONTINUATIONS_OPT:
149			gl_ctx.Const.DisableGLSLLineContinuations = true;
150			break;
151                case 'd':
152			glcpp_parser_debug = 1;
153			break;
154		default:
155			usage ();
156			exit (1);
157		}
158	}
159
160	if (optind + 1 < argc) {
161		printf ("Unexpected argument: %s\n", argv[optind+1]);
162		usage ();
163		exit (1);
164	}
165	if (optind < argc) {
166		filename = argv[optind];
167	}
168
169	shader = load_text_file (ctx, filename);
170	if (shader == NULL)
171	   return 1;
172
173	_mesa_locale_init();
174
175	ret = glcpp_preprocess(ctx, &shader, &info_log, NULL, NULL, &gl_ctx);
176
177	printf("%s", shader);
178	fprintf(stderr, "%s", info_log);
179
180	ralloc_free(ctx);
181
182	return ret;
183}
184