/*
===========================================================================
Copyright (C) 2015 the OpenMoHAA team

This file is part of OpenMoHAA source code.

OpenMoHAA source code 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.

OpenMoHAA source code 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 OpenMoHAA source code; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
===========================================================================
*/

// gamescript.h: Subclass of script that preprocesses labels

#ifndef __GAMESCRIPT_H__
#define __GAMESCRIPT_H__

#include "class.h"
#include "script.h"
#include "archive.h"

class Listener;
class ScriptThread;
class GameScript;

typedef struct {
	byte			*codepos;		// code position pointer
	const_str		key;			// label name
	bool			isprivate;		// new script engine implementation
} script_label_t;

typedef struct {
	unsigned int		sourcePos;
	int					column;
	int					line;
} sourceinfo_t;

class AbstractScript {
public:
	// File variables
	const_str		m_Filename;
	char			*m_SourceBuffer;
	size_t			m_SourceLength;

	// Developper variable
	con_set< uchar *, sourceinfo_t > *m_ProgToSource;

public:

	AbstractScript();

	str&					Filename( void );
	const_str				ConstFilename( void );
	bool					GetSourceAt( size_t sourcePos, str &sourceLine, int &column, int &line );
	void					PrintSourcePos( sourceinfo_t *sourcePos, bool dev );
	void					PrintSourcePos( size_t sourcePos, bool dev );
	void					PrintSourcePos( unsigned char *m_pCodePos, bool dev );
	void					PrintSourcePos( str sourceLine, int column, int line, bool dev );
};

class StateScript : public Class
{
	friend class GameScript;

private:
	// Label list
	con_set< const_str, script_label_t > label_list;
	Container< script_label_t * > reverse_label_list;

public:
	// Parent gamescript
	GameScript *m_Parent;

public:
	StateScript();

	virtual void		Archive( Archiver& arc );

	bool				AddLabel( str label, unsigned char *pos, bool private_section = false );
	bool				AddLabel( const_str label, unsigned char *pos, bool private_section = false );
	unsigned char		*FindLabel( str label );
	unsigned char		*FindLabel( const_str label );
	const_str			NearestLabel( unsigned char *pos );
};

class CatchBlock {
public:
	// program variable
	StateScript m_StateScript;

	// code position variables
	unsigned char *m_TryStartCodePos;
	unsigned char *m_TryEndCodePos;
};

class GameScript : public AbstractScript {
protected:
	// try/throw variable
	Container<CatchBlock *> m_CatchBlocks;

public:
		// program variables
		StateScript		m_State;
		unsigned char	*m_ProgBuffer;
		size_t			m_ProgLength;

		// compile variables
		bool			successCompile;
		bool			m_bPrecompiled;

		// stack variables
		unsigned int	requiredStackSize;

public:

	GameScript();
	GameScript( const char *filename );
	~GameScript();

	virtual void			Archive( Archiver& arc );
	static void				Archive( Archiver& arc, GameScript *&scr );
	void					ArchiveCodePos( Archiver& arc, unsigned char **codePos );

	void					Close( void );
	void					Load( const void *sourceBuffer, size_t sourceLength );


	bool					GetCodePos( unsigned char *codePos, str& filename, int& pos );
	bool					SetCodePos( unsigned char *&codePos, str& filename, int pos );

	unsigned int			GetRequiredStackSize( void );

	qboolean				labelExists( const char *name );

	StateScript				*CreateCatchStateScript( unsigned char *try_begin_code_pos, unsigned char *try_end_code_pos );
	StateScript				*CreateSwitchStateScript( void );

	StateScript				*GetCatchStateScript( unsigned char *in, unsigned char *&out );
};

class ScriptThreadLabel {
private:
	GameScript		*m_Script;
	const_str		m_Label;

public:
	ScriptThreadLabel();

	ScriptThread		*Create( Listener *listener );
	void				Execute( Listener *listener = NULL );
	void				Execute( Listener *listener, Event &ev );
	void				Execute( Listener *listener, Event *ev );

	void				Set( const char *label );
	void				SetScript( const ScriptVariable& label );
	void				SetScript( const char *label );
	void				SetThread( const ScriptVariable& label );

	bool				TrySet( const_str label );
	bool				TrySet( const char *label );
	bool				TrySetScript( const_str label );
	bool				TrySetScript( const char *label );

	void				GetScriptValue(Event *ev);
	void				SetScriptValue(Event *ev);

	bool				IsSet( void );

	void				Archive( Archiver& arc );
};

#endif