-
Notifications
You must be signed in to change notification settings - Fork 76
Expand file tree
/
Copy pathFFTPacketQueue.m
More file actions
136 lines (113 loc) · 2.47 KB
/
FFTPacketQueue.m
File metadata and controls
136 lines (113 loc) · 2.47 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
//
// FFTPacketQueue.m
// FFmpegTutorial
//
// Created by qianlongxu on 2022/7/16.
//
#import "FFTPacketQueue.h"
#import <libavcodec/avcodec.h>
@interface FFTPacketQueueItem : NSObject
@property (nonatomic, assign) AVPacket pkt;
@end
@implementation FFTPacketQueueItem
- (instancetype)initWithAVPacket:(AVPacket *)pkt
{
self = [super init];
if (self) {
self.pkt = *pkt;
}
return self;
}
- (void)dealloc
{
av_packet_unref(&_pkt);
}
- (AVPacket *)pktPtr
{
return &_pkt;
}
@end
@interface FFTPacketQueue ()
@property (nonatomic, strong) NSMutableArray *queue;
@property (nonatomic, strong) NSRecursiveLock *lock;
@property (nonatomic, strong) NSCondition *condition;
@property (atomic, assign) BOOL canceled;
@end
@implementation FFTPacketQueue
- (void)dealloc
{
}
- (instancetype)init
{
self = [super init];
if (self) {
self.queue = [NSMutableArray array];
self.lock = [[NSRecursiveLock alloc] init];
self.condition = [NSCondition new];
}
return self;
}
- (void)cancel
{
self.canceled = YES;
}
- (void)enQueue:(AVPacket *)frame
{
if (self.canceled) {
return;
}
FFTPacketQueueItem *item = [[FFTPacketQueueItem alloc] initWithAVPacket:frame];
[self.lock lock];
[self.queue addObject:item];
[self.lock unlock];
}
- (NSUInteger)count
{
NSUInteger count = 0;
[self.lock lock];
count = [self.queue count];
[self.lock unlock];
return count;
}
- (FFTPacketQueueItem *)waitAitem
{
FFTPacketQueueItem *item = nil;
while (!self.canceled) {
[self.lock lock];
if ([self.queue count] > 0) {
item = [self.queue firstObject];
[self.queue removeObjectAtIndex:0];
[self.lock unlock];
break;
} else {
[self.lock unlock];
[self.condition waitUntilDate:[NSDate dateWithTimeIntervalSinceNow:0.01]];
}
}
return item;
}
- (FFTPacketQueueItem *)popAitem
{
FFTPacketQueueItem *item = nil;
if (!self.canceled) {
[self.lock lock];
if ([self.queue count] > 0) {
item = [self.queue firstObject];
[self.queue removeObjectAtIndex:0];
}
[self.lock unlock];
}
return item;
}
- (void)deQueue:(void (^)(AVPacket * _Nullable))block
{
FFTPacketQueueItem *item = [self waitAitem];
AVPacket *pkt = [item pktPtr];
if (block) {
block(pkt);
}
if (pkt) {
av_packet_unref(pkt);
}
}
@end