1
0
mirror of https://github.com/VCMP-SqMod/SqMod.git synced 2025-01-19 03:57:14 +01:00

Global wide switch to extended format support.

This commit is contained in:
Sandu Liviu Catalin 2021-04-02 19:12:06 +03:00
parent cb9c5019f8
commit 950d684f9a
5 changed files with 104 additions and 16 deletions

View File

@ -134,10 +134,11 @@ SQInteger FormatContext::Proc(HSQUIRRELVM vm, SQInteger text, SQInteger args, SQ
if (SQ_FAILED(mRes)) if (SQ_FAILED(mRes))
{ {
return mRes; return mRes;
} // Drop previous arguments, if any
} else mArgs.clear();
// Remember the format string and assume it lives on the stack // Remember the format string and assume it lives on the stack
fmt::basic_string_view< SQChar > f_str(s, static_cast< size_t >(i)); fmt::basic_string_view< SQChar > f_str(s, static_cast< size_t >(i));
// // Process the arguments in the specified range
for(SQInteger idx = args; SQ_SUCCEEDED(mRes) && idx <= end; ++idx) for(SQInteger idx = args; SQ_SUCCEEDED(mRes) && idx <= end; ++idx)
{ {
switch(sq_gettype(vm, idx)) switch(sq_gettype(vm, idx))
@ -406,7 +407,38 @@ inline auto FormatObjectInContext(HSQUIRRELVM vm, SQInteger idx, SQInteger src,
} }
// ------------------------------------------------------------------------------------------------ // ------------------------------------------------------------------------------------------------
SQInteger SqFormat(HSQUIRRELVM vm) void ExtendedFormatProcess(StackStrF & ss, SQInteger top)
{
FormatContext ctx;
// Attempt to perform the format
ss.mRes = ctx.Proc(ss.mVM, ss.mIdx, ss.mIdx + 1, top);
// Did format succeed?
if (SQ_SUCCEEDED(ss.mRes))
{
// Transform the string into a script object
sq_pushstring(ss.mVM, ctx.mOut.data(), static_cast<SQInteger>(ctx.mOut.size()));
// At this point we have a new object on the stack that must be removed
SqPopTopGuard spg(ss.mVM);
// Obtain a reference to the string object
ss.mRes = sq_getstackobj(ss.mVM, -1, &ss.mObj);
// Could we retrieve the object from the stack?
if (SQ_SUCCEEDED(ss.mRes))
{
// Keep a strong reference to the object
sq_addref(ss.mVM, &ss.mObj);
// Attempt to retrieve the string value from the stack
ss.mRes = sq_getstringandsize(ss.mVM, -1, &ss.mPtr, &ss.mLen);
}
// Did the retrieval succeeded but ended up with a null string pointer?
if (SQ_SUCCEEDED(ss.mRes) && !ss.mPtr)
{
ss.mRes = sq_throwerror(ss.mVM, "Unable to retrieve the string");
}
}
}
// ------------------------------------------------------------------------------------------------
static SQInteger SqFormat(HSQUIRRELVM vm)
{ {
FormatContext ctx; FormatContext ctx;
// Attempt to generate the formatted string // Attempt to generate the formatted string

View File

@ -71,4 +71,9 @@ struct FormatContext
SQMOD_NODISCARD SQInteger Proc(HSQUIRRELVM vm, SQInteger text, SQInteger args, SQInteger end = -1); SQMOD_NODISCARD SQInteger Proc(HSQUIRRELVM vm, SQInteger text, SQInteger args, SQInteger end = -1);
}; };
/* ------------------------------------------------------------------------------------------------
* Helper function used to process a formatted string into the specified StackStrF instance.
*/
void ExtendedFormatProcess(StackStrF & ss, SQInteger top);
} // Namespace:: SqMod } // Namespace:: SqMod

View File

@ -1,5 +1,6 @@
// ------------------------------------------------------------------------------------------------ // ------------------------------------------------------------------------------------------------
#include "Library/Utils/String.hpp" #include "Library/Utils/String.hpp"
#include "Library/Format.hpp"
// ------------------------------------------------------------------------------------------------ // ------------------------------------------------------------------------------------------------
namespace SqMod { namespace SqMod {
@ -7,6 +8,25 @@ namespace SqMod {
// ------------------------------------------------------------------------------------------------ // ------------------------------------------------------------------------------------------------
SQMOD_DECL_TYPENAME(Typename, _SC("SqString")) SQMOD_DECL_TYPENAME(Typename, _SC("SqString"))
// ------------------------------------------------------------------------------------------------
static SQInteger SqStringFormat(HSQUIRRELVM vm)
{
FormatContext ctx;
// Attempt to generate the formatted string
if (SQ_FAILED(ctx.Proc(vm, 2, 3, sq_gettop(vm))))
{
return ctx.mRes;
}
// Create the instance and guard it to make sure it gets deleted in case of exceptions
DeleteGuard< SqString > instance(new SqString(std::move(ctx.mOut)));
// Push the instance on the stack
ClassType< SqString >::PushInstance(vm, instance);
// Stop guarding the instance
instance.Release();
// Specify that we returned a value
return 1;
}
// ================================================================================================ // ================================================================================================
void Register_Native_String(HSQUIRRELVM vm, Table & ns) void Register_Native_String(HSQUIRRELVM vm, Table & ns)
{ {
@ -17,7 +37,9 @@ void Register_Native_String(HSQUIRRELVM vm, Table & ns)
.Ctor< StackStrF & >() .Ctor< StackStrF & >()
// Meta-methods // Meta-methods
.SquirrelFunc(_SC("_typename"), &Typename::Fn) .SquirrelFunc(_SC("_typename"), &Typename::Fn)
.Func(_SC("_tostring"), &SqString::GetString)
// Properties // Properties
.Prop(_SC("Str"), &SqString::GetString, &SqString::SetString)
.Prop(_SC("Front"), &SqString::Front) .Prop(_SC("Front"), &SqString::Front)
.Prop(_SC("Back"), &SqString::Back) .Prop(_SC("Back"), &SqString::Back)
.Prop(_SC("Empty"), &SqString::Empty) .Prop(_SC("Empty"), &SqString::Empty)
@ -58,7 +80,10 @@ void Register_Native_String(HSQUIRRELVM vm, Table & ns)
.Func(_SC("GenerateBetween"), &SqString::GenerateBetween) .Func(_SC("GenerateBetween"), &SqString::GenerateBetween)
.Func(_SC("Sort"), &SqString::Sort) .Func(_SC("Sort"), &SqString::Sort)
.Func(_SC("Shuffle"), &SqString::Shuffle) .Func(_SC("Shuffle"), &SqString::Shuffle)
.Func(_SC("Assign"), &SqString::SetString)
); );
// --------------------------------------------------------------------------------------------
RootTable(vm).SquirrelFunc(_SC("SqStringF"), SqStringFormat);
} }
} // Namespace:: SqMod } // Namespace:: SqMod

View File

@ -158,6 +158,39 @@ struct SqString
return mS; return mS;
} }
/* --------------------------------------------------------------------------------------------
* Retrieve the internal string container.
*/
SQMOD_NODISCARD String & ToString()
{
return mS;
}
/* --------------------------------------------------------------------------------------------
* Retrieve the internal string container.
*/
SQMOD_NODISCARD const String & ToString() const
{
return mS;
}
/* --------------------------------------------------------------------------------------------
* Retrieve the internal string container.
*/
SQMOD_NODISCARD const String & GetString() const
{
return mS;
}
/* --------------------------------------------------------------------------------------------
* Modify the internal string container.
*/
SQMOD_NODISCARD SqString & SetString(StackStrF & str)
{
mS.assign(str.mPtr, static_cast< size_t >(str.mLen));
return *this; // Allow chaining
}
/* -------------------------------------------------------------------------------------------- /* --------------------------------------------------------------------------------------------
* Retrieve a value from the container. * Retrieve a value from the container.
*/ */

