mirror of
https://github.com/VCMP-SqMod/SqMod.git
synced 2024-11-08 08:47:17 +01:00
554 lines
18 KiB
C++
554 lines
18 KiB
C++
#ifndef _ROUTINE_HPP_
|
|
#define _ROUTINE_HPP_
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
#include "Base/Shared.hpp"
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
namespace SqMod {
|
|
|
|
/* ------------------------------------------------------------------------------------------------
|
|
* Execute callbacks after specific intervals of time.
|
|
*/
|
|
class Routine
|
|
{
|
|
public:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Simplify future changes to a single point of change.
|
|
*/
|
|
typedef Int64 Time;
|
|
typedef SQInteger Interval;
|
|
typedef Uint32 Iterator;
|
|
typedef LightObj Argument;
|
|
|
|
private:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Structure that represents an active routine and keeps track of the routine information.
|
|
*/
|
|
struct Instance
|
|
{
|
|
// ----------------------------------------------------------------------------------------
|
|
LightObj mEnv; // A reference to the managed environment object.
|
|
LightObj mFunc; // A reference to the managed function object.
|
|
LightObj mInst; // Reference to the routine associated with this instance.
|
|
LightObj mData; // A reference to the arbitrary data associated with this instance.
|
|
String mTag; // An arbitrary string which represents the tag.
|
|
Iterator mIterations; // Number of iterations before self destruct.
|
|
Interval mInterval; // Interval between routine invocations.
|
|
bool mSuspended; // Whether this instance is allowed to receive calls.
|
|
bool mQuiet; // Whether this instance is allowed to handle errors.
|
|
bool mEndure; // Whether this instance is allowed to terminate itself on errors.
|
|
Uint8 mArgc; // The number of arguments that the routine must forward.
|
|
Argument mArgv[14]; // The arguments that the routine must forward.
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Default constructor.
|
|
*/
|
|
Instance()
|
|
: mEnv()
|
|
, mFunc()
|
|
, mInst()
|
|
, mData()
|
|
, mTag()
|
|
, mIterations(0)
|
|
, mInterval(0)
|
|
, mSuspended(false)
|
|
, mQuiet(ErrorHandling::IsEnabled())
|
|
, mEndure(false)
|
|
, mArgc(0)
|
|
, mArgv()
|
|
{
|
|
/* ... */
|
|
}
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Copy constructor. (disabled)
|
|
*/
|
|
Instance(const Instance & o) = delete;
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Move constructor. (disabled)
|
|
*/
|
|
Instance(Instance && o) = delete;
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Destructor.
|
|
*/
|
|
~Instance()
|
|
{
|
|
Terminate();
|
|
}
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Copy assignment operator. (disabled)
|
|
*/
|
|
Instance & operator = (const Instance & o) = delete;
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Move assignment operator. (disabled)
|
|
*/
|
|
Instance & operator = (Instance && o) = delete;
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Initializes the routine parameters. (assumes previous values are already released)
|
|
*/
|
|
void Init(HSQOBJECT & env, HSQOBJECT & func, HSQOBJECT & inst, Interval intrv, Iterator itr)
|
|
{
|
|
// Initialize the callback objects
|
|
mEnv = LightObj(env);
|
|
mFunc = LightObj(func);
|
|
// Associate with the routine instance
|
|
mInst = LightObj(inst);
|
|
// Initialize the routine options
|
|
mIterations = itr;
|
|
mInterval = intrv;
|
|
}
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Release managed script resources.
|
|
*/
|
|
void Release()
|
|
{
|
|
mEnv.Release();
|
|
mFunc.Release();
|
|
mInst.Release();
|
|
mData.Release();
|
|
mIterations = 0;
|
|
mInterval = 0;
|
|
mTag.clear();
|
|
}
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Execute the managed routine.
|
|
*/
|
|
Interval Execute()
|
|
{
|
|
// is this even a valid routine?
|
|
if (mInst.IsNull())
|
|
{
|
|
return 0; // Dunno how we got here but it ends now
|
|
}
|
|
// Are we allowed to forward calls?
|
|
else if (!mSuspended)
|
|
{
|
|
// Grab the virtual machine once
|
|
HSQUIRRELVM vm = DefaultVM::Get();
|
|
// Push the function on the stack
|
|
sq_pushobject(vm, mFunc);
|
|
// Push the environment on the stack
|
|
sq_pushobject(vm, mEnv);
|
|
// Push function parameters, if any
|
|
for (Uint32 n = 0; n < mArgc; ++n)
|
|
{
|
|
sq_pushobject(vm, mArgv[n].mObj);
|
|
}
|
|
// Make the function call and store the result
|
|
const SQRESULT res = sq_call(vm, mArgc + 1, false, !mQuiet);
|
|
// Pop the callback object from the stack
|
|
sq_pop(vm, 1);
|
|
// Validate the result
|
|
if (SQ_FAILED(res))
|
|
{
|
|
// Should we endure the errors?
|
|
if (!mEndure)
|
|
{
|
|
Terminate(); // Destroy ourself on error
|
|
}
|
|
}
|
|
}
|
|
// Decrease the number of iterations if necessary
|
|
if (mIterations && (--mIterations) == 0)
|
|
{
|
|
Terminate(); // This routine reached the end of it's life
|
|
}
|
|
// Return the current interval
|
|
return mInterval;
|
|
}
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Clear the arguments.
|
|
*/
|
|
void Clear()
|
|
{
|
|
// Now release the arguments
|
|
for (auto & a : mArgv)
|
|
{
|
|
a.Release();
|
|
}
|
|
// Reset the counter
|
|
mArgc = 0;
|
|
}
|
|
|
|
/* ----------------------------------------------------------------------------------------
|
|
* Terminate the routine.
|
|
*/
|
|
void Terminate()
|
|
{
|
|
Release();
|
|
Clear();
|
|
}
|
|
};
|
|
|
|
private:
|
|
|
|
// --------------------------------------------------------------------------------------------
|
|
static Time s_Last; // Last time point.
|
|
static Time s_Prev; // Previous time point.
|
|
static Interval s_Intervals[SQMOD_MAX_ROUTINES]; // List of intervals to be processed.
|
|
static Instance s_Instances[SQMOD_MAX_ROUTINES]; // List of routines to be executed.
|
|
|
|
private:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* The index of the slot in the pool of active routines.
|
|
*/
|
|
Uint32 m_Slot;
|
|
|
|
protected:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Default constructor.
|
|
*/
|
|
Routine()
|
|
: m_Slot(SQMOD_MAX_ROUTINES)
|
|
{
|
|
/* ... */
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Default constructor.
|
|
*/
|
|
Routine(Uint32 slot)
|
|
: m_Slot(slot)
|
|
{
|
|
/* ... */
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Copy constructor. (disabled)
|
|
*/
|
|
Routine(const Routine & o) = delete;
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Move constructor. (disabled)
|
|
*/
|
|
Routine(Routine && o) = delete;
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Copy assignment operator. (disabled)
|
|
*/
|
|
Routine & operator = (const Routine & o) = delete;
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Move assignment operator. (disabled)
|
|
*/
|
|
Routine & operator = (Routine && o) = delete;
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Find an unoccupied routine slot.
|
|
*/
|
|
static SQInteger FindUnused()
|
|
{
|
|
for (const auto & r : s_Instances)
|
|
{
|
|
if (r.mInst.IsNull())
|
|
{
|
|
return (&r - s_Instances); // Return the index of this element
|
|
}
|
|
}
|
|
// No available slot
|
|
return -1;
|
|
}
|
|
|
|
public:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the number of used routine slots.
|
|
*/
|
|
static SQInteger GetUsed()
|
|
{
|
|
SQInteger n = 0;
|
|
// Iterate routine list
|
|
for (const auto & r : s_Instances)
|
|
{
|
|
if (!r.mInst.IsNull())
|
|
{
|
|
++n;
|
|
}
|
|
}
|
|
// Return the final count
|
|
return n;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the number of used routine slots.
|
|
*/
|
|
static const LightObj & FindByTag(const StackStrF & tag)
|
|
{
|
|
// Is the specified tag valid?
|
|
if (!tag.mPtr)
|
|
{
|
|
STHROWF("Invalid routine tag");
|
|
}
|
|
// Iterate routine list
|
|
for (const auto & r : s_Instances)
|
|
{
|
|
if (!r.mInst.IsNull() && r.mTag.compare(tag.mPtr) == 0)
|
|
{
|
|
return r.mInst; // Return this routine instance
|
|
}
|
|
}
|
|
// Unable to find such routine
|
|
STHROWF("Unable to find a routine with tag (%s)", tag.mPtr);
|
|
// Should not reach this point but if it did, we have to return something
|
|
return s_Instances[SQMOD_MAX_ROUTINES].mInst; // Intentional Buffer overflow!
|
|
}
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Check if a routine with a certain tag exists.
|
|
*/
|
|
static bool IsWithTag(const StackStrF & tag);
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Process all active routines and update elapsed time.
|
|
*/
|
|
static void Process();
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Initialize all resources and prepare for startup.
|
|
*/
|
|
static void Initialize();
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Release all resources and prepare for shutdown.
|
|
*/
|
|
static void Deinitialize();
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Create a routine with the specified parameters.
|
|
*/
|
|
static SQInteger Create(HSQUIRRELVM vm);
|
|
|
|
protected:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* See whether this routine is valid otherwise throw an exception.
|
|
*/
|
|
void Validate() const
|
|
{
|
|
if (m_Slot >= SQMOD_MAX_ROUTINES)
|
|
{
|
|
STHROWF("This instance does not reference a valid routine");
|
|
}
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* See whether this routine is valid otherwise throw an exception.
|
|
*/
|
|
Instance & GetValid() const
|
|
{
|
|
if (m_Slot >= SQMOD_MAX_ROUTINES)
|
|
{
|
|
STHROWF("This instance does not reference a valid routine");
|
|
}
|
|
// We know it's valid so let's return it
|
|
return s_Instances[m_Slot];
|
|
}
|
|
|
|
public:
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Used by the script engine to convert an instance of this type to a string.
|
|
*/
|
|
const String & ToString() const
|
|
{
|
|
return (m_Slot >= SQMOD_MAX_ROUTINES) ? NullString() : s_Instances[m_Slot].mTag;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Terminate the routine.
|
|
*/
|
|
void Terminate()
|
|
{
|
|
GetValid().Terminate();
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the associated user tag.
|
|
*/
|
|
const String & GetTag() const
|
|
{
|
|
return GetValid().mTag;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Modify the associated user tag.
|
|
*/
|
|
void SetTag(const StackStrF & tag)
|
|
{
|
|
GetValid().mTag.assign(tag.mPtr, ClampMin(tag.mLen, 0));
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the environment object.
|
|
*/
|
|
const LightObj & GetEnv() const
|
|
{
|
|
return GetValid().mEnv;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Modify the environment object.
|
|
*/
|
|
void SetEnv(const LightObj & env)
|
|
{
|
|
GetValid().mEnv = env.IsNull() ? LightObj(RootTable().GetObject()) : env;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the function object.
|
|
*/
|
|
const LightObj & GetFunc() const
|
|
{
|
|
return GetValid().mFunc;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Modify the function object.
|
|
*/
|
|
void SetFunc(const Function & func)
|
|
{
|
|
// Validate the specified
|
|
if (!sq_isclosure(func.GetFunc()) && !sq_isnativeclosure(func.GetFunc()))
|
|
{
|
|
STHROWF("Invalid callback type %s", SqTypeName(GetValid().mFunc.GetType()));
|
|
}
|
|
// Store the function without the environment
|
|
GetValid().mFunc = LightObj(func.GetFunc());
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the arbitrary user data object.
|
|
*/
|
|
const LightObj & GetData() const
|
|
{
|
|
return GetValid().mData;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Modify the arbitrary user data object.
|
|
*/
|
|
void SetData(const LightObj & data)
|
|
{
|
|
GetValid().mData = data;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the execution interval.
|
|
*/
|
|
SQInteger GetInterval() const
|
|
{
|
|
return ConvTo< SQInteger >::From(GetValid().mInterval);
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Modify the execution interval.
|
|
*/
|
|
void SetInterval(SQInteger itr)
|
|
{
|
|
GetValid().mInterval = ClampMin(ConvTo< Interval >::From(itr), static_cast< Interval >(0));
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the number of iterations.
|
|
*/
|
|
SQInteger GetIterations() const
|
|
{
|
|
return ConvTo< SQInteger >::From(GetValid().mIterations);
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Modify the number of iterations.
|
|
*/
|
|
void SetIterations(SQInteger itr)
|
|
{
|
|
GetValid().mIterations = ConvTo< Iterator >::From(itr);
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* See whether the routine is suspended.
|
|
*/
|
|
bool GetSuspended() const
|
|
{
|
|
return GetValid().mSuspended;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Set whether the routine should be suspended.
|
|
*/
|
|
void SetSuspended(bool toggle)
|
|
{
|
|
GetValid().mSuspended = toggle;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* See whether the routine is quite.
|
|
*/
|
|
bool GetQuiet() const
|
|
{
|
|
return GetValid().mQuiet;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Set whether the routine should be quiet.
|
|
*/
|
|
void SetQuiet(bool toggle)
|
|
{
|
|
GetValid().mQuiet = toggle;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* See whether the routine endures.
|
|
*/
|
|
bool GetEndure() const
|
|
{
|
|
return GetValid().mEndure;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Set whether the routine should endure.
|
|
*/
|
|
void SetEndure(bool toggle)
|
|
{
|
|
GetValid().mEndure = toggle;
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve the number of arguments to be forwarded.
|
|
*/
|
|
SQInteger GetArguments() const
|
|
{
|
|
return ConvTo< SQInteger >::From(GetValid().mArgc);
|
|
}
|
|
|
|
/* --------------------------------------------------------------------------------------------
|
|
* Retrieve a certain argument.
|
|
*/
|
|
const Argument & GetArgument(SQInteger arg) const
|
|
{
|
|
// Cast the index to the proper value
|
|
Uint8 idx = ConvTo< Uint8 >::From(arg);
|
|
// Validate the specified index
|
|
if (idx >= 14)
|
|
{
|
|
STHROWF("The specified index is out of range: %u >= %u", idx, 14);
|
|
}
|
|
// Return the requested argument
|
|
return GetValid().mArgv[idx];
|
|
}
|
|
};
|
|
|
|
} // Namespace:: SqMod
|
|
|
|
#endif // _ROUTINE_HPP_
|