Linux平台x86_64|aarch64架构RTMP推送|轻量级RTSP服务模块集成说明

NT_PB_SetSDKClientKey("xxx", "xxxx", 0, nullptr);

auto ret = push_api.Init(0, nullptr);

void LogInit()
{
	SmartLogAPI log_api;
	memset(&log_api, 0, sizeof(log_api));
	GetSmartLogAPI(&log_api);

	log_api.SetLevel(SL_INFO_LEVEL);
	log_api.SetPath((NT_PVOID)"./");
}

bool PushSDKInit(NT_SmartPublisherSDKAPI& push_api)
{
	memset(&push_api, 0, sizeof(push_api));
	NT_GetSmartPublisherSDKAPI(&push_api);

	auto ret = push_api.Init(0, nullptr);
	if (NT_ERC_OK != ret)
	{
		fprintf(stderr, "push_api.Init failed!\n");
		return false;
	}
	else
	{
		fprintf(stdout, "push_api.Init ok!\n");
	}

	return true;
}

/*
这个是最后一个调用的接口
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *UnInit)();

/*
video_option 请参考 NT_PB_E_VIDEO_OPTION,
auido_option 请参考 NT_PB_E_AUDIO_OPTION
flag目前传0，后面扩展用， pReserve传NULL,扩展用,
获取Handle
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *Open)(NT_PHANDLE pHandle, NT_UINT32 video_option, NT_UINT32 auido_option, NT_UINT32 flag, NT_PVOID pReserve);

/*
调用这个接口之后handle失效，
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *Close)(NT_HANDLE handle);

/*定义Video源选项*/
typedef enum _NT_PB_E_VIDEO_OPTION
{
	NT_PB_E_VIDEO_OPTION_NO_VIDEO = 0x0,
	NT_PB_E_VIDEO_OPTION_SCREEN   = 0x1, // 采集屏幕
	NT_PB_E_VIDEO_OPTION_CAMERA	  = 0x2, // 摄像头采集
	NT_PB_E_VIDEO_OPTION_LAYER    = 0x3, // 视频合并，比如桌面叠加摄像头等
	NT_PB_E_VIDEO_OPTION_ENCODED_DATA = 0x4, // 已经编码的视频数据，目前支持H264
	NT_PB_E_VIDEO_OPTION_WINDOW   = 0x5, // 采集窗口
} NT_PB_E_VIDEO_OPTION;

/*定义Auido源选项*/
typedef enum _NT_PB_E_AUDIO_OPTION
{
	NT_PB_E_AUDIO_OPTION_NO_AUDIO					= 0x0,
	NT_PB_E_AUDIO_OPTION_CAPTURE_MIC				= 0x1, // 采集麦克风音频
	NT_PB_E_AUDIO_OPTION_CAPTURE_SPEAKER			= 0x2, // 采集扬声器
	NT_PB_E_AUDIO_OPTION_CAPTURE_MIC_SPEAKER_MIXER	= 0x3, // 麦克风扬声器混音
	NT_PB_E_AUDIO_OPTION_ENCODED_DATA				= 0x4, // 编码后的音频数据，目前支持AAC, speex宽带(wideband mode)
	NT_PB_E_AUDIO_OPTION_EXTERNAL_PCM_DATA			= 0x5, /*外部PCM数据*/
	NT_PB_E_AUDIO_OPTION_MIC_EXTERNAL_PCM_MIXER     = 0x6, /* 麦克风和外部PCM数据混音 当前只支持一路外部音频和内置麦克风混音*/
	NT_PB_E_AUDIO_OPTION_TWO_EXTERNAL_PCM_MIXER     = 0x7, /* 两路外部PCM数据混音*/
} NT_PB_E_AUDIO_OPTION;

/*
 * nt_linux_smart_publisher_sdk.h
 * Author: daniusdk.com
 * WeChat: xinsheng120
 */
