Windows平台RTMP、RTSP播放器录像模块精细化控制

精选原创

音视频牛哥 2022-12-24 19:37:45 博主文章分类：大牛直播SDK ©著作权

文章标签 RTSP播放器 RTMP播放器 RTSP录制MP4 RTMP录制MP4 windows播放器 文章分类 C/C++ 后端开发 yyds干货盘点

©著作权归作者所有：来自51CTO博客作者音视频牛哥的原创作品，请联系作者获取转载授权，否则将追究法律责任

技术背景

上篇文章，我们介绍了Unity平台RTMP、RTSP播放器录像功能，这里，我们详细的介绍下，做个RTSP或RTMP拉流端录像模块有哪些需要考虑的技术点？

在我们常规的考量，RTMP或RTSP流录制，无非就是拉取数据写文件而已，接口设计StartRecorder()/StopRecorder()足矣。

是的，一般场景下，两个接口足够了，但如果是做个更加通用的模块，以下几点是可以酌情考虑的：

支持设置单个录像文件大小，比如单个录像文件最大设置到200M，到了200M，可自动切分到下个录像文件；
支持设置录像路径；
支持设置录像文件前缀：录像文件前缀，是为了更友好的做特定文件的分类；
支持文件名增加日期；
支持文件名增加时间；
支持设置纯音频、纯视频、音视频录制模式；
支持音频(PCMU/PCMA,Speex等)转AAC后再录像；
支持录像事件回调，从开始录像，到录像结束均有event callback上来。

除了上述的设计，还需要确保和RTSP、RTMP播放在一个实例下，确保播放的过程中可以随时录像，录像的过程中，可以随时播放。

录像模块设计

无图无真相，先看录像设置：

Windows平台RTMP、RTSP播放器录像模块精细化控制_RTMP播放器

启动录像、停止录像：

Windows平台RTMP、RTSP播放器录像模块精细化控制_windows播放器_02

Windows平台，我们提供了C++和C#的接口，本文以C++接口设计为例：

先说录像设置：

设置录制纯音频或纯视频：

/*
    * 设置是否录视频，默认的话，如果视频源有视频就录，没有就没得录, 但有些场景下可能不想录制视频，只想录音频，所以增加个开关
    * is_record_video: 1 表示录制视频, 0 表示不录制视频, 默认是1
    */
    NT_UINT32(NT_API *SetRecorderVideo)(NT_HANDLE handle, NT_INT32 is_record_video);


    /*
    * 设置是否录音频，默认的话，如果视频源有音频就录，没有就没得录, 但有些场景下可能不想录制音频，只想录视频，所以增加个开关
    * is_record_audio: 1 表示录制音频, 0 表示不录制音频, 默认是1
    */
    NT_UINT32(NT_API *SetRecorderAudio)(NT_HANDLE handle, NT_INT32 is_record_audio);

设置录像目录：

/*
    设置本地录像目录, 必须是英文目录，否则会失败
    */
    NT_UINT32(NT_API *SetRecorderDirectory)(NT_HANDLE handle, NT_PCSTR dir);

设置单个录像文件最大大小：

/*
    设置单个录像文件最大大小, 当超过这个值的时候，将切割成第二个文件
    size: 单位是KB(1024Byte), 当前范围是 [5MB-800MB], 超出将被设置到范围内
    */
    NT_UINT32(NT_API *SetRecorderFileMaxSize)(NT_HANDLE handle, NT_UINT32 size);

设置录像文件名生成规则：

/*
    设置录像文件名生成规则
    */
    NT_UINT32(NT_API *SetRecorderFileNameRuler)(NT_HANDLE handle, NT_SP_RecorderFileNameRuler* ruler);

对应的NT_SP_RecorderFileNameRuler设计：

// 如果三项都是0的话，将不能启动录像
typedef struct _NT_SP_RecorderFileNameRuler
{
  NT_UINT32  type_; // 这个值目前默认是0，将来扩展用
  NT_PCSTR  file_name_prefix_; // 设置一个录像文件名前缀, 例如:daniulive
  NT_INT32  append_date_; // 如果是1的话，将在文件名上加日期, 例如:daniulive-2017-01-17
  NT_INT32  append_time_; //  如果是1的话，将增加时间，例如:daniulive-2017-01-17-17-10-36
} NT_SP_RecorderFileNameRuler;

设置录像回调接口：

