src/gallium/auxiliary/gallivm/lp_bld_printf.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2010 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 #include <stdio.h>
  29
  30 #include "util/u_debug.h"
  31 #include "util/u_memory.h"
  32 #include "lp_bld_printf.h"
  33
  34
  35 static int
  36 lp_get_printf_arg_count(const char *fmt)
  37 {
  38    int count =0;
  39    const char *p = fmt;
  40    int c;
  41
  42    while ((c = *p++)) {
  43       if (c != '%')
  44          continue;
  45       switch (*p) {
  46          case '\0':
  47             continue;
  48          case '%':
  49             p++;
  50             continue;
  51          case '.':
  52             if (p[1] == '*' && p[2] == 's') {
  53                count += 2;
  54                p += 3;
  55                continue;
  56             }
  57          default:
  58             count ++;
  59       }
  60    }
  61    return count;
  62 }
  63
  64 LLVMValueRef
  65 lp_build_const_string_variable(LLVMModuleRef module, const char *str, int len)
  66 {
  67    LLVMValueRef string = LLVMAddGlobal(module, LLVMArrayType(LLVMInt8Type(), len + 1), "");
  68    LLVMSetGlobalConstant(string, TRUE);
  69    LLVMSetLinkage(string, LLVMInternalLinkage);
  70    LLVMSetInitializer(string, LLVMConstString(str, len + 1, TRUE));
  71    return string;
  72 }
  73
  74
  75 /**
  76  * lp_build_printf.
  77  *
  78  * Build printf call in LLVM IR. The output goes to stdout.
  79  * The additional variable arguments need to have type
  80  * LLVMValueRef.
  81  */
  82 LLVMValueRef
  83 lp_build_printf(LLVMBuilderRef builder, const char *fmt, ...)
  84 {
  85    va_list arglist;
  86    int i = 0;
  87    int argcount = lp_get_printf_arg_count(fmt);
  88    LLVMModuleRef module = LLVMGetGlobalParent(LLVMGetBasicBlockParent(LLVMGetInsertBlock(builder)));
  89    LLVMValueRef params[50];
  90    LLVMValueRef fmtarg = lp_build_const_string_variable(module, fmt, strlen(fmt) + 1);
  91    LLVMValueRef int0 = LLVMConstInt(LLVMInt32Type(), 0, 0);
  92    LLVMValueRef index[2];
  93    LLVMValueRef func_printf = LLVMGetNamedFunction(module, "printf");
  94
  95    assert(Elements(params) >= argcount + 1);
  96
  97    index[0] = index[1] = int0;
  98
  99    if (!func_printf) {
 100       LLVMTypeRef printf_type = LLVMFunctionType(LLVMIntType(32), NULL, 0, 1);
 101       func_printf = LLVMAddFunction(module, "printf", printf_type);
 102    }
 103
 104    params[0] = LLVMBuildGEP(builder, fmtarg, index, 2, "");
 105
 106    va_start(arglist, fmt);
 107    for (i = 1; i <= argcount; i++)
 108       params[i] = va_arg(arglist, LLVMValueRef);
 109    va_end(arglist);
 110
 111    return LLVMBuildCall(builder, func_printf, params, argcount + 1, "");
 112 }
 113