/*
设置软硬编码类型, 编码器, codec_id, 编码器其他参数.
type: 0为软编码, 1为硬编码, 默认是软编码. linux当前不支持硬编码
encoder_id: 如果是软编码, 并且用h264, 可以设置0, 0用默认编码器, 也可以设置1, 设置1将使用OpenH264编码.
param1: 如果是软编码,请设置0;
codec_id: 设置h264或h265编码, 默认是h264, 请参考NT_MEDIA_CODEC_ID,  h264填 NT_MEDIA_CODEC_ID_H264, h265填 NT_MEDIA_CODEC_ID_H265.
注意: 软编码不支持h265, 硬编码根据实际硬件情况决定是否支持h265.
如果调用了这个接口，请不要再调用SetVideoEncoderType接口
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *SetVideoEncoder)(NT_HANDLE handle, NT_INT32 type, NT_INT32 encoder_id, NT_UINT32 codec_id, NT_INT32 param1);

/*
注意，码率控制两种方式，要么是 SetVideoQuality(或者 SetVideoQualityV2) + SetVideoMaxBitRate
要么是 SetVideoMaxBitRate  + SetVideoBitRate
对于OpenH264, 只支持 SetVideoMaxBitRate  + SetVideoBitRate方式
*/

/*
设置视频质量, 范围[1-50], 值越小视频质量越好,但码率会越大. 请优先考虑默认值,
对于H264, 默认值23
对于H265, 默认值28
注意:用OpenH264编码器时调用这个接口无用
*/
NT_UINT32(NT_API *SetVideoQuality)(NT_HANDLE handle, NT_INT32 quality);


/*
设置最大视频码率, 单位kbps
*/
NT_UINT32(NT_API *SetVideoMaxBitRate)(NT_HANDLE handle, NT_INT32 kbit_rate);

/*
设置视频码率, 单位kbps, 默认是0，不使用平均码率方式
注意:用OpenH264编码器时调需要设置适当的值
*/
NT_UINT32(NT_API *SetVideoBitRate)(NT_HANDLE handle, NT_INT32 kbit_rate);


/*
* 在一些特殊场景下, 视频分辨率会改变, 如果设置一个固定码率的的话,当视频分辨率变大的时候会变的模糊,变小的话又会浪费码率
* 所以提供可以设置一组码率的接口,满足不同分辨率切换的需求
* 规则: 比如设置两组分辨率 640*360, 640*480, 那么当分辨率小于等于640*360时都使用640*360的码率,
* 当分辨率大于640*360且小于等于640*480时，就使用640*480的码率,如果分辨率大于640*480 那就使用640*480的分辨率
* 为了设置的更准确, 建议多划分几组, 让区间变小
* 调用这个接口每次设置一组,设置多组就调用多次
*
*/
NT_UINT32(NT_API* AddVideoEncoderBitrateGroupItem)(NT_HANDLE handle, const NT_PB_VideoEncoderBitrateGroupItem* item);


/*
清除视频码率组
*/
NT_UINT32(NT_API* ClearVideoEncoderBitrateGroup)(NT_HANDLE handle);


/*
设置关键帧间隔, 比如1表示所有帧都是关键帧，10表示每10帧里面一个关键帧，25表示每25帧一个关键帧
*/
NT_UINT32(NT_API *SetVideoKeyFrameInterval)(NT_HANDLE handle, NT_INT32 interval);


/*
设置H264 profile.
profile: 1: H264 baseline(默认值). 2: H264 main. 3. H264 high
注意:用OpenH264编码器时不支持main profile, 如果设置成main profile的话，会回落到baseline profile.
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *SetVideoEncoderProfile)(NT_HANDLE handle, NT_INT32 profile);

/*
设置视频编码速度
speed: 范围是 1 到 6,  值越小，速度越快，质量也越差
*/
NT_UINT32(NT_API *SetVideoEncoderSpeed)(NT_HANDLE handle, NT_INT32 speed);

