libctru/libctru/source/services/mvd.c

/*
  mvd.c - code for using this: http://3dbrew.org/wiki/MVD_Services
*/

#include <stdlib.h>
#include <string.h>
#include <3ds/types.h>
#include <3ds/svc.h>
#include <3ds/srv.h>
#include <3ds/os.h>
#include <3ds/linear.h>
#include <3ds/services/mvd.h>

Handle mvdstdHandle;
static u32 mvdstdInitialized = 0;
static mvdstdMode mvdstd_mode;
static mvdstdTypeInput mvdstd_input_type;
static mvdstdTypeOutput mvdstd_output_type;
static u32 *mvdstd_workbuf = NULL;
static size_t mvdstd_workbufsize = 0;

static Result mvdstdipc_Initialize(u32 *buf, u32 bufsize, Handle kprocess)
{
	u32* cmdbuf = getThreadCommandBuffer();
	cmdbuf[0] = 0x00010082; //request header code
	cmdbuf[1] = (u32)buf;
	cmdbuf[2] = bufsize;
	cmdbuf[3] = 0;
	cmdbuf[4] = kprocess;

	Result ret=0;
	if((ret=svcSendSyncRequest(mvdstdHandle)))return ret;

	return cmdbuf[1];
}

static Result mvdstdipc_Shutdown(void)
{
	u32* cmdbuf = getThreadCommandBuffer();
	cmdbuf[0] = 0x00020000; //request header code

	Result ret=0;
	if((ret = svcSendSyncRequest(mvdstdHandle)))return ret;

	return cmdbuf[1];
}

static Result mvdstdipc_cmd18(void)
{
	u32* cmdbuf = getThreadCommandBuffer();
	cmdbuf[0] = 0x00180000; //request header code

	Result ret=0;
	if((ret=svcSendSyncRequest(mvdstdHandle)))return ret;

	return cmdbuf[1];
}

static Result mvdstdipc_cmd19(void)
{
	u32* cmdbuf = getThreadCommandBuffer();
	cmdbuf[0] = 0x00190000; //request header code

	Result ret=0;
	if((ret=svcSendSyncRequest(mvdstdHandle)))return ret;

	return cmdbuf[1];
}

static Result mvdstdipc_cmd1a(void)
{
	u32* cmdbuf = getThreadCommandBuffer();
	cmdbuf[0] = 0x001A0000; //request header code

	Result ret=0;
	if((ret=svcSendSyncRequest(mvdstdHandle)))return ret;

	return cmdbuf[1];
}

Result mvdstdSetConfig(mvdstdConfig *config)
{
	u32* cmdbuf = getThreadCommandBuffer();
	cmdbuf[0] = 0x001E0044; //request header code
	cmdbuf[1] = sizeof(mvdstdConfig);
	cmdbuf[2] = 0;
	cmdbuf[3] = 0xffff8001;
	cmdbuf[4] = (sizeof(mvdstdConfig)<<4) | 10;
	cmdbuf[5] = (u32)config;

	Result ret=0;
	if((ret=svcSendSyncRequest(mvdstdHandle)))return ret;

	return cmdbuf[1];
}

void mvdstdGenerateDefaultConfig(mvdstdConfig *config, u32 input_width, u32 input_height, u32 output_width, u32 output_height, u32 *vaddr_colorconv_indata, u32 *vaddr_outdata0, u32 *vaddr_outdata1_colorconv)
{
	memset(config, 0, sizeof(mvdstdConfig));

	config->input_type = mvdstd_input_type;

	config->inwidth = input_width;
	config->inheight = input_height;

	if(mvdstd_mode==MVDMODE_COLORFORMATCONV)config->physaddr_colorconv_indata = osConvertVirtToPhys((u32)vaddr_colorconv_indata);

	if(mvdstd_mode==MVDMODE_VIDEOPROCESSING)
	{
		config->flag_x40 = 1;
		config->outheight0 = output_height;
		config->outwidth0 = output_width;
	}

	config->output_type = mvdstd_output_type;

	config->outwidth1 = output_width;
	config->outheight1 = output_height;

	config->physaddr_outdata0 = osConvertVirtToPhys((u32)vaddr_outdata0);
	if(mvdstd_mode==MVDMODE_COLORFORMATCONV)config->physaddr_outdata1_colorconv = osConvertVirtToPhys((u32)vaddr_outdata1_colorconv);

	config->unk_x6c[0] = 0x1;
	config->unk_x6c[(0x84-0x6c)>>2] = 0x12a;
	config->unk_x6c[(0x88-0x6c)>>2] = 0x199;
	config->unk_x6c[(0x8c-0x6c)>>2] = 0xd0;
	config->unk_x6c[(0x90-0x6c)>>2] = 0x64;
	config->unk_x6c[(0x94-0x6c)>>2] = 0x204;
	config->unk_x6c[(0xa8-0x6c)>>2] = 0x1;
	config->unk_x6c[(0x104-0x6c)>>2] = 0x1;
	config->unk_x6c[(0x110-0x6c)>>2] = 0x200;
	config->unk_x6c[(0x114-0x6c)>>2] = 0x100;
}

Result mvdstdInit(mvdstdMode mode, mvdstdTypeInput input_type, mvdstdTypeOutput output_type, u32 size)
{
	Result ret=0;

	if(mvdstdInitialized)return 0;

	mvdstd_workbufsize = size;
	mvdstd_mode = mode;
	mvdstd_input_type = input_type;
	mvdstd_output_type = output_type;

	if(mvdstd_mode==MVDMODE_COLORFORMATCONV)mvdstd_workbufsize = 1;
	if(mvdstd_mode!=MVDMODE_COLORFORMATCONV)return -2;//Video processing / H.264 isn't supported atm.

	if((ret=srvGetServiceHandle(&mvdstdHandle, "mvd:STD")))return ret;

	mvdstd_workbuf = linearAlloc(mvdstd_workbufsize);
	if(mvdstd_workbuf==NULL)return -1;

	ret = mvdstdipc_Initialize((u32*)osConvertOldLINEARMemToNew((u32)mvdstd_workbuf), mvdstd_workbufsize, 0xffff8001);
	if(ret<0)
	{
		svcCloseHandle(mvdstdHandle);
		linearFree(mvdstd_workbuf);
		return ret;
	}

	ret = mvdstdipc_cmd18();
	if(ret<0)
	{
		mvdstdipc_Shutdown();
		svcCloseHandle(mvdstdHandle);
		linearFree(mvdstd_workbuf);
		return ret;
	}

	mvdstdInitialized = 1;

	return 0;
}

Result mvdstdShutdown(void)
{
	if(!mvdstdInitialized)return 0;

	if(mvdstd_mode==MVDMODE_COLORFORMATCONV)
	{
		mvdstdipc_cmd19();
	}

	mvdstdipc_Shutdown();

	svcCloseHandle(mvdstdHandle);

	linearFree(mvdstd_workbuf);

	return 0;
}

Result mvdstdProcessFrame(mvdstdConfig *config, u32 *h264_vaddr_inframe, u32 h264_inframesize, u32 h264_frameid)
{
	Result ret;

	if(!mvdstdInitialized)return 0;
	if(config==NULL)return -1;
	if(mvdstd_mode!=MVDMODE_COLORFORMATCONV)return -2;

	ret = mvdstdSetConfig(config);
	if(ret<0)return ret;

	return mvdstdipc_cmd1a();
}