/*
    设置录像回调接口
    */
    NT_UINT32(NT_API *SetRecorderCallBack)(NT_HANDLE handle,
      NT_PVOID call_back_data, SP_SDKRecorderCallBack call_back);

对应录像回调：

/*
录像回调
status: 1:表示开始写一个新录像文件. 2:表示已经写好一个录像文件
file_name: 实际录像文件名
*/
typedef NT_VOID(NT_CALLBACK* SP_SDKRecorderCallBack)(NT_HANDLE handle, NT_PVOID user_data, NT_UINT32 status,
  NT_PCSTR file_name);

设置音频转AAC开关：

/*
    设置录像时音频转AAC编码的开关, aac比较通用，sdk增加其他音频编码(比如speex, pcmu, pcma等)转aac的功能.
    is_transcode: 设置为1的话，如果音频编码不是aac，则转成aac, 如果是aac，则不做转换. 设置为0的话，则不做任何转换. 默认是0.
    注意: 转码会增加性能消耗
    */
    NT_UINT32(NT_API *SetRecorderAudioTranscodeAAC)(NT_HANDLE handle, NT_INT32 is_transcode);

启动录像、停止录像：

/*
    启动录像
    */
    NT_UINT32(NT_API *StartRecorder)(NT_HANDLE handle);

    /*
    停止录像
    */
    NT_UINT32(NT_API *StopRecorder)(NT_HANDLE handle);

接口调用实例

/*
 * Author: https://daniusdk.com
 */
void CSmartPlayerDlg::OnBnClickedButtonRecord()
{
  if ( player_handle_ == NULL )
    return;

  CString btn_record_str;
  btn_record_.GetWindowTextW(btn_record_str);

  if ( btn_record_str == _T("录像") )
  {
    if ( !rec_conf_info_.is_record_video_ && !rec_conf_info_.is_record_audio_ )
    {
      AfxMessageBox(_T("音频录制选项和视频录制选项至少需要选择一个!"));
      return;
    }

    if ( !is_playing_ )
    {
      if ( !InitCommonSDKParam() )
      {
        AfxMessageBox(_T("设置参数错误!"));
        return;
      }
    }

    player_api_.SetRecorderVideo(player_handle_, rec_conf_info_.is_record_video_ ? 1 : 0);
    player_api_.SetRecorderAudio(player_handle_, rec_conf_info_.is_record_audio_ ? 1 : 0);

    auto ret = player_api_.SetRecorderDirectory(player_handle_, rec_conf_info_.dir_.c_str());
    if ( NT_ERC_OK != ret )
    {
      AfxMessageBox(_T("设置录像目录失败，请确保目录存在且是英文目录"));
      return;
    }

    player_api_.SetRecorderFileMaxSize(player_handle_, rec_conf_info_.file_max_size_);

    NT_SP_RecorderFileNameRuler rec_name_ruler = { 0 };

    rec_name_ruler.type_ = 0;
    rec_name_ruler.file_name_prefix_ = rec_conf_info_.file_name_prefix_.c_str();
    rec_name_ruler.append_date_     = rec_conf_info_.is_append_date_ ? 1 : 0;
    rec_name_ruler.append_time_     = rec_conf_info_.is_append_time_ ? 1 : 0;

    player_api_.SetRecorderFileNameRuler(player_handle_, &rec_name_ruler);

    player_api_.SetRecorderCallBack(player_handle_, GetSafeHwnd(), &SP_SDKRecorderHandle);

    player_api_.SetRecorderAudioTranscodeAAC(player_handle_, rec_conf_info_.is_audio_transcode_aac_ ? 1 : 0);

    if ( NT_ERC_OK != player_api_.StartRecorder(player_handle_) )
    {
      AfxMessageBox(_T("录像失败!"));
      return;
    }

    btn_record_.SetWindowTextW(_T("停止录像"));
    is_recording_ = true;
  }
  else
  {
    StopRecorder();
  }
}

停止录像：

void CSmartPlayerDlg::StopRecorder()
{
  if (player_handle_ == NULL)
    return;

  player_api_.StopRecorder(player_handle_);

  btn_record_.SetWindowTextW(_T("录像"));
  is_recording_ = false;

  if (!is_playing_)
  {
    SetWindowText(base_title_);
    edit_duration_.SetWindowText(_T(""));
    btn_pause_.SetWindowText(_T("暂停"));
  }

  RefreshLogo(true);
}