70 lines
1.9 KiB
C++
70 lines
1.9 KiB
C++
|
#include "testEnv.hpp"
|
||
|
|
|
||
|
|
namespace {
|
||
|
|
|
||
|
|
class Creature {
|
||
|
|
int health;
|
||
|
|
public:
|
||
|
|
enum {MaxHealth = 100};
|
||
|
|
Creature() : health(MaxHealth) {}
|
||
|
|
int GetMaxHealth(void) {
|
||
|
|
return MaxHealth;
|
||
|
|
}
|
||
|
|
int GetHealth(void) {
|
||
|
|
return health;
|
||
|
|
}
|
||
|
|
void SetHealth(int newHealth) {
|
||
|
|
health = newHealth;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
DECLARE_INSTANCE_TYPE(Creature);
|
||
|
|
|
||
|
|
|
||
|
|
SQPLUS_TEST(Test_PassByReference)
|
||
|
|
{
|
||
|
|
SQPLUS_TEST_TRACE();
|
||
|
|
|
||
|
|
using namespace SqPlus;
|
||
|
|
|
||
|
|
// === BEGIN example from forum post ===
|
||
|
|
|
||
|
|
SQClassDef<Creature>(_SC("Creature"))
|
||
|
|
.func(&Creature::GetMaxHealth, _SC("GetMaxHealth"))
|
||
|
|
.func(&Creature::GetHealth, _SC("GetHealth"))
|
||
|
|
.func(&Creature::SetHealth, _SC("SetHealth"))
|
||
|
|
;
|
||
|
|
Creature frodo;
|
||
|
|
frodo.SetHealth(frodo.GetMaxHealth() / 2);
|
||
|
|
|
||
|
|
RUN_SCRIPT(_SC("function HealthPotionUse(Target) { \n\
|
||
|
|
local curHealth = Target.GetHealth(); \n\
|
||
|
|
local maxHealth = Target.GetMaxHealth(); \n\
|
||
|
|
if ((maxHealth - curHealth) > 15) { \n\
|
||
|
|
curHealth += 15; \n\
|
||
|
|
} else { \n\
|
||
|
|
curHealth = maxHealth; \n\
|
||
|
|
} \n\
|
||
|
|
Target.SetHealth(curHealth); \n\
|
||
|
|
print(typeof Target); \n\
|
||
|
|
assert(typeof(Target) == \"instance\"); \n\
|
||
|
|
return Target; \n\
|
||
|
|
}"));
|
||
|
|
|
||
|
|
|
||
|
|
// Pass by value and return a copy
|
||
|
|
// (Must return by reference due to template system design).
|
||
|
|
Creature newFrodo =
|
||
|
|
SquirrelFunction<Creature&>(_SC("HealthPotionUse"))(frodo);
|
||
|
|
|
||
|
|
// Pass the address to directly modify frodo.
|
||
|
|
SquirrelFunction<void>(_SC("HealthPotionUse"))(&frodo);
|
||
|
|
|
||
|
|
scprintf(_SC("Frodo's health: %d %d\n"),
|
||
|
|
frodo.GetHealth(), newFrodo.GetHealth());
|
||
|
|
|
||
|
|
CHECK_EQUAL(frodo.GetHealth(), newFrodo.GetHealth());
|
||
|
|
|
||
|
|
// === END example from forum post ===
|
||
|
|
}
|