470 lines
15 KiB
C++
470 lines
15 KiB
C++
|
// Copyright 2017 Citra Emulator Project
|
||
|
// Licensed under GPLv2 or any later version
|
||
|
// Refer to the license.txt file included.
|
||
|
|
||
|
#include <cstring>
|
||
|
#include <string>
|
||
|
#include <vector>
|
||
|
#include <cryptopp/hex.h>
|
||
|
#include "common/bit_field.h"
|
||
|
#include "common/common_types.h"
|
||
|
#include "common/file_util.h"
|
||
|
#include "common/logging/log.h"
|
||
|
#include "common/scm_rev.h"
|
||
|
#include "common/string_util.h"
|
||
|
#include "common/swap.h"
|
||
|
#include "core/core.h"
|
||
|
#include "core/hle/service/hid/hid.h"
|
||
|
#include "core/hle/service/ir/extra_hid.h"
|
||
|
#include "core/hle/service/ir/ir_rst.h"
|
||
|
#include "core/movie.h"
|
||
|
|
||
|
namespace Movie {
|
||
|
|
||
|
enum class PlayMode { None, Recording, Playing };
|
||
|
|
||
|
enum class ControllerStateType : u8 {
|
||
|
PadAndCircle,
|
||
|
Touch,
|
||
|
Accelerometer,
|
||
|
Gyroscope,
|
||
|
IrRst,
|
||
|
ExtraHidResponse
|
||
|
};
|
||
|
|
||
|
#pragma pack(push, 1)
|
||
|
struct ControllerState {
|
||
|
ControllerStateType type;
|
||
|
|
||
|
union {
|
||
|
struct {
|
||
|
union {
|
||
|
u16_le hex;
|
||
|
|
||
|
BitField<0, 1, u16_le> a;
|
||
|
BitField<1, 1, u16_le> b;
|
||
|
BitField<2, 1, u16_le> select;
|
||
|
BitField<3, 1, u16_le> start;
|
||
|
BitField<4, 1, u16_le> right;
|
||
|
BitField<5, 1, u16_le> left;
|
||
|
BitField<6, 1, u16_le> up;
|
||
|
BitField<7, 1, u16_le> down;
|
||
|
BitField<8, 1, u16_le> r;
|
||
|
BitField<9, 1, u16_le> l;
|
||
|
BitField<10, 1, u16_le> x;
|
||
|
BitField<11, 1, u16_le> y;
|
||
|
// Bits 12-15 are currently unused
|
||
|
};
|
||
|
s16_le circle_pad_x;
|
||
|
s16_le circle_pad_y;
|
||
|
} pad_and_circle;
|
||
|
|
||
|
struct {
|
||
|
u16_le x;
|
||
|
u16_le y;
|
||
|
// This is a bool, u8 for platform compatibility
|
||
|
u8 valid;
|
||
|
} touch;
|
||
|
|
||
|
struct {
|
||
|
s16_le x;
|
||
|
s16_le y;
|
||
|
s16_le z;
|
||
|
} accelerometer;
|
||
|
|
||
|
struct {
|
||
|
s16_le x;
|
||
|
s16_le y;
|
||
|
s16_le z;
|
||
|
} gyroscope;
|
||
|
|
||
|
struct {
|
||
|
s16_le x;
|
||
|
s16_le y;
|
||
|
// These are bool, u8 for platform compatibility
|
||
|
u8 zl;
|
||
|
u8 zr;
|
||
|
} ir_rst;
|
||
|
|
||
|
struct {
|
||
|
union {
|
||
|
u32_le hex;
|
||
|
|
||
|
BitField<0, 5, u32_le> battery_level;
|
||
|
BitField<5, 1, u32_le> zl_not_held;
|
||
|
BitField<6, 1, u32_le> zr_not_held;
|
||
|
BitField<7, 1, u32_le> r_not_held;
|
||
|
BitField<8, 12, u32_le> c_stick_x;
|
||
|
BitField<20, 12, u32_le> c_stick_y;
|
||
|
};
|
||
|
} extra_hid_response;
|
||
|
};
|
||
|
};
|
||
|
static_assert(sizeof(ControllerState) == 7, "ControllerState should be 7 bytes");
|
||
|
#pragma pack(pop)
|
||
|
|
||
|
constexpr std::array<u8, 4> header_magic_bytes{{'C', 'T', 'M', 0x1B}};
|
||
|
|
||
|
#pragma pack(push, 1)
|
||
|
struct CTMHeader {
|
||
|
std::array<u8, 4> filetype; /// Unique Identifier to check the file type (always "CTM"0x1B)
|
||
|
u64_le program_id; /// ID of the ROM being executed. Also called title_id
|
||
|
std::array<u8, 20> revision; /// Git hash of the revision this movie was created with
|
||
|
|
||
|
std::array<u8, 224> reserved; /// Make heading 256 bytes so it has consistent size
|
||
|
};
|
||
|
static_assert(sizeof(CTMHeader) == 256, "CTMHeader should be 256 bytes");
|
||
|
#pragma pack(pop)
|
||
|
|
||
|
static PlayMode play_mode = PlayMode::None;
|
||
|
static std::vector<u8> recorded_input;
|
||
|
static size_t current_byte = 0;
|
||
|
|
||
|
static bool IsPlayingInput() {
|
||
|
return play_mode == PlayMode::Playing;
|
||
|
}
|
||
|
static bool IsRecordingInput() {
|
||
|
return play_mode == PlayMode::Recording;
|
||
|
}
|
||
|
|
||
|
static void CheckInputEnd() {
|
||
|
if (current_byte + sizeof(ControllerState) > recorded_input.size()) {
|
||
|
LOG_INFO(Movie, "Playback finished");
|
||
|
play_mode = PlayMode::None;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
static void Play(Service::HID::PadState& pad_state, s16& circle_pad_x, s16& circle_pad_y) {
|
||
|
ControllerState s;
|
||
|
std::memcpy(&s, &recorded_input[current_byte], sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
|
||
|
if (s.type != ControllerStateType::PadAndCircle) {
|
||
|
LOG_ERROR(Movie,
|
||
|
"Expected to read type %d, but found %d. Your playback will be out of sync",
|
||
|
ControllerStateType::PadAndCircle, s.type);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
pad_state.a.Assign(s.pad_and_circle.a);
|
||
|
pad_state.b.Assign(s.pad_and_circle.b);
|
||
|
pad_state.select.Assign(s.pad_and_circle.select);
|
||
|
pad_state.start.Assign(s.pad_and_circle.start);
|
||
|
pad_state.right.Assign(s.pad_and_circle.right);
|
||
|
pad_state.left.Assign(s.pad_and_circle.left);
|
||
|
pad_state.up.Assign(s.pad_and_circle.up);
|
||
|
pad_state.down.Assign(s.pad_and_circle.down);
|
||
|
pad_state.r.Assign(s.pad_and_circle.r);
|
||
|
pad_state.l.Assign(s.pad_and_circle.l);
|
||
|
pad_state.x.Assign(s.pad_and_circle.x);
|
||
|
pad_state.y.Assign(s.pad_and_circle.y);
|
||
|
|
||
|
circle_pad_x = s.pad_and_circle.circle_pad_x;
|
||
|
circle_pad_y = s.pad_and_circle.circle_pad_y;
|
||
|
}
|
||
|
|
||
|
static void Play(Service::HID::TouchDataEntry& touch_data) {
|
||
|
ControllerState s;
|
||
|
std::memcpy(&s, &recorded_input[current_byte], sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
|
||
|
if (s.type != ControllerStateType::Touch) {
|
||
|
LOG_ERROR(Movie,
|
||
|
"Expected to read type %d, but found %d. Your playback will be out of sync",
|
||
|
ControllerStateType::Touch, s.type);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
touch_data.x = s.touch.x;
|
||
|
touch_data.y = s.touch.y;
|
||
|
touch_data.valid.Assign(s.touch.valid);
|
||
|
}
|
||
|
|
||
|
static void Play(Service::HID::AccelerometerDataEntry& accelerometer_data) {
|
||
|
ControllerState s;
|
||
|
std::memcpy(&s, &recorded_input[current_byte], sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
|
||
|
if (s.type != ControllerStateType::Accelerometer) {
|
||
|
LOG_ERROR(Movie,
|
||
|
"Expected to read type %d, but found %d. Your playback will be out of sync",
|
||
|
ControllerStateType::Accelerometer, s.type);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
accelerometer_data.x = s.accelerometer.x;
|
||
|
accelerometer_data.y = s.accelerometer.y;
|
||
|
accelerometer_data.z = s.accelerometer.z;
|
||
|
}
|
||
|
|
||
|
static void Play(Service::HID::GyroscopeDataEntry& gyroscope_data) {
|
||
|
ControllerState s;
|
||
|
std::memcpy(&s, &recorded_input[current_byte], sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
|
||
|
if (s.type != ControllerStateType::Gyroscope) {
|
||
|
LOG_ERROR(Movie,
|
||
|
"Expected to read type %d, but found %d. Your playback will be out of sync",
|
||
|
ControllerStateType::Gyroscope, s.type);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
gyroscope_data.x = s.gyroscope.x;
|
||
|
gyroscope_data.y = s.gyroscope.y;
|
||
|
gyroscope_data.z = s.gyroscope.z;
|
||
|
}
|
||
|
|
||
|
static void Play(Service::IR::PadState& pad_state, s16& c_stick_x, s16& c_stick_y) {
|
||
|
ControllerState s;
|
||
|
std::memcpy(&s, &recorded_input[current_byte], sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
|
||
|
if (s.type != ControllerStateType::IrRst) {
|
||
|
LOG_ERROR(Movie,
|
||
|
"Expected to read type %d, but found %d. Your playback will be out of sync",
|
||
|
ControllerStateType::IrRst, s.type);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
c_stick_x = s.ir_rst.x;
|
||
|
c_stick_y = s.ir_rst.y;
|
||
|
pad_state.zl.Assign(s.ir_rst.zl);
|
||
|
pad_state.zr.Assign(s.ir_rst.zr);
|
||
|
}
|
||
|
|
||
|
static void Play(Service::IR::ExtraHIDResponse& extra_hid_response) {
|
||
|
ControllerState s;
|
||
|
std::memcpy(&s, &recorded_input[current_byte], sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
|
||
|
if (s.type != ControllerStateType::ExtraHidResponse) {
|
||
|
LOG_ERROR(Movie,
|
||
|
"Expected to read type %d, but found %d. Your playback will be out of sync",
|
||
|
ControllerStateType::ExtraHidResponse, s.type);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
extra_hid_response.buttons.battery_level.Assign(s.extra_hid_response.battery_level);
|
||
|
extra_hid_response.c_stick.c_stick_x.Assign(s.extra_hid_response.c_stick_x);
|
||
|
extra_hid_response.c_stick.c_stick_y.Assign(s.extra_hid_response.c_stick_y);
|
||
|
extra_hid_response.buttons.r_not_held.Assign(s.extra_hid_response.r_not_held);
|
||
|
extra_hid_response.buttons.zl_not_held.Assign(s.extra_hid_response.zl_not_held);
|
||
|
extra_hid_response.buttons.zr_not_held.Assign(s.extra_hid_response.zr_not_held);
|
||
|
}
|
||
|
|
||
|
static void Record(const ControllerState& controller_state) {
|
||
|
recorded_input.resize(current_byte + sizeof(ControllerState));
|
||
|
std::memcpy(&recorded_input[current_byte], &controller_state, sizeof(ControllerState));
|
||
|
current_byte += sizeof(ControllerState);
|
||
|
}
|
||
|
|
||
|
static void Record(const Service::HID::PadState& pad_state, const s16& circle_pad_x,
|
||
|
const s16& circle_pad_y) {
|
||
|
ControllerState s;
|
||
|
s.type = ControllerStateType::PadAndCircle;
|
||
|
|
||
|
s.pad_and_circle.a.Assign(static_cast<u16>(pad_state.a));
|
||
|
s.pad_and_circle.b.Assign(static_cast<u16>(pad_state.b));
|
||
|
s.pad_and_circle.select.Assign(static_cast<u16>(pad_state.select));
|
||
|
s.pad_and_circle.start.Assign(static_cast<u16>(pad_state.start));
|
||
|
s.pad_and_circle.right.Assign(static_cast<u16>(pad_state.right));
|
||
|
s.pad_and_circle.left.Assign(static_cast<u16>(pad_state.left));
|
||
|
s.pad_and_circle.up.Assign(static_cast<u16>(pad_state.up));
|
||
|
s.pad_and_circle.down.Assign(static_cast<u16>(pad_state.down));
|
||
|
s.pad_and_circle.r.Assign(static_cast<u16>(pad_state.r));
|
||
|
s.pad_and_circle.l.Assign(static_cast<u16>(pad_state.l));
|
||
|
s.pad_and_circle.x.Assign(static_cast<u16>(pad_state.x));
|
||
|
s.pad_and_circle.y.Assign(static_cast<u16>(pad_state.y));
|
||
|
|
||
|
s.pad_and_circle.circle_pad_x = circle_pad_x;
|
||
|
s.pad_and_circle.circle_pad_y = circle_pad_y;
|
||
|
|
||
|
Record(s);
|
||
|
}
|
||
|
|
||
|
static void Record(const Service::HID::TouchDataEntry& touch_data) {
|
||
|
ControllerState s;
|
||
|
s.type = ControllerStateType::Touch;
|
||
|
|
||
|
s.touch.x = touch_data.x;
|
||
|
s.touch.y = touch_data.y;
|
||
|
s.touch.valid = static_cast<u8>(touch_data.valid);
|
||
|
|
||
|
Record(s);
|
||
|
}
|
||
|
|
||
|
static void Record(const Service::HID::AccelerometerDataEntry& accelerometer_data) {
|
||
|
ControllerState s;
|
||
|
s.type = ControllerStateType::Accelerometer;
|
||
|
|
||
|
s.accelerometer.x = accelerometer_data.x;
|
||
|
s.accelerometer.y = accelerometer_data.y;
|
||
|
s.accelerometer.z = accelerometer_data.z;
|
||
|
|
||
|
Record(s);
|
||
|
}
|
||
|
|
||
|
static void Record(const Service::HID::GyroscopeDataEntry& gyroscope_data) {
|
||
|
ControllerState s;
|
||
|
s.type = ControllerStateType::Gyroscope;
|
||
|
|
||
|
s.gyroscope.x = gyroscope_data.x;
|
||
|
s.gyroscope.y = gyroscope_data.y;
|
||
|
s.gyroscope.z = gyroscope_data.z;
|
||
|
|
||
|
Record(s);
|
||
|
}
|
||
|
|
||
|
static void Record(const Service::IR::PadState& pad_state, const s16& c_stick_x,
|
||
|
const s16& c_stick_y) {
|
||
|
ControllerState s;
|
||
|
s.type = ControllerStateType::IrRst;
|
||
|
|
||
|
s.ir_rst.x = c_stick_x;
|
||
|
s.ir_rst.y = c_stick_y;
|
||
|
s.ir_rst.zl = static_cast<u8>(pad_state.zl);
|
||
|
s.ir_rst.zr = static_cast<u8>(pad_state.zr);
|
||
|
|
||
|
Record(s);
|
||
|
}
|
||
|
|
||
|
static void Record(const Service::IR::ExtraHIDResponse& extra_hid_response) {
|
||
|
ControllerState s;
|
||
|
s.type = ControllerStateType::ExtraHidResponse;
|
||
|
|
||
|
s.extra_hid_response.battery_level.Assign(extra_hid_response.buttons.battery_level);
|
||
|
s.extra_hid_response.c_stick_x.Assign(extra_hid_response.c_stick.c_stick_x);
|
||
|
s.extra_hid_response.c_stick_y.Assign(extra_hid_response.c_stick.c_stick_y);
|
||
|
s.extra_hid_response.r_not_held.Assign(extra_hid_response.buttons.r_not_held);
|
||
|
s.extra_hid_response.zl_not_held.Assign(extra_hid_response.buttons.zl_not_held);
|
||
|
s.extra_hid_response.zr_not_held.Assign(extra_hid_response.buttons.zr_not_held);
|
||
|
|
||
|
Record(s);
|
||
|
}
|
||
|
|
||
|
static bool ValidateHeader(const CTMHeader& header) {
|
||
|
if (header_magic_bytes != header.filetype) {
|
||
|
LOG_ERROR(Movie, "Playback file does not have valid header");
|
||
|
return false;
|
||
|
}
|
||
|
|
||
|
std::string revision =
|
||
|
Common::ArrayToString(header.revision.data(), header.revision.size(), 21, false);
|
||
|
revision = Common::ToLower(revision);
|
||
|
|
||
|
if (revision != Common::g_scm_rev) {
|
||
|
LOG_WARNING(Movie,
|
||
|
"This movie was created on a different version of Citra, playback may desync");
|
||
|
}
|
||
|
|
||
|
u64 program_id;
|
||
|
Core::System::GetInstance().GetAppLoader().ReadProgramId(program_id);
|
||
|
if (program_id != header.program_id) {
|
||
|
LOG_WARNING(Movie, "This movie was recorded using a ROM with a different program id");
|
||
|
}
|
||
|
|
||
|
return true;
|
||
|
}
|
||
|
|
||
|
static void SaveMovie() {
|
||
|
LOG_INFO(Movie, "Saving movie");
|
||
|
FileUtil::IOFile save_record(Settings::values.movie_record, "wb");
|
||
|
|
||
|
if (!save_record.IsGood()) {
|
||
|
LOG_ERROR(Movie, "Unable to open file to save movie");
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
CTMHeader header = {};
|
||
|
header.filetype = header_magic_bytes;
|
||
|
|
||
|
Core::System::GetInstance().GetAppLoader().ReadProgramId(header.program_id);
|
||
|
|
||
|
std::string rev_bytes;
|
||
|
CryptoPP::StringSource(Common::g_scm_rev, true,
|
||
|
new CryptoPP::HexDecoder(new CryptoPP::StringSink(rev_bytes)));
|
||
|
std::memcpy(header.revision.data(), rev_bytes.data(), sizeof(CTMHeader::revision));
|
||
|
|
||
|
save_record.WriteBytes(&header, sizeof(CTMHeader));
|
||
|
save_record.WriteBytes(recorded_input.data(), recorded_input.size());
|
||
|
|
||
|
if (!save_record.IsGood()) {
|
||
|
LOG_ERROR(Movie, "Error saving movie");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void Init() {
|
||
|
if (!Settings::values.movie_play.empty()) {
|
||
|
LOG_INFO(Movie, "Loading Movie for playback");
|
||
|
FileUtil::IOFile save_record(Settings::values.movie_play, "rb");
|
||
|
u64 size = save_record.GetSize();
|
||
|
|
||
|
if (save_record.IsGood() && size > sizeof(CTMHeader)) {
|
||
|
CTMHeader header;
|
||
|
save_record.ReadArray(&header, 1);
|
||
|
if (ValidateHeader(header)) {
|
||
|
play_mode = PlayMode::Playing;
|
||
|
recorded_input.resize(size - sizeof(CTMHeader));
|
||
|
save_record.ReadArray(recorded_input.data(), recorded_input.size());
|
||
|
current_byte = 0;
|
||
|
}
|
||
|
} else {
|
||
|
LOG_ERROR(Movie, "Failed to playback movie: Unable to open '%s'",
|
||
|
Settings::values.movie_play.c_str());
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if (!Settings::values.movie_record.empty()) {
|
||
|
LOG_INFO(Movie, "Enabling Movie recording");
|
||
|
play_mode = PlayMode::Recording;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void Shutdown() {
|
||
|
if (!IsRecordingInput()) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
SaveMovie();
|
||
|
|
||
|
play_mode = PlayMode::None;
|
||
|
recorded_input.resize(0);
|
||
|
current_byte = 0;
|
||
|
}
|
||
|
|
||
|
template <typename... Targs>
|
||
|
static void Handle(Targs&... Fargs) {
|
||
|
if (IsPlayingInput()) {
|
||
|
Play(Fargs...);
|
||
|
CheckInputEnd();
|
||
|
} else if (IsRecordingInput()) {
|
||
|
Record(Fargs...);
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void HandlePadAndCircleStatus(Service::HID::PadState& pad_state, s16& circle_pad_x,
|
||
|
s16& circle_pad_y) {
|
||
|
Handle(pad_state, circle_pad_x, circle_pad_y);
|
||
|
}
|
||
|
|
||
|
void HandleTouchStatus(Service::HID::TouchDataEntry& touch_data) {
|
||
|
Handle(touch_data);
|
||
|
}
|
||
|
|
||
|
void HandleAccelerometerStatus(Service::HID::AccelerometerDataEntry& accelerometer_data) {
|
||
|
Handle(accelerometer_data);
|
||
|
}
|
||
|
|
||
|
void HandleGyroscopeStatus(Service::HID::GyroscopeDataEntry& gyroscope_data) {
|
||
|
Handle(gyroscope_data);
|
||
|
}
|
||
|
|
||
|
void HandleIrRst(Service::IR::PadState& pad_state, s16& c_stick_x, s16& c_stick_y) {
|
||
|
Handle(pad_state, c_stick_x, c_stick_y);
|
||
|
}
|
||
|
|
||
|
void HandleExtraHidResponse(Service::IR::ExtraHIDResponse& extra_hid_response) {
|
||
|
Handle(extra_hid_response);
|
||
|
}
|
||
|
}
|