tristanr
/
SS13ArcadeTCP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: Updated to latest Gymnasium spec

This commit is contained in:
Tristan Russell 2025-01-17 10:55:22 -05:00
parent fec8a1007e
commit 87ab7cbcb1
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
SS13ArcadeEnv.c
View File

@ -17,7 +17,8 @@ void CreateEnvironment(struct SS13ArcadeEnv* env) {
void Step(struct SS13ArcadeEnv* instance, const int action, struct Observation* obs) { void Step(struct SS13ArcadeEnv* instance, const int action, struct Observation* obs) {
int reward = 0; int reward = 0;
bool done = false; bool terminated = false;
bool truncated = false;
switch (action) { switch (action) {
case 0: // Attack case 0: // Attack
@ -34,7 +35,7 @@ void Step(struct SS13ArcadeEnv* instance, const int action, struct Observation*
if(instance->enemy_hp <= 0 || instance->enemy_mp <= 0) { // Enemy Defeated if(instance->enemy_hp <= 0 || instance->enemy_mp <= 0) { // Enemy Defeated
reward = 1; reward = 1;
done = true; terminated = true;
} else if(instance->enemy_mp <= 5 && rand() % 1 + 9 >= 7) { // Enemy Drain Player MP } else if(instance->enemy_mp <= 5 && rand() % 1 + 9 >= 7) { // Enemy Drain Player MP
instance->player_mp -= rand() % 2 + 1; instance->player_mp -= rand() % 2 + 1;
} else if(instance->enemy_hp <= 10 && instance->enemy_mp > 4) { // Enemy Heal } else if(instance->enemy_hp <= 10 && instance->enemy_mp > 4) { // Enemy Heal
@ -44,9 +45,11 @@ void Step(struct SS13ArcadeEnv* instance, const int action, struct Observation*
instance->player_hp -= rand() % 3 + 3; instance->player_hp -= rand() % 3 + 3;
} }
if(instance->player_hp <= 0 || instance->player_mp <= 0 || instance->steps == 200) { if(instance->player_hp <= 0 || instance->player_mp <= 0) {
reward = -1; reward = -1;
done = true; terminated = true;
} else if(instance->steps == 200) {
truncated = true;
} }
instance->steps += 1; instance->steps += 1;
@ -59,7 +62,8 @@ void Step(struct SS13ArcadeEnv* instance, const int action, struct Observation*
obs->state = state; obs->state = state;
obs->reward = reward; obs->reward = reward;
obs->done = done; obs->terminated = terminated;
obs->truncated = truncated;
} }
void GetState(struct SS13ArcadeEnv* instance, struct State* state) { void GetState(struct SS13ArcadeEnv* instance, struct State* state) {

3
SS13ArcadeEnv.h
View File

@ -9,7 +9,8 @@ struct __attribute__((packed)) State {
struct __attribute__((packed)) Observation { struct __attribute__((packed)) Observation {
struct State state; struct State state;
int reward; int reward;
bool done; bool terminated;
bool truncated;
}; };
struct SS13ArcadeEnv { struct SS13ArcadeEnv {