#ifndef _TCUCOMMANDLINE_HPP
#define _TCUCOMMANDLINE_HPP
/*-------------------------------------------------------------------------
 * drawElements Quality Program Tester Core
 * ----------------------------------------
 *
 * Copyright 2014 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.
 *
 *//*!
 * \file
 * \brief Command line parsing.
 *//*--------------------------------------------------------------------*/

#include "tcuDefs.hpp"
#include "deCommandLine.hpp"
#include "tcuTestCase.hpp"
#include "deUniquePtr.hpp"

#include <string>
#include <vector>
#include <istream>

namespace tcu
{

/*--------------------------------------------------------------------*//*!
 * \brief Run mode tells whether the test program should run the tests or
 *		  dump out metadata about the tests.
 *//*--------------------------------------------------------------------*/
enum RunMode
{
	RUNMODE_EXECUTE = 0,			//! Test program executes the tests.
	RUNMODE_DUMP_XML_CASELIST,		//! Test program dumps the list of contained test cases in XML format.
	RUNMODE_DUMP_TEXT_CASELIST,		//! Test program dumps the list of contained test cases in plain-text format.
	RUNMODE_DUMP_STDOUT_CASELIST,	//! Test program dumps the list of contained test cases in plain-text format into stdout.
	RUNMODE_VERIFY_AMBER_COHERENCY, //! Test program verifies that amber tests have coherent capability requirements

	RUNMODE_LAST
};

/*--------------------------------------------------------------------*//*!
 * \brief Should graphical tests show rendering results on screen.
 *//*--------------------------------------------------------------------*/
enum WindowVisibility
{
	WINDOWVISIBILITY_WINDOWED = 0,
	WINDOWVISIBILITY_FULLSCREEN,
	WINDOWVISIBILITY_HIDDEN,

	WINDOWVISIBILITY_LAST
};

/*--------------------------------------------------------------------*//*!
 * \brief The type of rendering surface the tests should be executed on.
 *//*--------------------------------------------------------------------*/
enum SurfaceType
{
	SURFACETYPE_WINDOW = 0,			//!< Native window.
	SURFACETYPE_OFFSCREEN_NATIVE,	//!< Native offscreen surface, such as pixmap.
	SURFACETYPE_OFFSCREEN_GENERIC,	//!< Generic offscreen surface, such as pbuffer.
	SURFACETYPE_FBO,				//!< Framebuffer object.

	SURFACETYPE_LAST
};

/*--------------------------------------------------------------------*//*!
 * \brief Screen rotation, always to clockwise direction.
 *//*--------------------------------------------------------------------*/
enum ScreenRotation
{
	SCREENROTATION_UNSPECIFIED,		//!< Use default / current orientation.
	SCREENROTATION_0,				//!< Set rotation to 0 degrees from baseline.
	SCREENROTATION_90,
	SCREENROTATION_180,
	SCREENROTATION_270,

	SCREENROTATION_LAST
};

class CaseTreeNode;
class CasePaths;
class Archive;

// Match a single path component against a pattern component that may contain *-wildcards.
bool matchWildcards(std::string::const_iterator		patternStart,
					std::string::const_iterator		patternEnd,
					std::string::const_iterator		pathStart,
					std::string::const_iterator		pathEnd,
					bool							allowPrefix);

class CaseListFilter
{
public:
									CaseListFilter				(const de::cmdline::CommandLine& cmdLine, const tcu::Archive& archive);
									CaseListFilter				(void);
									~CaseListFilter				(void);

	//! Check if test group is in supplied test case list.
	bool							checkTestGroupName			(const char* groupName) const;

	//! Check if test case is in supplied test case list.
	bool							checkTestCaseName			(const char* caseName) const;

	//! Check if test group passes the case fraction filter.
	bool							checkCaseFraction			(int i, const std::string& testCaseName) const;

	//! Check if test case runner is of supplied type
	bool							checkRunnerType				(tcu::TestRunnerType type) const { return ((m_runnerType & type) == m_runnerType); }

private:
	CaseListFilter												(const CaseListFilter&);	// not allowed!
	CaseListFilter&					operator=					(const CaseListFilter&);	// not allowed!

