/* ============================================================
 * cl_info.c
 * Written by Steffen Boerm <boerm@email.uni-kiel.de>
 * All rights reserved.
 *
 * $Revision: 62$
 * ============================================================ */

#include <assert.h>
#include <stdio.h>
#include <stdlib.h>

#define CL_USE_DEPRECATED_OPENCL_1_2_APIS
#define CL_TARGET_OPENCL_VERSION 120
#include <CL/cl.h>

int
main()
{
  cl_uint platforms, devices;
  cl_platform_id *platform;
  cl_device_id *device;
  cl_int result;
  cl_uint max_compute_units, max_clock_frequency;
  char *info;
  size_t infosize;
  cl_uint i, j;

  printf("========================================\n"
	 "Querying number of platforms\n");
  /* Calling clGetPlatformIDs with 0 in the first two arguments and
   * a pointer in the third returns the number of platforms */
  result = clGetPlatformIDs(0, 0, &platforms);
  if(result != CL_SUCCESS) {
    printf("Error code %d\n", result);
    abort();
  }

  /* Allocate storage for all platform id's */
  platform = (cl_platform_id *) malloc((size_t) sizeof(cl_platform_id) * platforms);

  /* Calling clGetPlatformIDs with a number of platforms and a pointer
   * to enough storage returns the id's of the platforms */
  result = clGetPlatformIDs(platforms, platform, 0);
  assert(result == CL_SUCCESS);
  printf("  %d platform(s)\n\n", platforms);

  for(i=0; i<platforms; i++) {
    printf("----------------------------------------\n"
	   "Querying platform %u\n", i);
    /* Calling clGetPlatformInfo with 0 in the third and fourth argument
     * and a pointer in the fifth returns the length of the answer to
     * our question */
    result = clGetPlatformInfo(platform[i], CL_PLATFORM_NAME, 0, 0, &infosize);

    /* Allocate storage for the platform name */
    info = (char *) malloc((size_t) sizeof(char) * infosize);

    /* Get the platform name */
    result = clGetPlatformInfo(platform[i], CL_PLATFORM_NAME, infosize, info, 0);
    assert(result == CL_SUCCESS);
    printf("  Name: \"%s\"\n", info);
    free(info);

    /* Get the platform version */
    result = clGetPlatformInfo(platform[i], CL_PLATFORM_VERSION, 0, 0, &infosize);
    assert(result == CL_SUCCESS);
    info = (char *) malloc((size_t) sizeof(char) * infosize);
    result = clGetPlatformInfo(platform[i], CL_PLATFORM_VERSION, infosize, info, 0);
    assert(result == CL_SUCCESS);
    printf("  OpenCL version: \"%s\"\n", info);
    free(info);

    /* Get the supported profile */
    result = clGetPlatformInfo(platform[i], CL_PLATFORM_PROFILE, 0, 0, &infosize);
    assert(result == CL_SUCCESS);
    info = (char *) malloc((size_t) sizeof(char) * infosize);
    result = clGetPlatformInfo(platform[i], CL_PLATFORM_PROFILE, infosize, info, 0);
    assert(result == CL_SUCCESS);
    printf("  OpenCL profile: \"%s\"\n", info);
    free(info);

    /* Calling clGetDeviceIDs with 0 in the third and fourth argument
     * and a pointer in the fifth returns the number of devices managed
     * by this platform */
    result = clGetDeviceIDs(platform[i], CL_DEVICE_TYPE_ALL, 0, 0, &devices);
    if(result == CL_DEVICE_NOT_FOUND) {
      (void) printf("  No devices\n");
    }
    else {
      assert(result == CL_SUCCESS);
      
      /* Allocate storage for the device id's */
      device = (cl_device_id *) malloc((size_t) sizeof(cl_device_id) * devices);
      
      /* Calling clGetDeviceIDs with the number of devices in the third argument
       * and a pointer in the fourth returns the device id's */
      result = clGetDeviceIDs(platform[i], CL_DEVICE_TYPE_ALL, devices, device, 0);
      assert(result == CL_SUCCESS);
      printf("  %d device(s)\n", devices);
  
      for(j=0; j<devices; j++) {
	/* clGetDeviceInfo called with 0 in the third and fourth argument
	 * and a pointer in the fifth returns the required storage */
	printf("  Querying device %u\n", j);
	result = clGetDeviceInfo(device[j], CL_DEVICE_NAME, 0, 0, &infosize);
	assert(result == CL_SUCCESS);
	
	/* Allocate storage for the device name */
	info = (char *) malloc((size_t) sizeof(char) * infosize);
	
	/* Get the device name */
	result = clGetDeviceInfo(device[j], CL_DEVICE_NAME, infosize, info, 0);
	assert(result == CL_SUCCESS);
	printf("    Name: \"%s\"\n", info);
	free(info);
	
	/* Get the number of compute units */
	result = clGetDeviceInfo(device[j], CL_DEVICE_MAX_COMPUTE_UNITS, sizeof(cl_uint), &max_compute_units, 0);
	assert(result == CL_SUCCESS);
	printf("    Compute units: %d\n", max_compute_units);
	
	/* Get the clock frequency */
	result = clGetDeviceInfo(device[j], CL_DEVICE_MAX_CLOCK_FREQUENCY, sizeof(cl_uint), &max_clock_frequency, 0);
	assert(result == CL_SUCCESS);
	printf("    Frequency: %d\n", max_clock_frequency);
      }
      free(device);
    }
  }
  free(platform);
  printf("========================================\n");

  return 0;
}