path: root/src/libsok/sok-ctx.c

#include <stdbool.h>
#include <string.h> // memset()
#include "sok.h"

#define UNUSED(a) ((void) (a))

#define VALID_POS(pos) ( \
  (pos).x < SOK_LEVEL_MAX_WIDTH && \
  (pos).y < SOK_LEVEL_MAX_HEIGHT \
)

#define POINTS_EQUAL(a, b) ( \
  ((a).x == (b).x) && \
  ((a).y == (b).y) \
)

static bool
on_size(
  const sok_level_parser_t * const parser,
  const sok_pos_t pos
) {
  sok_ctx_t *ctx = (sok_ctx_t*) parser->user_data;

  if (!VALID_POS(pos)) {
    // size out of bounds
    return false;
  }

  // save width and height
  ctx->level.size = pos;

  // return success
  return true;
}

static bool
on_home(
  const sok_level_parser_t * const parser,
  const sok_pos_t pos
) {
  sok_ctx_t *ctx = (sok_ctx_t*) parser->user_data;

  if (!VALID_POS(pos)) {
    // position out of bounds
    return false;
  }

  // save start position
  ctx->level.home = pos;

  // return success
  return true;
}

static bool
on_wall(
  const sok_level_parser_t * const parser,
  const sok_pos_t pos
) {
  sok_ctx_t *ctx = (sok_ctx_t*) parser->user_data;

  if (!VALID_POS(pos)) {
    // position out of bounds
    return false;
  }

  // save wall
  ctx->level.walls[pos.y * ctx->level.size.x + pos.x] = true;

  // return sucess
  return true;
}

static bool
on_goal(
  const sok_level_parser_t * const parser,
  const sok_pos_t pos
) {
  sok_ctx_t *ctx = (sok_ctx_t*) parser->user_data;

  if (!VALID_POS(pos)) {
    // position out of bounds
    return false;
  }

  if (ctx->level.num_goals >= SOK_LEVEL_MAX_GOALS - 1) {
    // too many goals
    return false;
  }

  // save goal
  ctx->level.goals[ctx->level.num_goals++] = pos;

  // return sucess
  return true;
}

static bool
on_box(
  const sok_level_parser_t * const parser,
  const sok_pos_t pos
) {
  sok_ctx_t *ctx = (sok_ctx_t*) parser->user_data;

  if (!VALID_POS(pos)) {
    // position out of bounds
    return false;
  }

  if (ctx->level.num_boxes >= SOK_LEVEL_MAX_BOXES - 1) {
    // too many boxes
    return false;
  }

  // save box
  ctx->level.boxes[ctx->level.num_boxes++] = pos;

  // return sucess
  return true;
}

static bool
on_junk(
  const sok_level_parser_t * const parser,
  const size_t ofs,
  const char tile
) {
  UNUSED(parser);
  UNUSED(ofs);
  UNUSED(tile);

  // TODO

  // return sucess
  return true;
}

static const sok_level_parser_cbs_t
LEVEL_PARSER_CBS = {
  .on_size = on_size,
  .on_home = on_home,
  .on_wall = on_wall,
  .on_goal = on_goal,
  .on_box  = on_box,
  .on_junk = on_junk,
};

static size_t
pos_find(
  const sok_pos_t pos,
  const sok_pos_t * const set,
  const size_t len
) {
  for (size_t i = 0; i < len; i++) {
    if (POINTS_EQUAL(pos, set[i])) {
      return i;
    }
  }

  return len;
}

static size_t
sok_ctx_count_goals_left(
  sok_ctx_t * const ctx
) {
  const size_t num_boxes = ctx->level.num_boxes;
  size_t r = 0;

  for (size_t i = 0; i < ctx->level.num_goals; i++) {
    r += pos_find(
      ctx->level.goals[i],
      ctx->boxes,
      num_boxes
    ) < num_boxes ? 0 : 1;
  }

  return r;
}


void
sok_ctx_init(sok_ctx_t * const ctx, void *user_data) {
  memset(ctx, 0, sizeof(sok_ctx_t));
  ctx->user_data = user_data;
}

bool
sok_ctx_set_level(
  sok_ctx_t * const ctx,
  const char * const str
) {

  // clear level
  ctx->level.num_goals = 0;
  ctx->level.num_boxes = 0;
  memset(ctx->level.walls, 0, SOK_LEVEL_MAX_WIDTH * SOK_LEVEL_MAX_HEIGHT);

  // parse level
  sok_level_parser_t parser;
  sok_level_parser_init(&parser, &LEVEL_PARSER_CBS, ctx);
  if (!sok_level_parser_parse(&parser, str, strlen(str))) {
    // return failure
    return false;
  }

  // init moves, player, and boxes
  ctx->num_moves = 0;
  ctx->home = ctx->level.home;
  memcpy(ctx->boxes, ctx->level.boxes, ctx->level.num_boxes * sizeof(sok_pos_t));

  // count number of goals left
  ctx->num_goals_left = sok_ctx_count_goals_left(ctx);

  // return success
  return true;
}