	CaseTreeNode*					m_caseTree;
	de::MovePtr<const CasePaths>	m_casePaths;
	std::vector<int>				m_caseFraction;
	de::MovePtr<const CasePaths>	m_caseFractionMandatoryTests;
	tcu::TestRunnerType				m_runnerType;
};

/*--------------------------------------------------------------------*//*!
 * \brief Test command line
 *
 * CommandLine handles argument parsing and provides convinience functions
 * for querying test parameters.
 *//*--------------------------------------------------------------------*/
class CommandLine
{
public:
									CommandLine						(void);
									CommandLine						(int argc, const char* const* argv);
	explicit						CommandLine						(const std::string& cmdLine);
	virtual							~CommandLine					(void);

	bool							parse							(int argc, const char* const* argv);
	bool							parse							(const std::string& cmdLine);

	const std::string&				getApplicationName				(void) const;
	const std::string&				getInitialCmdLine				(void) const;

	//! Is quiet mode active?
	bool							quietMode						(void) const;

	//! Get log file name (--deqp-log-filename)
	const char*						getLogFileName					(void) const;

	//! Get logging flags
	deUint32						getLogFlags						(void) const;

	//! Get run mode (--deqp-runmode)
	RunMode							getRunMode						(void) const;

	//! Get caselist dump target file pattern (--deqp-caselist-export-file)
	const char*						getCaseListExportFile			(void) const;

	//! Get default window visibility (--deqp-visibility)
	WindowVisibility				getVisibility					(void) const;

	//! Get watchdog enable status (--deqp-watchdog)
	bool							isWatchDogEnabled				(void) const;

	//! Get crash handling enable status (--deqp-crashhandler)
	bool							isCrashHandlingEnabled			(void) const;

	//! Get base seed for randomization (--deqp-base-seed)
	int								getBaseSeed						(void) const;

	//! Get test iteration count (--deqp-test-iteration-count)
	int								getTestIterationCount			(void) const;

	//! Get rendering target width (--deqp-surface-width)
	int								getSurfaceWidth					(void) const;

	//! Get rendering target height (--deqp-surface-height)
	int								getSurfaceHeight				(void) const;

	//! Get rendering taget type (--deqp-surface-type)
	SurfaceType						getSurfaceType					(void) const;

	//! Get screen rotation (--deqp-screen-rotation)
	ScreenRotation					getScreenRotation				(void) const;

	//! Get GL context factory name (--deqp-gl-context-type)
	const char*						getGLContextType				(void) const;

	//! Get GL config ID (--deqp-gl-config-id)
	int								getGLConfigId					(void) const;

	//! Get GL config name (--deqp-gl-config-name)
	const char*						getGLConfigName					(void) const;

	//! Get GL context flags (--deqp-gl-context-flags)
	const char*						getGLContextFlags				(void) const;

	//! Get OpenCL platform ID (--deqp-cl-platform-id)
	int								getCLPlatformId					(void) const;

	//! Get OpenCL device IDs (--deqp-cl-device-ids)
	void							getCLDeviceIds					(std::vector<int>& deviceIds) const	{ deviceIds = getCLDeviceIds(); }
	const std::vector<int>&			getCLDeviceIds					(void) const;

	//! Get extra OpenCL program build options (--deqp-cl-build-options)
	const char*						getCLBuildOptions				(void) const;

	//! Get EGL native display factory (--deqp-egl-display-type)
	const char*						getEGLDisplayType				(void) const;

	//! Get EGL native window factory (--deqp-egl-window-type)
	const char*						getEGLWindowType				(void) const;

	//! Get EGL native pixmap factory (--deqp-egl-pixmap-type)
	const char*						getEGLPixmapType				(void) const;

	//! Get Vulkan device ID (--deqp-vk-device-id)
	int								getVKDeviceId					(void) const;

	//! Get Vulkan device group ID (--deqp-vk-device-group-id)
	int								getVKDeviceGroupId				(void) const;

	//! Enable development-time test case validation checks
	bool							isValidationEnabled				(void) const;

	//! Print validation errors to standard error or keep them in the log only.
	bool							printValidationErrors			(void) const;

	//! Log of decompiled SPIR-V shader source (--deqp-log-decompiled-spirv)
	bool							isLogDecompiledSpirvEnabled		(void) const;

	//! Should we run tests that exhaust memory (--deqp-test-oom)
	bool							isOutOfMemoryTestEnabled		(void) const;

	//! Should the shader cache be enabled (--deqp-shadercache)
	bool							isShadercacheEnabled			(void) const;

	//! Get the filename for shader cache (--deqp-shadercache-filename)
	const char*						getShaderCacheFilename			(void) const;

	//! Should the shader cache be truncated before run (--deqp-shadercache-truncate)
	bool							isShaderCacheTruncateEnabled	(void) const;

