1f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org//
2f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// Copyright 2012 Francisco Jerez
3f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org//
4f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// Permission is hereby granted, free of charge, to any person obtaining a
5f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// copy of this software and associated documentation files (the "Software"),
6f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// to deal in the Software without restriction, including without limitation
7f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// the rights to use, copy, modify, merge, publish, distribute, sublicense,
8f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// and/or sell copies of the Software, and to permit persons to whom the
9f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// Software is furnished to do so, subject to the following conditions:
10f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org//
11f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// The above copyright notice and this permission notice shall be included in
12f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// all copies or substantial portions of the Software.
13f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org//
14f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
18f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF
19f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
20f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org// SOFTWARE.
21f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org//
22f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
23f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "api/util.hpp"
24f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org#include "core/context.hpp"
25f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
26f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgusing namespace clover;
27f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
28f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgPUBLIC cl_context
29f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgclCreateContext(const cl_context_properties *props, cl_uint num_devs,
30f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                const cl_device_id *devs,
31f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                void (CL_CALLBACK *pfn_notify)(const char *, const void *,
32f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                               size_t, void *),
33f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                void *user_data, cl_int *errcode_ret) try {
34f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   auto mprops = property_map(props);
35f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
36f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (!devs || !num_devs ||
37f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org       (!pfn_notify && user_data))
38f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      throw error(CL_INVALID_VALUE);
39f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
40f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (any_of(is_zero<cl_device_id>(), devs, devs + num_devs))
41f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      throw error(CL_INVALID_DEVICE);
42f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
43f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   for (auto p : mprops) {
44f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      if (!(p.first == CL_CONTEXT_PLATFORM &&
45f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org            (cl_platform_id)p.second == NULL))
46f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org         throw error(CL_INVALID_PROPERTY);
47f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   }
48f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
49f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   ret_error(errcode_ret, CL_SUCCESS);
50f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   return new context(
51f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      property_vector(mprops),
52f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      std::vector<cl_device_id>(devs, devs + num_devs));
53f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
54f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org} catch(error &e) {
55f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   ret_error(errcode_ret, e);
56f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   return NULL;
57f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
58f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
59f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgPUBLIC cl_context
60f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgclCreateContextFromType(const cl_context_properties *props,
61f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                        cl_device_type type,
62f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                        void (CL_CALLBACK *pfn_notify)(
63f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                           const char *, const void *, size_t, void *),
64f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                        void *user_data, cl_int *errcode_ret) {
65f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   cl_device_id dev;
66f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   cl_int ret;
67f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
68f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   ret = clGetDeviceIDs(0, type, 1, &dev, 0);
69f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (ret) {
70f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      ret_error(errcode_ret, ret);
71f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return NULL;
72f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   }
73f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
74f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   return clCreateContext(props, 1, &dev, pfn_notify, user_data, errcode_ret);
75f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
76f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
77f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgPUBLIC cl_int
78f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgclRetainContext(cl_context ctx) {
79f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (!ctx)
80f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return CL_INVALID_CONTEXT;
81f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
82f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   ctx->retain();
83f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   return CL_SUCCESS;
84f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
85f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
86f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgPUBLIC cl_int
87f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgclReleaseContext(cl_context ctx) {
88f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (!ctx)
89f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return CL_INVALID_CONTEXT;
90f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
91f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (ctx->release())
92f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      delete ctx;
93f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
94f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   return CL_SUCCESS;
95f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
96f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
97f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgPUBLIC cl_int
98f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.orgclGetContextInfo(cl_context ctx, cl_context_info param,
99f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                 size_t size, void *buf, size_t *size_ret) {
100f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   if (!ctx)
101f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return CL_INVALID_CONTEXT;
102f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
103f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   switch (param) {
104f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   case CL_CONTEXT_REFERENCE_COUNT:
105f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return scalar_property<cl_uint>(buf, size, size_ret, ctx->ref_count());
106f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
107f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   case CL_CONTEXT_NUM_DEVICES:
108f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return scalar_property<cl_uint>(buf, size, size_ret, ctx->devs.size());
109f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
110f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   case CL_CONTEXT_DEVICES:
111f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return vector_property<cl_device_id>(buf, size, size_ret, ctx->devs);
112f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
113f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   case CL_CONTEXT_PROPERTIES:
114f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return vector_property<cl_context_properties>(buf, size, size_ret,
115f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org                                                    ctx->props());
116f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org
117f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   default:
118f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org      return CL_INVALID_VALUE;
119f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org   }
120f2ba7591b1407a7ee9209f842c50696914dc2dedkbr@chromium.org}
121