/*
    V4L2 API codec ioctl tests.

    Copyright (C) 2012  Hans Verkuil <hverkuil-cisco@xs4all.nl>

    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 2 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program; if not, write to the Free Software
    Foundation, Inc., 51 Franklin Street, Suite 500, Boston, MA  02110-1335  USA
 */

#include <map>
#include <set>

#include <sys/types.h>

#include "v4l2-compliance.h"

int testEncoder(struct node *node)
{
	struct v4l2_encoder_cmd cmd;
	bool is_encoder = node->codec_mask & (STATEFUL_ENCODER | JPEG_ENCODER);
	int ret;

	fail_on_test((node->codec_mask & STATELESS_ENCODER) && !node->has_media);
	if (IS_ENCODER(node) && node->has_media)
		fail_on_test(node->function != MEDIA_ENT_F_PROC_VIDEO_ENCODER);
	memset(&cmd, 0xff, sizeof(cmd));
	memset(&cmd.raw, 0, sizeof(cmd.raw));
	ret = doioctl(node, VIDIOC_ENCODER_CMD, &cmd);
	if (ret == ENOTTY) {
		fail_on_test(is_encoder);
		fail_on_test(doioctl(node, VIDIOC_TRY_ENCODER_CMD, &cmd) != ENOTTY);
		return ret;
	}
	fail_on_test(IS_DECODER(node));
	fail_on_test(ret != EINVAL);
	ret = doioctl(node, VIDIOC_TRY_ENCODER_CMD, &cmd);
	fail_on_test(ret == ENOTTY);
	fail_on_test(ret != EINVAL);

	cmd.cmd = V4L2_ENC_CMD_STOP;
	cmd.flags = ~0U;
	ret = doioctl(node, VIDIOC_TRY_ENCODER_CMD, &cmd);
	fail_on_test(ret != 0);
	fail_on_test(cmd.flags & ~V4L2_ENC_CMD_STOP_AT_GOP_END);
	fail_on_test(is_encoder && cmd.flags);
	cmd.flags = 0;
	ret = doioctl(node, VIDIOC_TRY_ENCODER_CMD, &cmd);
	fail_on_test(ret != 0);
	ret = doioctl(node, VIDIOC_ENCODER_CMD, &cmd);
	fail_on_test(ret != 0);

	cmd.cmd = V4L2_ENC_CMD_START;
	cmd.flags = ~0U;
	ret = doioctl(node, VIDIOC_TRY_ENCODER_CMD, &cmd);
	fail_on_test(ret != 0);
	fail_on_test(cmd.flags);

	cmd.cmd = V4L2_ENC_CMD_PAUSE;
	ret = doioctl(node, VIDIOC_ENCODER_CMD, &cmd);
	fail_on_test(ret != EPERM && ret != EINVAL);
	fail_on_test(is_encoder && ret != EINVAL);

	cmd.cmd = V4L2_ENC_CMD_RESUME;
	ret = doioctl(node, VIDIOC_ENCODER_CMD, &cmd);
	fail_on_test(ret != EPERM && ret != EINVAL);
	fail_on_test(is_encoder && ret != EINVAL);
	return 0;
}

int testEncIndex(struct node *node)
{
	struct v4l2_enc_idx idx;
	int ret;

	memset(&idx, 0xff, sizeof(idx));
	ret = doioctl(node, VIDIOC_G_ENC_INDEX, &idx);
	if (ret == ENOTTY)
		return ret;
	if (check_0(idx.reserved, sizeof(idx.reserved)))
		return fail("idx.reserved not zeroed\n");
	fail_on_test(ret);
	fail_on_test(idx.entries != 0);
	fail_on_test(idx.entries_cap == 0);
	return 0;
}

int testDecoder(struct node *node)
{
	struct v4l2_decoder_cmd cmd;
	bool is_decoder = node->codec_mask & (STATEFUL_DECODER | JPEG_DECODER);
	bool is_stateless = node->codec_mask & STATELESS_DECODER;
	int ret;

	fail_on_test((node->codec_mask & STATELESS_DECODER) && !node->has_media);
	if (IS_DECODER(node) && node->has_media)
		fail_on_test(node->function != MEDIA_ENT_F_PROC_VIDEO_DECODER);
	memset(&cmd, 0xff, sizeof(cmd));
	memset(&cmd.raw, 0, sizeof(cmd.raw));
	ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
	if (ret == ENOTTY) {
		fail_on_test(is_decoder);
		fail_on_test(doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd) != ENOTTY);
		return ret;
	}
	fail_on_test(IS_ENCODER(node));
	fail_on_test(ret != EINVAL);
	ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
	fail_on_test(ret == ENOTTY);
	fail_on_test(ret != EINVAL);

	if (is_stateless) {
		cmd.cmd = V4L2_DEC_CMD_FLUSH;
		cmd.flags = 0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(ret);
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(ret);

		cmd.cmd = V4L2_DEC_CMD_STOP;
		cmd.flags = 0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(!ret);
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(!ret);

		cmd.cmd = V4L2_DEC_CMD_START;
		cmd.flags = 0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(!ret);
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(!ret);

		cmd.cmd = V4L2_DEC_CMD_PAUSE;
		cmd.flags = 0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(!ret);
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(!ret);

		cmd.cmd = V4L2_DEC_CMD_RESUME;
		cmd.flags = 0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(!ret);
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(!ret);
	} else {
		cmd.cmd = V4L2_DEC_CMD_STOP;
		cmd.flags = ~0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(ret);
		fail_on_test(cmd.flags & ~(V4L2_DEC_CMD_STOP_TO_BLACK | V4L2_DEC_CMD_STOP_IMMEDIATELY));
		fail_on_test(is_decoder && cmd.flags);
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(ret);

		cmd.cmd = V4L2_DEC_CMD_START;
		cmd.flags = ~0;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(ret);
		fail_on_test(cmd.flags & ~V4L2_DEC_CMD_START_MUTE_AUDIO);
		fail_on_test(is_decoder && cmd.flags);

		cmd.cmd = V4L2_DEC_CMD_START;
		cmd.flags = 0;
		cmd.start.speed = ~0;
		cmd.start.format = ~0U;
		ret = doioctl(node, VIDIOC_TRY_DECODER_CMD, &cmd);
		fail_on_test(ret);
		fail_on_test(cmd.start.format == ~0U);
		fail_on_test(cmd.start.speed == ~0);
		fail_on_test(is_decoder && cmd.start.format);
		fail_on_test(is_decoder && cmd.start.speed);

		cmd.cmd = V4L2_DEC_CMD_PAUSE;
		cmd.flags = 0;
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(ret != EPERM && ret != EINVAL);
		fail_on_test(is_decoder && ret != EINVAL);

		cmd.cmd = V4L2_DEC_CMD_RESUME;
		ret = doioctl(node, VIDIOC_DECODER_CMD, &cmd);
		fail_on_test(ret != EPERM && ret != EINVAL);
		fail_on_test(is_decoder && ret != EINVAL);
	}

	return 0;
}