libdrm_radeon: Optimize cs_gem_reloc to do less looping.
[glsdk/libdrm.git] / radeon / radeon_bo_gem.c
1 /*
2  * Copyright © 2008 Dave Airlie
3  * Copyright © 2008 Jérôme Glisse
4  * All Rights Reserved.
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining
7  * a 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 SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
15  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
16  * OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
17  * NON-INFRINGEMENT. IN NO EVENT SHALL THE COPYRIGHT HOLDERS, AUTHORS
18  * AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
20  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE
21  * USE OR OTHER DEALINGS IN THE SOFTWARE.
22  *
23  * The above copyright notice and this permission notice (including the
24  * next paragraph) shall be included in all copies or substantial portions
25  * of the Software.
26  */
27 /*
28  * Authors:
29  *      Dave Airlie
30  *      Jérôme Glisse <glisse@freedesktop.org>
31  */
32 #ifdef HAVE_CONFIG_H
33 #include <config.h>
34 #endif
35 #include <stdio.h>
36 #include <stdint.h>
37 #include <stdlib.h>
38 #include <string.h>
39 #include <sys/mman.h>
40 #include <errno.h>
41 #include "xf86drm.h"
42 #include "xf86atomic.h"
43 #include "drm.h"
44 #include "radeon_drm.h"
45 #include "radeon_bo.h"
46 #include "radeon_bo_int.h"
47 #include "radeon_bo_gem.h"
49 struct radeon_bo_gem {
50     struct radeon_bo_int base;
51     uint32_t            name;
52     int                 map_count;
53     atomic_t            reloc_in_cs;
54     void *priv_ptr;
55 };
57 struct bo_manager_gem {
58     struct radeon_bo_manager    base;
59 };
61 static int bo_wait(struct radeon_bo_int *boi);
62     
63 static struct radeon_bo *bo_open(struct radeon_bo_manager *bom,
64                                  uint32_t handle,
65                                  uint32_t size,
66                                  uint32_t alignment,
67                                  uint32_t domains,
68                                  uint32_t flags)
69 {
70     struct radeon_bo_gem *bo;
71     int r;
73     bo = (struct radeon_bo_gem*)calloc(1, sizeof(struct radeon_bo_gem));
74     if (bo == NULL) {
75         return NULL;
76     }
78     bo->base.bom = bom;
79     bo->base.handle = 0;
80     bo->base.size = size;
81     bo->base.alignment = alignment;
82     bo->base.domains = domains;
83     bo->base.flags = flags;
84     bo->base.ptr = NULL;
85     atomic_set(&bo->reloc_in_cs, 0);
86     bo->map_count = 0;
87     if (handle) {
88         struct drm_gem_open open_arg;
90         memset(&open_arg, 0, sizeof(open_arg));
91         open_arg.name = handle;
92         r = drmIoctl(bom->fd, DRM_IOCTL_GEM_OPEN, &open_arg);
93         if (r != 0) {
94             free(bo);
95             return NULL;
96         }
97         bo->base.handle = open_arg.handle;
98         bo->base.size = open_arg.size;
99         bo->name = handle;
100     } else {
101         struct drm_radeon_gem_create args;
103         args.size = size;
104         args.alignment = alignment;
105         args.initial_domain = bo->base.domains;
106         args.flags = 0;
107         args.handle = 0;
108         r = drmCommandWriteRead(bom->fd, DRM_RADEON_GEM_CREATE,
109                                 &args, sizeof(args));
110         bo->base.handle = args.handle;
111         if (r) {
112             fprintf(stderr, "Failed to allocate :\n");
113             fprintf(stderr, "   size      : %d bytes\n", size);
114             fprintf(stderr, "   alignment : %d bytes\n", alignment);
115             fprintf(stderr, "   domains   : %d\n", bo->base.domains);
116             free(bo);
117             return NULL;
118         }
119     }
120     radeon_bo_ref((struct radeon_bo*)bo);
121     return (struct radeon_bo*)bo;
124 static void bo_ref(struct radeon_bo_int *boi)
128 static struct radeon_bo *bo_unref(struct radeon_bo_int *boi)
130     struct radeon_bo_gem *bo_gem = (struct radeon_bo_gem*)boi;
131     struct drm_gem_close args;
133     if (boi->cref) {
134         return (struct radeon_bo *)boi;
135     }
136     if (bo_gem->priv_ptr) {
137         munmap(bo_gem->priv_ptr, boi->size);
138     }
140     /* Zero out args to make valgrind happy */
141     memset(&args, 0, sizeof(args));
143     /* close object */
144     args.handle = boi->handle;
145     drmIoctl(boi->bom->fd, DRM_IOCTL_GEM_CLOSE, &args);
146     memset(bo_gem, 0, sizeof(struct radeon_bo_gem));
147     free(bo_gem);
148     return NULL;
151 static int bo_map(struct radeon_bo_int *boi, int write)
153     struct radeon_bo_gem *bo_gem = (struct radeon_bo_gem*)boi;
154     struct drm_radeon_gem_mmap args;
155     int r;
156     void *ptr;
158     if (bo_gem->map_count++ != 0) {
159         return 0;
160     }
161     if (bo_gem->priv_ptr) {
162         goto wait;
163     }
165     boi->ptr = NULL;
167     /* Zero out args to make valgrind happy */
168     memset(&args, 0, sizeof(args));
169     args.handle = boi->handle;
170     args.offset = 0;
171     args.size = (uint64_t)boi->size;
172     r = drmCommandWriteRead(boi->bom->fd,
173                             DRM_RADEON_GEM_MMAP,
174                             &args,
175                             sizeof(args));
176     if (r) {
177         fprintf(stderr, "error mapping %p 0x%08X (error = %d)\n",
178                 boi, boi->handle, r);
179         return r;
180     }
181     ptr = mmap(0, args.size, PROT_READ|PROT_WRITE, MAP_SHARED, boi->bom->fd, args.addr_ptr);
182     if (ptr == MAP_FAILED)
183         return -errno;
184     bo_gem->priv_ptr = ptr;
185 wait:
186     boi->ptr = bo_gem->priv_ptr;
187     r = bo_wait(boi);
188     if (r)
189         return r;
190     return 0;
193 static int bo_unmap(struct radeon_bo_int *boi)
195     struct radeon_bo_gem *bo_gem = (struct radeon_bo_gem*)boi;
197     if (--bo_gem->map_count > 0) {
198         return 0;
199     }
200     //munmap(bo->ptr, bo->size);
201     boi->ptr = NULL;
202     return 0;
205 static int bo_wait(struct radeon_bo_int *boi)
207     struct drm_radeon_gem_wait_idle args;
208     int ret;
210     /* Zero out args to make valgrind happy */
211     memset(&args, 0, sizeof(args));
212     args.handle = boi->handle;
213     do {
214         ret = drmCommandWriteRead(boi->bom->fd, DRM_RADEON_GEM_WAIT_IDLE,
215                                   &args, sizeof(args));
216     } while (ret == -EBUSY);
217     return ret;
220 static int bo_is_busy(struct radeon_bo_int *boi, uint32_t *domain)
222     struct drm_radeon_gem_busy args;
223     int ret;
225     args.handle = boi->handle;
226     args.domain = 0;
228     ret = drmCommandWriteRead(boi->bom->fd, DRM_RADEON_GEM_BUSY,
229                               &args, sizeof(args));
231     *domain = args.domain;
232     return ret;
235 static int bo_set_tiling(struct radeon_bo_int *boi, uint32_t tiling_flags,
236                          uint32_t pitch)
238     struct drm_radeon_gem_set_tiling args;
239     int r;
241     args.handle = boi->handle;
242     args.tiling_flags = tiling_flags;
243     args.pitch = pitch;
245     r = drmCommandWriteRead(boi->bom->fd,
246                             DRM_RADEON_GEM_SET_TILING,
247                             &args,
248                             sizeof(args));
249     return r;
252 static int bo_get_tiling(struct radeon_bo_int *boi, uint32_t *tiling_flags,
253                          uint32_t *pitch)
255     struct drm_radeon_gem_set_tiling args;
256     int r;
258     args.handle = boi->handle;
260     r = drmCommandWriteRead(boi->bom->fd,
261                             DRM_RADEON_GEM_GET_TILING,
262                             &args,
263                             sizeof(args));
265     if (r)
266         return r;
268     *tiling_flags = args.tiling_flags;
269     *pitch = args.pitch;
270     return r;
273 static struct radeon_bo_funcs bo_gem_funcs = {
274     bo_open,
275     bo_ref,
276     bo_unref,
277     bo_map,
278     bo_unmap,
279     bo_wait,
280     NULL,
281     bo_set_tiling,
282     bo_get_tiling,
283     bo_is_busy,
284 };
286 struct radeon_bo_manager *radeon_bo_manager_gem_ctor(int fd)
288     struct bo_manager_gem *bomg;
290     bomg = (struct bo_manager_gem*)calloc(1, sizeof(struct bo_manager_gem));
291     if (bomg == NULL) {
292         return NULL;
293     }
294     bomg->base.funcs = &bo_gem_funcs;
295     bomg->base.fd = fd;
296     return (struct radeon_bo_manager*)bomg;
299 void radeon_bo_manager_gem_dtor(struct radeon_bo_manager *bom)
301     struct bo_manager_gem *bomg = (struct bo_manager_gem*)bom;
303     if (bom == NULL) {
304         return;
305     }
306     free(bomg);
309 uint32_t radeon_gem_name_bo(struct radeon_bo *bo)
311     struct radeon_bo_gem *bo_gem = (struct radeon_bo_gem*)bo;
312     return bo_gem->name;
315 void *radeon_gem_get_reloc_in_cs(struct radeon_bo *bo)
317     struct radeon_bo_gem *bo_gem = (struct radeon_bo_gem*)bo;
318     return &bo_gem->reloc_in_cs;
321 int radeon_gem_get_kernel_name(struct radeon_bo *bo, uint32_t *name)
323     struct radeon_bo_int *boi = (struct radeon_bo_int *)bo;
324     struct drm_gem_flink flink;
325     int r;
327     flink.handle = bo->handle;
328     r = drmIoctl(boi->bom->fd, DRM_IOCTL_GEM_FLINK, &flink);
329     if (r) {
330         return r;
331     }
332     *name = flink.name;
333     return 0;
336 int radeon_gem_set_domain(struct radeon_bo *bo, uint32_t read_domains, uint32_t write_domain)
338     struct radeon_bo_int *boi = (struct radeon_bo_int *)bo;
339     struct drm_radeon_gem_set_domain args;
340     int r;
342     args.handle = bo->handle;
343     args.read_domains = read_domains;
344     args.write_domain = write_domain;
346     r = drmCommandWriteRead(boi->bom->fd,
347                             DRM_RADEON_GEM_SET_DOMAIN,
348                             &args,
349                             sizeof(args));
350     return r;