/*
设置视频编码量化范围, 设置最小值
qp_min: h264和h265设置范围都是[0, 51], 设置-1的话，编码器使用默认值;
注意目前只对OpenH264有效
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *SetVideoEncoderQPMin)(NT_HANDLE handle, NT_INT32 qp_min);


/*
设置视频编码量化范围, 设置最小值
qp_max: h264和h265设置范围都是[0, 51], 设置-1的话，编码器使用默认值;
注意目前只对OpenH264有效
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *SetVideoEncoderQPMax)(NT_HANDLE handle, NT_INT32 qp_max);


/*
不同的视频编码器一般会有一些特定的选项, 这个接口可以设置这些选项
option_name: 选项名称
option_value: 选项值
成功返回 NT_ERC_OK
********************************************
***********OpenH264选项说明******************
可以设置的选项有: "usage_type", "rc_mode", "enable_frame_skip" 
"usage_type": 编码场景, 可以设置0和1, 0是实时摄像头编码, 1是实时屏幕编码, sdk默认是0;
"rc_mode": 码率控制模式, 可以设置0和1, 0是质量模式, 1是码率模式, sdk默认是1;
"enable_frame_skip": 通过跳帧的方式来控制码率, 设置0是关闭跳帧, 1是打开跳帧, sdk默认是0;
***********OpenH264选项说明******************
********************************************
*/
NT_UINT32(NT_API *SetVideoEncoderSpecialInt32Option)(NT_HANDLE handle, NT_PCSTR option_name, NT_INT32 option_value);


/*
清除编码器所有设置的特定选项
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API *ClearVideoEncoderSpecialOptions)(NT_HANDLE handle);

/*
 * nt_linux_smart_publisher_sdk.h
 * Author: daniusdk.com
 * WeChat: xinsheng120
 */
/*
获取系统麦克风输入设备数
audio_input_layer: 1:alsa lib, 2:PulseAudio, 其他值返回错误, 优先推荐使用PulseAudio
注意:当使用alsa lib, 可能会返回很多PCM设备, 具体可以用 "arecord -L" 命令自己比对下
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* GetAuidoInputDeviceNumber)(NT_INT32 audio_input_layer, NT_INT32* pNumer);

/*
获取麦克风设备名称
audio_input_layer: 1:alsa lib, 2:PulseAudio, 其他值返回错误.
device_id: 设备ID，需要输入, 从0开始，最大值不能超过设备数, 注意: 这个id一定要和audio_input_layer对应
device_name: 设备名称，返回的字符串以0结尾
device_name_size: 设备名称buffer的大小,建议512
成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* GetAuidoInputDeviceName)(NT_INT32 audio_input_layer, NT_INT32 device_id, NT_PSTR device_name, NT_UINT32 device_name_size);


/*
*设置麦克风采集层
*audio_input_layer: 1:使用alsa lib采集麦克风, 2:使用PulseAudio采集麦克风, 其他值返回失败, SDK默认使用PulseAudio采集麦克风
*成功返回NT_ERC_OK
*/
NT_UINT32(NT_API *SetAudioInputLayer)(NT_HANDLE handle, NT_INT32 audio_input_layer);


/*
设置麦克风采集设备ID
*device_id: 设备id, 一般从0开始, 这个id要根据SetAudioInputLayer设置的值来设置, 
*就是说如果使用alsa lib采集麦克风, 那就对应alsa lib获取到的id,
*如果使用PulseAudio, 那就对应PulseAudio获取到的id,
*不能随意设置，否则采集结果未定义, 或者返回失败
*成功返回NT_ERC_OK
*/
NT_UINT32(NT_API* SetAuidoInputDeviceId)(NT_HANDLE handle, NT_INT32 device_id);


