rpcs3/rpcs3/Emu/RSX/Capture/rsx_replay.cpp
2018-05-14 21:14:39 +04:00

340 lines
11 KiB
C++

#include "stdafx.h"
#include "rsx_replay.h"
#include "Emu/System.h"
#include "Emu/Cell/lv2/sys_rsx.h"
#include "Emu/Memory/Memory.h"
#include "Emu/RSX/GSRender.h"
#include <map>
namespace rsx
{
be_t<u32> rsx_replay_thread::allocate_context()
{
const u32 contextAddr = vm::alloc(sizeof(rsx_context), vm::main);
if (contextAddr == 0)
fmt::throw_exception("Capture Replay: context alloc failed");
const auto& contextInfo = vm::_ref<rsx_context>(contextAddr);
if (sys_rsx_memory_allocate(vm::get_addr(&contextInfo.mem_handle), vm::get_addr(&contextInfo.mem_addr), 0x0F900000, 0, 0, 0, 0) != CELL_OK)
fmt::throw_exception("Capture Replay: sys_rsx_memory_allocate failed!");
if (sys_rsx_context_allocate(vm::get_addr(&contextInfo.context_id), vm::get_addr(&contextInfo.dma_addr), vm::get_addr(&contextInfo.driver_info), vm::get_addr(&contextInfo.reports_addr), contextInfo.mem_handle, 0) != CELL_OK)
fmt::throw_exception("Capture Replay: sys_rsx_context_allocate failed!");
// 'fake' initialize usermemory
// todo: seriously, need to probly watch the replay memory map and just make sure its mapped before we copy rather than do this
vm::falloc(0x20000000, 0x10000000, vm::user_space);
return contextInfo.context_id;
}
std::tuple<u32, u32> rsx_replay_thread::get_usable_fifo_range()
{
u32 fifo_size = 4;
// run through replay commands to figure out how big command buffer needs to be
// technically we could do this in batches if it gets too big, but we should be fine
// as we aren't allocating anything on main memory, although it may make issues with iooffset later
for (const auto& rc : frame->replay_commands)
{
const u32 count = (rc.rsx_command.first >> 18) & 0x7ff;
// allocate for register plus w/e number of arguments it has
fifo_size += (count + 1) * 4;
}
// safety check for now
// since we are allocating iobuffer, we need to make sure that any memory we use is not being used by the replay
std::map<u32, u32> ioOffsets;
u32 lowest_iooffset = 0xFFFFFFFF;
for (const auto& mm : frame->memory_map)
{
u32 offset = mm.second.ioOffset;
lowest_iooffset = std::min(lowest_iooffset, offset);
if (offset != 0xFFFFFFFF)
{
u32 iosize = mm.second.size + mm.second.offset;
auto it = ioOffsets.find(offset);
if (it == ioOffsets.end())
ioOffsets[offset] = iosize;
else
ioOffsets[offset] = std::max(ioOffsets[offset], iosize);
}
}
// if we cant use fifo starting at 0, try to find a block between them
u32 fifo_start_addr = 0;
if (fifo_size >= lowest_iooffset)
{
u32 largest_free_block = 0;
u32 largest_cur_end = 0; // this keeps track of largest end, in case allocations 'overlap'
for (auto io = ioOffsets.begin(); io != ioOffsets.end(); ++io)
{
auto next = std::next(io);
// last 'offset' is just end of memory
u32 nextOffset = 0x0F900000;
if (next != ioOffsets.end())
nextOffset = next->first - 4;
largest_cur_end = std::max(largest_cur_end, io->first + io->second + 4);
if (largest_cur_end < nextOffset)
{
u32 freeSize = nextOffset - largest_cur_end;
if (freeSize > largest_free_block)
{
fifo_start_addr = largest_cur_end;
largest_free_block = freeSize;
}
if (largest_free_block > fifo_size)
break;
}
if (next == ioOffsets.end())
break;
}
// todo: figure out *another* way to inject fifo if both ideas above fail
if (largest_free_block < fifo_size)
fmt::throw_exception("Capture Replay: no space in io for fifo commands! size: 0x%x, lowest in capture: 0x%x, largest_free_block: 0x%x", fifo_size, lowest_iooffset, largest_free_block);
}
return std::make_tuple(fifo_start_addr, fifo_size);
}
std::vector<u32> rsx_replay_thread::alloc_write_fifo(be_t<u32> context_id, u32 fifo_start_addr, u32 fifo_size)
{
const u32 fifo_mem = vm::alloc(fifo_size, vm::main);
if (fifo_mem == 0)
fmt::throw_exception("Capture Replay: fifo alloc failed! size: 0x%x", fifo_size);
// copy commands into fifo buffer
// todo: could change rsx_command to just be values to avoid this loop,
auto fifo_addr = vm::ptr<u32>::make(fifo_mem);
u32 count = 0;
std::vector<u32> fifo_stops;
u32 currentOffset = fifo_start_addr;
for (const auto& rc : frame->replay_commands)
{
bool hasState = (rc.memory_state.size() > 0) || (rc.display_buffer_state != 0) || (rc.tile_state != 0);
if (hasState)
{
if (count != 0)
{
// todo: support memory state in the middle of incremented command
// This shouldn't ever happen as long as captures stay in 'strict' aka non-multidraw mode
fmt::throw_exception("capture replay: state change not supported between increment commands");
}
fifo_stops.emplace_back(currentOffset);
}
// spit out command
if (count == 0)
{
count = (rc.rsx_command.first >> 18) & 0x7ff;
*fifo_addr = rc.rsx_command.first;
fifo_addr++;
currentOffset += 4;
}
if (count != 0)
{
*fifo_addr = rc.rsx_command.second;
fifo_addr++;
count--;
currentOffset += 4;
}
}
fifo_stops.emplace_back(currentOffset);
if (sys_rsx_context_iomap(context_id, fifo_start_addr, fifo_mem, fifo_size, 0) != CELL_OK)
fmt::throw_exception("Capture Replay: fifo mapping failed");
return fifo_stops;
}
void rsx_replay_thread::apply_frame_state(be_t<u32> context_id, const frame_capture_data::replay_command& replay_cmd)
{
// apply memory needed for command
for (const auto& state : replay_cmd.memory_state)
{
auto it = frame->memory_map.find(state);
if (it == frame->memory_map.end())
fmt::throw_exception("requested memory state for command not found in memory_map");
if (it->second.data_state != 0)
{
const auto& memblock = it->second;
auto it_data = frame->memory_data_map.find(it->second.data_state);
if (it_data == frame->memory_data_map.end())
fmt::throw_exception("requested memory data state for command not found in memory_data_map");
const auto& data_block = it_data->second;
std::memcpy(vm::base(memblock.addr + memblock.offset), data_block.data.data(), data_block.data.size());
}
}
if (replay_cmd.display_buffer_state != 0 && replay_cmd.display_buffer_state != cs.display_buffer_hash)
{
auto it = frame->display_buffers_map.find(replay_cmd.display_buffer_state);
if (it == frame->display_buffers_map.end())
fmt::throw_exception("requested display buffer for command not found");
const auto& dbstate = it->second;
for (u32 i = 0; i < dbstate.count; ++i)
{
const auto& buf = dbstate.buffers[i];
if (cs.display_buffer_hash != 0 && memcmp(&cs.buffer_state.buffers[i], &buf, sizeof(rsx::frame_capture_data::buffer_state)) == 0)
continue;
cs.buffer_state.buffers[i] = buf;
sys_rsx_context_attribute(context_id, 0x104, i,
(u64)dbstate.buffers[i].width << 32 | dbstate.buffers[i].height, (u64)dbstate.buffers[i].pitch << 32 | dbstate.buffers[i].offset, 0);
}
cs.display_buffer_hash = replay_cmd.display_buffer_state;
}
if (replay_cmd.tile_state != 0 && replay_cmd.tile_state != cs.tile_hash)
{
auto it = frame->tile_map.find(replay_cmd.tile_state);
if (it == frame->tile_map.end())
fmt::throw_exception("requested tile state command not found");
const auto& tstate = it->second;
for (u32 i = 0; i < limits::tiles_count; ++i)
{
const auto& tstile = tstate.tiles[i];
if (cs.tile_hash != 0 && memcmp(&cs.tile_state.tiles[i], &tstile, sizeof(rsx::frame_capture_data::tile_info)) == 0)
continue;
cs.tile_state.tiles[i] = tstile;
GcmTileInfo t;
t.bank = tstile.bank;
t.base = tstile.base;
t.binded = tstile.binded;
t.comp = tstile.comp;
t.location = tstile.location;
t.offset = tstile.offset;
t.pitch = tstile.pitch;
t.size = tstile.size;
const auto& ti = t.pack();
sys_rsx_context_attribute(context_id, 0x300, i, (u64)ti.tile << 32 | ti.limit, (u64)ti.pitch << 32 | ti.format, 0);
}
for (u32 i = 0; i < limits::zculls_count; ++i)
{
const auto& zctile = tstate.zculls[i];
if (cs.tile_hash != 0 && memcmp(&cs.tile_state.zculls[i], &zctile, sizeof(rsx::frame_capture_data::zcull_info)) == 0)
continue;
cs.tile_state.zculls[i] = zctile;
GcmZcullInfo zc;
zc.aaFormat = zctile.aaFormat;
zc.binded = zctile.binded;
zc.cullStart = zctile.cullStart;
zc.height = zctile.height;
zc.offset = zctile.offset;
zc.sFunc = zctile.sFunc;
zc.sMask = zctile.sMask;
zc.sRef = zctile.sRef;
zc.width = zctile.width;
zc.zcullDir = zctile.zcullDir;
zc.zcullFormat = zctile.zcullFormat;
zc.zFormat = zctile.zFormat;
const auto& zci = zc.pack();
sys_rsx_context_attribute(context_id, 0x301, i, (u64)zci.region << 32 | zci.size, (u64)zci.start << 32 | zci.offset, (u64)zci.status0 << 32 | zci.status1);
}
cs.tile_hash = replay_cmd.tile_state;
}
}
void rsx_replay_thread::cpu_task()
{
be_t<u32> context_id = allocate_context();
auto fifo_info = get_usable_fifo_range();
const u32 fifo_start_addr = std::get<0>(fifo_info);
const u32 fifo_size = std::get<1>(fifo_info);
auto fifo_stops = alloc_write_fifo(context_id, fifo_start_addr, fifo_size);
// map game io
for (const auto it : frame->memory_map)
{
const auto& memblock = it.second;
if (memblock.ioOffset == 0xFFFFFFFF)
continue;
// sanity check
if (memblock.ioOffset <= fifo_start_addr + fifo_size && fifo_start_addr <= memblock.size + memblock.offset)
fmt::throw_exception("Capture Replay: overlap detected between game io allocs and fifo alloc, algorithms botched.");
if (sys_rsx_context_iomap(context_id, memblock.ioOffset, memblock.addr, memblock.size + memblock.offset, 0) != CELL_OK)
fmt::throw_exception("rsx io map failed for block");
}
while (!Emu.IsStopped())
{
// start up fifo buffer by dumping the put ptr to first stop
sys_rsx_context_attribute(context_id, 0x001, fifo_start_addr, fifo_stops[0], 0, 0);
auto renderer = fxm::get<GSRender>();
size_t stopIdx = 0;
for (const auto& replay_cmd : frame->replay_commands)
{
while (Emu.IsPaused())
std::this_thread::sleep_for(10ms);
if (Emu.IsStopped())
break;
// Loop and hunt down our next state change that needs to be done
if (!((replay_cmd.memory_state.size() > 0) || (replay_cmd.display_buffer_state != 0) || (replay_cmd.tile_state != 0)))
continue;
// wait until rsx idle and at our first 'stop' to apply state
while (!Emu.IsStopped() && (renderer->ctrl->get != renderer->ctrl->put) && (renderer->ctrl->get != fifo_stops[stopIdx]))
{
while (Emu.IsPaused())
std::this_thread::sleep_for(10ms);
std::this_thread::yield();
}
stopIdx++;
apply_frame_state(context_id, replay_cmd);
// move put ptr to next stop
if (stopIdx >= fifo_stops.size())
fmt::throw_exception("Capture Replay: StopIdx greater than size of fifo_stops");
renderer->ctrl->put = fifo_stops[stopIdx];
}
// dump put to end of stops, which should have actual end
u32 end = fifo_stops.back();
renderer->ctrl->put = end;
while (renderer->ctrl->get != end && !Emu.IsStopped())
{
while (Emu.IsPaused())
std::this_thread::sleep_for(10ms);
}
// random pause to not destroy gpu
std::this_thread::sleep_for(10ms);
}
state += cpu_flag::exit;
}
}