path: root/src/kernels/oclconv/oclconv.cpp

/******************************************************************************
 * Copyright (c) 2013-2014, Texas Instruments Incorporated - http://www.ti.com/
 *   All rights reserved.
 *
 *   Redistribution and use in source and binary forms, with or without
 *   modification, are permitted provided that the following conditions are met:
 *       * Redistributions of source code must retain the above copyright
 *         notice, this list of conditions and the following disclaimer.
 *       * 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.
 *       * Neither the name of Texas Instruments Incorporated nor the
 *         names of its contributors may be used to endorse or promote products
 *         derived from this software without specific prior written permission.
 *
 *   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 OWNER 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.
 *****************************************************************************/
#define __CL_ENABLE_EXCEPTIONS
#include <CL/cl.hpp>
#include <iostream>
#include <fstream>
#include <cstdlib>
#include "ocl_util.h"
#include "conv.dsp_h"

//#define VERBOSE
using namespace cl;
using namespace std;

static int oclconv_imgproc(char *kernelName, unsigned char *data_in, unsigned char *data_out, int width, int height, int sstride, int dstride)
{
   cl_int err     = CL_SUCCESS;
   int  bufsize   = sstride * height * sizeof(unsigned char);
#ifdef VERBOSE
   ofstream logfile;
   logfile.open ("/home/root/oclconv_log.txt", ios::out | ios::app); 
   logfile << "Entered oclconv_test, width=" << width << " height=" << height << " dstride=" << dstride << " sstride=" << sstride << '\n';
   logfile.close();
#endif
   try 
   {
     Context context(CL_DEVICE_TYPE_ACCELERATOR);
     std::vector<Device> devices = context.getInfo<CL_CONTEXT_DEVICES>();
     devices.resize(1); // resize to 1 since we are only running on 1 DSP

     Buffer bufA   (context, CL_MEM_READ_ONLY,  bufsize);
     Buffer bufDst (context, CL_MEM_WRITE_ONLY, bufsize);

     Program::Binaries binary(1, make_pair(conv_dsp_bin,sizeof(conv_dsp_bin)));
     Program           program = Program(context, devices, binary);
     program.build(devices);
     Kernel kernel(program, kernelName);
     kernel.setArg(0, bufA);
     kernel.setArg(1, bufDst);
     kernel.setArg(2, width);
     kernel.setArg(3, height);
     kernel.setArg(4, dstride);
     kernel.setArg(5, sstride);

     Event ev1,ev2,ev3,ev4;

     CommandQueue Q(context, devices[0], CL_QUEUE_PROFILING_ENABLE);

     Q.enqueueWriteBuffer(bufA, CL_FALSE, 0, bufsize, data_in, NULL, &ev1);
     Q.enqueueTask (kernel, NULL, &ev3);
     Q.enqueueReadBuffer (bufDst, CL_TRUE, 0, bufsize, data_out, NULL, &ev4);
   }
   catch (Error err) 
   { cerr << "ERROR: " << err.what() << "(" << err.err() << ")" << endl; return -1; }
#ifdef VERBOSE
   logfile.open ("/home/root/oclconv_log.txt", ios::out | ios::app); 
   logfile << "Success!" << endl; 
   logfile.close();
#endif
   return 0;
}


#ifdef __cplusplus
extern "C" {
#endif
int oclconv_kernel(int kernel_type, int filter_size, 
                   unsigned char *data_in, unsigned char *data_out, 
                   int width, int height, int dstride, int sstride)
{
  int retval = -1;
  switch(kernel_type)
  {
    case 0: /* Median */
      if(filter_size == 5) { 
        retval = oclconv_imgproc("Median2x2", data_in, data_out, width, height, sstride, dstride);
      } else if(filter_size == 9) { 
        retval = oclconv_imgproc("Median3x3", data_in, data_out, width, height, sstride, dstride);
      }
      break;
    case 1: /* Sobel */
      if(filter_size == 9) { 
        retval = oclconv_imgproc("Sobel3x3", data_in, data_out, width, height, sstride, dstride);
      }
      break;
    case 2: /* conv */
      if(filter_size == 25) { 
        retval = oclconv_imgproc("Conv5x5", data_in, data_out, width, height, sstride, dstride);
        return 0;
      }
      break;
    default:
      break;
  }
  return retval;
}
#ifdef __cplusplus
}
#endif