/*
 * Copyright (C) 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#pragma once

#include <aidl/android/hardware/common/Ashmem.h>

#include <media/audiohal/EffectBufferHalInterface.h>
#include <system/audio_effect.h>

namespace android {
namespace effect {

class EffectBufferHalAidl : public EffectBufferHalInterface {
  public:
    static status_t allocate(size_t size, sp<EffectBufferHalInterface>* buffer);
    static status_t mirror(void* external, size_t size, sp<EffectBufferHalInterface>* buffer);

    audio_buffer_t* audioBuffer() override;
    void* externalData() const override;

    size_t getSize() const override { return mBufferSize; }

    void setExternalData(void* external) override;
    void setFrameCount(size_t frameCount) override;
    bool checkFrameCountChange() override;

    void update() override;
    void commit() override;
    void update(size_t size) override;
    void commit(size_t size) override;

  private:
    friend class EffectBufferHalInterface;

    // buffer size in bytes
    const size_t mBufferSize;
    bool mFrameCountChanged;
    void* mExternalData;
    aidl::android::hardware::common::Ashmem mMemory;
    audio_buffer_t mAudioBuffer;

    // Can not be constructed directly by clients.
    explicit EffectBufferHalAidl(size_t size);

    ~EffectBufferHalAidl();
    void copy(void* dst, const void* src, size_t n) const;
    status_t init();
};

} // namespace effect
} // namespace android