gba-link-connection/lib/LinkCable.h
2023-02-23 15:22:55 -03:00

419 lines
11 KiB
C++

#ifndef LINK_CABLE_H
#define LINK_CABLE_H
// --------------------------------------------------------------------------
// A Link Cable connection for Multi-Play mode.
// --------------------------------------------------------------------------
// Usage:
// - 1) Include this header in your main.cpp file and add:
// LinkCable* linkCable = new LinkCable();
// - 2) Add the required interrupt service routines: (*)
// irq_init(NULL);
// irq_add(II_VBLANK, LINK_CABLE_ISR_VBLANK);
// irq_add(II_SERIAL, LINK_CABLE_ISR_SERIAL);
// irq_add(II_TIMER3, LINK_CABLE_ISR_TIMER);
// - 3) Initialize the library with:
// linkCable->activate();
// - 4) Send/read messages by using:
// bool isConnected = linkCable->isConnected();
// u8 playerCount = linkCable->playerCount();
// u8 currentPlayerId = linkCable->currentPlayerId();
// linkCable->send(0x1234);
// if (isConnected && linkCable->canRead(!currentPlayerId)) {
// u16 message = linkCable->read(!currentPlayerId);
// // ...
// }
// - 5) Mark the current state copy (front buffer) as consumed:
// linkCable->consume();
// // (put this line at the end of your game loop)
// --------------------------------------------------------------------------
// (*) libtonc's interrupt handler sometimes ignores interrupts due to a bug.
// That can cause packet loss. You might want to use libugba's instead.
// (see examples)
// --------------------------------------------------------------------------
// `send(...)` restrictions:
// - 0xFFFF and 0x0 are reserved values, so don't send them!
// (they mean 'disconnected' and 'no data' respectively)
// --------------------------------------------------------------------------
#include <tonc_core.h>
// Buffer size
#define LINK_CABLE_QUEUE_SIZE 30
#define LINK_CABLE_MAX_PLAYERS 4
#define LINK_CABLE_DISCONNECTED 0xFFFF
#define LINK_CABLE_NO_DATA 0x0
#define LINK_CABLE_DEFAULT_TIMEOUT 3
#define LINK_CABLE_DEFAULT_REMOTE_TIMEOUT 5
#define LINK_CABLE_DEFAULT_INTERVAL 50
#define LINK_CABLE_DEFAULT_SEND_TIMER_ID 3
#define LINK_CABLE_BASE_FREQUENCY TM_FREQ_1024
#define LINK_CABLE_REMOTE_TIMEOUT_OFFLINE -1
#define LINK_CABLE_BIT_SLAVE 2
#define LINK_CABLE_BIT_READY 3
#define LINK_CABLE_BITS_PLAYER_ID 4
#define LINK_CABLE_BIT_ERROR 6
#define LINK_CABLE_BIT_START 7
#define LINK_CABLE_BIT_MULTIPLAYER 13
#define LINK_CABLE_BIT_IRQ 14
#define LINK_CABLE_BIT_GENERAL_PURPOSE_LOW 14
#define LINK_CABLE_BIT_GENERAL_PURPOSE_HIGH 15
#define LINK_CABLE_SET_HIGH(REG, BIT) REG |= 1 << BIT
#define LINK_CABLE_SET_LOW(REG, BIT) REG &= ~(1 << BIT)
#define LINK_CABLE_BARRIER asm volatile("" ::: "memory")
static volatile char LINK_CABLE_VERSION[] = "LinkCable/v5.0.2";
void LINK_CABLE_ISR_VBLANK();
void LINK_CABLE_ISR_SERIAL();
void LINK_CABLE_ISR_TIMER();
const u16 LINK_CABLE_TIMER_IRQ_IDS[] = {IRQ_TIMER0, IRQ_TIMER1, IRQ_TIMER2,
IRQ_TIMER3};
class LinkCable {
public:
enum BaudRate {
BAUD_RATE_0, // 9600 bps
BAUD_RATE_1, // 38400 bps
BAUD_RATE_2, // 57600 bps
BAUD_RATE_3 // 115200 bps
};
class U16Queue {
public:
void push(u16 item) {
if (isFull())
pop();
rear = (rear + 1) % LINK_CABLE_QUEUE_SIZE;
arr[rear] = item;
count++;
}
u16 pop() {
if (isEmpty())
return LINK_CABLE_NO_DATA;
auto x = arr[front];
front = (front + 1) % LINK_CABLE_QUEUE_SIZE;
count--;
return x;
}
void clear() {
while (!isEmpty())
pop();
}
int size() { return count; }
bool isEmpty() { return size() == 0; }
bool isFull() { return size() == LINK_CABLE_QUEUE_SIZE; }
private:
u16 arr[LINK_CABLE_QUEUE_SIZE];
vs32 front = 0;
vs32 rear = -1;
vu32 count = 0;
};
explicit LinkCable(BaudRate baudRate = BAUD_RATE_1,
u32 timeout = LINK_CABLE_DEFAULT_TIMEOUT,
u32 remoteTimeout = LINK_CABLE_DEFAULT_REMOTE_TIMEOUT,
u16 interval = LINK_CABLE_DEFAULT_INTERVAL,
u8 sendTimerId = LINK_CABLE_DEFAULT_SEND_TIMER_ID) {
this->config.baudRate = baudRate;
this->config.timeout = timeout;
this->config.remoteTimeout = remoteTimeout;
this->config.interval = interval;
this->config.sendTimerId = sendTimerId;
}
bool isActive() { return isEnabled; }
void activate() {
reset();
isEnabled = true;
}
void deactivate() {
isEnabled = false;
isStateReady = false;
isStateConsumed = false;
isResetting = false;
resetState();
stop();
}
bool isConnected() {
return $state.playerCount > 1 &&
$state.currentPlayerId < $state.playerCount;
}
u8 playerCount() { return $state.playerCount; }
u8 currentPlayerId() { return $state.currentPlayerId; }
bool canRead(u8 playerId) {
if (!isStateReady || isStateConsumed)
return false;
LINK_CABLE_BARRIER;
return !$state.incomingMessages[playerId].isEmpty();
}
u16 read(u8 playerId) {
if (!isStateReady || isStateConsumed)
return LINK_CABLE_NO_DATA;
LINK_CABLE_BARRIER;
return $state.incomingMessages[playerId].pop();
}
void consume() { isStateConsumed = true; }
void send(u16 data) {
if (data == LINK_CABLE_DISCONNECTED || data == LINK_CABLE_NO_DATA)
return;
LINK_CABLE_BARRIER;
isAddingMessage = true;
LINK_CABLE_BARRIER;
_state.outgoingMessages.push(data);
LINK_CABLE_BARRIER;
isAddingMessage = false;
LINK_CABLE_BARRIER;
if (isResetting) {
_state.outgoingMessages.clear();
isResetting = false;
}
}
void _onVBlank() {
if (!isEnabled)
return;
if (!_state.IRQFlag)
_state.IRQTimeout++;
_state.IRQFlag = false;
copyState();
}
void _onSerial() {
if (!isEnabled)
return;
if (resetIfNeeded()) {
copyState();
return;
}
_state.IRQFlag = true;
_state.IRQTimeout = 0;
u8 newPlayerCount = 0;
for (u32 i = 0; i < LINK_CABLE_MAX_PLAYERS; i++) {
u16 data = REG_SIOMULTI[i];
if (data != LINK_CABLE_DISCONNECTED) {
if (data != LINK_CABLE_NO_DATA && i != state.currentPlayerId)
state.incomingMessages[i].push(data);
newPlayerCount++;
_state.timeouts[i] = 0;
} else if (_state.timeouts[i] > LINK_CABLE_REMOTE_TIMEOUT_OFFLINE) {
_state.timeouts[i]++;
if (_state.timeouts[i] >= (int)config.remoteTimeout) {
state.incomingMessages[i].clear();
_state.timeouts[i] = LINK_CABLE_REMOTE_TIMEOUT_OFFLINE;
} else
newPlayerCount++;
}
}
state.playerCount = newPlayerCount;
state.currentPlayerId =
(REG_SIOCNT & (0b11 << LINK_CABLE_BITS_PLAYER_ID)) >>
LINK_CABLE_BITS_PLAYER_ID;
if (!isMaster())
sendPendingData();
copyState();
}
void _onTimer() {
if (!isEnabled)
return;
if (didTimeout()) {
reset();
copyState();
return;
}
if (isMaster() && isReady() && !isSending())
sendPendingData();
copyState();
}
private:
struct Config {
BaudRate baudRate;
u32 timeout;
u32 remoteTimeout;
u32 interval;
u8 sendTimerId;
};
struct ExternalState {
U16Queue incomingMessages[LINK_CABLE_MAX_PLAYERS];
u8 playerCount;
u8 currentPlayerId;
};
struct InternalState {
U16Queue outgoingMessages;
int timeouts[LINK_CABLE_MAX_PLAYERS];
bool IRQFlag;
u32 IRQTimeout;
};
ExternalState state; // (updated state / back buffer)
ExternalState $state; // (visible state / front buffer)
InternalState _state; // (internal state)
Config config;
bool isEnabled = false;
volatile bool isStateReady = false;
volatile bool isStateConsumed = false;
volatile bool isAddingMessage = false;
volatile bool isResetting = false;
bool isReady() { return isBitHigh(LINK_CABLE_BIT_READY); }
bool hasError() { return isBitHigh(LINK_CABLE_BIT_ERROR); }
bool isMaster() { return !isBitHigh(LINK_CABLE_BIT_SLAVE); }
bool isSending() { return isBitHigh(LINK_CABLE_BIT_START); }
bool didTimeout() { return _state.IRQTimeout >= config.timeout; }
void sendPendingData() {
if (isAddingMessage)
return;
LINK_CABLE_BARRIER;
transfer(_state.outgoingMessages.pop());
}
void transfer(u16 data) {
REG_SIOMLT_SEND = data;
if (isMaster())
setBitHigh(LINK_CABLE_BIT_START);
}
bool resetIfNeeded() {
if (!isReady() || hasError()) {
reset();
return true;
}
return false;
}
void reset() {
resetState();
stop();
start();
}
void resetState() {
state.playerCount = 0;
state.currentPlayerId = 0;
for (u32 i = 0; i < LINK_CABLE_MAX_PLAYERS; i++) {
state.incomingMessages[i].clear();
_state.timeouts[i] = LINK_CABLE_REMOTE_TIMEOUT_OFFLINE;
}
_state.IRQFlag = false;
_state.IRQTimeout = 0;
if (isAddingMessage || isResetting)
isResetting = true;
else
_state.outgoingMessages.clear();
}
void stop() {
stopTimer();
LINK_CABLE_SET_LOW(REG_RCNT, LINK_CABLE_BIT_GENERAL_PURPOSE_LOW);
LINK_CABLE_SET_HIGH(REG_RCNT, LINK_CABLE_BIT_GENERAL_PURPOSE_HIGH);
}
void start() {
startTimer();
LINK_CABLE_SET_LOW(REG_RCNT, LINK_CABLE_BIT_GENERAL_PURPOSE_HIGH);
REG_SIOCNT = config.baudRate;
REG_SIOMLT_SEND = 0;
setBitHigh(LINK_CABLE_BIT_MULTIPLAYER);
setBitHigh(LINK_CABLE_BIT_IRQ);
}
void stopTimer() {
REG_TM[config.sendTimerId].cnt =
REG_TM[config.sendTimerId].cnt & (~TM_ENABLE);
}
void startTimer() {
REG_TM[config.sendTimerId].start = -config.interval;
REG_TM[config.sendTimerId].cnt =
TM_ENABLE | TM_IRQ | LINK_CABLE_BASE_FREQUENCY;
}
void copyState() {
if (isStateReady && !isStateConsumed)
return;
LINK_CABLE_BARRIER;
$state.playerCount = state.playerCount;
$state.currentPlayerId = state.currentPlayerId;
for (u32 i = 0; i < LINK_CABLE_MAX_PLAYERS; i++) {
$state.incomingMessages[i].clear();
while (!state.incomingMessages[i].isEmpty())
$state.incomingMessages[i].push(state.incomingMessages[i].pop());
}
LINK_CABLE_BARRIER;
isStateReady = true;
isStateConsumed = false;
LINK_CABLE_BARRIER;
}
bool isBitHigh(u8 bit) { return (REG_SIOCNT >> bit) & 1; }
void setBitHigh(u8 bit) { LINK_CABLE_SET_HIGH(REG_SIOCNT, bit); }
void setBitLow(u8 bit) { LINK_CABLE_SET_LOW(REG_SIOCNT, bit); }
};
extern LinkCable* linkCable;
inline void LINK_CABLE_ISR_VBLANK() {
linkCable->_onVBlank();
}
inline void LINK_CABLE_ISR_SERIAL() {
linkCable->_onSerial();
}
inline void LINK_CABLE_ISR_TIMER() {
linkCable->_onTimer();
}
#endif // LINK_CABLE_H