static bool
tile_is_floor(
  const sok_ctx_t * const ctx,
  const sok_pos_t pos
) {
  return (
    (pos.x < SOK_LEVEL_MAX_WIDTH - 1) &&
    (pos.y < SOK_LEVEL_MAX_HEIGHT - 1) &&
    !(ctx->level.walls[pos.y * ctx->level.size.x + pos.x])
  );
}

static bool
tile_is_box(
  const sok_ctx_t * const ctx,
  const sok_pos_t pos
) {
  const size_t num_boxes = ctx->level.num_boxes;
  return pos_find(pos, ctx->boxes, num_boxes) < num_boxes;
}

static bool
tile_is_goal(
  const sok_ctx_t * const ctx,
  const sok_pos_t pos
) {
  const size_t num_goals = ctx->level.num_goals;
  return pos_find(pos, ctx->level.goals, num_goals) < num_goals;
}

static bool
sok_ctx_push_move(
  sok_ctx_t * const ctx,
  const sok_dir_t dir,
  const bool is_push
) {
  if (ctx->num_moves >= SOK_CTX_MAX_MOVES - 1) {
    return false;
  }

  ctx->moves[ctx->num_moves].pos = ctx->home;
  ctx->moves[ctx->num_moves].dir = dir;
  ctx->moves[ctx->num_moves].is_push = is_push;
  ctx->num_moves++;

  return true;
}

static bool
sok_ctx_pop_move(
  sok_ctx_t * const ctx,
  sok_move_t * const ret
) {
  if (!ctx->num_moves) {
    // return failure
    return false;
  }

  // decriment moves
  ctx->num_moves--;

  if (ret) {
    // save popped move
    *ret = ctx->moves[ctx->num_moves];
  }

  // return success
  return true;
}

static bool
sok_ctx_move_box(
  sok_ctx_t * const ctx,
  const sok_pos_t old_pos,
  const sok_pos_t new_pos
) {
  const size_t num_boxes = ctx->level.num_boxes,
               box_ofs = pos_find(old_pos, ctx->boxes, num_boxes);

  if (box_ofs < num_boxes) {
    // update box and goal count
    ctx->boxes[box_ofs] = new_pos;
    ctx->num_goals_left = sok_ctx_count_goals_left(ctx);

    // return success
    return true;
  }

  // return failure
  return false;
}

bool
sok_ctx_is_done(
  sok_ctx_t * const ctx
) {
  return ctx->num_goals_left == 0;
}

