slangin/slangin.ino

/*
Slangin'
Author: Rudis Muiznieks
License: WTFPL
*/

#include <Arduboy2.h>
#include "src/fonts/Font3x5.h"
#include "src/fonts/Font4x6.h"

Arduboy2 arduboy;
Font4x6 font4x6 = Font4x6();
Font3x5 font3x5 = Font3x5();

enum GameState {
  STATE_TITLE = 0,
  STATE_TURN_MENU,
  STATE_JET_MENU,
  STATE_INVENTORY,
  STATE_BUY_MENU,
  STATE_SELL_MENU,
  STATE_FIGHT_MENU,
  STATE_SHARK_MENU,
  STATE_BANK_MENU,
  STATE_INFO_DIALOG,
  STATE_QUESTION_DIALOG,
  STATE_AMOUNT_DIALOG,
  STATE_SUBWAY
};

enum GameLocation {
  LOC_BRONX = 0,
  LOC_GHETTO,
  LOC_CENTRAL_PARK,
  LOC_MANHATTEN,
  LOC_CONEY_ISLAND,
  LOC_BROOKLYN
};

enum Drug {
  DRUG_COCAINE = 0,
  DRUG_HEROINE,
  DRUG_ACID,
  DRUG_WEED,
  DRUG_SPEED,
  DRUG_LUDES
};

enum DialogType {
  DIALOG_INFO = 0,
  DIALOG_YESNO,
  DIALOG_AMOUNT
};

const uint8_t PROGMEM spriteDollar[] = {
  5, 5,
  0x12, 0x15, 0x1f, 0x15, 0x09,
};

/**************/
/* game state */
/**************/
bool screenInitialized;
GameState sGameState;
GameState sPreviousGameState;
int sCurrentDay;
int sDrugPrices[6];
int sRandomEvent;

/****************/
/* player state */
/****************/
int pMoney;
int pLoanAmount;
int pSavingsAmount;
GameLocation pLocation;
int pGuns;
int pCapacity;
int pHealth;
int pInventory[6];

String menu[8];
int menuLength = 0;
bool menuCols = false;
bool menuSmall = false;
int menuSelected = 0;

/***********************/
/* game setup and loop */
/***********************/
void setup() {
  sGameState = STATE_TITLE;
  screenInitialized = false;
  arduboy.begin();
  arduboy.setFrameRate(15);
}

void loop() {
  if (!(arduboy.nextFrame()))
    return;

  // draw screen if needed
  if (!screenInitialized) {
    arduboy.clear();
    menuSelected = 0;
    switch (sGameState) {
      case STATE_TITLE:
        arduboy.setCursor(25, 10);
        arduboy.print(F("Slangin' v0.9"));
        menu[0] = F("New Game");
        menuLength = 1;
        menuSmall = false;
        menuCols = true;
        drawMenu(menuSmall, menuCols, menuLength, menu);
        drawMenuIndicator(menuSelected, menuSmall, menuCols, menuLength, menu, false);
        break;

      case STATE_TURN_MENU:
        drawStatusBar();
        drawTitle(lookupLocation(pLocation));
        menu[0] = F("Trenchcoat");
        menu[1] = F("Buy Drugs");
        menu[2] = F("Sell Drugs");
        menu[3] = F("Jet");
        if (pLocation == LOC_BRONX) {
          menu[4] = F("Loan Shark");
          menu[5] = F("Bank");
          menuLength = 6;
        } else {
          menuLength = 4;
        }
        menuSmall = false;
        menuCols = true;
        drawMenu(menuSmall, menuCols, menuLength, menu);
        drawMenuIndicator(menuSelected, menuSmall, menuCols, menuLength, menu, false);
        break;

      case STATE_INVENTORY:
        drawStatusBar();
        drawTitle(F("Trenchcoat"));
        break;
    }
    screenInitialized = true;
  }

  // handle user input
  arduboy.pollButtons();
  switch (sGameState) {
    // menu screens
    case STATE_TITLE:
    case STATE_TURN_MENU:
    case STATE_JET_MENU:
    case STATE_BUY_MENU:
    case STATE_SELL_MENU:
    case STATE_FIGHT_MENU:
    case STATE_SHARK_MENU:
    case STATE_BANK_MENU:
      drawMenuIndicator(menuSelected, menuSmall, menuCols, menuLength, menu, true);
      int col1Count = round(menuLength / 2.0);
      int inCol = menuSelected < col1Count ? 1 : 2;

      if (arduboy.justPressed(DOWN_BUTTON)) {
        menuSelected++;
        if (menuSelected >= menuLength) menuSelected = menuLength - 1;

      } else if (arduboy.justPressed(UP_BUTTON)) {
        menuSelected--;
        if (menuSelected < 0) menuSelected = 0;

      } else if (arduboy.justPressed(RIGHT_BUTTON)) {
        if (menuCols && inCol == 1) {
          menuSelected += col1Count;
        } else {
          menuSelected++;
        }
        if (menuSelected >= menuLength) menuSelected = menuLength - 1;

      } else if (arduboy.justPressed(LEFT_BUTTON)) {
        if (menuCols && inCol == 2) {
          menuSelected -= col1Count;
        } else {
          menuSelected--;
        }
        if (menuSelected < 0) menuSelected = 0;

      } else if (arduboy.justPressed(A_BUTTON)) {

        // advance to next state
        handleMenuAction();

      } else if (arduboy.justPressed(B_BUTTON)) {
        // return to turn menu from the other menus
        if (sGameState == STATE_JET_MENU ||
            sGameState == STATE_BUY_MENU ||
            sGameState == STATE_SELL_MENU ||
            sGameState == STATE_SHARK_MENU ||
            sGameState == STATE_BANK_MENU) {
          sGameState = STATE_TURN_MENU;
        }
      }
      drawMenuIndicator(menuSelected, menuSmall, menuCols, menuLength, menu, false);
      break;
    case STATE_INVENTORY:
      // TODO: not working
      if (arduboy.justPressed(B_BUTTON)) {
        screenInitialized = false;
        sGameState = STATE_TURN_MENU;
      }
      break;
  }

  arduboy.display();
}

/***********************/
/* screen draw heplers */
/***********************/
void drawStatusBar() {
  const int chars = log10(pMoney);
  const int x = 128 - (chars * 4);
  Sprites::drawOverwrite(x - 6, 1, spriteDollar, 0);
  arduboy.fillRect(0, 8, 128, 1, WHITE);
  font3x5.setCursor(1, 0);
  font3x5.print("Day");
  font3x5.setCursor(14, 0);
  font3x5.print(sCurrentDay);
  font3x5.setCursor(x, 0);
  font3x5.print(pMoney);
}

void drawTitle(const String title) {
  font4x6.setCursor(6, 11);
  font4x6.print(title);
}