You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

76 lines
2.1 KiB

/*
* Copyright 2019 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.
*/
#ifndef FRAME_REASSEMBLER_H_
#define FRAME_REASSEMBLER_H_
#include <set>
#include <memory>
#include <media/MediaCodecBuffer.h>
#include <C2Config.h>
#include <C2Work.h>
namespace android {
class FrameReassembler {
public:
FrameReassembler();
void init(
const std::shared_ptr<C2BlockPool> &pool,
C2MemoryUsage usage,
uint32_t frameSize,
uint32_t sampleRate,
uint32_t channelCount,
C2Config::pcm_encoding_t encoding);
void updateFrameSize(uint32_t frameSize);
void updateSampleRate(uint32_t sampleRate);
void updateChannelCount(uint32_t channelCount);
void updatePcmEncoding(C2Config::pcm_encoding_t encoding);
void reset();
void flush();
explicit operator bool() const;
c2_status_t process(
const sp<MediaCodecBuffer> &buffer,
std::list<std::unique_ptr<C2Work>> *items);
private:
std::shared_ptr<C2BlockPool> mBlockPool;
C2MemoryUsage mUsage;
std::optional<uint32_t> mFrameSize;
uint32_t mSampleRate;
uint32_t mChannelCount;
C2Config::pcm_encoding_t mEncoding;
std::list<std::unique_ptr<C2Work>> mPendingWork;
C2WorkOrdinalStruct mCurrentOrdinal;
std::shared_ptr<C2LinearBlock> mCurrentBlock;
std::optional<C2WriteView> mWriteView;
uint64_t bytesToSamples(size_t numBytes) const;
size_t usToSamples(uint64_t us) const;
uint32_t bytesPerSample() const;
void finishCurrentBlock(std::list<std::unique_ptr<C2Work>> *items);
};
} // namespace android
#endif // FRAME_REASSEMBLER_H_