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             /* fallthrough */
  58          default:
  59             count ++;
  60       }
  61    }
  62    return count;
  63 }
  64
  65 LLVMValueRef
  66 lp_build_const_string_variable(LLVMModuleRef module, const char *str, int len)
  67 {
  68    LLVMValueRef string = LLVMAddGlobal(module, LLVMArrayType(LLVMInt8Type(), len + 1), "");
  69    LLVMSetGlobalConstant(string, TRUE);
  70    LLVMSetLinkage(string, LLVMInternalLinkage);
  71    LLVMSetInitializer(string, LLVMConstString(str, len + 1, TRUE));
  72    return string;
  73 }
  74
  75
  76 /**
  77  * lp_build_printf.
  78  *
  79  * Build printf call in LLVM IR. The output goes to stdout.
  80  * The additional variable arguments need to have type
  81  * LLVMValueRef.
  82  */
  83 LLVMValueRef
  84 lp_build_printf(LLVMBuilderRef builder, const char *fmt, ...)
  85 {
  86    va_list arglist;
  87    int i = 0;
  88    int argcount = lp_get_printf_arg_count(fmt);
  89    LLVMModuleRef module = LLVMGetGlobalParent(LLVMGetBasicBlockParent(LLVMGetInsertBlock(builder)));
  90    LLVMValueRef params[50];
  91    LLVMValueRef fmtarg = lp_build_const_string_variable(module, fmt, strlen(fmt) + 1);
  92    LLVMValueRef int0 = LLVMConstInt(LLVMInt32Type(), 0, 0);
  93    LLVMValueRef index[2];
  94    LLVMValueRef func_printf = LLVMGetNamedFunction(module, "printf");
  95
  96    assert(Elements(params) >= argcount + 1);
  97
  98    index[0] = index[1] = int0;
  99
 100    if (!func_printf) {
 101       LLVMTypeRef printf_type = LLVMFunctionType(LLVMIntType(32), NULL, 0, 1);
 102       func_printf = LLVMAddFunction(module, "printf", printf_type);
 103    }
 104
 105    params[0] = LLVMBuildGEP(builder, fmtarg, index, 2, "");
 106
 107    va_start(arglist, fmt);
 108    for (i = 1; i <= argcount; i++) {
 109       LLVMValueRef val = va_arg(arglist, LLVMValueRef);
 110       LLVMTypeRef type = LLVMTypeOf(val);
 111       /* printf wants doubles, so lets convert so that
 112        * we can actually print them */
 113       if (LLVMGetTypeKind(type) == LLVMFloatTypeKind)
 114          val = LLVMBuildFPExt(builder, val, LLVMDoubleType(), "");
 115       params[i] = val;
 116    }
 117    va_end(arglist);
 118
 119    return LLVMBuildCall(builder, func_printf, params, argcount + 1, "");
 120 }
 121