/*
检查是否能采集扬声器音频, 当前SDK只使用PulseAudio采集
audio_input_layer: 2:使用PulseAudio采集扬声器声音, 这个值目前必须是2, 其他返回错误
out_flag: 1表示可以采集扬声器，0:表示不可以采集扬声器
*/
NT_UINT32(NT_API* IsCanCaptureSpeaker)(NT_INT32 audio_input_layer, NT_INT32* out_flag);


/*
*设置推送音频编码类型
*type: 1:使用AAC编码, 2:使用speex编码, 其他值返回错误
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetPublisherAudioCodecType)(NT_HANDLE handle, NT_INT32 type);


/*
*设置推送Speex编码质量
*quality: 范围是0-10, 默认是8:大约28kbps, 值越大，质量越好，码率也越大
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetPublisherSpeexEncoderQuality)(NT_HANDLE handle, NT_INT32 quality);


/*
设置推送静音
is_mute: 0表示不静音, 1表示静音
*/
NT_UINT32(NT_API* SetMute)(NT_HANDLE handle, NT_INT32 is_mute);


/*
*设置回音消除
*isCancel： 1表示消除回音, 0表示不消除回音
*delay: 回音时延，单位是毫秒,目前推荐设置100ms, 如果设置为0的话, 系统自动处理
*注意这个只对麦克风捕获有作用，扬声器捕获无效
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetEchoCancellation)(NT_HANDLE handle, NT_INT32 isCancel, NT_INT32 delay);

	
/*
*设置音频噪音抑制
*isNS: 1表示噪音抑制, 0表示不抑制
*注意，这个一般用在采集麦克风上，采集系统播放声音时必须要性不大
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetNoiseSuppression)(NT_HANDLE handle, NT_INT32 isNS);

/*
*设置音频自动增益控制
*isAGC: 1表示增益控制, 0表示不控制
*注意，这个一般用在采集麦克风上，采集系统播放声音时必须要性不大
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetAGC)(NT_HANDLE handle, NT_INT32 isAGC);


/*
*设置端点检测(Voice Activity Detection (VAD))
*isVAD: 1表示端点检测, 0表示不检测
*注意，这个一般用在采集麦克风上，采集系统播放音乐时不要使用
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetVAD)(NT_HANDLE handle, NT_INT32 isVAD);


/*
*设置输入音量, 这个接口一般不建议调用, 在一些特殊情况下可能会用, 一般不建议放大音量, 优先考虑AGC
*index: 一般是0和1, 如果没有混音的只用0, 有混音的话, 0,1分别设置音量
*volume: 音量，默认是1.0，范围是[0.0, 2.0], 设置成0静音, 1音量不变
*成功返回 NT_ERC_OK
*/
NT_UINT32(NT_API* SetInputAudioVolume)(NT_HANDLE handle, NT_INT32 index, float volume);

push_api->SetEventCallBack(push_handle, nullptr, OnSDKEventHandle);