	//! Should the shader cache use inter process communication (IPC) (--deqp-shadercache-ipc)
	bool							isShaderCacheIPCEnabled	(void) const;

	//! Get shader optimization recipe (--deqp-optimization-recipe)
	int								getOptimizationRecipe		(void) const;

	//! Enable optimizing of spir-v (--deqp-optimize-spirv)
	bool							isSpirvOptimizationEnabled	(void) const;

	//! Enable RenderDoc frame markers (--deqp-renderdoc)
	bool							isRenderDocEnabled			(void) const;

	//! Get waiver file name (--deqp-waiver-file)
	const char*						getWaiverFileName			(void) const;

	//! Get case list fraction
	const std::vector<int>&			getCaseFraction				(void) const;

	//! Get must-list filename
	const char*						getCaseFractionMandatoryTests(void) const;

	//! Get archive directory path
	const char*						getArchiveDir				(void) const;

	//! Get runner type (--deqp-runner-type)
	tcu::TestRunnerType				getRunnerType				(void) const;

	//! Should the run be terminated on first failure (--deqp-terminate-on-fail)
	bool							isTerminateOnFailEnabled	(void) const;

	//! Should the run be terminated on first device lost (--deqp-terminate-on-device-lost)
	bool							isTerminateOnDeviceLostEnabled	(void) const;

	//! Start as subprocess ( Vulkan SC )
	bool							isSubProcess				(void) const;

	//! Define default number of tests performed in main process ( Vulkan SC )
	int								getSubprocessTestCount		(void) const;

	//! Config file defining number of tests performed in subprocess for specific test branches
	const char*						getSubprocessConfigFile		(void) const;

	//! Optional server address that will be responsible for (among other things) compiling shaders ( Vulkan SC )
	const char*						getServerAddress			(void) const;

	//! Define minimum size of a single command pool ( Vulkan SC )
	int								getCommandPoolMinSize(void) const;

	//! Define minimum size of a single command buffer ( Vulkan SC )
	int								getCommandBufferMinSize(void) const;

	//! Define default size for single command in command buffer ( Vulkan SC )
	int								getCommandDefaultSize(void) const;

	//! Define default size for single pipeline ( Vulkan SC )
	int								getPipelineDefaultSize(void) const;

	//! Path to offline pipeline compiler executable
	const char*						getPipelineCompilerPath		(void) const;

	//! Directory containing input and output pipeline compiler files
	const char*						getPipelineCompilerDataDir	(void) const;

	//! Additional args for offline pipeline compiler
	const char*						getPipelineCompilerArgs		(void) const;

	//! Output pipeline cache file produced by offline pipeline compiler
	const char*						getPipelineCompilerOutputFile(void) const;

	//! Log file for offline pipeline compiler
	const char*						getPipelineCompilerLogFile	(void) const;

	//! Prefix for offline pipeline compiler input files
	const char*						getPipelineCompilerFilePrefix(void) const;

	//! Path to Vulkan library (e.g. loader library vulkan-1.dll)
	const char*						getVkLibraryPath			(void) const;

	//! File that provides a default set of application parameters
	const char*						getAppParamsInputFilePath(void) const;

	//! Perform tests for devices implementing compute-only functionality
	bool							isComputeOnly				(void) const;

	/*--------------------------------------------------------------------*//*!
	 * \brief Creates case list filter
	 * \param archive Resources
	 *
	 * Creates case list filter based on one of the following parameters:
	 *
	 * --deqp-case
	 * --deqp-caselist
	 * --deqp-caselist-file
	 * --deqp-caselist-resource
	 * --deqp-stdin-caselist
	 *
	 * Throws std::invalid_argument if parsing fails.
	 *//*--------------------------------------------------------------------*/
	de::MovePtr<CaseListFilter>		createCaseListFilter		(const tcu::Archive& archive) const;

protected:
	const de::cmdline::CommandLine&	getCommandLine				(void) const;

private:
									CommandLine					(const CommandLine&);	// not allowed!
	CommandLine&					operator=					(const CommandLine&);	// not allowed!

	void							clear						(void);

	virtual void					registerExtendedOptions		(de::cmdline::Parser& parser);

	std::string						m_appName;
	de::cmdline::CommandLine		m_cmdLine;
	deUint32						m_logFlags;

	std::string						m_initialCmdLine;
	bool							m_hadHelpSpecified;
};

} // tcu

#endif // _TCUCOMMANDLINE_HPP