zircon/system/dev/audio/sherlock-pdm-input/audio-stream-in.h

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#pragma once

#include <audio-proto/audio-proto.h>

#include <lib/device-protocol/pdev.h>

#include <dispatcher-pool/dispatcher-timer.h>

#include <lib/fzl/pinned-vmo.h>
#include <lib/simple-audio-stream/simple-audio-stream.h>
#include <lib/zx/bti.h>
#include <lib/zx/vmo.h>

#include <soc/aml-common/aml-pdm-audio.h>

#include <lib/zircon-internal/thread_annotations.h>

#include <optional>

namespace audio {
namespace sherlock {

class SherlockAudioStreamIn : public SimpleAudioStream {
 public:
  static zx_status_t Create(void* ctx, zx_device_t* parent);

 protected:
  zx_status_t Init() __TA_REQUIRES(domain_->token()) override;
  zx_status_t ChangeFormat(const audio_proto::StreamSetFmtReq& req)
      __TA_REQUIRES(domain_->token()) override;
  zx_status_t GetBuffer(const audio_proto::RingBufGetBufferReq& req, uint32_t* out_num_rb_frames,
                        zx::vmo* out_buffer) __TA_REQUIRES(domain_->token()) override;
  zx_status_t Start(uint64_t* out_start_time) __TA_REQUIRES(domain_->token()) override;
  zx_status_t Stop() __TA_REQUIRES(domain_->token()) override;
  zx_status_t InitPost() override;

 private:
  friend class SimpleAudioStream;
  friend class fbl::RefPtr<SherlockAudioStreamIn>;

  SherlockAudioStreamIn(zx_device_t* parent);
  ~SherlockAudioStreamIn() {}

  zx_status_t AddFormats() __TA_REQUIRES(domain_->token());
  zx_status_t InitBuffer(size_t size) TA_REQ(domain_->token());
  zx_status_t InitPDev() TA_REQ(domain_->token());
  zx_status_t ProcessRingNotification();

  uint32_t us_per_notification_ = 0;
  fbl::RefPtr<dispatcher::Timer> notify_timer_;
  std::optional<ddk::PDev> pdev_ TA_GUARDED(domain_->token());
  zx::vmo ring_buffer_vmo_ TA_GUARDED(domain_->token());
  fzl::PinnedVmo pinned_ring_buffer_ TA_GUARDED(domain_->token());
  fbl::unique_ptr<AmlPdmDevice> pdm_;
  zx::bti bti_ TA_GUARDED(domain_->token());
};
}  // namespace sherlock
}  // namespace audio