Mesen2/Core/NES/Mappers/FDS/Fds.cpp

622 lines
18 KiB
C++

#include "pch.h"
#include <assert.h>
#include "NES/NesConsole.h"
#include "NES/NesCpu.h"
#include "NES/NesPpu.h"
#include "NES/NesMemoryManager.h"
#include "NES/Mappers/FDS/Fds.h"
#include "NES/Mappers/FDS/FdsAudio.h"
#include "NES/Mappers/FDS/FdsInputButtons.h"
#include "Shared/EmuSettings.h"
#include "Shared/BatteryManager.h"
#include "Shared/BaseControlManager.h"
#include "Shared/Movies/MovieManager.h"
#include "Shared/NotificationManager.h"
#include "Shared/FirmwareHelper.h"
#include "Utilities/Patches/IpsPatcher.h"
#include "Utilities/Serializer.h"
#include "Utilities/StringUtilities.h"
void Fds::InitMapper()
{
//Replace PRG ROM data with FDS bios, if available (otherwise set an empty 8kb block)
_prgSize = 0x2000;
delete[] _prgRom;
if(!FirmwareHelper::LoadFdsFirmware(_emu, &_prgRom)) {
_prgRom = new uint8_t[_prgSize];
memset(_prgRom, 0, _prgSize);
}
_emu->RegisterMemory(MemoryType::NesPrgRom, _prgRom, _prgSize);
_settings = &_console->GetNesConfig();
_cpu = _console->GetCpu();
_memoryManager = _console->GetMemoryManager();
//FDS BIOS
SetCpuMemoryMapping(0xE000, 0xFFFF, 0, PrgMemoryType::PrgRom, MemoryAccessType::Read);
//Work RAM
SetCpuMemoryMapping(0x6000, 0xDFFF, 0, PrgMemoryType::WorkRam, MemoryAccessType::ReadWrite);
//8k of CHR RAM
SelectChrPage(0, 0);
}
void Fds::InitMapper(RomData &romData)
{
_audio.reset(new FdsAudio(_console));
_fdsDiskSides = romData.FdsDiskData;
_fdsDiskHeaders = romData.FdsDiskHeaders;
_fdsRawData = romData.RawData;
string filename = StringUtilities::ToLower(romData.Info.Filename);
_useQdFormat = StringUtilities::EndsWith(filename, ".qd");
FdsLoader loader(_useQdFormat);
loader.LoadDiskData(_fdsRawData, _orgDiskSides, _orgDiskHeaders);
//Apply save data (saved as an IPS file), if found
vector<uint8_t> ipsData = _emu->GetBatteryManager()->LoadBattery(".ips");
LoadDiskData(ipsData);
_input.reset(new FdsInputButtons(this, _emu));
_emu->GetNotificationManager()->RegisterNotificationListener(_input);
_console->GetControlManager()->AddSystemControlDevice(_input);
if(_settings->FdsAutoLoadDisk) {
_input->InsertDisk(0);
}
}
void Fds::LoadDiskData(vector<uint8_t> ipsData)
{
_fdsDiskSides.clear();
_fdsDiskHeaders.clear();
FdsLoader loader(_useQdFormat);
if(_settings->OverwriteOriginalRom) {
loader.LoadDiskData(_fdsRawData, _fdsDiskSides, _fdsDiskHeaders);
} else {
vector<uint8_t> patchedData;
if(ipsData.size() > 0 && IpsPatcher::PatchBuffer(ipsData, _fdsRawData, patchedData)) {
loader.LoadDiskData(patchedData, _fdsDiskSides, _fdsDiskHeaders);
} else {
loader.LoadDiskData(_fdsRawData, _fdsDiskSides, _fdsDiskHeaders);
}
}
}
void Fds::SaveBattery()
{
if(_needSave) {
FdsLoader loader(_useQdFormat);
bool needHeader = (memcmp(_fdsRawData.data(), "FDS\x1a", 4) == 0);
vector<uint8_t> newData = loader.RebuildFdsFile(_fdsDiskSides, needHeader);
if(_settings->OverwriteOriginalRom) {
ofstream file(_emu->GetRomInfo().RomFile, ios::binary);
file.write((char*)newData.data(), newData.size());
} else {
vector<uint8_t> ipsData = IpsPatcher::CreatePatch(_fdsRawData, newData);
_emu->GetBatteryManager()->SaveBattery(".ips", ipsData.data(), (uint32_t)ipsData.size());
}
_needSave = false;
}
}
void Fds::Reset(bool softReset)
{
_autoDiskEjectCounter = -1;
_autoDiskSwitchCounter = -1;
_disableAutoInsertDisk = false;
_gameStarted = false;
}
uint32_t Fds::GetFdsDiskSideSize(uint8_t side)
{
assert(side < _fdsDiskSides.size());
return (uint32_t)_fdsDiskSides[side].size();
}
uint8_t Fds::ReadFdsDisk()
{
assert(_diskNumber < _fdsDiskSides.size());
assert(_diskPosition < _fdsDiskSides[_diskNumber].size());
return _fdsDiskSides[_diskNumber][_diskPosition];
}
void Fds::WriteFdsDisk(uint8_t value)
{
assert(_diskNumber < _fdsDiskSides.size());
assert(_diskPosition < _fdsDiskSides[_diskNumber].size());
uint8_t currentValue = _fdsDiskSides[_diskNumber][_diskPosition];
if(currentValue != value) {
_fdsDiskSides[_diskNumber][_diskPosition] = value;
_needSave = true;
}
}
void Fds::ClockIrq()
{
if(_irqEnabled) {
if(_irqCounter == 0) {
_cpu->SetIrqSource(IRQSource::External);
_irqCounter = _irqReloadValue;
if(!_irqRepeatEnabled) {
_irqEnabled = false;
}
} else {
_irqCounter--;
}
}
}
uint8_t Fds::ReadRam(uint16_t addr)
{
if(addr == 0xE18C && !_gameStarted && (_memoryManager->DebugRead(0x100) & 0xC0) != 0) {
//$E18B is the NMI entry point (using $E18C due to dummy reads)
//When NMI occurs while $100 & $C0 != 0, it typically means that the game is starting.
_gameStarted = true;
} else if(addr == 0xE445 && IsAutoInsertDiskEnabled()) {
//Game is trying to check if a specific disk/side is inserted
//Find the matching disk and insert it automatically
uint16_t bufferAddr = _memoryManager->DebugReadWord(0);
uint8_t buffer[10];
for(int i = 0; i < 10; i++) {
//Prevent infinite recursion
if(bufferAddr + i != 0xE445) {
buffer[i] = _memoryManager->DebugRead(bufferAddr + i);
} else {
buffer[i] = 0;
}
}
int matchCount = 0;
int matchIndex = -1;
for(int j = 0; j < (int)_fdsDiskHeaders.size(); j++) {
bool match = true;
for(int i = 0; i < 10; i++) {
if(buffer[i] != 0xFF && buffer[i] != _fdsDiskHeaders[j][i + 14]) {
match = false;
break;
}
}
if(match) {
matchCount++;
matchIndex = matchCount > 1 ? -1 : j;
}
}
if(matchCount > 1) {
//More than 1 disk matches, can happen in unlicensed games - disable auto insert logic
_disableAutoInsertDisk = true;
}
if(matchIndex >= 0) {
//Found a single match, insert it
_diskNumber = matchIndex;
if(_diskNumber != _previousDiskNumber) {
MessageManager::Log("[FDS] Disk automatically inserted: Disk " + std::to_string((_diskNumber / 2) + 1) + ((_diskNumber & 0x01) ? " Side B" : " Side A"));
_previousDiskNumber = _diskNumber;
}
if(matchIndex > 0) {
//Make sure we disable fast forward
_gameStarted = true;
}
}
//Prevent disk from being switched again until the disk is actually read
_autoDiskSwitchCounter = -1;
_restartAutoInsertCounter = -1;
}
return BaseMapper::ReadRam(addr);
}
void Fds::ProcessAutoDiskInsert()
{
if(IsAutoInsertDiskEnabled()) {
bool fastForwardEnabled = _settings->FdsFastForwardOnLoad;
uint32_t currentFrame = _emu->GetFrameCount();
if(_previousFrame != currentFrame) {
_previousFrame = currentFrame;
if(_autoDiskEjectCounter > 0) {
//After reading a disk, wait until this counter reaches 0 before
//automatically ejecting the disk the next time $4032 is read
_autoDiskEjectCounter--;
_emu->GetSettings()->SetFlagState(EmulationFlags::MaximumSpeed, fastForwardEnabled && _autoDiskEjectCounter != 0);
} else if(_autoDiskSwitchCounter > 0) {
//After ejecting the disk, wait a bit before we insert a new one
_autoDiskSwitchCounter--;
_emu->GetSettings()->SetFlagState(EmulationFlags::MaximumSpeed, fastForwardEnabled && _autoDiskSwitchCounter != 0);
if(_autoDiskSwitchCounter == 0) {
//Insert a disk (real disk/side will be selected when game executes $E445
MessageManager::Log("[FDS] Auto-inserted dummy disk.");
InsertDisk(0);
//Restart process after 200 frames if the game hasn't read the disk yet
_restartAutoInsertCounter = 200;
}
} else if(_restartAutoInsertCounter > 0) {
//After ejecting the disk, wait a bit before we insert a new one
_restartAutoInsertCounter--;
_emu->GetSettings()->SetFlagState(EmulationFlags::MaximumSpeed, fastForwardEnabled && _restartAutoInsertCounter != 0);
if(_restartAutoInsertCounter == 0) {
//Wait a bit before ejecting the disk (eject in ~34 frames)
MessageManager::Log("[FDS] Game failed to load disk, try again.");
_previousDiskNumber = Fds::NoDiskInserted;
_autoDiskEjectCounter = 34;
_autoDiskSwitchCounter = -1;
}
}
}
}
}
void Fds::ProcessCpuClock()
{
BaseProcessCpuClock();
if(_settings->FdsFastForwardOnLoad) {
_emu->GetSettings()->SetFlagState(EmulationFlags::MaximumSpeed, _scanningDisk || !_gameStarted);
} else {
_emu->GetSettings()->ClearFlag(EmulationFlags::MaximumSpeed);
}
ProcessAutoDiskInsert();
ClockIrq();
_audio->Clock();
if(_diskNumber == Fds::NoDiskInserted || !_motorOn) {
//Disk has been ejected
_endOfHead = true;
_scanningDisk = false;
return;
}
if(_resetTransfer && !_scanningDisk) {
return;
}
if(_endOfHead) {
_delay = 50000;
_endOfHead = false;
_diskPosition = 0;
_gapEnded = false;
return;
}
if(_delay > 0) {
_delay--;
} else {
_scanningDisk = true;
_autoDiskEjectCounter = -1;
_autoDiskSwitchCounter = -1;
uint8_t diskData = 0;
bool needIrq = _diskIrqEnabled;
if(_readMode) {
diskData = ReadFdsDisk();
if(!_previousCrcControlFlag) {
UpdateCrc(diskData);
}
if(!_diskReady) {
_gapEnded = false;
_crcAccumulator = 0;
_badCrc = false;
} else if(diskData && !_gapEnded) {
_gapEnded = true;
needIrq = false;
}
if(_gapEnded) {
_transferComplete = true;
_readDataReg = diskData;
if(needIrq) {
_cpu->SetIrqSource(IRQSource::FdsDisk);
}
}
if(!_previousCrcControlFlag && _crcControl) {
_badCrc = _crcAccumulator != 0;
}
} else {
if(!_crcControl) {
_transferComplete = true;
diskData = _writeDataReg;
if(needIrq) {
_cpu->SetIrqSource(IRQSource::FdsDisk);
}
}
if(!_diskReady) {
diskData = 0x00;
_crcAccumulator = 0;
}
if(!_crcControl) {
UpdateCrc(diskData);
} else {
diskData = _crcAccumulator & 0xFF;
_crcAccumulator >>= 8;
}
WriteFdsDisk(diskData);
_gapEnded = false;
_badCrc = false;
}
_previousCrcControlFlag = _crcControl;
_diskPosition++;
if(_diskPosition >= GetFdsDiskSideSize(_diskNumber)) {
_motorOn = false;
if(_diskIrqEnabled) {
//"Kosodate Gokko" disk copier seems to expect an IRQ to happen
//around the time the drive reaches the end of the disk
//Without this, the software locks up on a black screen
_cpu->SetIrqSource(IRQSource::FdsDisk);
}
//Wait a bit before ejecting the disk (eject in ~77 frames)
_autoDiskEjectCounter = 77;
} else {
//This delay used to be 150, but this triggers a bug in Ai Senshi Nicol
//during the transition from level 2 to 3 - the 150 value causes an NMI
//to occur between 2 writes to $2006 (vram addr), which ends up breaking
//the PPU update logic and causes broken graphics when stage 3 loads
//Both 149 or 151 fix the issue because they change the timing enough
//that the NMI no longer interrupts the vram update routine.
_delay = 149;
}
}
}
void Fds::UpdateCrc(uint8_t value)
{
_crcAccumulator ^= value;
for(uint16_t n = 0; n < 8; n++) {
uint8_t carry = (_crcAccumulator & 1);
_crcAccumulator >>= 1;
if(carry) {
_crcAccumulator ^= 0x8408;
}
}
}
void Fds::WriteRegister(uint16_t addr, uint8_t value)
{
if((!_diskRegEnabled && addr >= 0x4024 && addr <= 0x4026) || (!_soundRegEnabled && addr >= 0x4040)) {
return;
}
switch(addr) {
case 0x4020:
_irqReloadValue = (_irqReloadValue & 0xFF00) | value;
break;
case 0x4021:
_irqReloadValue = (_irqReloadValue & 0x00FF) | (value << 8);
break;
case 0x4022:
_irqRepeatEnabled = (value & 0x01) == 0x01;
_irqEnabled = (value & 0x02) == 0x02 && _diskRegEnabled;
if(_irqEnabled) {
_irqCounter = _irqReloadValue;
} else {
_cpu->ClearIrqSource(IRQSource::External);
}
break;
case 0x4023:
_diskRegEnabled = (value & 0x01) == 0x01;
_soundRegEnabled = (value & 0x02) == 0x02;
if(!_diskRegEnabled) {
_irqEnabled = false;
_cpu->ClearIrqSource(IRQSource::External);
_cpu->ClearIrqSource(IRQSource::FdsDisk);
}
break;
case 0x4024:
_writeDataReg = value;
_transferComplete = false;
//Unsure about clearing irq here: FCEUX/Nintendulator don't do this, puNES does.
_cpu->ClearIrqSource(IRQSource::FdsDisk);
break;
case 0x4025:
_motorOn = (value & 0x01) == 0x01;
_resetTransfer = (value & 0x02) == 0x02;
_readMode = (value & 0x04) == 0x04;
SetMirroringType(value & 0x08 ? MirroringType::Horizontal : MirroringType::Vertical);
_crcControl = (value & 0x10) == 0x10;
//Bit 6 is not used, always 1
_diskReady = (value & 0x40) == 0x40;
_diskIrqEnabled = (value & 0x80) == 0x80;
//Writing to $4025 clears IRQ according to FCEUX, puNES & Nintendulator
//Fixes issues in some unlicensed games (error $20 at power on)
_cpu->ClearIrqSource(IRQSource::FdsDisk);
break;
case 0x4026:
_extConWriteReg = value;
break;
default:
if(addr >= 0x4040) {
_audio->WriteRegister(addr, value);
}
break;
}
}
uint8_t Fds::ReadRegister(uint16_t addr)
{
uint8_t value = _memoryManager->GetOpenBus();
if(_soundRegEnabled && addr >= 0x4040) {
return _audio->ReadRegister(addr);
} else if(_diskRegEnabled && addr <= 0x4033) {
switch(addr) {
case 0x4030:
//These 3 pins are open bus
value &= 0x24;
value |= _cpu->HasIrqSource(IRQSource::External) ? 0x01 : 0x00;
value |= _transferComplete ? 0x02 : 0x00;
value |= GetMirroringType() == MirroringType::Horizontal ? 0x08 : 0;
value |= _useQdFormat && _badCrc ? 0x10 : 0x00;
//value |= _endOfHead ? 0x40 : 0x00;
//value |= _diskRegEnabled ? 0x80 : 0x00;
_transferComplete = false;
_cpu->ClearIrqSource(IRQSource::External);
_cpu->ClearIrqSource(IRQSource::FdsDisk);
return value;
case 0x4031:
_transferComplete = false;
_cpu->ClearIrqSource(IRQSource::FdsDisk);
return _readDataReg;
case 0x4032:
//These 5 pins are open bus
value &= 0xF8;
value |= !IsDiskInserted() ? 0x01 : 0x00; //Disk not in drive
value |= (!IsDiskInserted() || !_scanningDisk) ? 0x02 : 0x00; //Disk not ready
value |= !IsDiskInserted() ? 0x04 : 0x00; //Disk not writable
if(IsAutoInsertDiskEnabled()) {
if(_emu->GetFrameCount() - _lastDiskCheckFrame < 100) {
_successiveChecks++;
} else {
_successiveChecks = 0;
}
_lastDiskCheckFrame = _emu->GetFrameCount();
if(_successiveChecks > 20 && _autoDiskEjectCounter == 0 && _autoDiskSwitchCounter == -1) {
//Game tried to check if a disk was inserted or not - this is usually done when the disk needs to be changed
//Eject the current disk and insert the new one in ~77 frames
_lastDiskCheckFrame = 0;
_successiveChecks = 0;
_autoDiskSwitchCounter = 77;
_previousDiskNumber = _diskNumber;
_diskNumber = NoDiskInserted;
MessageManager::Log("[FDS] Disk automatically ejected.");
}
}
return value;
case 0x4033:
//Always return good battery
return _extConWriteReg;
}
}
return _memoryManager->GetOpenBus();
}
void Fds::Serialize(Serializer& s)
{
BaseMapper::Serialize(s);
SV(_audio);
SV(_irqReloadValue); SV(_irqCounter); SV(_irqEnabled); SV(_irqRepeatEnabled); SV(_diskRegEnabled); SV(_soundRegEnabled); SV(_writeDataReg); SV(_motorOn); SV(_resetTransfer);
SV(_readMode); SV(_crcControl); SV(_diskReady); SV(_diskIrqEnabled); SV(_extConWriteReg); SV(_badCrc); SV(_endOfHead); SV(_readWriteEnabled); SV(_readDataReg); SV(_diskWriteProtected);
SV(_diskNumber); SV(_diskPosition); SV(_delay); SV(_previousCrcControlFlag); SV(_gapEnded); SV(_scanningDisk); SV(_transferComplete);
SV(_autoDiskEjectCounter); SV(_autoDiskSwitchCounter); SV(_restartAutoInsertCounter); SV(_previousFrame); SV(_lastDiskCheckFrame);
SV(_successiveChecks); SV(_previousDiskNumber); SV(_crcAccumulator);
if(s.IsSaving()) {
for(int i = 0; i < (int)_fdsDiskSides.size(); i++) {
vector<uint8_t> ipsData = IpsPatcher::CreatePatch(_orgDiskSides[i], _fdsDiskSides[i]);
SVVectorI(ipsData);
}
} else {
for(int i = 0; i < (int)_fdsDiskSides.size(); i++) {
vector<uint8_t> ipsData;
SVVectorI(ipsData);
IpsPatcher::PatchBuffer(ipsData, _orgDiskSides[i], _fdsDiskSides[i]);
}
if(!_emu->IsRunAheadFrame()) {
//Make sure we disable fast forwarding when loading a state
//Otherwise it's possible to get stuck in fast forward mode
_gameStarted = true;
}
}
}
Fds::~Fds()
{
//Restore emulation speed to normal when closing
_emu->GetSettings()->ClearFlag(EmulationFlags::MaximumSpeed);
}
uint32_t Fds::GetSideCount()
{
return (uint32_t)_fdsDiskSides.size();
}
void Fds::EjectDisk()
{
_diskNumber = Fds::NoDiskInserted;
}
void Fds::InsertDisk(uint32_t diskNumber)
{
if(_diskNumber == Fds::NoDiskInserted) {
_diskNumber = diskNumber % GetSideCount();
}
}
uint32_t Fds::GetCurrentDisk()
{
return _diskNumber;
}
bool Fds::IsDiskInserted()
{
return _diskNumber != Fds::NoDiskInserted;
}
bool Fds::IsAutoInsertDiskEnabled()
{
return !_disableAutoInsertDisk && _settings->FdsAutoInsertDisk && !_emu->GetMovieManager()->Playing() && !_emu->GetMovieManager()->Recording();
}
vector<MapperStateEntry> Fds::GetMapperStateEntries()
{
vector<MapperStateEntry> entries;
entries.push_back(MapperStateEntry("$4020/1", "IRQ Reload Value", _irqReloadValue, MapperStateValueType::Number16));
entries.push_back(MapperStateEntry("$4022.0", "IRQ Repeat", _irqRepeatEnabled, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("$4022.1", "IRQ Enabled", _irqEnabled, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("", "IRQ Counter", _irqCounter, MapperStateValueType::Number16));
entries.push_back(MapperStateEntry("$4023.0", "Disk Registers Enabled", _diskRegEnabled, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("$4023.1", "Sound Registers Enabled", _soundRegEnabled, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("$4025.0", "Motor Enabled", _motorOn, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("$4025.1", "Reset Transfer", _resetTransfer, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("$4025.2", "Read Mode", _readMode, MapperStateValueType::Bool));
entries.push_back(MapperStateEntry("$4025.3", "Mirroring", GetMirroringType() == MirroringType::Horizontal ? "Horizontal" : "Vertical", GetMirroringType() == MirroringType::Horizontal ? 1 : 0));
entries.push_back(MapperStateEntry("$4025.7", "Disk IRQ Enabled", _diskIrqEnabled, MapperStateValueType::Bool));
_audio->GetMapperStateEntries(entries);
return entries;
}