FFmpeg coverage


Directory: ../../../ffmpeg/
File: src/fftools/objpool.c
Date: 2022-10-02 18:56:10
Exec Total Coverage
Lines: 50 53 94.3%
Branches: 10 14 71.4%

Line Branch Exec Source
1 /*
2 * This file is part of FFmpeg.
3 *
4 * FFmpeg is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * FFmpeg is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with FFmpeg; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
17 */
18
19 #include <stdint.h>
20
21 #include "libavcodec/packet.h"
22
23 #include "libavutil/common.h"
24 #include "libavutil/error.h"
25 #include "libavutil/frame.h"
26 #include "libavutil/mem.h"
27
28 #include "objpool.h"
29
30 struct ObjPool {
31 void *pool[32];
32 unsigned int pool_count;
33
34 ObjPoolCBAlloc alloc;
35 ObjPoolCBReset reset;
36 ObjPoolCBFree free;
37 };
38
39 9004 ObjPool *objpool_alloc(ObjPoolCBAlloc cb_alloc, ObjPoolCBReset cb_reset,
40 ObjPoolCBFree cb_free)
41 {
42 9004 ObjPool *op = av_mallocz(sizeof(*op));
43
44
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9004 times.
9004 if (!op)
45 return NULL;
46
47 9004 op->alloc = cb_alloc;
48 9004 op->reset = cb_reset;
49 9004 op->free = cb_free;
50
51 9004 return op;
52 }
53
54 9004 void objpool_free(ObjPool **pop)
55 {
56 9004 ObjPool *op = *pop;
57
58
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 9004 times.
9004 if (!op)
59 return;
60
61
2/2
✓ Branch 0 taken 18373 times.
✓ Branch 1 taken 9004 times.
27377 for (unsigned int i = 0; i < op->pool_count; i++)
62 18373 op->free(&op->pool[i]);
63
64 9004 av_freep(pop);
65 }
66
67 489487 int objpool_get(ObjPool *op, void **obj)
68 {
69
2/2
✓ Branch 0 taken 471093 times.
✓ Branch 1 taken 18394 times.
489487 if (op->pool_count) {
70 471093 *obj = op->pool[--op->pool_count];
71 471093 op->pool[op->pool_count] = NULL;
72 } else
73 18394 *obj = op->alloc();
74
75
1/2
✓ Branch 0 taken 489487 times.
✗ Branch 1 not taken.
489487 return *obj ? 0 : AVERROR(ENOMEM);
76 }
77
78 489487 void objpool_release(ObjPool *op, void **obj)
79 {
80
1/2
✗ Branch 0 not taken.
✓ Branch 1 taken 489487 times.
489487 if (!*obj)
81 return;
82
83 489487 op->reset(*obj);
84
85
2/2
✓ Branch 0 taken 489466 times.
✓ Branch 1 taken 21 times.
489487 if (op->pool_count < FF_ARRAY_ELEMS(op->pool))
86 489466 op->pool[op->pool_count++] = *obj;
87 else
88 21 op->free(obj);
89
90 489487 *obj = NULL;
91 }
92
93 15817 static void *alloc_packet(void)
94 {
95 15817 return av_packet_alloc();
96 }
97 2577 static void *alloc_frame(void)
98 {
99 2577 return av_frame_alloc();
100 }
101
102 479134 static void reset_packet(void *obj)
103 {
104 479134 av_packet_unref(obj);
105 479134 }
106 10353 static void reset_frame(void *obj)
107 {
108 10353 av_frame_unref(obj);
109 10353 }
110
111 15817 static void free_packet(void **obj)
112 {
113 15817 AVPacket *pkt = *obj;
114 15817 av_packet_free(&pkt);
115 15817 *obj = NULL;
116 15817 }
117 2577 static void free_frame(void **obj)
118 {
119 2577 AVFrame *frame = *obj;
120 2577 av_frame_free(&frame);
121 2577 *obj = NULL;
122 2577 }
123
124 6430 ObjPool *objpool_alloc_packets(void)
125 {
126 6430 return objpool_alloc(alloc_packet, reset_packet, free_packet);
127 }
128 2574 ObjPool *objpool_alloc_frames(void)
129 {
130 2574 return objpool_alloc(alloc_frame, reset_frame, free_frame);
131 }
132