media/codec2/components/g711/C2SoftG711Dec.cpp - LeafOS-Project/android_frameworks_av - Gitiles

 /*
  * Copyright (C) 2018 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.
  */

 //#define LOG_NDEBUG 0
 #define LOG_TAG "C2SoftG711Dec"
 #include <log/log.h>

 #include <media/stagefright/foundation/MediaDefs.h>

 #include <C2PlatformSupport.h>
 #include <SimpleC2Interface.h>
 #include <g711Dec.h>
 #include "C2SoftG711Dec.h"

 namespace android {

 namespace {

 #ifdef ALAW
 constexpr char COMPONENT_NAME[] = "c2.android.g711.alaw.decoder";
 #else
 constexpr char COMPONENT_NAME[] = "c2.android.g711.mlaw.decoder";
 #endif

 }  // namespace

 class C2SoftG711Dec::IntfImpl : public SimpleInterface<void>::BaseParams {
 public:
     explicit IntfImpl(const std::shared_ptr<C2ReflectorHelper> &helper)
         : SimpleInterface<void>::BaseParams(
                 helper,
                 COMPONENT_NAME,
                 C2Component::KIND_DECODER,
                 C2Component::DOMAIN_AUDIO,
 #ifdef ALAW
                 MEDIA_MIMETYPE_AUDIO_G711_ALAW
 #else
                 MEDIA_MIMETYPE_AUDIO_G711_MLAW
 #endif
                 ) {
         noPrivateBuffers();
         noInputReferences();
         noOutputReferences();
         noInputLatency();
         noTimeStretch();
         setDerivedInstance(this);

         addParameter(
                 DefineParam(mAttrib, C2_PARAMKEY_COMPONENT_ATTRIBUTES)
                 .withConstValue(new C2ComponentAttributesSetting(
                     C2Component::ATTRIB_IS_TEMPORAL))
                 .build());

         addParameter(
                 DefineParam(mSampleRate, C2_PARAMKEY_SAMPLE_RATE)
                 .withDefault(new C2StreamSampleRateInfo::output(0u, 8000))
                 .withFields({C2F(mSampleRate, value).inRange(8000, 48000)})
                 .withSetter((Setter<decltype(*mSampleRate)>::StrictValueWithNoDeps))
                 .build());

         addParameter(
                 DefineParam(mChannelCount, C2_PARAMKEY_CHANNEL_COUNT)
                 .withDefault(new C2StreamChannelCountInfo::output(0u, 1))
                 .withFields({C2F(mChannelCount, value).inRange(1, 6)})
                 .withSetter(Setter<decltype(*mChannelCount)>::StrictValueWithNoDeps)
                 .build());

         addParameter(
                 DefineParam(mBitrate, C2_PARAMKEY_BITRATE)
                 .withDefault(new C2StreamBitrateInfo::input(0u, 64000))
                 .withFields({C2F(mBitrate, value).equalTo(64000)})
                 .withSetter(Setter<decltype(*mBitrate)>::NonStrictValueWithNoDeps)
                 .build());

         addParameter(
                 DefineParam(mInputMaxBufSize, C2_PARAMKEY_INPUT_MAX_BUFFER_SIZE)
                 .withConstValue(new C2StreamMaxBufferSizeInfo::input(0u, 8192))
                 .build());
     }

 private:
     std::shared_ptr<C2StreamSampleRateInfo::output> mSampleRate;
     std::shared_ptr<C2StreamChannelCountInfo::output> mChannelCount;
     std::shared_ptr<C2StreamBitrateInfo::input> mBitrate;
     std::shared_ptr<C2StreamMaxBufferSizeInfo::input> mInputMaxBufSize;
 };

 C2SoftG711Dec::C2SoftG711Dec(
         const char *name,
         c2_node_id_t id,
         const std::shared_ptr<IntfImpl> &intfImpl)
     : SimpleC2Component(std::make_shared<SimpleInterface<IntfImpl>>(name, id, intfImpl)),
       mIntf(intfImpl) {
 }

 C2SoftG711Dec::~C2SoftG711Dec() {
     onRelease();
 }

 c2_status_t C2SoftG711Dec::onInit() {
     mSignalledOutputEos = false;
     return C2_OK;
 }

 c2_status_t C2SoftG711Dec::onStop() {
     mSignalledOutputEos = false;
     return C2_OK;
 }

 void C2SoftG711Dec::onReset() {
     (void)onStop();
 }

 void C2SoftG711Dec::onRelease() {
 }

 c2_status_t C2SoftG711Dec::onFlush_sm() {
     return onStop();
 }

 void C2SoftG711Dec::process(
         const std::unique_ptr<C2Work> &work,
         const std::shared_ptr<C2BlockPool> &pool) {
     // Initialize output work
     work->result = C2_OK;
     work->workletsProcessed = 1u;
     work->worklets.front()->output.flags = work->input.flags;

     if (mSignalledOutputEos) {
         work->result = C2_BAD_VALUE;
         return;
     }

     C2ReadView rView = mDummyReadView;
     size_t inOffset = 0u;
     size_t inSize = 0u;
     if (!work->input.buffers.empty()) {
         rView = work->input.buffers[0]->data().linearBlocks().front().map().get();
         inSize = rView.capacity();
         if (inSize && rView.error()) {
             ALOGE("read view map failed %d", rView.error());
             work->result = C2_CORRUPTED;
             return;
         }
     }
     bool eos = (work->input.flags & C2FrameData::FLAG_END_OF_STREAM) != 0;
     int outSize = inSize * sizeof(int16_t);

     ALOGV("in buffer attr. size %zu timestamp %d frameindex %d", inSize,
           (int)work->input.ordinal.timestamp.peeku(), (int)work->input.ordinal.frameIndex.peeku());

     if (inSize == 0) {
         work->worklets.front()->output.flags = work->input.flags;
         work->worklets.front()->output.buffers.clear();
         work->worklets.front()->output.ordinal = work->input.ordinal;
         if (eos) {
             mSignalledOutputEos = true;
             ALOGV("signalled EOS");
         }
         return;
     }

     uint8_t *inputptr = const_cast<uint8_t *>(rView.data() + inOffset);

     std::shared_ptr<C2LinearBlock> block;
     C2MemoryUsage usage = { C2MemoryUsage::CPU_READ, C2MemoryUsage::CPU_WRITE };
     c2_status_t err = pool->fetchLinearBlock(outSize, usage, &block);
     if (err != C2_OK) {
         ALOGE("fetchLinearBlock for Output failed with status %d", err);
         work->result = C2_NO_MEMORY;
         return;
     }
     C2WriteView wView = block->map().get();
     if (wView.error()) {
         ALOGE("write view map failed %d", wView.error());
         work->result = C2_CORRUPTED;
         return;
     }
     int16_t *outputptr = reinterpret_cast<int16_t *>(wView.data());

 #ifdef ALAW
     DecodeALaw(outputptr, inputptr, inSize);
 #else
     DecodeMLaw(outputptr, inputptr, inSize);
 #endif

     work->worklets.front()->output.flags = work->input.flags;
     work->worklets.front()->output.buffers.clear();
     work->worklets.front()->output.buffers.push_back(createLinearBuffer(block, 0, outSize));
     work->worklets.front()->output.ordinal = work->input.ordinal;

     if (eos) {
         mSignalledOutputEos = true;
         ALOGV("signalled EOS");
     }
 }

 c2_status_t C2SoftG711Dec::drain(
         uint32_t drainMode,
         const std::shared_ptr<C2BlockPool> &pool) {
     (void) pool;
     if (drainMode == NO_DRAIN) {
         ALOGW("drain with NO_DRAIN: no-op");
         return C2_OK;
     }
     if (drainMode == DRAIN_CHAIN) {
         ALOGW("DRAIN_CHAIN not supported");
         return C2_OMITTED;
     }

     return C2_OK;
 }

 class C2SoftG711DecFactory : public C2ComponentFactory {
 public:
     C2SoftG711DecFactory() : mHelper(std::static_pointer_cast<C2ReflectorHelper>(
             GetCodec2PlatformComponentStore()->getParamReflector())) {
     }

     virtual c2_status_t createComponent(
             c2_node_id_t id,
             std::shared_ptr<C2Component>* const component,
             std::function<void(C2Component*)> deleter) override {
         *component = std::shared_ptr<C2Component>(
                 new C2SoftG711Dec(COMPONENT_NAME, id,
                                std::make_shared<C2SoftG711Dec::IntfImpl>(mHelper)),
                 deleter);
         return C2_OK;
     }

     virtual c2_status_t createInterface(
             c2_node_id_t id, std::shared_ptr<C2ComponentInterface>* const interface,
             std::function<void(C2ComponentInterface*)> deleter) override {
         *interface = std::shared_ptr<C2ComponentInterface>(
                 new SimpleInterface<C2SoftG711Dec::IntfImpl>(
                         COMPONENT_NAME, id, std::make_shared<C2SoftG711Dec::IntfImpl>(mHelper)),
                 deleter);
         return C2_OK;
     }

     virtual ~C2SoftG711DecFactory() override = default;

 private:
     std::shared_ptr<C2ReflectorHelper> mHelper;
 };

 }  // namespace android

 __attribute__((cfi_canonical_jump_table))
 extern "C" ::C2ComponentFactory* CreateCodec2Factory() {
     ALOGV("in %s", __func__);
     return new ::android::C2SoftG711DecFactory();
 }

 __attribute__((cfi_canonical_jump_table))
 extern "C" void DestroyCodec2Factory(::C2ComponentFactory* factory) {
     ALOGV("in %s", __func__);
     delete factory;
 }
	/*
	* Copyright (C) 2018 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.
	*/

	//#define LOG_NDEBUG 0
	#define LOG_TAG "C2SoftG711Dec"
	#include <log/log.h>

	#include <media/stagefright/foundation/MediaDefs.h>

	#include <C2PlatformSupport.h>
	#include <SimpleC2Interface.h>
	#include <g711Dec.h>
	#include "C2SoftG711Dec.h"

	namespace android {

	namespace {

	#ifdef ALAW
	constexpr char COMPONENT_NAME[] = "c2.android.g711.alaw.decoder";
	#else
	constexpr char COMPONENT_NAME[] = "c2.android.g711.mlaw.decoder";
	#endif

	} // namespace

	class C2SoftG711Dec::IntfImpl : public SimpleInterface<void>::BaseParams {
	public:
	explicit IntfImpl(const std::shared_ptr<C2ReflectorHelper> &helper)
	: SimpleInterface<void>::BaseParams(
	helper,
	COMPONENT_NAME,
	C2Component::KIND_DECODER,
	C2Component::DOMAIN_AUDIO,
	#ifdef ALAW
	MEDIA_MIMETYPE_AUDIO_G711_ALAW
	#else
	MEDIA_MIMETYPE_AUDIO_G711_MLAW
	#endif
	) {
	noPrivateBuffers();
	noInputReferences();
	noOutputReferences();
	noInputLatency();
	noTimeStretch();
	setDerivedInstance(this);

	addParameter(
	DefineParam(mAttrib, C2_PARAMKEY_COMPONENT_ATTRIBUTES)
	.withConstValue(new C2ComponentAttributesSetting(
	C2Component::ATTRIB_IS_TEMPORAL))
	.build());

	addParameter(
	DefineParam(mSampleRate, C2_PARAMKEY_SAMPLE_RATE)
	.withDefault(new C2StreamSampleRateInfo::output(0u, 8000))
	.withFields({C2F(mSampleRate, value).inRange(8000, 48000)})
	.withSetter((Setter<decltype(*mSampleRate)>::StrictValueWithNoDeps))
	.build());

	addParameter(
	DefineParam(mChannelCount, C2_PARAMKEY_CHANNEL_COUNT)
	.withDefault(new C2StreamChannelCountInfo::output(0u, 1))
	.withFields({C2F(mChannelCount, value).inRange(1, 6)})
	.withSetter(Setter<decltype(*mChannelCount)>::StrictValueWithNoDeps)
	.build());

	addParameter(
	DefineParam(mBitrate, C2_PARAMKEY_BITRATE)
	.withDefault(new C2StreamBitrateInfo::input(0u, 64000))
	.withFields({C2F(mBitrate, value).equalTo(64000)})
	.withSetter(Setter<decltype(*mBitrate)>::NonStrictValueWithNoDeps)
	.build());

	addParameter(
	DefineParam(mInputMaxBufSize, C2_PARAMKEY_INPUT_MAX_BUFFER_SIZE)
	.withConstValue(new C2StreamMaxBufferSizeInfo::input(0u, 8192))
	.build());
	}

	private:
	std::shared_ptr<C2StreamSampleRateInfo::output> mSampleRate;
	std::shared_ptr<C2StreamChannelCountInfo::output> mChannelCount;
	std::shared_ptr<C2StreamBitrateInfo::input> mBitrate;
	std::shared_ptr<C2StreamMaxBufferSizeInfo::input> mInputMaxBufSize;
	};

	C2SoftG711Dec::C2SoftG711Dec(
	const char *name,
	c2_node_id_t id,
	const std::shared_ptr<IntfImpl> &intfImpl)
	: SimpleC2Component(std::make_shared<SimpleInterface<IntfImpl>>(name, id, intfImpl)),
	mIntf(intfImpl) {
	}

	C2SoftG711Dec::~C2SoftG711Dec() {
	onRelease();
	}

	c2_status_t C2SoftG711Dec::onInit() {
	mSignalledOutputEos = false;
	return C2_OK;
	}

	c2_status_t C2SoftG711Dec::onStop() {
	mSignalledOutputEos = false;
	return C2_OK;
	}

	void C2SoftG711Dec::onReset() {
	(void)onStop();
	}

	void C2SoftG711Dec::onRelease() {
	}

	c2_status_t C2SoftG711Dec::onFlush_sm() {
	return onStop();
	}

	void C2SoftG711Dec::process(
	const std::unique_ptr<C2Work> &work,
	const std::shared_ptr<C2BlockPool> &pool) {
	// Initialize output work
	work->result = C2_OK;
	work->workletsProcessed = 1u;
	work->worklets.front()->output.flags = work->input.flags;

	if (mSignalledOutputEos) {
	work->result = C2_BAD_VALUE;
	return;
	}

	C2ReadView rView = mDummyReadView;
	size_t inOffset = 0u;
	size_t inSize = 0u;
	if (!work->input.buffers.empty()) {
	rView = work->input.buffers[0]->data().linearBlocks().front().map().get();
	inSize = rView.capacity();
	if (inSize && rView.error()) {
	ALOGE("read view map failed %d", rView.error());
	work->result = C2_CORRUPTED;
	return;
	}
	}
	bool eos = (work->input.flags & C2FrameData::FLAG_END_OF_STREAM) != 0;
	int outSize = inSize * sizeof(int16_t);

	ALOGV("in buffer attr. size %zu timestamp %d frameindex %d", inSize,
	(int)work->input.ordinal.timestamp.peeku(), (int)work->input.ordinal.frameIndex.peeku());

	if (inSize == 0) {
	work->worklets.front()->output.flags = work->input.flags;
	work->worklets.front()->output.buffers.clear();
	work->worklets.front()->output.ordinal = work->input.ordinal;
	if (eos) {
	mSignalledOutputEos = true;
	ALOGV("signalled EOS");
	}
	return;
	}

	uint8_t inputptr = const_cast<uint8_t >(rView.data() + inOffset);

	std::shared_ptr<C2LinearBlock> block;
	C2MemoryUsage usage = { C2MemoryUsage::CPU_READ, C2MemoryUsage::CPU_WRITE };
	c2_status_t err = pool->fetchLinearBlock(outSize, usage, &block);
	if (err != C2_OK) {
	ALOGE("fetchLinearBlock for Output failed with status %d", err);
	work->result = C2_NO_MEMORY;
	return;
	}
	C2WriteView wView = block->map().get();
	if (wView.error()) {
	ALOGE("write view map failed %d", wView.error());
	work->result = C2_CORRUPTED;
	return;
	}
	int16_t outputptr = reinterpret_cast<int16_t >(wView.data());

	#ifdef ALAW
	DecodeALaw(outputptr, inputptr, inSize);
	#else
	DecodeMLaw(outputptr, inputptr, inSize);
	#endif

	work->worklets.front()->output.flags = work->input.flags;
	work->worklets.front()->output.buffers.clear();
	work->worklets.front()->output.buffers.push_back(createLinearBuffer(block, 0, outSize));
	work->worklets.front()->output.ordinal = work->input.ordinal;

	if (eos) {
	mSignalledOutputEos = true;
	ALOGV("signalled EOS");
	}
	}

	c2_status_t C2SoftG711Dec::drain(
	uint32_t drainMode,
	const std::shared_ptr<C2BlockPool> &pool) {
	(void) pool;
	if (drainMode == NO_DRAIN) {
	ALOGW("drain with NO_DRAIN: no-op");
	return C2_OK;
	}
	if (drainMode == DRAIN_CHAIN) {
	ALOGW("DRAIN_CHAIN not supported");
	return C2_OMITTED;
	}

	return C2_OK;
	}

	class C2SoftG711DecFactory : public C2ComponentFactory {
	public:
	C2SoftG711DecFactory() : mHelper(std::static_pointer_cast<C2ReflectorHelper>(
	GetCodec2PlatformComponentStore()->getParamReflector())) {
	}

	virtual c2_status_t createComponent(
	c2_node_id_t id,
	std::shared_ptr<C2Component>* const component,
	std::function<void(C2Component*)> deleter) override {
	*component = std::shared_ptr<C2Component>(
	new C2SoftG711Dec(COMPONENT_NAME, id,
	std::make_shared<C2SoftG711Dec::IntfImpl>(mHelper)),
	deleter);
	return C2_OK;
	}

	virtual c2_status_t createInterface(
	c2_node_id_t id, std::shared_ptr<C2ComponentInterface>* const interface,
	std::function<void(C2ComponentInterface*)> deleter) override {
	*interface = std::shared_ptr<C2ComponentInterface>(
	new SimpleInterface<C2SoftG711Dec::IntfImpl>(
	COMPONENT_NAME, id, std::make_shared<C2SoftG711Dec::IntfImpl>(mHelper)),
	deleter);
	return C2_OK;
	}

	virtual ~C2SoftG711DecFactory() override = default;

	private:
	std::shared_ptr<C2ReflectorHelper> mHelper;
	};

	} // namespace android

	__attribute__((cfi_canonical_jump_table))
	extern "C" ::C2ComponentFactory* CreateCodec2Factory() {
	ALOGV("in %s", __func__);
	return new ::android::C2SoftG711DecFactory();
	}

	__attribute__((cfi_canonical_jump_table))
	extern "C" void DestroyCodec2Factory(::C2ComponentFactory* factory) {
	ALOGV("in %s", __func__);
	delete factory;
	}