blob: c709c67de4eb18167a02c873410aa266bd6de6b0 [file] [log] [blame]
// Copyright 2018 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <cras_client.h>
#include <cras_helpers.h>
#include <cras_types.h>
#include <cras_util.h>
#include <map>
#include <stdint.h>
#include <stdio.h>
#include <string>
#include "media_perception/chrome_audio_service_client.h"
#include "media_perception/device_management.pb.h"
namespace mri {
// Handles interaction with a single audio device and piping outputs to the
// registered handlers. By itself, this class is not thread safe.
class AudioReceiver {
explicit AudioReceiver(const std::string& device_id);
~AudioReceiver() { DestroyParams(); }
// Returns true if the |requested| params match the current |params_|.
bool CaptureFormatMatches(const AudioStreamParams& requested);
// Returns the size of the frame handler map.
int GetFrameHandlerCount();
// Sets the audio stream |params_| on the AudioReceiver.
bool SetAudioStreamParams(const AudioStreamParams& params);
AudioStreamParams GetAudioStreamParams();
// Attempts to start audio capture.
bool StartAudioCaptureForDeviceIdx(struct cras_client* client, int dev_idx);
// Adds an audio frame handler into the frame handler map. Return value is the
// frame handler id.
int AddAudioFrameHandler(ChromeAudioServiceClient::AudioFrameHandler handler);
// Callback when getting a new audio frame.
void ProcessAudioSamples(const uint8_t* samples, unsigned int num_samples);
// Attempts to remove a frame handler based on id value. Return value
// indicates success or failure.
bool RemoveFrameHandler(int frame_handler_id);
void StopAudioCapture(struct cras_client* client);
void DestroyParams();
// Keeps track of the current capture parameters for an audio device.
AudioStreamParams params_;
// Cras structures managed by the AudioReceiver class.
struct cras_audio_format* audio_format_;
struct cras_stream_params* audio_capture_params_;
cras_stream_id_t stream_id_;
// The device id associated with this device.
std::string device_id_;
// Keeps a counter to ensure that each frame handler has a unique id.
int frame_handler_counter_;
// Stores a map of frame handler ids to AudioFrameHandlers.
std::map<int /* frame_handler_id */,
} // namespace mri