asr_online.h

Go to the documentation of this file.

#ifndef ASR_ONLINE_H_
#define ASR_ONLINE_H_
#include <curl/curl.h>
#include "common_config.h"
#include "file_operation.h"
#include "iostream"
#include "linuxrec.h"

struct AsrConfig
{
  std::string api_key;
  std::string secret_key;
  std::string format;
  int rate;
  int dev_pid;
  std::string cuid;
  std::string scope;
  std::string url;
};

class BaiduAsrOnline
{
public:
  void setAsrParams(const std::string base_path, const std::string pcm_file, const int channel);

  void initAndConfigAsr();

  void speechGetToken();

  void getPcmFileData();

  char* runAsrAndRecog();

  std::string resultFromJson();

  struct DataBuff recordThroughMIC(const float record_time, bool enable_audio_save);

  void getPCMData(struct DataBuff pcm_buff);

  void stopRecordThroughMIC();

  void uninitAsr();

private:
  struct DataBuff getOneChannelData();

  void saveRecordDataToFile();

  void praseToken();

  static size_t writefunc(void* ptr, size_t size, size_t nmemb, char** result);

  struct DataBuff pcm_data_ = { NULL, 0 };  // 存储pcm音频数据
  char* rec_result_ = NULL;                 // 保存语音识别结果
  std::string online_final_result;
  std::string base_path_;
  std::string pcm_file_;
  int channel_;
  int record_loops_ = 0;
  std::string token_;
  char* url_response_ = NULL;
  const std::string API_TOKEN_URL = "http://openapi.baidu.com/oauth/2.0/token";
  struct recorder* asr_record_ = NULL;
  FileOperation pcm_file_operation;
  RecordAlsaAPI record_alsa;
  struct AsrConfig asr_config;
};

#endif