View File

@ -39,6 +39,10 @@
#include <Poco/Exception.h> #include <Poco/Exception.h>
namespace SqMod {
extern void ExtendedFormatProcess(StackStrF & ss, SQInteger top);
} // Namespace:: SqMod
namespace Sqrat { namespace Sqrat {
#if defined(__GNUC__) #if defined(__GNUC__)
@ -1980,19 +1984,7 @@ struct StackStrF
// Do we have enough values to call the format function and are we allowed to? // Do we have enough values to call the format function and are we allowed to?
else if (fmt && (top - 1) >= mIdx) else if (fmt && (top - 1) >= mIdx)
{ {
// Pointer to the generated string ::SqMod::ExtendedFormatProcess(*this, top);
SQChar * str = nullptr;
// Attempt to generate the specified string format
mRes = sqstd_format(mVM, mIdx, &mLen, &str);
// Did the format succeeded but ended up with a null string pointer?
if (SQ_SUCCEEDED(mRes) && !str)
{
mRes = sq_throwerror(mVM, "Unable to generate the string");
}
else
{
mPtr = const_cast< const SQChar * >(str);
}
} }
// Is the value on the stack an actual string? // Is the value on the stack an actual string?
else if (sq_gettype(mVM, mIdx) == OT_STRING) else if (sq_gettype(mVM, mIdx) == OT_STRING)
@ -2040,6 +2032,7 @@ struct StackStrF
mRes = sq_throwerror(mVM, "Unable to retrieve the value"); mRes = sq_throwerror(mVM, "Unable to retrieve the value");
} }
} }
// Return last known result
return mRes; return mRes;
} }
///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////