xref: /OK3568_Linux_fs/external/gstreamer-rockchip/gst/rkximage/gstkmsallocator.c (revision 4882a59341e53eb6f0b4789bf948001014eff981)
1*4882a593Smuzhiyun /* GStreamer
2*4882a593Smuzhiyun  *
3*4882a593Smuzhiyun  * Copyright (C) 2016 Igalia
4*4882a593Smuzhiyun  *
5*4882a593Smuzhiyun  * Authors:
6*4882a593Smuzhiyun  *  Víctor Manuel Jáquez Leal <vjaquez@igalia.com>
7*4882a593Smuzhiyun  *  Javier Martin <javiermartin@by.com.es>
8*4882a593Smuzhiyun  *
9*4882a593Smuzhiyun  * This library is free software; you can redistribute it and/or
10*4882a593Smuzhiyun  * modify it under the terms of the GNU Library General Public
11*4882a593Smuzhiyun  * License as published by the Free Software Foundation; either
12*4882a593Smuzhiyun  * version 2 of the License, or (at your option) any later version.
13*4882a593Smuzhiyun  *
14*4882a593Smuzhiyun  * This library is distributed in the hope that it will be useful,
15*4882a593Smuzhiyun  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16*4882a593Smuzhiyun  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17*4882a593Smuzhiyun  * Library General Public License for more details.
18*4882a593Smuzhiyun  *
19*4882a593Smuzhiyun  * You should have received a copy of the GNU Library General Public
20*4882a593Smuzhiyun  * License along with this library; if not, write to the
21*4882a593Smuzhiyun  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
22*4882a593Smuzhiyun  * Boston, MA 02110-1301, USA.
23*4882a593Smuzhiyun  *
24*4882a593Smuzhiyun  */
25*4882a593Smuzhiyun 
26*4882a593Smuzhiyun #ifdef HAVE_CONFIG_H
27*4882a593Smuzhiyun #include "config.h"
28*4882a593Smuzhiyun #endif
29*4882a593Smuzhiyun 
30*4882a593Smuzhiyun #include <fcntl.h>
31*4882a593Smuzhiyun #include <xf86drm.h>
32*4882a593Smuzhiyun #include <xf86drmMode.h>
33*4882a593Smuzhiyun #include <stdlib.h>
34*4882a593Smuzhiyun #include <string.h>
35*4882a593Smuzhiyun #include <sys/mman.h>
36*4882a593Smuzhiyun #include <unistd.h>
37*4882a593Smuzhiyun 
38*4882a593Smuzhiyun /* it needs to be below because is internal to libdrm */
39*4882a593Smuzhiyun #include <drm.h>
40*4882a593Smuzhiyun #include <drm_fourcc.h>
41*4882a593Smuzhiyun 
42*4882a593Smuzhiyun #include <gst/allocators/gstdmabuf.h>
43*4882a593Smuzhiyun 
44*4882a593Smuzhiyun #include "gstkmsallocator.h"
45*4882a593Smuzhiyun #include "gstkmsutils.h"
46*4882a593Smuzhiyun 
47*4882a593Smuzhiyun #ifndef DRM_RDWR
48*4882a593Smuzhiyun #define DRM_RDWR O_RDWR
49*4882a593Smuzhiyun #endif
50*4882a593Smuzhiyun 
51*4882a593Smuzhiyun #define GST_CAT_DEFAULT kmsallocator_debug
52*4882a593Smuzhiyun GST_DEBUG_CATEGORY_STATIC (GST_CAT_DEFAULT);
53*4882a593Smuzhiyun 
54*4882a593Smuzhiyun #define GST_KMS_MEMORY_TYPE "KMSMemory"
55*4882a593Smuzhiyun 
56*4882a593Smuzhiyun struct kms_bo
57*4882a593Smuzhiyun {
58*4882a593Smuzhiyun   void *ptr;
59*4882a593Smuzhiyun   size_t size;
60*4882a593Smuzhiyun   unsigned handle;
61*4882a593Smuzhiyun   unsigned int refs;
62*4882a593Smuzhiyun };
63*4882a593Smuzhiyun 
64*4882a593Smuzhiyun struct _GstKMSAllocatorPrivate
65*4882a593Smuzhiyun {
66*4882a593Smuzhiyun   int fd;
67*4882a593Smuzhiyun   /* protected by GstKMSAllocator object lock */
68*4882a593Smuzhiyun   GList *mem_cache;
69*4882a593Smuzhiyun   GstAllocator *dmabuf_alloc;
70*4882a593Smuzhiyun };
71*4882a593Smuzhiyun 
72*4882a593Smuzhiyun #define parent_class gst_kms_allocator_parent_class
73*4882a593Smuzhiyun G_DEFINE_TYPE_WITH_CODE (GstKMSAllocator, gst_kms_allocator, GST_TYPE_ALLOCATOR,
74*4882a593Smuzhiyun     G_ADD_PRIVATE (GstKMSAllocator);
75*4882a593Smuzhiyun     GST_DEBUG_CATEGORY_INIT (GST_CAT_DEFAULT, "kmsallocator", 0,
76*4882a593Smuzhiyun         "KMS allocator"));
77*4882a593Smuzhiyun 
78*4882a593Smuzhiyun enum
79*4882a593Smuzhiyun {
80*4882a593Smuzhiyun   PROP_DRM_FD = 1,
81*4882a593Smuzhiyun   PROP_N,
82*4882a593Smuzhiyun };
83*4882a593Smuzhiyun 
84*4882a593Smuzhiyun static GParamSpec *g_props[PROP_N] = { NULL, };
85*4882a593Smuzhiyun 
86*4882a593Smuzhiyun gboolean
gst_is_kms_memory(GstMemory * mem)87*4882a593Smuzhiyun gst_is_kms_memory (GstMemory * mem)
88*4882a593Smuzhiyun {
89*4882a593Smuzhiyun   return gst_memory_is_type (mem, GST_KMS_MEMORY_TYPE);
90*4882a593Smuzhiyun }
91*4882a593Smuzhiyun 
92*4882a593Smuzhiyun guint32
gst_kms_memory_get_fb_id(GstMemory * mem)93*4882a593Smuzhiyun gst_kms_memory_get_fb_id (GstMemory * mem)
94*4882a593Smuzhiyun {
95*4882a593Smuzhiyun   if (!gst_is_kms_memory (mem))
96*4882a593Smuzhiyun     return 0;
97*4882a593Smuzhiyun   return ((GstKMSMemory *) mem)->fb_id;
98*4882a593Smuzhiyun }
99*4882a593Smuzhiyun 
100*4882a593Smuzhiyun static gboolean
check_fd(GstKMSAllocator * alloc)101*4882a593Smuzhiyun check_fd (GstKMSAllocator * alloc)
102*4882a593Smuzhiyun {
103*4882a593Smuzhiyun   return alloc->priv->fd > -1;
104*4882a593Smuzhiyun }
105*4882a593Smuzhiyun 
106*4882a593Smuzhiyun static void
gst_kms_allocator_memory_reset(GstKMSAllocator * allocator,GstKMSMemory * mem)107*4882a593Smuzhiyun gst_kms_allocator_memory_reset (GstKMSAllocator * allocator, GstKMSMemory * mem)
108*4882a593Smuzhiyun {
109*4882a593Smuzhiyun   int err;
110*4882a593Smuzhiyun   struct drm_mode_destroy_dumb arg = { 0, };
111*4882a593Smuzhiyun 
112*4882a593Smuzhiyun   if (!check_fd (allocator))
113*4882a593Smuzhiyun     return;
114*4882a593Smuzhiyun 
115*4882a593Smuzhiyun   if (mem->fb_id) {
116*4882a593Smuzhiyun     GST_DEBUG_OBJECT (allocator, "removing fb id %d", mem->fb_id);
117*4882a593Smuzhiyun     drmModeRmFB (allocator->priv->fd, mem->fb_id);
118*4882a593Smuzhiyun     mem->fb_id = 0;
119*4882a593Smuzhiyun   }
120*4882a593Smuzhiyun 
121*4882a593Smuzhiyun   if (!mem->bo)
122*4882a593Smuzhiyun     return;
123*4882a593Smuzhiyun 
124*4882a593Smuzhiyun   if (mem->bo->ptr != NULL) {
125*4882a593Smuzhiyun     GST_WARNING_OBJECT (allocator, "destroying mapped bo (refcount=%d)",
126*4882a593Smuzhiyun         mem->bo->refs);
127*4882a593Smuzhiyun     munmap (mem->bo->ptr, mem->bo->size);
128*4882a593Smuzhiyun     mem->bo->ptr = NULL;
129*4882a593Smuzhiyun   }
130*4882a593Smuzhiyun 
131*4882a593Smuzhiyun   arg.handle = mem->bo->handle;
132*4882a593Smuzhiyun 
133*4882a593Smuzhiyun   err = drmIoctl (allocator->priv->fd, DRM_IOCTL_MODE_DESTROY_DUMB, &arg);
134*4882a593Smuzhiyun   if (err)
135*4882a593Smuzhiyun     GST_WARNING_OBJECT (allocator,
136*4882a593Smuzhiyun         "Failed to destroy dumb buffer object: %s %d", strerror (errno), errno);
137*4882a593Smuzhiyun 
138*4882a593Smuzhiyun   g_free (mem->bo);
139*4882a593Smuzhiyun   mem->bo = NULL;
140*4882a593Smuzhiyun }
141*4882a593Smuzhiyun 
142*4882a593Smuzhiyun /* Copied from gst_v4l2_object_extrapolate_stride() */
143*4882a593Smuzhiyun static gint
extrapolate_stride(const GstVideoFormatInfo * finfo,gint plane,gint stride)144*4882a593Smuzhiyun extrapolate_stride (const GstVideoFormatInfo * finfo, gint plane, gint stride)
145*4882a593Smuzhiyun {
146*4882a593Smuzhiyun   gint estride;
147*4882a593Smuzhiyun 
148*4882a593Smuzhiyun   switch (finfo->format) {
149*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV12:
150*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV12_64Z32:
151*4882a593Smuzhiyun #ifdef HAVE_NV12_10LE40
152*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV12_10LE40:
153*4882a593Smuzhiyun #endif
154*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV21:
155*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV16:
156*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV61:
157*4882a593Smuzhiyun     case GST_VIDEO_FORMAT_NV24:
158*4882a593Smuzhiyun       estride = (plane == 0 ? 1 : 2) *
159*4882a593Smuzhiyun           GST_VIDEO_FORMAT_INFO_SCALE_WIDTH (finfo, plane, stride);
160*4882a593Smuzhiyun       break;
161*4882a593Smuzhiyun     default:
162*4882a593Smuzhiyun       estride = GST_VIDEO_FORMAT_INFO_SCALE_WIDTH (finfo, plane, stride);
163*4882a593Smuzhiyun       break;
164*4882a593Smuzhiyun   }
165*4882a593Smuzhiyun 
166*4882a593Smuzhiyun   return estride;
167*4882a593Smuzhiyun }
168*4882a593Smuzhiyun 
169*4882a593Smuzhiyun static gboolean
gst_kms_allocator_memory_create(GstKMSAllocator * allocator,GstKMSMemory * kmsmem,GstVideoInfo * vinfo)170*4882a593Smuzhiyun gst_kms_allocator_memory_create (GstKMSAllocator * allocator,
171*4882a593Smuzhiyun     GstKMSMemory * kmsmem, GstVideoInfo * vinfo)
172*4882a593Smuzhiyun {
173*4882a593Smuzhiyun   gint i, ret, h;
174*4882a593Smuzhiyun   struct drm_mode_create_dumb arg = { 0, };
175*4882a593Smuzhiyun   guint32 fmt;
176*4882a593Smuzhiyun   gint num_planes = GST_VIDEO_INFO_N_PLANES (vinfo);
177*4882a593Smuzhiyun   gsize offs = 0;
178*4882a593Smuzhiyun 
179*4882a593Smuzhiyun   if (kmsmem->bo)
180*4882a593Smuzhiyun     return TRUE;
181*4882a593Smuzhiyun 
182*4882a593Smuzhiyun   if (!check_fd (allocator))
183*4882a593Smuzhiyun     return FALSE;
184*4882a593Smuzhiyun 
185*4882a593Smuzhiyun   kmsmem->bo = g_malloc0 (sizeof (*kmsmem->bo));
186*4882a593Smuzhiyun   if (!kmsmem->bo)
187*4882a593Smuzhiyun     return FALSE;
188*4882a593Smuzhiyun 
189*4882a593Smuzhiyun   fmt = gst_drm_format_from_video (GST_VIDEO_INFO_FORMAT (vinfo));
190*4882a593Smuzhiyun   arg.bpp = gst_drm_bpp_from_drm (fmt);
191*4882a593Smuzhiyun   arg.width = GST_VIDEO_INFO_WIDTH (vinfo);
192*4882a593Smuzhiyun   h = GST_VIDEO_INFO_HEIGHT (vinfo);
193*4882a593Smuzhiyun   arg.height = gst_drm_height_from_drm (fmt, h);
194*4882a593Smuzhiyun 
195*4882a593Smuzhiyun   ret = drmIoctl (allocator->priv->fd, DRM_IOCTL_MODE_CREATE_DUMB, &arg);
196*4882a593Smuzhiyun   if (ret)
197*4882a593Smuzhiyun     goto create_failed;
198*4882a593Smuzhiyun 
199*4882a593Smuzhiyun   if (!arg.pitch)
200*4882a593Smuzhiyun     goto done;
201*4882a593Smuzhiyun 
202*4882a593Smuzhiyun   for (i = 0; i < num_planes; i++) {
203*4882a593Smuzhiyun     guint32 pitch;
204*4882a593Smuzhiyun 
205*4882a593Smuzhiyun     if (!arg.pitch)
206*4882a593Smuzhiyun       continue;
207*4882a593Smuzhiyun 
208*4882a593Smuzhiyun     /* Overwrite the video info's stride and offset using the pitch calculcated
209*4882a593Smuzhiyun      * by the kms driver. */
210*4882a593Smuzhiyun     pitch = extrapolate_stride (vinfo->finfo, i, arg.pitch);
211*4882a593Smuzhiyun     GST_VIDEO_INFO_PLANE_STRIDE (vinfo, i) = pitch;
212*4882a593Smuzhiyun     GST_VIDEO_INFO_PLANE_OFFSET (vinfo, i) = offs;
213*4882a593Smuzhiyun 
214*4882a593Smuzhiyun     /* Note that we cannot negotiate special padding betweem each planes,
215*4882a593Smuzhiyun      * hence using the display height here. */
216*4882a593Smuzhiyun     offs += pitch * GST_VIDEO_FORMAT_INFO_SCALE_HEIGHT (vinfo->finfo, i, h);
217*4882a593Smuzhiyun 
218*4882a593Smuzhiyun     GST_DEBUG_OBJECT (allocator, "Created BO plane %i with stride %i and "
219*4882a593Smuzhiyun         "offset %" G_GSIZE_FORMAT, i,
220*4882a593Smuzhiyun         GST_VIDEO_INFO_PLANE_STRIDE (vinfo, i),
221*4882a593Smuzhiyun         GST_VIDEO_INFO_PLANE_OFFSET (vinfo, i));
222*4882a593Smuzhiyun   }
223*4882a593Smuzhiyun 
224*4882a593Smuzhiyun   /* Update with the size use for display, excluding any padding at the end */
225*4882a593Smuzhiyun   GST_VIDEO_INFO_SIZE (vinfo) = offs;
226*4882a593Smuzhiyun 
227*4882a593Smuzhiyun done:
228*4882a593Smuzhiyun   kmsmem->bo->handle = arg.handle;
229*4882a593Smuzhiyun   /* will be used a memory maxsize */
230*4882a593Smuzhiyun   kmsmem->bo->size = arg.size;
231*4882a593Smuzhiyun 
232*4882a593Smuzhiyun   /* Validate the size to prevent overflow */
233*4882a593Smuzhiyun   if (kmsmem->bo->size < GST_VIDEO_INFO_SIZE (vinfo)) {
234*4882a593Smuzhiyun     GST_ERROR_OBJECT (allocator,
235*4882a593Smuzhiyun         "DUMB buffer has a size of %" G_GSIZE_FORMAT
236*4882a593Smuzhiyun         " but we require at least %" G_GSIZE_FORMAT " to hold a frame",
237*4882a593Smuzhiyun         kmsmem->bo->size, GST_VIDEO_INFO_SIZE (vinfo));
238*4882a593Smuzhiyun     return FALSE;
239*4882a593Smuzhiyun   }
240*4882a593Smuzhiyun 
241*4882a593Smuzhiyun   return TRUE;
242*4882a593Smuzhiyun 
243*4882a593Smuzhiyun   /* ERRORS */
244*4882a593Smuzhiyun create_failed:
245*4882a593Smuzhiyun   {
246*4882a593Smuzhiyun     GST_ERROR_OBJECT (allocator, "Failed to create buffer object: %s (%d)",
247*4882a593Smuzhiyun         strerror (-ret), ret);
248*4882a593Smuzhiyun     g_free (kmsmem->bo);
249*4882a593Smuzhiyun     kmsmem->bo = NULL;
250*4882a593Smuzhiyun     return FALSE;
251*4882a593Smuzhiyun   }
252*4882a593Smuzhiyun }
253*4882a593Smuzhiyun 
254*4882a593Smuzhiyun static void
gst_kms_allocator_free(GstAllocator * allocator,GstMemory * mem)255*4882a593Smuzhiyun gst_kms_allocator_free (GstAllocator * allocator, GstMemory * mem)
256*4882a593Smuzhiyun {
257*4882a593Smuzhiyun   GstKMSAllocator *alloc;
258*4882a593Smuzhiyun   GstKMSMemory *kmsmem;
259*4882a593Smuzhiyun 
260*4882a593Smuzhiyun   alloc = GST_KMS_ALLOCATOR (allocator);
261*4882a593Smuzhiyun   kmsmem = (GstKMSMemory *) mem;
262*4882a593Smuzhiyun 
263*4882a593Smuzhiyun   gst_kms_allocator_memory_reset (alloc, kmsmem);
264*4882a593Smuzhiyun   g_slice_free (GstKMSMemory, kmsmem);
265*4882a593Smuzhiyun }
266*4882a593Smuzhiyun 
267*4882a593Smuzhiyun static void
gst_kms_allocator_set_property(GObject * object,guint prop_id,const GValue * value,GParamSpec * pspec)268*4882a593Smuzhiyun gst_kms_allocator_set_property (GObject * object, guint prop_id,
269*4882a593Smuzhiyun     const GValue * value, GParamSpec * pspec)
270*4882a593Smuzhiyun {
271*4882a593Smuzhiyun   GstKMSAllocator *alloc;
272*4882a593Smuzhiyun 
273*4882a593Smuzhiyun   alloc = GST_KMS_ALLOCATOR (object);
274*4882a593Smuzhiyun 
275*4882a593Smuzhiyun   switch (prop_id) {
276*4882a593Smuzhiyun     case PROP_DRM_FD:{
277*4882a593Smuzhiyun       int fd = g_value_get_int (value);
278*4882a593Smuzhiyun       if (fd > -1)
279*4882a593Smuzhiyun         alloc->priv->fd = dup (fd);
280*4882a593Smuzhiyun       break;
281*4882a593Smuzhiyun     }
282*4882a593Smuzhiyun     default:
283*4882a593Smuzhiyun       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
284*4882a593Smuzhiyun       break;
285*4882a593Smuzhiyun   }
286*4882a593Smuzhiyun }
287*4882a593Smuzhiyun 
288*4882a593Smuzhiyun static void
gst_kms_allocator_get_property(GObject * object,guint prop_id,GValue * value,GParamSpec * pspec)289*4882a593Smuzhiyun gst_kms_allocator_get_property (GObject * object, guint prop_id,
290*4882a593Smuzhiyun     GValue * value, GParamSpec * pspec)
291*4882a593Smuzhiyun {
292*4882a593Smuzhiyun   GstKMSAllocator *alloc;
293*4882a593Smuzhiyun 
294*4882a593Smuzhiyun   alloc = GST_KMS_ALLOCATOR (object);
295*4882a593Smuzhiyun 
296*4882a593Smuzhiyun   switch (prop_id) {
297*4882a593Smuzhiyun     case PROP_DRM_FD:
298*4882a593Smuzhiyun       g_value_set_int (value, alloc->priv->fd);
299*4882a593Smuzhiyun       break;
300*4882a593Smuzhiyun     default:
301*4882a593Smuzhiyun       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
302*4882a593Smuzhiyun       break;
303*4882a593Smuzhiyun   }
304*4882a593Smuzhiyun }
305*4882a593Smuzhiyun 
306*4882a593Smuzhiyun static void
gst_kms_allocator_finalize(GObject * obj)307*4882a593Smuzhiyun gst_kms_allocator_finalize (GObject * obj)
308*4882a593Smuzhiyun {
309*4882a593Smuzhiyun   GstKMSAllocator *alloc;
310*4882a593Smuzhiyun 
311*4882a593Smuzhiyun   alloc = GST_KMS_ALLOCATOR (obj);
312*4882a593Smuzhiyun 
313*4882a593Smuzhiyun   gst_kms_allocator_clear_cache (GST_ALLOCATOR (alloc));
314*4882a593Smuzhiyun 
315*4882a593Smuzhiyun   if (alloc->priv->dmabuf_alloc)
316*4882a593Smuzhiyun     gst_object_unref (alloc->priv->dmabuf_alloc);
317*4882a593Smuzhiyun 
318*4882a593Smuzhiyun   if (check_fd (alloc))
319*4882a593Smuzhiyun     close (alloc->priv->fd);
320*4882a593Smuzhiyun 
321*4882a593Smuzhiyun   G_OBJECT_CLASS (parent_class)->finalize (obj);
322*4882a593Smuzhiyun }
323*4882a593Smuzhiyun 
324*4882a593Smuzhiyun static void
gst_kms_allocator_class_init(GstKMSAllocatorClass * klass)325*4882a593Smuzhiyun gst_kms_allocator_class_init (GstKMSAllocatorClass * klass)
326*4882a593Smuzhiyun {
327*4882a593Smuzhiyun   GObjectClass *gobject_class;
328*4882a593Smuzhiyun   GstAllocatorClass *allocator_class;
329*4882a593Smuzhiyun 
330*4882a593Smuzhiyun   allocator_class = GST_ALLOCATOR_CLASS (klass);
331*4882a593Smuzhiyun   gobject_class = G_OBJECT_CLASS (klass);
332*4882a593Smuzhiyun 
333*4882a593Smuzhiyun   allocator_class->free = gst_kms_allocator_free;
334*4882a593Smuzhiyun 
335*4882a593Smuzhiyun   gobject_class->set_property = gst_kms_allocator_set_property;
336*4882a593Smuzhiyun   gobject_class->get_property = gst_kms_allocator_get_property;
337*4882a593Smuzhiyun   gobject_class->finalize = gst_kms_allocator_finalize;
338*4882a593Smuzhiyun 
339*4882a593Smuzhiyun   g_props[PROP_DRM_FD] = g_param_spec_int ("drm-fd", "DRM fd",
340*4882a593Smuzhiyun       "DRM file descriptor", -1, G_MAXINT, -1,
341*4882a593Smuzhiyun       G_PARAM_READWRITE | G_PARAM_CONSTRUCT);
342*4882a593Smuzhiyun 
343*4882a593Smuzhiyun   g_object_class_install_properties (gobject_class, PROP_N, g_props);
344*4882a593Smuzhiyun }
345*4882a593Smuzhiyun 
346*4882a593Smuzhiyun static gpointer
gst_kms_memory_map(GstMemory * mem,gsize maxsize,GstMapFlags flags)347*4882a593Smuzhiyun gst_kms_memory_map (GstMemory * mem, gsize maxsize, GstMapFlags flags)
348*4882a593Smuzhiyun {
349*4882a593Smuzhiyun   GstKMSMemory *kmsmem;
350*4882a593Smuzhiyun   GstKMSAllocator *alloc;
351*4882a593Smuzhiyun   int err;
352*4882a593Smuzhiyun   gpointer out;
353*4882a593Smuzhiyun   struct drm_mode_map_dumb arg = { 0, };
354*4882a593Smuzhiyun 
355*4882a593Smuzhiyun   alloc = (GstKMSAllocator *) mem->allocator;
356*4882a593Smuzhiyun 
357*4882a593Smuzhiyun   if (!check_fd (alloc))
358*4882a593Smuzhiyun     return NULL;
359*4882a593Smuzhiyun 
360*4882a593Smuzhiyun   kmsmem = (GstKMSMemory *) mem;
361*4882a593Smuzhiyun   if (!kmsmem->bo)
362*4882a593Smuzhiyun     return NULL;
363*4882a593Smuzhiyun 
364*4882a593Smuzhiyun   /* Reuse existing buffer object mapping if possible */
365*4882a593Smuzhiyun   if (kmsmem->bo->ptr != NULL) {
366*4882a593Smuzhiyun     goto out;
367*4882a593Smuzhiyun   }
368*4882a593Smuzhiyun 
369*4882a593Smuzhiyun   arg.handle = kmsmem->bo->handle;
370*4882a593Smuzhiyun 
371*4882a593Smuzhiyun   err = drmIoctl (alloc->priv->fd, DRM_IOCTL_MODE_MAP_DUMB, &arg);
372*4882a593Smuzhiyun   if (err) {
373*4882a593Smuzhiyun     GST_ERROR_OBJECT (alloc, "Failed to get offset of buffer object: %s %d",
374*4882a593Smuzhiyun         strerror (-err), err);
375*4882a593Smuzhiyun     return NULL;
376*4882a593Smuzhiyun   }
377*4882a593Smuzhiyun 
378*4882a593Smuzhiyun   out = mmap (0, kmsmem->bo->size,
379*4882a593Smuzhiyun       PROT_READ | PROT_WRITE, MAP_SHARED, alloc->priv->fd, arg.offset);
380*4882a593Smuzhiyun   if (out == MAP_FAILED) {
381*4882a593Smuzhiyun     GST_ERROR_OBJECT (alloc, "Failed to map dumb buffer object: %s %d",
382*4882a593Smuzhiyun         strerror (errno), errno);
383*4882a593Smuzhiyun     return NULL;
384*4882a593Smuzhiyun   }
385*4882a593Smuzhiyun   kmsmem->bo->ptr = out;
386*4882a593Smuzhiyun 
387*4882a593Smuzhiyun out:
388*4882a593Smuzhiyun   g_atomic_int_inc (&kmsmem->bo->refs);
389*4882a593Smuzhiyun   return kmsmem->bo->ptr;
390*4882a593Smuzhiyun }
391*4882a593Smuzhiyun 
392*4882a593Smuzhiyun static void
gst_kms_memory_unmap(GstMemory * mem)393*4882a593Smuzhiyun gst_kms_memory_unmap (GstMemory * mem)
394*4882a593Smuzhiyun {
395*4882a593Smuzhiyun   GstKMSMemory *kmsmem;
396*4882a593Smuzhiyun 
397*4882a593Smuzhiyun   if (!check_fd ((GstKMSAllocator *) mem->allocator))
398*4882a593Smuzhiyun     return;
399*4882a593Smuzhiyun 
400*4882a593Smuzhiyun   kmsmem = (GstKMSMemory *) mem;
401*4882a593Smuzhiyun   if (!kmsmem->bo)
402*4882a593Smuzhiyun     return;
403*4882a593Smuzhiyun 
404*4882a593Smuzhiyun   if (g_atomic_int_dec_and_test (&kmsmem->bo->refs)) {
405*4882a593Smuzhiyun     munmap (kmsmem->bo->ptr, kmsmem->bo->size);
406*4882a593Smuzhiyun     kmsmem->bo->ptr = NULL;
407*4882a593Smuzhiyun   }
408*4882a593Smuzhiyun }
409*4882a593Smuzhiyun 
410*4882a593Smuzhiyun static void
gst_kms_allocator_init(GstKMSAllocator * allocator)411*4882a593Smuzhiyun gst_kms_allocator_init (GstKMSAllocator * allocator)
412*4882a593Smuzhiyun {
413*4882a593Smuzhiyun   GstAllocator *alloc;
414*4882a593Smuzhiyun 
415*4882a593Smuzhiyun   alloc = GST_ALLOCATOR_CAST (allocator);
416*4882a593Smuzhiyun 
417*4882a593Smuzhiyun   allocator->priv = gst_kms_allocator_get_instance_private (allocator);
418*4882a593Smuzhiyun   allocator->priv->fd = -1;
419*4882a593Smuzhiyun 
420*4882a593Smuzhiyun   alloc->mem_type = GST_KMS_MEMORY_TYPE;
421*4882a593Smuzhiyun   alloc->mem_map = gst_kms_memory_map;
422*4882a593Smuzhiyun   alloc->mem_unmap = gst_kms_memory_unmap;
423*4882a593Smuzhiyun   /* Use the default, fallback copy function */
424*4882a593Smuzhiyun 
425*4882a593Smuzhiyun   GST_OBJECT_FLAG_SET (allocator, GST_ALLOCATOR_FLAG_CUSTOM_ALLOC);
426*4882a593Smuzhiyun }
427*4882a593Smuzhiyun 
428*4882a593Smuzhiyun GstAllocator *
gst_kms_allocator_new(int fd)429*4882a593Smuzhiyun gst_kms_allocator_new (int fd)
430*4882a593Smuzhiyun {
431*4882a593Smuzhiyun   GstAllocator *alloc;
432*4882a593Smuzhiyun 
433*4882a593Smuzhiyun   alloc = g_object_new (GST_TYPE_KMS_ALLOCATOR, "name",
434*4882a593Smuzhiyun       "KMSMemory::allocator", "drm-fd", fd, NULL);
435*4882a593Smuzhiyun   gst_object_ref_sink (alloc);
436*4882a593Smuzhiyun 
437*4882a593Smuzhiyun   return alloc;
438*4882a593Smuzhiyun }
439*4882a593Smuzhiyun 
440*4882a593Smuzhiyun /* The mem_offsets are relative to the GstMemory start, unlike the vinfo->offset
441*4882a593Smuzhiyun  * which are relative to the GstBuffer start. */
442*4882a593Smuzhiyun static gboolean
gst_kms_allocator_add_fb(GstKMSAllocator * alloc,GstKMSMemory * kmsmem,gsize in_offsets[GST_VIDEO_MAX_PLANES],GstVideoInfo * vinfo)443*4882a593Smuzhiyun gst_kms_allocator_add_fb (GstKMSAllocator * alloc, GstKMSMemory * kmsmem,
444*4882a593Smuzhiyun     gsize in_offsets[GST_VIDEO_MAX_PLANES], GstVideoInfo * vinfo)
445*4882a593Smuzhiyun {
446*4882a593Smuzhiyun   gint i, ret = -1;
447*4882a593Smuzhiyun   gint num_planes = GST_VIDEO_INFO_N_PLANES (vinfo);
448*4882a593Smuzhiyun   guint32 w, h, fmt, bo_handles[4] = { 0, };
449*4882a593Smuzhiyun   guint32 pitches[4] = { 0, };
450*4882a593Smuzhiyun   guint32 offsets[4] = { 0, };
451*4882a593Smuzhiyun 
452*4882a593Smuzhiyun   if (kmsmem->fb_id)
453*4882a593Smuzhiyun     return TRUE;
454*4882a593Smuzhiyun 
455*4882a593Smuzhiyun   w = GST_VIDEO_INFO_WIDTH (vinfo);
456*4882a593Smuzhiyun   h = GST_VIDEO_INFO_HEIGHT (vinfo);
457*4882a593Smuzhiyun   fmt = gst_drm_format_from_video (GST_VIDEO_INFO_FORMAT (vinfo));
458*4882a593Smuzhiyun 
459*4882a593Smuzhiyun   for (i = 0; i < num_planes; i++) {
460*4882a593Smuzhiyun     if (kmsmem->bo)
461*4882a593Smuzhiyun       bo_handles[i] = kmsmem->bo->handle;
462*4882a593Smuzhiyun     else
463*4882a593Smuzhiyun       bo_handles[i] = kmsmem->gem_handle[i];
464*4882a593Smuzhiyun 
465*4882a593Smuzhiyun     pitches[i] = GST_VIDEO_INFO_PLANE_STRIDE (vinfo, i);
466*4882a593Smuzhiyun     offsets[i] = in_offsets[i];
467*4882a593Smuzhiyun   }
468*4882a593Smuzhiyun 
469*4882a593Smuzhiyun   GST_DEBUG_OBJECT (alloc, "bo handles: %d, %d, %d, %d", bo_handles[0],
470*4882a593Smuzhiyun       bo_handles[1], bo_handles[2], bo_handles[3]);
471*4882a593Smuzhiyun 
472*4882a593Smuzhiyun   if (GST_VIDEO_INFO_IS_AFBC (vinfo)) {
473*4882a593Smuzhiyun     guint64 modifiers[4] = { 0 };
474*4882a593Smuzhiyun 
475*4882a593Smuzhiyun     for (i = 0; i < num_planes; i++)
476*4882a593Smuzhiyun       modifiers[i] = DRM_AFBC_MODIFIER;
477*4882a593Smuzhiyun 
478*4882a593Smuzhiyun     if (fmt == DRM_FORMAT_NV12 || fmt == DRM_FORMAT_NV12_10 ||
479*4882a593Smuzhiyun         fmt == DRM_FORMAT_NV16) {
480*4882a593Smuzhiyun       /* The newer kernel might use new formats instead */
481*4882a593Smuzhiyun       guint32 _handles[4] = { bo_handles[0], 0, };
482*4882a593Smuzhiyun       guint32 _pitches[4] = { pitches[0], 0, };
483*4882a593Smuzhiyun       guint32 _offsets[4] = { offsets[0], 0, };
484*4882a593Smuzhiyun       guint64 _modifiers[4] = { modifiers[0], 0, };
485*4882a593Smuzhiyun       guint32 _fmt;
486*4882a593Smuzhiyun 
487*4882a593Smuzhiyun       if (fmt == DRM_FORMAT_NV12) {
488*4882a593Smuzhiyun         _fmt = DRM_FORMAT_YUV420_8BIT;
489*4882a593Smuzhiyun         /* The bpp of YUV420_8BIT is 12 */
490*4882a593Smuzhiyun         _pitches[0] *= 1.5;
491*4882a593Smuzhiyun       } else if (fmt == DRM_FORMAT_NV12_10) {
492*4882a593Smuzhiyun         _fmt = DRM_FORMAT_YUV420_10BIT;
493*4882a593Smuzhiyun         /* The bpp of YUV420_10BIT is 15 */
494*4882a593Smuzhiyun         _pitches[0] *= 1.5;
495*4882a593Smuzhiyun       } else {
496*4882a593Smuzhiyun         _fmt = DRM_FORMAT_YUYV;
497*4882a593Smuzhiyun         /* The bpp of YUYV (AFBC) is 16 */
498*4882a593Smuzhiyun         _pitches[0] *= 2;
499*4882a593Smuzhiyun       }
500*4882a593Smuzhiyun 
501*4882a593Smuzhiyun       ret = drmModeAddFB2WithModifiers (alloc->priv->fd, w, h, _fmt, _handles,
502*4882a593Smuzhiyun           _pitches, _offsets, _modifiers, &kmsmem->fb_id,
503*4882a593Smuzhiyun           DRM_MODE_FB_MODIFIERS);
504*4882a593Smuzhiyun     }
505*4882a593Smuzhiyun 
506*4882a593Smuzhiyun     if (ret)
507*4882a593Smuzhiyun       ret = drmModeAddFB2WithModifiers (alloc->priv->fd, w, h, fmt, bo_handles,
508*4882a593Smuzhiyun           pitches, offsets, modifiers, &kmsmem->fb_id, DRM_MODE_FB_MODIFIERS);
509*4882a593Smuzhiyun   } else {
510*4882a593Smuzhiyun     ret = drmModeAddFB2 (alloc->priv->fd, w, h, fmt, bo_handles, pitches,
511*4882a593Smuzhiyun         offsets, &kmsmem->fb_id, 0);
512*4882a593Smuzhiyun     if (ret && fmt == DRM_FORMAT_NV12_10)
513*4882a593Smuzhiyun       ret = drmModeAddFB2 (alloc->priv->fd, w, h, DRM_FORMAT_NV15, bo_handles,
514*4882a593Smuzhiyun           pitches, offsets, &kmsmem->fb_id, 0);
515*4882a593Smuzhiyun   }
516*4882a593Smuzhiyun   if (ret) {
517*4882a593Smuzhiyun     GST_ERROR_OBJECT (alloc, "Failed to bind to framebuffer: %s (%d)",
518*4882a593Smuzhiyun         strerror (-ret), ret);
519*4882a593Smuzhiyun     return FALSE;
520*4882a593Smuzhiyun   }
521*4882a593Smuzhiyun 
522*4882a593Smuzhiyun   return TRUE;
523*4882a593Smuzhiyun }
524*4882a593Smuzhiyun 
525*4882a593Smuzhiyun GstMemory *
gst_kms_allocator_bo_alloc(GstAllocator * allocator,GstVideoInfo * vinfo)526*4882a593Smuzhiyun gst_kms_allocator_bo_alloc (GstAllocator * allocator, GstVideoInfo * vinfo)
527*4882a593Smuzhiyun {
528*4882a593Smuzhiyun   GstKMSAllocator *alloc;
529*4882a593Smuzhiyun   GstKMSMemory *kmsmem;
530*4882a593Smuzhiyun   GstMemory *mem;
531*4882a593Smuzhiyun 
532*4882a593Smuzhiyun   kmsmem = g_slice_new0 (GstKMSMemory);
533*4882a593Smuzhiyun   if (!kmsmem)
534*4882a593Smuzhiyun     return NULL;
535*4882a593Smuzhiyun 
536*4882a593Smuzhiyun   alloc = GST_KMS_ALLOCATOR (allocator);
537*4882a593Smuzhiyun 
538*4882a593Smuzhiyun   mem = GST_MEMORY_CAST (kmsmem);
539*4882a593Smuzhiyun 
540*4882a593Smuzhiyun   if (!gst_kms_allocator_memory_create (alloc, kmsmem, vinfo)) {
541*4882a593Smuzhiyun     g_slice_free (GstKMSMemory, kmsmem);
542*4882a593Smuzhiyun     return NULL;
543*4882a593Smuzhiyun   }
544*4882a593Smuzhiyun 
545*4882a593Smuzhiyun   gst_memory_init (mem, GST_MEMORY_FLAG_NO_SHARE, allocator, NULL,
546*4882a593Smuzhiyun       kmsmem->bo->size, 0, 0, GST_VIDEO_INFO_SIZE (vinfo));
547*4882a593Smuzhiyun 
548*4882a593Smuzhiyun   if (!gst_kms_allocator_add_fb (alloc, kmsmem, vinfo->offset, vinfo))
549*4882a593Smuzhiyun     goto fail;
550*4882a593Smuzhiyun 
551*4882a593Smuzhiyun   return mem;
552*4882a593Smuzhiyun 
553*4882a593Smuzhiyun   /* ERRORS */
554*4882a593Smuzhiyun fail:
555*4882a593Smuzhiyun   gst_memory_unref (mem);
556*4882a593Smuzhiyun   return NULL;
557*4882a593Smuzhiyun }
558*4882a593Smuzhiyun 
559*4882a593Smuzhiyun GstKMSMemory *
gst_kms_allocator_dmabuf_import(GstAllocator * allocator,gint * prime_fds,gint n_planes,gsize offsets[GST_VIDEO_MAX_PLANES],GstVideoInfo * vinfo)560*4882a593Smuzhiyun gst_kms_allocator_dmabuf_import (GstAllocator * allocator, gint * prime_fds,
561*4882a593Smuzhiyun     gint n_planes, gsize offsets[GST_VIDEO_MAX_PLANES], GstVideoInfo * vinfo)
562*4882a593Smuzhiyun {
563*4882a593Smuzhiyun   GstKMSAllocator *alloc;
564*4882a593Smuzhiyun   GstKMSMemory *kmsmem;
565*4882a593Smuzhiyun   GstMemory *mem;
566*4882a593Smuzhiyun   gint i, ret;
567*4882a593Smuzhiyun   guint32 handle = 0;
568*4882a593Smuzhiyun 
569*4882a593Smuzhiyun   g_return_val_if_fail (n_planes <= GST_VIDEO_MAX_PLANES, FALSE);
570*4882a593Smuzhiyun 
571*4882a593Smuzhiyun   kmsmem = g_slice_new0 (GstKMSMemory);
572*4882a593Smuzhiyun   if (!kmsmem)
573*4882a593Smuzhiyun     return FALSE;
574*4882a593Smuzhiyun 
575*4882a593Smuzhiyun   mem = GST_MEMORY_CAST (kmsmem);
576*4882a593Smuzhiyun   gst_memory_init (mem, GST_MEMORY_FLAG_NO_SHARE, allocator, NULL,
577*4882a593Smuzhiyun       GST_VIDEO_INFO_SIZE (vinfo), 0, 0, GST_VIDEO_INFO_SIZE (vinfo));
578*4882a593Smuzhiyun 
579*4882a593Smuzhiyun   alloc = GST_KMS_ALLOCATOR (allocator);
580*4882a593Smuzhiyun   for (i = 0; i < n_planes; i++) {
581*4882a593Smuzhiyun     ret = drmPrimeFDToHandle (alloc->priv->fd, prime_fds[i],
582*4882a593Smuzhiyun         &kmsmem->gem_handle[i]);
583*4882a593Smuzhiyun     if (ret)
584*4882a593Smuzhiyun       goto import_fd_failed;
585*4882a593Smuzhiyun   }
586*4882a593Smuzhiyun 
587*4882a593Smuzhiyun   if (!gst_kms_allocator_add_fb (alloc, kmsmem, offsets, vinfo))
588*4882a593Smuzhiyun     goto failed;
589*4882a593Smuzhiyun 
590*4882a593Smuzhiyun   for (i = 0; i < n_planes; i++) {
591*4882a593Smuzhiyun     struct drm_gem_close arg = { kmsmem->gem_handle[i], };
592*4882a593Smuzhiyun     gint err;
593*4882a593Smuzhiyun 
594*4882a593Smuzhiyun     if (handle == arg.handle)
595*4882a593Smuzhiyun       break;
596*4882a593Smuzhiyun 
597*4882a593Smuzhiyun     handle = arg.handle;
598*4882a593Smuzhiyun 
599*4882a593Smuzhiyun     err = drmIoctl (alloc->priv->fd, DRM_IOCTL_GEM_CLOSE, &arg);
600*4882a593Smuzhiyun     if (err)
601*4882a593Smuzhiyun       GST_WARNING_OBJECT (allocator,
602*4882a593Smuzhiyun           "Failed to close GEM handle: %s %d", strerror (errno), errno);
603*4882a593Smuzhiyun 
604*4882a593Smuzhiyun     kmsmem->gem_handle[i] = 0;
605*4882a593Smuzhiyun   }
606*4882a593Smuzhiyun 
607*4882a593Smuzhiyun   return kmsmem;
608*4882a593Smuzhiyun 
609*4882a593Smuzhiyun   /* ERRORS */
610*4882a593Smuzhiyun import_fd_failed:
611*4882a593Smuzhiyun   {
612*4882a593Smuzhiyun     GST_ERROR_OBJECT (alloc, "Failed to import prime fd %d: %s (%d)",
613*4882a593Smuzhiyun         prime_fds[i], strerror (-ret), ret);
614*4882a593Smuzhiyun     /* fallback */
615*4882a593Smuzhiyun   }
616*4882a593Smuzhiyun 
617*4882a593Smuzhiyun failed:
618*4882a593Smuzhiyun   {
619*4882a593Smuzhiyun     gst_memory_unref (mem);
620*4882a593Smuzhiyun     return NULL;
621*4882a593Smuzhiyun   }
622*4882a593Smuzhiyun }
623*4882a593Smuzhiyun 
624*4882a593Smuzhiyun GstMemory *
gst_kms_allocator_dmabuf_export(GstAllocator * allocator,GstMemory * _kmsmem)625*4882a593Smuzhiyun gst_kms_allocator_dmabuf_export (GstAllocator * allocator, GstMemory * _kmsmem)
626*4882a593Smuzhiyun {
627*4882a593Smuzhiyun   GstKMSMemory *kmsmem = (GstKMSMemory *) _kmsmem;
628*4882a593Smuzhiyun   GstKMSAllocator *alloc = GST_KMS_ALLOCATOR (allocator);
629*4882a593Smuzhiyun   GstMemory *mem;
630*4882a593Smuzhiyun   gint ret;
631*4882a593Smuzhiyun   gint prime_fd;
632*4882a593Smuzhiyun 
633*4882a593Smuzhiyun   /* We can only export DUMB buffers */
634*4882a593Smuzhiyun   g_return_val_if_fail (kmsmem->bo, NULL);
635*4882a593Smuzhiyun 
636*4882a593Smuzhiyun 
637*4882a593Smuzhiyun   ret = drmPrimeHandleToFD (alloc->priv->fd, kmsmem->bo->handle,
638*4882a593Smuzhiyun       DRM_CLOEXEC | DRM_RDWR, &prime_fd);
639*4882a593Smuzhiyun   if (ret)
640*4882a593Smuzhiyun     goto export_fd_failed;
641*4882a593Smuzhiyun 
642*4882a593Smuzhiyun   if (G_UNLIKELY (alloc->priv->dmabuf_alloc == NULL))
643*4882a593Smuzhiyun     alloc->priv->dmabuf_alloc = gst_dmabuf_allocator_new ();
644*4882a593Smuzhiyun 
645*4882a593Smuzhiyun   mem = gst_dmabuf_allocator_alloc (alloc->priv->dmabuf_alloc, prime_fd,
646*4882a593Smuzhiyun       gst_memory_get_sizes (_kmsmem, NULL, NULL));
647*4882a593Smuzhiyun 
648*4882a593Smuzhiyun   /* Populate the cache so KMSSink can find the kmsmem back when it receives
649*4882a593Smuzhiyun    * one of these DMABuf. This call takes ownership of the kmsmem. */
650*4882a593Smuzhiyun   gst_kms_allocator_cache (allocator, mem, _kmsmem);
651*4882a593Smuzhiyun 
652*4882a593Smuzhiyun   GST_DEBUG_OBJECT (alloc, "Exported bo handle %d as %d", kmsmem->bo->handle,
653*4882a593Smuzhiyun       prime_fd);
654*4882a593Smuzhiyun 
655*4882a593Smuzhiyun   return mem;
656*4882a593Smuzhiyun 
657*4882a593Smuzhiyun   /* ERRORS */
658*4882a593Smuzhiyun export_fd_failed:
659*4882a593Smuzhiyun   {
660*4882a593Smuzhiyun     GST_ERROR_OBJECT (alloc, "Failed to export bo handle %d: %s (%d)",
661*4882a593Smuzhiyun         kmsmem->bo->handle, g_strerror (errno), ret);
662*4882a593Smuzhiyun     return NULL;
663*4882a593Smuzhiyun   }
664*4882a593Smuzhiyun }
665*4882a593Smuzhiyun 
666*4882a593Smuzhiyun /* FIXME, using gdata for caching on upstream memory is not tee safe */
667*4882a593Smuzhiyun GstMemory *
gst_kms_allocator_get_cached(GstMemory * mem)668*4882a593Smuzhiyun gst_kms_allocator_get_cached (GstMemory * mem)
669*4882a593Smuzhiyun {
670*4882a593Smuzhiyun   return gst_mini_object_get_qdata (GST_MINI_OBJECT (mem),
671*4882a593Smuzhiyun       g_quark_from_static_string ("kmsmem"));
672*4882a593Smuzhiyun }
673*4882a593Smuzhiyun 
674*4882a593Smuzhiyun static void
cached_kmsmem_disposed_cb(GstKMSAllocator * alloc,GstMiniObject * obj)675*4882a593Smuzhiyun cached_kmsmem_disposed_cb (GstKMSAllocator * alloc, GstMiniObject * obj)
676*4882a593Smuzhiyun {
677*4882a593Smuzhiyun   GST_OBJECT_LOCK (alloc);
678*4882a593Smuzhiyun   alloc->priv->mem_cache = g_list_remove (alloc->priv->mem_cache, obj);
679*4882a593Smuzhiyun   GST_OBJECT_UNLOCK (alloc);
680*4882a593Smuzhiyun }
681*4882a593Smuzhiyun 
682*4882a593Smuzhiyun void
gst_kms_allocator_clear_cache(GstAllocator * allocator)683*4882a593Smuzhiyun gst_kms_allocator_clear_cache (GstAllocator * allocator)
684*4882a593Smuzhiyun {
685*4882a593Smuzhiyun   GstKMSAllocator *alloc = GST_KMS_ALLOCATOR (allocator);
686*4882a593Smuzhiyun   GList *iter;
687*4882a593Smuzhiyun 
688*4882a593Smuzhiyun   GST_OBJECT_LOCK (alloc);
689*4882a593Smuzhiyun 
690*4882a593Smuzhiyun   iter = alloc->priv->mem_cache;
691*4882a593Smuzhiyun   while (iter) {
692*4882a593Smuzhiyun     GstMiniObject *obj = iter->data;
693*4882a593Smuzhiyun     gst_mini_object_weak_unref (obj,
694*4882a593Smuzhiyun         (GstMiniObjectNotify) cached_kmsmem_disposed_cb, alloc);
695*4882a593Smuzhiyun     gst_mini_object_set_qdata (obj,
696*4882a593Smuzhiyun         g_quark_from_static_string ("kmsmem"), NULL, NULL);
697*4882a593Smuzhiyun     iter = iter->next;
698*4882a593Smuzhiyun   }
699*4882a593Smuzhiyun 
700*4882a593Smuzhiyun   g_list_free (alloc->priv->mem_cache);
701*4882a593Smuzhiyun   alloc->priv->mem_cache = NULL;
702*4882a593Smuzhiyun 
703*4882a593Smuzhiyun   GST_OBJECT_UNLOCK (alloc);
704*4882a593Smuzhiyun }
705*4882a593Smuzhiyun 
706*4882a593Smuzhiyun /* @kmsmem is transfer-full */
707*4882a593Smuzhiyun void
gst_kms_allocator_cache(GstAllocator * allocator,GstMemory * mem,GstMemory * kmsmem)708*4882a593Smuzhiyun gst_kms_allocator_cache (GstAllocator * allocator, GstMemory * mem,
709*4882a593Smuzhiyun     GstMemory * kmsmem)
710*4882a593Smuzhiyun {
711*4882a593Smuzhiyun   GstKMSAllocator *alloc = GST_KMS_ALLOCATOR (allocator);
712*4882a593Smuzhiyun 
713*4882a593Smuzhiyun   GST_OBJECT_LOCK (alloc);
714*4882a593Smuzhiyun   gst_mini_object_weak_ref (GST_MINI_OBJECT (mem),
715*4882a593Smuzhiyun       (GstMiniObjectNotify) cached_kmsmem_disposed_cb, alloc);
716*4882a593Smuzhiyun   alloc->priv->mem_cache = g_list_prepend (alloc->priv->mem_cache, mem);
717*4882a593Smuzhiyun   GST_OBJECT_UNLOCK (alloc);
718*4882a593Smuzhiyun 
719*4882a593Smuzhiyun   gst_mini_object_set_qdata (GST_MINI_OBJECT (mem),
720*4882a593Smuzhiyun       g_quark_from_static_string ("kmsmem"), kmsmem,
721*4882a593Smuzhiyun       (GDestroyNotify) gst_memory_unref);
722*4882a593Smuzhiyun }
723