/*事件ID*/
typedef enum _NT_PB_E_EVENT_ID
{
	NT_PB_E_EVENT_ID_BASE = NT_EVENT_ID_SMART_PUBLISHER_SDK,

	NT_PB_E_EVENT_ID_CONNECTING			= NT_PB_E_EVENT_ID_BASE | 0x2,	/*连接中, param5表示推送URL */
	NT_PB_E_EVENT_ID_CONNECTION_FAILED	= NT_PB_E_EVENT_ID_BASE | 0x3,	/*连接失败, param5表示推送URL*/
	NT_PB_E_EVENT_ID_CONNECTED			= NT_PB_E_EVENT_ID_BASE | 0x4,	/*已连接, param5表示推送URL*/
	NT_PB_E_EVENT_ID_DISCONNECTED		= NT_PB_E_EVENT_ID_BASE | 0x5,	/*断开连接, param5表示推送URL*/
	
	NT_PB_E_EVENT_ID_RECORDER_START_NEW_FILE    = NT_PB_E_EVENT_ID_BASE | 0x7,	/*录像写入新文件, param5表示录像文件名*/
	NT_PB_E_EVENT_ID_ONE_RECORDER_FILE_FINISHED = NT_PB_E_EVENT_ID_BASE | 0x8,	/*一个录像文件完成, param5表示录像文件名*/

	NT_PB_E_EVENT_ID_CAPTURE_WINDOW_INVALID = NT_PB_E_EVENT_ID_BASE | 0xd, /*捕获窗口时，如果窗口句柄无效则通知用户, param1为窗口句柄*/

	NT_PB_E_EVENT_ID_RTSP_URL = NT_PB_E_EVENT_ID_BASE | 0xe, /* 通知rtsp url, param1表示rtsp server handle, param5 表示rtsp url */

	NT_PB_E_EVENT_ID_PUSH_RTSP_SERVER_RESPONSE_STATUS_CODE = NT_PB_E_EVENT_ID_BASE | 0xf,  /* 推送rtsp时服务端相应的status code上报，目前只上报401, param1表示status code,  param5表示推送URL */
	NT_PB_E_EVENT_ID_PUSH_RTSP_SERVER_NOT_SUPPORT = NT_PB_E_EVENT_ID_BASE | 0x10,  /* 推送rtsp时服务器不支持rtsp推送,  param5表示推送URL */

} NT_PB_E_EVENT_ID;

bool start_rtmp(NT_SmartPublisherSDKAPI* push_api, NT_HANDLE handle, const std::string& rtmp_url) {
	if (NT_ERC_OK != push_api->SetURL(handle, rtmp_url.c_str(), NULL))
		return false;

	if (NT_ERC_OK != push_api->StartPublisher(handle, NULL))
		return false;

	return true;
}

push_api->StopPublisher(handle);

NT_HANDLE start_rtsp_server(NT_SmartPublisherSDKAPI* push_api, int port, std::string user_name, std::string password) {

	NT_HANDLE rtsp_server_handle = nullptr;
	if (NT_ERC_OK != push_api->OpenRtspServer(&rtsp_server_handle, 0)) {
		fprintf(stderr, "OpenRtspServer failed\n");
		return nullptr;
	}

	if (nullptr == rtsp_server_handle) {
		fprintf(stderr, "rtsp_server_handle is null\n");
		return nullptr;
	}

	if (NT_ERC_OK != push_api->SetRtspServerPort(rtsp_server_handle, port)) {
		push_api->CloseRtspServer(rtsp_server_handle);
		return nullptr;
	}

	if (!user_name.empty() && !password.empty())
		push_api->SetRtspServerUserNamePassword(rtsp_server_handle, user_name.c_str(), password.c_str());

	if (NT_ERC_OK == push_api->StartRtspServer(rtsp_server_handle, 0))
		return rtsp_server_handle;

	fprintf(stderr, "StartRtspServer failed\n");
	push_api->CloseRtspServer(rtsp_server_handle);
	
	return nullptr;
}

void stop_rtsp_server(NT_SmartPublisherSDKAPI* push_api, NT_HANDLE& rtsp_server_handle) {
	if (nullptr == rtsp_server_handle)
		return;

	NT_HANDLE handle = rtsp_server_handle;
	rtsp_server_handle = nullptr;

	push_api->StopRtspServer(handle);
	push_api->CloseRtspServer(handle);
}

bool start_rtsp_stream(NT_SmartPublisherSDKAPI* push_api, NT_HANDLE rtsp_server_handle, NT_HANDLE handle, const std::string stream_name) {

	push_api->SetRtspStreamName(handle, stream_name.c_str());

	push_api->ClearRtspStreamServer(handle);

	push_api->AddRtspStreamServer(handle, rtsp_server_handle, 0);
	
	if (NT_ERC_OK != push_api->StartRtspStream(handle, 0))
		return false;

	return true;
}

push_api.StopRtspStream(push_handle);