src/media/audio/tools/audio-driver-ctl/wav-sink.h

// Copyright 2017 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.

#ifndef SRC_MEDIA_AUDIO_TOOLS_AUDIO_DRIVER_CTL_WAV_SINK_H_
#define SRC_MEDIA_AUDIO_TOOLS_AUDIO_DRIVER_CTL_WAV_SINK_H_

#include <zircon/types.h>

#include <audio-utils/audio-stream.h>

#include "wav-common.h"

class WAVSink : public WAVCommon, public audio::utils::AudioSink {
 public:
  WAVSink() {}
  ~WAVSink() { Finalize(); }
  zx_status_t Initialize(const char* filename) {
    return WAVCommon::Initialize(filename, InitMode::SINK);
  }

  // AudioSink interface
  zx_status_t SetFormat(const AudioStream::Format& format) final;
  zx_status_t PutFrames(const void* buffer, uint32_t amt) final;
  zx_status_t Finalize() final;

 private:
  bool format_set_ = false;
  uint64_t bytes_written_ = 0;
};

#endif  // SRC_MEDIA_AUDIO_TOOLS_AUDIO_DRIVER_CTL_WAV_SINK_H_