media/libstagefright/OMXClient.cpp - platform/frameworks/av - Git at Google

 /*
  * Copyright (C) 2009 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 "OMXClient"
 #include <utils/Log.h>

 #include <binder/IServiceManager.h>
 #include <media/IMediaPlayerService.h>
 #include <media/stagefright/foundation/ADebug.h>
 #include <media/stagefright/OMXClient.h>
 #include <utils/KeyedVector.h>

 #include "include/OMX.h"

 namespace android {

 struct MuxOMX : public IOMX {
     MuxOMX(const sp<IOMX> &remoteOMX);
     virtual ~MuxOMX();

     virtual IBinder *onAsBinder() { return NULL; }

     virtual bool livesLocally(node_id node, pid_t pid);

     virtual status_t listNodes(List<ComponentInfo> *list);

     virtual status_t allocateNode(
             const char *name, const sp<IOMXObserver> &observer,
             node_id *node);

     virtual status_t freeNode(node_id node);

     virtual status_t sendCommand(
             node_id node, OMX_COMMANDTYPE cmd, OMX_S32 param);

     virtual status_t getParameter(
             node_id node, OMX_INDEXTYPE index,
             void *params, size_t size);

     virtual status_t setParameter(
             node_id node, OMX_INDEXTYPE index,
             const void *params, size_t size);

     virtual status_t getConfig(
             node_id node, OMX_INDEXTYPE index,
             void *params, size_t size);

     virtual status_t setConfig(
             node_id node, OMX_INDEXTYPE index,
             const void *params, size_t size);

     virtual status_t getState(
             node_id node, OMX_STATETYPE* state);

     virtual status_t storeMetaDataInBuffers(
             node_id node, OMX_U32 port_index, OMX_BOOL enable);

     virtual status_t enableGraphicBuffers(
             node_id node, OMX_U32 port_index, OMX_BOOL enable);

     virtual status_t getGraphicBufferUsage(
             node_id node, OMX_U32 port_index, OMX_U32* usage);

     virtual status_t useBuffer(
             node_id node, OMX_U32 port_index, const sp<IMemory> &params,
             buffer_id *buffer);

     virtual status_t useGraphicBuffer(
             node_id node, OMX_U32 port_index,
             const sp<GraphicBuffer> &graphicBuffer, buffer_id *buffer);

     virtual status_t allocateBuffer(
             node_id node, OMX_U32 port_index, size_t size,
             buffer_id *buffer, void **buffer_data);

     virtual status_t allocateBufferWithBackup(
             node_id node, OMX_U32 port_index, const sp<IMemory> &params,
             buffer_id *buffer);

     virtual status_t freeBuffer(
             node_id node, OMX_U32 port_index, buffer_id buffer);

     virtual status_t fillBuffer(node_id node, buffer_id buffer);

     virtual status_t emptyBuffer(
             node_id node,
             buffer_id buffer,
             OMX_U32 range_offset, OMX_U32 range_length,
             OMX_U32 flags, OMX_TICKS timestamp);

     virtual status_t getExtensionIndex(
             node_id node,
             const char *parameter_name,
             OMX_INDEXTYPE *index);

 private:
     mutable Mutex mLock;

     sp<IOMX> mRemoteOMX;
     sp<IOMX> mLocalOMX;

     KeyedVector<node_id, bool> mIsLocalNode;

     bool isLocalNode(node_id node) const;
     bool isLocalNode_l(node_id node) const;
     const sp<IOMX> &getOMX(node_id node) const;
     const sp<IOMX> &getOMX_l(node_id node) const;

     static bool IsSoftwareComponent(const char *name);

     DISALLOW_EVIL_CONSTRUCTORS(MuxOMX);
 };

 MuxOMX::MuxOMX(const sp<IOMX> &remoteOMX)
     : mRemoteOMX(remoteOMX) {
 }

 MuxOMX::~MuxOMX() {
 }

 bool MuxOMX::isLocalNode(node_id node) const {
     Mutex::Autolock autoLock(mLock);

     return isLocalNode_l(node);
 }

 bool MuxOMX::isLocalNode_l(node_id node) const {
     return mIsLocalNode.indexOfKey(node) >= 0;
 }

 // static
 bool MuxOMX::IsSoftwareComponent(const char *name) {
     return !strncasecmp(name, "OMX.google.", 11);
 }

 const sp<IOMX> &MuxOMX::getOMX(node_id node) const {
     return isLocalNode(node) ? mLocalOMX : mRemoteOMX;
 }

 const sp<IOMX> &MuxOMX::getOMX_l(node_id node) const {
     return isLocalNode_l(node) ? mLocalOMX : mRemoteOMX;
 }

 bool MuxOMX::livesLocally(node_id node, pid_t pid) {
     return getOMX(node)->livesLocally(node, pid);
 }

 status_t MuxOMX::listNodes(List<ComponentInfo> *list) {
     Mutex::Autolock autoLock(mLock);

     if (mLocalOMX == NULL) {
         mLocalOMX = new OMX;
     }

     return mLocalOMX->listNodes(list);
 }

 status_t MuxOMX::allocateNode(
         const char *name, const sp<IOMXObserver> &observer,
         node_id *node) {
     Mutex::Autolock autoLock(mLock);

     sp<IOMX> omx;

     if (IsSoftwareComponent(name)) {
         if (mLocalOMX == NULL) {
             mLocalOMX = new OMX;
         }
         omx = mLocalOMX;
     } else {
         omx = mRemoteOMX;
     }

     status_t err = omx->allocateNode(name, observer, node);

     if (err != OK) {
         return err;
     }

     if (omx == mLocalOMX) {
         mIsLocalNode.add(*node, true);
     }

     return OK;
 }

 status_t MuxOMX::freeNode(node_id node) {
     Mutex::Autolock autoLock(mLock);

     status_t err = getOMX_l(node)->freeNode(node);

     if (err != OK) {
         return err;
     }

     mIsLocalNode.removeItem(node);

     return OK;
 }

 status_t MuxOMX::sendCommand(
         node_id node, OMX_COMMANDTYPE cmd, OMX_S32 param) {
     return getOMX(node)->sendCommand(node, cmd, param);
 }

 status_t MuxOMX::getParameter(
         node_id node, OMX_INDEXTYPE index,
         void *params, size_t size) {
     return getOMX(node)->getParameter(node, index, params, size);
 }

 status_t MuxOMX::setParameter(
         node_id node, OMX_INDEXTYPE index,
         const void *params, size_t size) {
     return getOMX(node)->setParameter(node, index, params, size);
 }

 status_t MuxOMX::getConfig(
         node_id node, OMX_INDEXTYPE index,
         void *params, size_t size) {
     return getOMX(node)->getConfig(node, index, params, size);
 }

 status_t MuxOMX::setConfig(
         node_id node, OMX_INDEXTYPE index,
         const void *params, size_t size) {
     return getOMX(node)->setConfig(node, index, params, size);
 }

 status_t MuxOMX::getState(
         node_id node, OMX_STATETYPE* state) {
     return getOMX(node)->getState(node, state);
 }

 status_t MuxOMX::storeMetaDataInBuffers(
         node_id node, OMX_U32 port_index, OMX_BOOL enable) {
     return getOMX(node)->storeMetaDataInBuffers(node, port_index, enable);
 }

 status_t MuxOMX::enableGraphicBuffers(
         node_id node, OMX_U32 port_index, OMX_BOOL enable) {
     return getOMX(node)->enableGraphicBuffers(node, port_index, enable);
 }

 status_t MuxOMX::getGraphicBufferUsage(
         node_id node, OMX_U32 port_index, OMX_U32* usage) {
     return getOMX(node)->getGraphicBufferUsage(node, port_index, usage);
 }

 status_t MuxOMX::useBuffer(
         node_id node, OMX_U32 port_index, const sp<IMemory> &params,
         buffer_id *buffer) {
     return getOMX(node)->useBuffer(node, port_index, params, buffer);
 }

 status_t MuxOMX::useGraphicBuffer(
         node_id node, OMX_U32 port_index,
         const sp<GraphicBuffer> &graphicBuffer, buffer_id *buffer) {
     return getOMX(node)->useGraphicBuffer(
             node, port_index, graphicBuffer, buffer);
 }

 status_t MuxOMX::allocateBuffer(
         node_id node, OMX_U32 port_index, size_t size,
         buffer_id *buffer, void **buffer_data) {
     return getOMX(node)->allocateBuffer(
             node, port_index, size, buffer, buffer_data);
 }

 status_t MuxOMX::allocateBufferWithBackup(
         node_id node, OMX_U32 port_index, const sp<IMemory> &params,
         buffer_id *buffer) {
     return getOMX(node)->allocateBufferWithBackup(
             node, port_index, params, buffer);
 }

 status_t MuxOMX::freeBuffer(
         node_id node, OMX_U32 port_index, buffer_id buffer) {
     return getOMX(node)->freeBuffer(node, port_index, buffer);
 }

 status_t MuxOMX::fillBuffer(node_id node, buffer_id buffer) {
     return getOMX(node)->fillBuffer(node, buffer);
 }

 status_t MuxOMX::emptyBuffer(
         node_id node,
         buffer_id buffer,
         OMX_U32 range_offset, OMX_U32 range_length,
         OMX_U32 flags, OMX_TICKS timestamp) {
     return getOMX(node)->emptyBuffer(
             node, buffer, range_offset, range_length, flags, timestamp);
 }

 status_t MuxOMX::getExtensionIndex(
         node_id node,
         const char *parameter_name,
         OMX_INDEXTYPE *index) {
     return getOMX(node)->getExtensionIndex(node, parameter_name, index);
 }

 OMXClient::OMXClient() {
 }

 status_t OMXClient::connect() {
     sp<IServiceManager> sm = defaultServiceManager();
     sp<IBinder> binder = sm->getService(String16("media.player"));
     sp<IMediaPlayerService> service = interface_cast<IMediaPlayerService>(binder);

     CHECK(service.get() != NULL);

     mOMX = service->getOMX();
     CHECK(mOMX.get() != NULL);

     if (!mOMX->livesLocally(NULL /* node */, getpid())) {
         ALOGI("Using client-side OMX mux.");
         mOMX = new MuxOMX(mOMX);
     }

     return OK;
 }

 void OMXClient::disconnect() {
     if (mOMX.get() != NULL) {
         mOMX.clear();
         mOMX = NULL;
     }
 }

 }  // namespace android
	/*
	* Copyright (C) 2009 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 "OMXClient"
	#include <utils/Log.h>

	#include <binder/IServiceManager.h>
	#include <media/IMediaPlayerService.h>
	#include <media/stagefright/foundation/ADebug.h>
	#include <media/stagefright/OMXClient.h>
	#include <utils/KeyedVector.h>

	#include "include/OMX.h"

	namespace android {

	struct MuxOMX : public IOMX {
	MuxOMX(const sp<IOMX> &remoteOMX);
	virtual ~MuxOMX();

	virtual IBinder *onAsBinder() { return NULL; }

	virtual bool livesLocally(node_id node, pid_t pid);

	virtual status_t listNodes(List<ComponentInfo> *list);

	virtual status_t allocateNode(
	const char *name, const sp<IOMXObserver> &observer,
	node_id *node);

	virtual status_t freeNode(node_id node);

	virtual status_t sendCommand(
	node_id node, OMX_COMMANDTYPE cmd, OMX_S32 param);

	virtual status_t getParameter(
	node_id node, OMX_INDEXTYPE index,
	void *params, size_t size);

	virtual status_t setParameter(
	node_id node, OMX_INDEXTYPE index,
	const void *params, size_t size);

	virtual status_t getConfig(
	node_id node, OMX_INDEXTYPE index,
	void *params, size_t size);

	virtual status_t setConfig(
	node_id node, OMX_INDEXTYPE index,
	const void *params, size_t size);

	virtual status_t getState(
	node_id node, OMX_STATETYPE* state);

	virtual status_t storeMetaDataInBuffers(
	node_id node, OMX_U32 port_index, OMX_BOOL enable);

	virtual status_t enableGraphicBuffers(
	node_id node, OMX_U32 port_index, OMX_BOOL enable);

	virtual status_t getGraphicBufferUsage(
	node_id node, OMX_U32 port_index, OMX_U32* usage);

	virtual status_t useBuffer(
	node_id node, OMX_U32 port_index, const sp<IMemory> &params,
	buffer_id *buffer);

	virtual status_t useGraphicBuffer(
	node_id node, OMX_U32 port_index,
	const sp<GraphicBuffer> &graphicBuffer, buffer_id *buffer);

	virtual status_t allocateBuffer(
	node_id node, OMX_U32 port_index, size_t size,
	buffer_id buffer, void *buffer_data);

	virtual status_t allocateBufferWithBackup(
	node_id node, OMX_U32 port_index, const sp<IMemory> &params,
	buffer_id *buffer);

	virtual status_t freeBuffer(
	node_id node, OMX_U32 port_index, buffer_id buffer);

	virtual status_t fillBuffer(node_id node, buffer_id buffer);

	virtual status_t emptyBuffer(
	node_id node,
	buffer_id buffer,
	OMX_U32 range_offset, OMX_U32 range_length,
	OMX_U32 flags, OMX_TICKS timestamp);

	virtual status_t getExtensionIndex(
	node_id node,
	const char *parameter_name,
	OMX_INDEXTYPE *index);

	private:
	mutable Mutex mLock;

	sp<IOMX> mRemoteOMX;
	sp<IOMX> mLocalOMX;

	KeyedVector<node_id, bool> mIsLocalNode;

	bool isLocalNode(node_id node) const;
	bool isLocalNode_l(node_id node) const;
	const sp<IOMX> &getOMX(node_id node) const;
	const sp<IOMX> &getOMX_l(node_id node) const;

	static bool IsSoftwareComponent(const char *name);

	DISALLOW_EVIL_CONSTRUCTORS(MuxOMX);
	};

	MuxOMX::MuxOMX(const sp<IOMX> &remoteOMX)
	: mRemoteOMX(remoteOMX) {
	}

	MuxOMX::~MuxOMX() {
	}

	bool MuxOMX::isLocalNode(node_id node) const {
	Mutex::Autolock autoLock(mLock);

	return isLocalNode_l(node);
	}

	bool MuxOMX::isLocalNode_l(node_id node) const {
	return mIsLocalNode.indexOfKey(node) >= 0;
	}

	// static
	bool MuxOMX::IsSoftwareComponent(const char *name) {
	return !strncasecmp(name, "OMX.google.", 11);
	}

	const sp<IOMX> &MuxOMX::getOMX(node_id node) const {
	return isLocalNode(node) ? mLocalOMX : mRemoteOMX;
	}

	const sp<IOMX> &MuxOMX::getOMX_l(node_id node) const {
	return isLocalNode_l(node) ? mLocalOMX : mRemoteOMX;
	}

	bool MuxOMX::livesLocally(node_id node, pid_t pid) {
	return getOMX(node)->livesLocally(node, pid);
	}

	status_t MuxOMX::listNodes(List<ComponentInfo> *list) {
	Mutex::Autolock autoLock(mLock);

	if (mLocalOMX == NULL) {
	mLocalOMX = new OMX;
	}

	return mLocalOMX->listNodes(list);
	}

	status_t MuxOMX::allocateNode(
	const char *name, const sp<IOMXObserver> &observer,
	node_id *node) {
	Mutex::Autolock autoLock(mLock);

	sp<IOMX> omx;

	if (IsSoftwareComponent(name)) {
	if (mLocalOMX == NULL) {
	mLocalOMX = new OMX;
	}
	omx = mLocalOMX;
	} else {
	omx = mRemoteOMX;
	}

	status_t err = omx->allocateNode(name, observer, node);

	if (err != OK) {
	return err;
	}

	if (omx == mLocalOMX) {
	mIsLocalNode.add(*node, true);
	}

	return OK;
	}

	status_t MuxOMX::freeNode(node_id node) {
	Mutex::Autolock autoLock(mLock);

	status_t err = getOMX_l(node)->freeNode(node);

	if (err != OK) {
	return err;
	}

	mIsLocalNode.removeItem(node);

	return OK;
	}

	status_t MuxOMX::sendCommand(
	node_id node, OMX_COMMANDTYPE cmd, OMX_S32 param) {
	return getOMX(node)->sendCommand(node, cmd, param);
	}

	status_t MuxOMX::getParameter(
	node_id node, OMX_INDEXTYPE index,
	void *params, size_t size) {
	return getOMX(node)->getParameter(node, index, params, size);
	}

	status_t MuxOMX::setParameter(
	node_id node, OMX_INDEXTYPE index,
	const void *params, size_t size) {
	return getOMX(node)->setParameter(node, index, params, size);
	}

	status_t MuxOMX::getConfig(
	node_id node, OMX_INDEXTYPE index,
	void *params, size_t size) {
	return getOMX(node)->getConfig(node, index, params, size);
	}

	status_t MuxOMX::setConfig(
	node_id node, OMX_INDEXTYPE index,
	const void *params, size_t size) {
	return getOMX(node)->setConfig(node, index, params, size);
	}

	status_t MuxOMX::getState(
	node_id node, OMX_STATETYPE* state) {
	return getOMX(node)->getState(node, state);
	}

	status_t MuxOMX::storeMetaDataInBuffers(
	node_id node, OMX_U32 port_index, OMX_BOOL enable) {
	return getOMX(node)->storeMetaDataInBuffers(node, port_index, enable);
	}

	status_t MuxOMX::enableGraphicBuffers(
	node_id node, OMX_U32 port_index, OMX_BOOL enable) {
	return getOMX(node)->enableGraphicBuffers(node, port_index, enable);
	}

	status_t MuxOMX::getGraphicBufferUsage(
	node_id node, OMX_U32 port_index, OMX_U32* usage) {
	return getOMX(node)->getGraphicBufferUsage(node, port_index, usage);
	}

	status_t MuxOMX::useBuffer(
	node_id node, OMX_U32 port_index, const sp<IMemory> &params,
	buffer_id *buffer) {
	return getOMX(node)->useBuffer(node, port_index, params, buffer);
	}

	status_t MuxOMX::useGraphicBuffer(
	node_id node, OMX_U32 port_index,
	const sp<GraphicBuffer> &graphicBuffer, buffer_id *buffer) {
	return getOMX(node)->useGraphicBuffer(
	node, port_index, graphicBuffer, buffer);
	}

	status_t MuxOMX::allocateBuffer(
	node_id node, OMX_U32 port_index, size_t size,
	buffer_id buffer, void *buffer_data) {
	return getOMX(node)->allocateBuffer(
	node, port_index, size, buffer, buffer_data);
	}

	status_t MuxOMX::allocateBufferWithBackup(
	node_id node, OMX_U32 port_index, const sp<IMemory> &params,
	buffer_id *buffer) {
	return getOMX(node)->allocateBufferWithBackup(
	node, port_index, params, buffer);
	}

	status_t MuxOMX::freeBuffer(
	node_id node, OMX_U32 port_index, buffer_id buffer) {
	return getOMX(node)->freeBuffer(node, port_index, buffer);
	}

	status_t MuxOMX::fillBuffer(node_id node, buffer_id buffer) {
	return getOMX(node)->fillBuffer(node, buffer);
	}

	status_t MuxOMX::emptyBuffer(
	node_id node,
	buffer_id buffer,
	OMX_U32 range_offset, OMX_U32 range_length,
	OMX_U32 flags, OMX_TICKS timestamp) {
	return getOMX(node)->emptyBuffer(
	node, buffer, range_offset, range_length, flags, timestamp);
	}

	status_t MuxOMX::getExtensionIndex(
	node_id node,
	const char *parameter_name,
	OMX_INDEXTYPE *index) {
	return getOMX(node)->getExtensionIndex(node, parameter_name, index);
	}

	OMXClient::OMXClient() {
	}

	status_t OMXClient::connect() {
	sp<IServiceManager> sm = defaultServiceManager();
	sp<IBinder> binder = sm->getService(String16("media.player"));
	sp<IMediaPlayerService> service = interface_cast<IMediaPlayerService>(binder);

	CHECK(service.get() != NULL);

	mOMX = service->getOMX();
	CHECK(mOMX.get() != NULL);

	if (!mOMX->livesLocally(NULL /* node */, getpid())) {
	ALOGI("Using client-side OMX mux.");
	mOMX = new MuxOMX(mOMX);
	}

	return OK;
	}

	void OMXClient::disconnect() {
	if (mOMX.get() != NULL) {
	mOMX.clear();
	mOMX = NULL;
	}
	}

	} // namespace android