Subversion Repositories Kolibri OS

/*

 * Copyright © 2008, 2009 Intel Corporation

 *

 * Permission is hereby granted, free of charge, to any person obtaining a

 * copy of this software and associated documentation files (the "Software"),

 * to deal in the Software without restriction, including without limitation

 * the rights to use, copy, modify, merge, publish, distribute, sublicense,

 * and/or sell copies of the Software, and to permit persons to whom the

 * Software is furnished to do so, subject to the following conditions:

 *

 * The above copyright notice and this permission notice (including the next

 * paragraph) shall be included in all copies or substantial portions of the

 * Software.

 *

 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL

 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING

 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER

 * DEALINGS IN THE SOFTWARE.

 */

#include 

/** @file main.cpp

 *

 * This file is the main() routine and scaffolding for producing

 * builtin_compiler (which doesn't include builtins itself and is used

 * to generate the profile information for builtin_function.cpp), and

 * for glsl_compiler (which does include builtins and can be used to

 * offline compile GLSL code and examine the resulting GLSL IR.

 */

#include "ast.h"

#include "glsl_parser_extras.h"

#include "ir_optimization.h"

#include "program.h"

#include "loop_analysis.h"

#include "standalone_scaffolding.h"

static void

initialize_context(struct gl_context *ctx, gl_api api)

{

   initialize_context_to_defaults(ctx, api);

   /* The standalone compiler needs to claim support for almost

    * everything in order to compile the built-in functions.

    */

   ctx->Const.GLSLVersion = 150;

   ctx->Extensions.ARB_ES3_compatibility = true;

   ctx->Const.MaxClipPlanes = 8;

   ctx->Const.MaxDrawBuffers = 2;

   /* More than the 1.10 minimum to appease parser tests taken from

    * apps that (hopefully) already checked the number of coords.

    */

   ctx->Const.MaxTextureCoordUnits = 4;

   ctx->Driver.NewShader = _mesa_new_shader;

}

/* Returned string will have 'ctx' as its ralloc owner. */

static char *

load_text_file(void *ctx, const char *file_name)

{

	char *text = NULL;

	size_t size;

	size_t total_read = 0;

	FILE *fp = fopen(file_name, "rb");

	if (!fp) {

		return NULL;

	}

	fseek(fp, 0L, SEEK_END);

	size = ftell(fp);

	fseek(fp, 0L, SEEK_SET);

	text = (char *) ralloc_size(ctx, size + 1);

	if (text != NULL) {

		do {

			size_t bytes = fread(text + total_read,

					     1, size - total_read, fp);

			if (bytes < size - total_read) {

				free(text);

				text = NULL;

				break;

			}

			if (bytes == 0) {

				break;

			}

			total_read += bytes;

		} while (total_read < size);

		text[total_read] = '\0';

	}

	fclose(fp);

	return text;

}

int glsl_es = 0;

int dump_ast = 0;

int dump_hir = 0;

int dump_lir = 0;

int do_link = 0;

const struct option compiler_opts[] = {

   { "glsl-es",  0, &glsl_es,  1 },

   { "dump-ast", 0, &dump_ast, 1 },

   { "dump-hir", 0, &dump_hir, 1 },

   { "dump-lir", 0, &dump_lir, 1 },

   { "link",     0, &do_link,  1 },

   { NULL, 0, NULL, 0 }

};

/**

 * \brief Print proper usage and exit with failure.

 */

void

usage_fail(const char *name)

{

   const char *header =

      "usage: %s [options] \n"

      "\n"

      "Possible options are:\n";

   printf(header, name, name);

   for (const struct option *o = compiler_opts; o->name != 0; ++o) {

      printf("    --%s\n", o->name);

   }

   exit(EXIT_FAILURE);

}

void

compile_shader(struct gl_context *ctx, struct gl_shader *shader)

{

   struct _mesa_glsl_parse_state *state =

      new(shader) _mesa_glsl_parse_state(ctx, shader->Type, shader);

   _mesa_glsl_compile_shader(ctx, shader, dump_ast, dump_hir);

   /* Print out the resulting IR */

   if (!state->error && dump_lir) {

      _mesa_print_ir(shader->ir, state);

   }

   return;

}

int

main(int argc, char **argv)

{

   int status = EXIT_SUCCESS;

   struct gl_context local_ctx;

   struct gl_context *ctx = &local_ctx;

   int c;

   int idx = 0;

   while ((c = getopt_long(argc, argv, "", compiler_opts, &idx)) != -1)

      /* empty */ ;

   if (argc <= optind)

      usage_fail(argv[0]);

   initialize_context(ctx, (glsl_es) ? API_OPENGLES2 : API_OPENGL_COMPAT);

   struct gl_shader_program *whole_program;

   whole_program = rzalloc (NULL, struct gl_shader_program);

   assert(whole_program != NULL);

   whole_program->InfoLog = ralloc_strdup(whole_program, "");

   for (/* empty */; argc > optind; optind++) {

      whole_program->Shaders =

	 reralloc(whole_program, whole_program->Shaders,

		  struct gl_shader *, whole_program->NumShaders + 1);

      assert(whole_program->Shaders != NULL);

      struct gl_shader *shader = rzalloc(whole_program, gl_shader);

      whole_program->Shaders[whole_program->NumShaders] = shader;

      whole_program->NumShaders++;

      const unsigned len = strlen(argv[optind]);

      if (len < 6)

	 usage_fail(argv[0]);

      const char *const ext = & argv[optind][len - 5];

      if (strncmp(".vert", ext, 5) == 0 || strncmp(".glsl", ext, 5) == 0)

	 shader->Type = GL_VERTEX_SHADER;

      else if (strncmp(".geom", ext, 5) == 0)

	 shader->Type = GL_GEOMETRY_SHADER;

      else if (strncmp(".frag", ext, 5) == 0)

	 shader->Type = GL_FRAGMENT_SHADER;

      else

	 usage_fail(argv[0]);

      shader->Source = load_text_file(whole_program, argv[optind]);

      if (shader->Source == NULL) {

	 printf("File \"%s\" does not exist.\n", argv[optind]);

	 exit(EXIT_FAILURE);

      }

      compile_shader(ctx, shader);

      if (!shader->CompileStatus) {

	 printf("Info log for %s:\n%s\n", argv[optind], shader->InfoLog);

	 status = EXIT_FAILURE;

	 break;

      }

   }

   if ((status == EXIT_SUCCESS) && do_link)  {

      link_shaders(ctx, whole_program);

      status = (whole_program->LinkStatus) ? EXIT_SUCCESS : EXIT_FAILURE;

      if (strlen(whole_program->InfoLog) > 0)

	 printf("Info log for linking:\n%s\n", whole_program->InfoLog);

   }

   for (unsigned i = 0; i < MESA_SHADER_TYPES; i++)

      ralloc_free(whole_program->_LinkedShaders[i]);

   ralloc_free(whole_program);

   _mesa_glsl_release_types();

   _mesa_glsl_release_functions();

   return status;

}