| // |
| // Copyright (c) 2002-2010 The ANGLE Project Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| // |
| |
| #include "GLSLANG/ShaderLang.h" |
| |
| #include <math.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| |
| // |
| // Return codes from main. |
| // |
| enum TFailCode { |
| ESuccess = 0, |
| EFailUsage, |
| EFailCompile, |
| EFailCompilerCreate, |
| }; |
| |
| static EShLanguage FindLanguage(char *lang); |
| bool CompileFile(char *fileName, ShHandle, int); |
| void usage(); |
| void FreeFileData(char **data); |
| char** ReadFileData(char *fileName); |
| void LogMsg(char* msg, const char* name, const int num, const char* logName); |
| |
| //Added to accomodate the multiple strings. |
| int OutputMultipleStrings = 1; |
| |
| // |
| // Set up the per compile resources |
| // |
| void GenerateResources(TBuiltInResource* resources) |
| { |
| ShInitBuiltInResource(resources); |
| |
| resources->MaxVertexAttribs = 8; |
| resources->MaxVertexUniformVectors = 128; |
| resources->MaxVaryingVectors = 8; |
| resources->MaxVertexTextureImageUnits = 0; |
| resources->MaxCombinedTextureImageUnits = 8; |
| resources->MaxTextureImageUnits = 8; |
| resources->MaxFragmentUniformVectors = 16; |
| resources->MaxDrawBuffers = 1; |
| |
| resources->OES_standard_derivatives = 0; |
| } |
| |
| int main(int argc, char* argv[]) |
| { |
| TFailCode failCode = ESuccess; |
| |
| int compileOptions = 0; |
| int numCompiles = 0; |
| ShHandle vertexCompiler = 0; |
| ShHandle fragmentCompiler = 0; |
| char* buffer = 0; |
| int bufferLen = 0; |
| |
| ShInitialize(); |
| |
| TBuiltInResource resources; |
| GenerateResources(&resources); |
| |
| argc--; |
| argv++; |
| for (; (argc >= 1) && (failCode == ESuccess); argc--, argv++) { |
| if (argv[0][0] == '-' || argv[0][0] == '/') { |
| switch (argv[0][1]) { |
| case 'i': compileOptions |= EShOptIntermediateTree; break; |
| case 'o': compileOptions |= EShOptObjectCode; break; |
| default: failCode = EFailUsage; |
| } |
| } else { |
| ShHandle compiler = 0; |
| switch (FindLanguage(argv[0])) { |
| case EShLangVertex: |
| if (vertexCompiler == 0) |
| vertexCompiler = ShConstructCompiler(EShLangVertex, EShSpecGLES2, &resources); |
| compiler = vertexCompiler; |
| break; |
| case EShLangFragment: |
| if (fragmentCompiler == 0) |
| fragmentCompiler = ShConstructCompiler(EShLangFragment, EShSpecGLES2, &resources); |
| compiler = fragmentCompiler; |
| break; |
| default: break; |
| } |
| if (compiler) { |
| bool compiled = CompileFile(argv[0], compiler, compileOptions); |
| |
| LogMsg("BEGIN", "COMPILER", numCompiles, "INFO LOG"); |
| ShGetInfo(compiler, SH_INFO_LOG_LENGTH, &bufferLen); |
| buffer = (char*) realloc(buffer, bufferLen * sizeof(char)); |
| ShGetInfoLog(compiler, buffer); |
| puts(buffer); |
| LogMsg("END", "COMPILER", numCompiles, "INFO LOG"); |
| |
| if (compiled && (compileOptions & EShOptObjectCode)) { |
| LogMsg("BEGIN", "COMPILER", numCompiles, "OBJ CODE"); |
| ShGetInfo(compiler, SH_OBJECT_CODE_LENGTH, &bufferLen); |
| buffer = (char*) realloc(buffer, bufferLen * sizeof(char)); |
| ShGetObjectCode(compiler, buffer); |
| puts(buffer); |
| LogMsg("END", "COMPILER", numCompiles, "OBJ CODE"); |
| } |
| if (!compiled) |
| failCode = EFailCompile; |
| ++numCompiles; |
| } else { |
| failCode = EFailCompilerCreate; |
| } |
| } |
| } |
| |
| if ((vertexCompiler == 0) && (fragmentCompiler == 0)) |
| failCode = EFailUsage; |
| if (failCode == EFailUsage) |
| usage(); |
| |
| if (vertexCompiler) |
| ShDestruct(vertexCompiler); |
| if (fragmentCompiler) |
| ShDestruct(fragmentCompiler); |
| if (buffer) |
| free(buffer); |
| ShFinalize(); |
| |
| return failCode; |
| } |
| |
| // |
| // Deduce the language from the filename. Files must end in one of the |
| // following extensions: |
| // |
| // .frag* = fragment programs |
| // .vert* = vertex programs |
| // |
| static EShLanguage FindLanguage(char *name) |
| { |
| if (!name) |
| return EShLangFragment; |
| |
| char *ext = strrchr(name, '.'); |
| |
| if (ext && strcmp(ext, ".sl") == 0) |
| for (; ext > name && ext[0] != '.'; ext--); |
| |
| if (ext = strrchr(name, '.')) { |
| if (strncmp(ext, ".frag", 4) == 0) return EShLangFragment; |
| if (strncmp(ext, ".vert", 4) == 0) return EShLangVertex; |
| } |
| |
| return EShLangFragment; |
| } |
| |
| // |
| // Read a file's data into a string, and compile it using ShCompile |
| // |
| bool CompileFile(char *fileName, ShHandle compiler, int compileOptions) |
| { |
| int ret; |
| char **data = ReadFileData(fileName); |
| |
| if (!data) |
| return false; |
| |
| ret = ShCompile(compiler, data, OutputMultipleStrings, compileOptions); |
| |
| FreeFileData(data); |
| |
| return ret ? true : false; |
| } |
| |
| // |
| // print usage to stdout |
| // |
| void usage() |
| { |
| printf("Usage: translate [-i -o] file1 file2 ...\n" |
| "Where: filename = filename ending in .frag or .vert\n" |
| " -i = print intermediate tree\n" |
| " -o = print translated code\n"); |
| } |
| |
| // |
| // Malloc a string of sufficient size and read a string into it. |
| // |
| # define MAX_SOURCE_STRINGS 5 |
| char** ReadFileData(char *fileName) |
| { |
| FILE* in = fopen(fileName, "r"); |
| char* fdata = 0; |
| int count = 0; |
| char** return_data = (char**)malloc(MAX_SOURCE_STRINGS+1); |
| |
| if (!in) { |
| printf("Error: unable to open input file: %s\n", fileName); |
| return 0; |
| } |
| |
| while (fgetc(in) != EOF) |
| count++; |
| |
| fseek(in, 0, SEEK_SET); |
| |
| if (!(fdata = (char*)malloc(count+2))) { |
| printf("Error allocating memory\n"); |
| return 0; |
| } |
| if (fread(fdata, 1, count, in) != count) { |
| printf("Error reading input file: %s\n", fileName); |
| return 0; |
| } |
| fdata[count] = '\0'; |
| fclose(in); |
| if (count == 0){ |
| return_data[0] = (char*)malloc(count+2); |
| return_data[0][0] = '\0'; |
| OutputMultipleStrings = 0; |
| return return_data; |
| } |
| |
| int len = (int)(ceil)((float)count / (float)OutputMultipleStrings); |
| int ptr_len = 0, i = 0; |
| while (count > 0) { |
| return_data[i] = (char*)malloc(len+2); |
| memcpy(return_data[i], fdata+ptr_len, len); |
| return_data[i][len] = '\0'; |
| count -= (len); |
| ptr_len += (len); |
| if (count < len){ |
| if(count == 0){ |
| OutputMultipleStrings = (i+1); |
| break; |
| } |
| len = count; |
| } |
| ++i; |
| } |
| return return_data; |
| } |
| |
| void FreeFileData(char** data) |
| { |
| for(int i=0;i<OutputMultipleStrings;i++) |
| free(data[i]); |
| } |
| |
| void LogMsg(char* msg, const char* name, const int num, const char* logName) |
| { |
| printf(num >= 0 ? "#### %s %s %d %s ####\n" : |
| "#### %s %s INFO LOG ####\n", msg, name, num, logName); |
| } |
| |