bool
sok_ctx_move(
  sok_ctx_t * const ctx,
  const sok_dir_t dir
) {
  if (ctx->num_moves >= SOK_CTX_MAX_MOVES - 1) {
    // no more move slots available
    return false;
  }

  const sok_pos_t ps[2] = {{
    .x = ctx->home.x + ((dir == SOK_DIR_LEFT) ? -1 : ((dir == SOK_DIR_RIGHT) ? 1 : 0)),
    .y = ctx->home.y + ((dir == SOK_DIR_UP) ? -1 : ((dir == SOK_DIR_DOWN) ? 1 : 0))
  }, {
    .x = ctx->home.x + ((dir == SOK_DIR_LEFT) ? -2 : ((dir == SOK_DIR_RIGHT) ? 2 : 0)),
    .y = ctx->home.y + ((dir == SOK_DIR_UP) ? -2 : ((dir == SOK_DIR_DOWN) ? 2 : 0))
  }};

  const bool can_move = (
    (dir == SOK_DIR_UP) ? (ctx->home.y > 0) : 
    ((dir == SOK_DIR_DOWN) ? (ctx->home.y < SOK_LEVEL_MAX_HEIGHT - 1) :
    ((dir == SOK_DIR_LEFT) ? (ctx->home.x > 0) : 
    ((dir == SOK_DIR_RIGHT) ? (ctx->home.x < SOK_LEVEL_MAX_WIDTH - 1) : 
    false
  ))));

  const bool can_push = (
    (dir == SOK_DIR_UP) ? (ctx->home.y > 1) : 
    ((dir == SOK_DIR_DOWN) ? (ctx->home.y < SOK_LEVEL_MAX_HEIGHT - 2) :
    ((dir == SOK_DIR_LEFT) ? (ctx->home.x > 1) : 
    ((dir == SOK_DIR_RIGHT) ? (ctx->home.x < SOK_LEVEL_MAX_WIDTH - 2) : 
    false
  ))));

  if (
    can_move &&
    tile_is_floor(ctx, ps[0]) &&
    !tile_is_box(ctx, ps[0])
  ) { 
    // push move
    if (!sok_ctx_push_move(ctx, dir, false)) {
      return false;
    }

    // update position
    ctx->home.x += (dir == SOK_DIR_LEFT) ? -1 : ((dir == SOK_DIR_RIGHT) ? 1 : 0);
    ctx->home.y += (dir == SOK_DIR_UP) ? -1 : ((dir == SOK_DIR_DOWN) ? 1 : 0);

    // return success
    return true;
  } else if (
    can_push &&
    tile_is_floor(ctx, ps[0]) &&
    tile_is_box(ctx, ps[0]) &&
    tile_is_floor(ctx, ps[1]) &&
    !tile_is_box(ctx, ps[1])
  ) { 
    // push move
    if (!sok_ctx_push_move(ctx, dir, true)) {
      return false;
    }

    // move box
    if (!sok_ctx_move_box(ctx, ps[0], ps[1])) {
      return false;
    }

    // update position
    ctx->home.x += (dir == SOK_DIR_LEFT) ? -1 : ((dir == SOK_DIR_RIGHT) ? 1 : 0);
    ctx->home.y += (dir == SOK_DIR_UP) ? -1 : ((dir == SOK_DIR_DOWN) ? 1 : 0);


    // return success
    return true;
  }

  // return failure
  return false;
}

bool
sok_ctx_undo(
  sok_ctx_t * const ctx
) {
  sok_move_t move;
  if (!sok_ctx_pop_move(ctx, &move)) {
    // return failure
    return false;
  }

  if (move.is_push) {
    const sok_pos_t box_pos = {
      .x = move.pos.x + ((move.dir == SOK_DIR_LEFT) ? -2 : ((move.dir == SOK_DIR_RIGHT) ? 2 : 0)),
      .y = move.pos.y + ((move.dir == SOK_DIR_UP) ? -2 : ((move.dir == SOK_DIR_DOWN) ? 2 : 0)),
    };

    sok_ctx_move_box(ctx, box_pos, ctx->home);
  }

  // update position
  ctx->home = move.pos;

  // return success
  return true;
}

bool sok_ctx_walk(
  const sok_ctx_t * const ctx,
  const sok_ctx_walk_cbs_t * const cbs,
  void * const data
) {
  if (!cbs) {
    return false;
  }

  if (cbs->on_size) {
    // emit size
    if (!cbs->on_size(ctx, ctx->level.size, data)) {
      return false;
    }
  }

  if (cbs->on_home) {
    const bool has_goal = tile_is_goal(ctx, ctx->home);

    // emit home
    if (!cbs->on_home(ctx, ctx->home, has_goal, data)) {
      return false;
    }
  }

  if (cbs->on_wall) {
    // walk walls
    for (size_t i = 0; i < (ctx->level.size.x * ctx->level.size.y); i++) {
      if (ctx->level.walls[i]) {
        const sok_pos_t pos = {
          .y = i / ctx->level.size.x,
          .x = i % ctx->level.size.x,
        };

        // emit wall
        if (!cbs->on_wall(ctx, pos, data)) {
          return false;
        }
      }
    }
  }

  if (cbs->on_goal) {
    // walk goals
    for (size_t i = 0; i < ctx->level.num_goals; i++) {
      const bool has_player = POINTS_EQUAL(ctx->level.goals[i], ctx->home);
      const bool has_box = tile_is_box(ctx, ctx->level.goals[i]);

      // emit goal
      if (!cbs->on_goal(ctx, ctx->level.goals[i], has_player, has_box, data)) {
        return false;
      }
    }
  }

  if (cbs->on_box) {
    // walk boxes
    for (size_t i = 0; i < ctx->level.num_boxes; i++) {
      const bool has_goal = tile_is_goal(ctx, ctx->boxes[i]);

      // emit box
      if (!cbs->on_box(ctx, ctx->boxes[i], has_goal, data)) {
        return false;
      }
    }
  }

  if (cbs->on_move) {
    // walk moves
    for (size_t i = 0; i < ctx->num_moves; i++) {
      // emit move
      if (!cbs->on_move(ctx, ctx->moves[i], data)) {
        return false;
      }
    }
  }

  // return success
  return true;
}