/* 
 * Copyright (c) 2009 by Tor M. Aamodt, Ali Bakhoda, Joey Ting, Dan O'Connor, 
 * Clive Lin, George L. Yuan, Wilson W. L. Fung and the 
 * University of British Columbia
 * Vancouver, BC  V6T 1Z4
 * All Rights Reserved.
 * 
 * THIS IS A LEGAL DOCUMENT BY DOWNLOADING GPGPU-SIM, YOU ARE AGREEING TO THESE
 * TERMS AND CONDITIONS.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNERS OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 * 
 * NOTE: The files libcuda/cuda_runtime_api.c and src/cuda-sim/cuda-math.h
 * are derived from the CUDA Toolset available from http://www.nvidia.com/cuda
 * (property of NVIDIA).  The files benchmarks/BlackScholes/ and 
 * benchmarks/template/ are derived from the CUDA SDK available from 
 * http://www.nvidia.com/cuda (also property of NVIDIA).  The files from 
 * src/intersim/ are derived from Booksim (a simulator provided with the 
 * textbook "Principles and Practices of Interconnection Networks" available 
 * from http://cva.stanford.edu/books/ppin/). As such, those files are bound by 
 * the corresponding legal terms and conditions set forth separately (original 
 * copyright notices are left in files from these sources and where we have 
 * modified a file our copyright notice appears before the original copyright 
 * notice).  
 * 
 * Using this version of GPGPU-Sim requires a complete installation of CUDA 
 * which is distributed seperately by NVIDIA under separate terms and 
 * conditions.  To use this version of GPGPU-Sim with OpenCL requires a
 * recent version of NVIDIA's drivers which support OpenCL.
 * 
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * 
 * 2. Redistributions in binary form must reproduce the above copyright notice,
 * this list of conditions and the following disclaimer in the documentation
 * and/or other materials provided with the distribution.
 * 
 * 3. Neither the name of the University of British Columbia nor the names of
 * its contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * 
 * 4. This version of GPGPU-SIM is distributed freely for non-commercial use only.  
 *  
 * 5. No nonprofit user may place any restrictions on the use of this software,
 * including as modified by the user, by any other authorized user.
 * 
 * 6. GPGPU-SIM was developed primarily by Tor M. Aamodt, Wilson W. L. Fung, 
 * Ali Bakhoda, George L. Yuan, at the University of British Columbia, 
 * Vancouver, BC V6T 1Z4
 */

#include "cuda_device_printf.h"
#include "ptx_ir.h"

void decode_space( memory_space_t &space, const ptx_thread_info *thread, const operand_info &op, memory_space *&mem, addr_t &addr);

void my_cuda_printf(const char *fmtstr,const char *arg_list)
{
   FILE *fp = stdout;
   unsigned i=0,j=0;
   unsigned arg_offset=0;
   char buf[64];
   bool in_fmt=false;
   while( fmtstr[i] ) {
      char c = fmtstr[i++];
      if( !in_fmt ) {
         if( c != '%' ) {
            fprintf(fp,"%c",c);
         } else {
            in_fmt=true;
            buf[0] = c;
            j=1;
         }
      } else {
         if(!( c == 'u' || c == 'f' || c == 'd' )) {
            printf("GPGPU-Sim PTX: ERROR ** printf parsing support is limited to %%u, %%f, %%d at present");
            abort();
         }
         buf[j] = c;
         buf[j+1] = 0;
         unsigned long long value = ((unsigned long long*)arg_list)[arg_offset];
         if( c == 'u' || c == 'd' ) {
            fprintf(fp,buf,value);
         } else if( c == 'f' ) {
            double tmp = *(double*)(void*)&value;
            fprintf(fp,buf,tmp);
         }
         arg_offset++;
         in_fmt=false;
      }
   }
}

void gpgpusim_cuda_vprintf(const ptx_instruction * pI, const ptx_thread_info * thread, const function_info * target_func ) 
{
      char *fmtstr = NULL;
      char *arg_list = NULL;
      unsigned n_return = target_func->has_return();
      unsigned n_args = target_func->num_args();
      assert( n_args == 2 );
      for( unsigned arg=0; arg < n_args; arg ++ ) {
         const operand_info &actual_param_op = pI->operand_lookup(n_return+1+arg);
         const symbol *formal_param = target_func->get_arg(arg);
         unsigned size=formal_param->get_size_in_bytes();
         assert( formal_param->is_param_local() );
         assert( actual_param_op.is_param_local() );
         addr_t from_addr = actual_param_op.get_symbol()->get_address();
         char buffer[1024];
         assert(size<1024); 
         thread->m_local_mem->read(from_addr,size,buffer);
         addr_t addr = (addr_t)*(unsigned long long*)((void*)buffer); // should be pointer to generic memory location
         memory_space *mem=NULL;
         memory_space_t space = generic_space;
         decode_space(space,thread,actual_param_op,mem,addr); // figure out which space
         if( arg == 0 ) {
            unsigned len = 0;
            char b = 0;
            do { // figure out length
               mem->read(addr+len,1,&b);
               len++;
            } while(b);
            fmtstr = (char*)malloc(len+64);
            for( int i=0; i < len; i++ ) 
               mem->read(addr+i,1,fmtstr+i);
            //mem->read(addr,len,fmtstr);
         } else {
            unsigned len = thread->get_finfo()->local_mem_framesize();
            arg_list = (char*)malloc(len+64);
            for( int i=0; i < len; i++ ) 
               mem->read(addr+i,1,arg_list+i);
            //mem->read(addr,len,arg_list);
         }
      }
      my_cuda_printf(fmtstr,arg_list);
      free(fmtstr);
      free(arg_list);
}