forked from AgoraIO/API-Examples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
RTMPStreaming.swift
356 lines (324 loc) · 14.7 KB
/
RTMPStreaming.swift
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
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
//
// JoinChannelVC.swift
// APIExample
//
// Created by 张乾泽 on 2020/4/17.
// Copyright © 2020 Agora Corp. All rights reserved.
//
import Cocoa
import AgoraRtcKit
import AGEVideoLayout
let CANVAS_WIDTH = 640
let CANVAS_HEIGHT = 480
class RTMPStreaming: BaseViewController {
var videos: [VideoView] = []
@IBOutlet weak var Container: AGEVideoContainer!
var agoraKit: AgoraRtcEngineKit!
var transcoding = AgoraLiveTranscoding.default()
/**
--- rtmpUrls Picker ---
*/
@IBOutlet weak var selectRtmpUrlsPicker: Picker!
@IBOutlet weak var removeURLBtn: NSButton!
@IBOutlet weak var removeAllURLBtn: NSButton!
var rtmpURLs: [String] = []
var selectedrtmpUrl: String? {
let index = self.selectRtmpUrlsPicker.indexOfSelectedItem
if index >= 0 && index < rtmpURLs.count {
return rtmpURLs[index]
} else {
return nil
}
}
func initSelectRtmpUrlsPicker() {
selectRtmpUrlsPicker.label.stringValue = "urls"
selectRtmpUrlsPicker.picker.addItems(withTitles: rtmpURLs)
}
/// callback when remove streaming url button hit
@IBAction func onRemoveStreamingURL(_ sender: Any) {
guard let selectedURL = selectedrtmpUrl else { return }
agoraKit.removePublishStreamUrl(selectedURL)
rtmpURLs.remove(at: selectRtmpUrlsPicker.indexOfSelectedItem)
selectRtmpUrlsPicker.picker.removeItem(at: selectRtmpUrlsPicker.indexOfSelectedItem)
}
/// callback when remove all streaming url button hit
@IBAction func onRemoveAllStreamingURL(_ sender: Any) {
for url in rtmpURLs {
agoraKit.removePublishStreamUrl(url)
}
rtmpURLs = []
selectRtmpUrlsPicker.picker.removeAllItems()
}
/**
--- Channel TextField ---
*/
@IBOutlet weak var channelField: Input!
func initChannelField() {
channelField.label.stringValue = "Channel".localized
channelField.field.placeholderString = "Channel Name".localized
}
/**
--- rtmp TextField ---
*/
@IBOutlet weak var rtmpURLField: Input!
@IBOutlet weak var transcodingCheckBox: NSButton!
var transcodingEnabled: Bool {
get {
return transcodingCheckBox.state == .on
}
}
@IBOutlet weak var addURLBtn: NSButton!
func initRtmpURLField() {
rtmpURLField.label.stringValue = "rtmp"
rtmpURLField.field.placeholderString = "rtmp://"
}
/// callback when publish button hit
@IBAction func onAddStreamingURL(_ sender: Any) {
//let transcodingEnabled = transcodingCheckBox.state == .on
let rtmpURL = rtmpURLField.stringValue
if(rtmpURL.isEmpty || !rtmpURL.starts(with: "rtmp://")) {
showAlert(title: "Add Streaming URL Failed", message: "RTMP URL cannot be empty or not start with 'rtmp://'")
return
}
if transcodingEnabled {
// we will use transcoding to composite multiple hosts' video
// therefore we have to create a livetranscoding object and call before addPublishStreamUrl
transcoding.size = CGSize(width: CANVAS_WIDTH, height: CANVAS_HEIGHT)
agoraKit.setLiveTranscoding(transcoding)
}
// start publishing to this URL
agoraKit.addPublishStreamUrl(rtmpURL, transcodingEnabled: transcodingEnabled)
// update properties and UI
rtmpURLs.append(rtmpURL)
selectRtmpUrlsPicker.picker.addItem(withTitle: rtmpURL)
}
/**
--- Button ---
*/
@IBOutlet weak var joinChannelButton: NSButton!
func initJoinChannelButton() {
joinChannelButton.title = isJoined ? "Leave Channel".localized : "Join Channel".localized
}
// indicate if current instance has joined channel
var isJoined: Bool = false {
didSet {
channelField.isEnabled = !isJoined
initJoinChannelButton()
}
}
// indicate for doing something
var isProcessing: Bool = false {
didSet {
joinChannelButton.isEnabled = !isProcessing
}
}
override func viewDidLoad() {
super.viewDidLoad()
layoutVideos(2)
// Do view setup here.
let config = AgoraRtcEngineConfig()
config.appId = KeyCenter.AppId
config.areaCode = GlobalSettings.shared.area.rawValue
agoraKit = AgoraRtcEngineKit.sharedEngine(with: config, delegate: self)
agoraKit.enableVideo()
initSelectRtmpUrlsPicker()
initRtmpURLField()
initChannelField()
initJoinChannelButton()
}
override func viewWillBeRemovedFromSplitView() {
if isJoined {
agoraKit.leaveChannel { (stats:AgoraChannelStats) in
LogUtils.log(message: "Left channel", level: .info)
}
}
AgoraRtcEngineKit.destroy()
}
@IBAction func onJoinPressed(_ sender:Any) {
if !isJoined {
// check configuration
let channel = channelField.stringValue
if channel.isEmpty {
return
}
// set live broadcaster mode
agoraKit.setChannelProfile(.liveBroadcasting)
// set myself as broadcaster to stream video/audio
agoraKit.setClientRole(.broadcaster)
// set proxy configuration
let proxySetting = GlobalSettings.shared.proxySetting.selectedOption().value
agoraKit.setCloudProxy(AgoraCloudProxyType.init(rawValue: UInt(proxySetting)) ?? .noneProxy)
// enable video module and set up video encoding configs
agoraKit.setVideoEncoderConfiguration(
AgoraVideoEncoderConfiguration(
size: Configs.Resolutions[GlobalSettings.shared.resolutionSetting.selectedOption().value].size(),
frameRate: AgoraVideoFrameRate(rawValue: Configs.Fps[GlobalSettings.shared.fpsSetting.selectedOption().value]) ?? .fps15,
bitrate: AgoraVideoBitrateStandard,
orientationMode: .adaptative
)
)
// set up local video to render your local camera preview
let localVideo = videos[0]
let videoCanvas = AgoraRtcVideoCanvas()
videoCanvas.uid = 0
// the view to be binded
videoCanvas.view = localVideo.videocanvas
videoCanvas.renderMode = .hidden
agoraKit.setupLocalVideo(videoCanvas)
// start joining channel
// 1. Users can only see each other after they join the
// same channel successfully using the same app id.
// 2. If app certificate is turned on at dashboard, token is needed
// when joining channel. The channel name and uid used to calculate
// the token has to match the ones used for channel join
isProcessing = true
let option = AgoraRtcChannelMediaOptions()
let result = agoraKit.joinChannel(byToken: KeyCenter.Token, channelId: channel, info: nil, uid: 0, options: option)
if result != 0 {
isProcessing = false
// Usually happens with invalid parameters
// Error code description can be found at:
// en: https://docs.agora.io/en/Voice/API%20Reference/oc/Constants/AgoraErrorCode.html
// cn: https://docs.agora.io/cn/Voice/API%20Reference/oc/Constants/AgoraErrorCode.html
self.showAlert(title: "Error", message: "joinChannel call failed: \(result), please check your params")
}
} else {
isProcessing = true
agoraKit.leaveChannel { [unowned self] (stats:AgoraChannelStats) in
self.isProcessing = false
LogUtils.log(message: "Left channel", level: .info)
self.videos[0].uid = nil
self.isJoined = false
self.videos.forEach {
$0.uid = nil
$0.statsLabel.stringValue = ""
}
}
}
}
func layoutVideos(_ count: Int) {
videos = []
for i in 0...count - 1 {
let view = VideoView.createFromNib()!
if(i == 0) {
view.placeholder.stringValue = "Local"
} else {
view.placeholder.stringValue = "Remote \(i)"
}
videos.append(view)
}
// layout render view
Container.layoutStream(views: videos)
}
}
/// agora rtc engine delegate events
extension RTMPStreaming: AgoraRtcEngineDelegate {
/// callback when warning occured for agora sdk, warning can usually be ignored, still it's nice to check out
/// what is happening
/// Warning code description can be found at:
/// en: https://docs.agora.io/en/Voice/API%20Reference/oc/Constants/AgoraWarningCode.html
/// cn: https://docs.agora.io/cn/Voice/API%20Reference/oc/Constants/AgoraWarningCode.html
/// @param warningCode warning code of the problem
func rtcEngine(_ engine: AgoraRtcEngineKit, didOccurWarning warningCode: AgoraWarningCode) {
LogUtils.log(message: "warning: \(warningCode.rawValue)", level: .warning)
}
/// callback when error occured for agora sdk, you are recommended to display the error descriptions on demand
/// to let user know something wrong is happening
/// Error code description can be found at:
/// en: https://docs.agora.io/en/Voice/API%20Reference/oc/Constants/AgoraErrorCode.html
/// cn: https://docs.agora.io/cn/Voice/API%20Reference/oc/Constants/AgoraErrorCode.html
/// @param errorCode error code of the problem
func rtcEngine(_ engine: AgoraRtcEngineKit, didOccurError errorCode: AgoraErrorCode) {
LogUtils.log(message: "error: \(errorCode)", level: .error)
if isProcessing {
isProcessing = false
}
self.showAlert(title: "Error", message: "Error \(errorCode.rawValue) occur")
}
/// callback when the local user joins a specified channel.
/// @param channel
/// @param uid uid of local user
/// @param elapsed time elapse since current sdk instance join the channel in ms
func rtcEngine(_ engine: AgoraRtcEngineKit, didJoinChannel channel: String, withUid uid: UInt, elapsed: Int) {
isProcessing = false
isJoined = true
let localVideo = videos[0]
localVideo.uid = uid
LogUtils.log(message: "Join \(channel) with uid \(uid) elapsed \(elapsed)ms", level: .info)
// add transcoding user so the video stream will be involved
// in future RTMP Stream
let user = AgoraLiveTranscodingUser()
user.rect = CGRect(x: 0, y: 0, width: CANVAS_WIDTH / 2, height: CANVAS_HEIGHT)
user.uid = uid
transcoding.add(user)
}
/// callback when a remote user is joinning the channel, note audience in live broadcast mode will NOT trigger this event
/// @param uid uid of remote joined user
/// @param elapsed time elapse since current sdk instance join the channel in ms
func rtcEngine(_ engine: AgoraRtcEngineKit, didJoinedOfUid uid: UInt, elapsed: Int) {
LogUtils.log(message: "remote user join: \(uid) \(elapsed)ms", level: .info)
// find a VideoView w/o uid assigned
if let remoteVideo = videos.first(where: { $0.uid == nil }) {
let videoCanvas = AgoraRtcVideoCanvas()
videoCanvas.uid = uid
// the view to be binded
videoCanvas.view = remoteVideo.videocanvas
videoCanvas.renderMode = .hidden
agoraKit.setupRemoteVideo(videoCanvas)
remoteVideo.uid = uid
} else {
LogUtils.log(message: "no video canvas available for \(uid), cancel bind", level: .warning)
}
// update live transcoding
// add new user onto the canvas
let user = AgoraLiveTranscodingUser()
user.rect = CGRect(x: CANVAS_WIDTH / 2, y: 0, width: CANVAS_WIDTH / 2, height: CANVAS_HEIGHT)
user.uid = uid
self.transcoding.add(user)
// remember you need to call setLiveTranscoding again if you changed the layout
agoraKit.setLiveTranscoding(transcoding)
}
/// callback when a remote user is leaving the channel, note audience in live broadcast mode will NOT trigger this event
/// @param uid uid of remote joined user
/// @param reason reason why this user left, note this event may be triggered when the remote user
/// become an audience in live broadcasting profile
func rtcEngine(_ engine: AgoraRtcEngineKit, didOfflineOfUid uid: UInt, reason: AgoraUserOfflineReason) {
LogUtils.log(message: "remote user left: \(uid) reason \(reason)", level: .info)
// to unlink your view from sdk, so that your view reference will be released
// note the video will stay at its last frame, to completely remove it
// you will need to remove the EAGL sublayer from your binded view
if let remoteVideo = videos.first(where: { $0.uid == uid }) {
let videoCanvas = AgoraRtcVideoCanvas()
videoCanvas.uid = uid
// the view to be binded
videoCanvas.view = nil
videoCanvas.renderMode = .hidden
agoraKit.setupRemoteVideo(videoCanvas)
remoteVideo.uid = nil
} else {
LogUtils.log(message: "no matching video canvas for \(uid), cancel unbind", level: .warning)
}
// remove user from canvas if current cohost left channel
transcoding.removeUser(uid)
// remember you need to call setLiveTranscoding again if you changed the layout
agoraKit.setLiveTranscoding(transcoding)
}
/// callback for state of rtmp streaming, for both good and bad state
/// @param url rtmp streaming url
/// @param state state of rtmp streaming
/// @param reason
func rtcEngine(_ engine: AgoraRtcEngineKit, rtmpStreamingChangedToState url: String, state: AgoraRtmpStreamingState, errorCode: AgoraRtmpStreamingErrorCode) {
LogUtils.log(message: "rtmp streaming: \(url) state \(state.rawValue) error \(errorCode.rawValue)", level: .info)
if(state == .running) {
self.showAlert(title: "Notice", message: "\(url) Publish Success")
} else if(state == .failure) {
self.showAlert(title: "Error", message: "\(url) Publish Failed: \(errorCode.rawValue)")
} else if(state == .idle) {
self.showAlert(title: "Notice", message: "\(url) Publish Stopped")
}
}
/// callback when live transcoding is properly updated
func rtcEngineTranscodingUpdated(_ engine: AgoraRtcEngineKit) {
LogUtils.log(message: "live transcoding updated", level: .info)
}
}