mii: Add MiiManager class to manage Mii database
Provides serialization/deserialization to the database in system save files, accessors for database state and proper handling of both major Mii formats (MiiInfo and MiiStoreData)
This commit is contained in:
		
							
								
								
									
										369
									
								
								src/core/hle/service/mii/mii_manager.cpp
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										369
									
								
								src/core/hle/service/mii/mii_manager.cpp
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,369 @@
 | 
			
		||||
// Copyright 2018 yuzu emulator team
 | 
			
		||||
// Licensed under GPLv2 or any later version
 | 
			
		||||
// Refer to the license.txt file included.
 | 
			
		||||
 | 
			
		||||
#include <algorithm>
 | 
			
		||||
#include <cstring>
 | 
			
		||||
#include "common/assert.h"
 | 
			
		||||
#include "common/file_util.h"
 | 
			
		||||
#include "common/logging/log.h"
 | 
			
		||||
#include "common/string_util.h"
 | 
			
		||||
#include "core/hle/service/mii/mii_manager.h"
 | 
			
		||||
 | 
			
		||||
namespace Service::Mii {
 | 
			
		||||
 | 
			
		||||
constexpr char MII_SAVE_DATABASE_PATH[] = "/system/save/8000000000000030/MiiDatabase.dat";
 | 
			
		||||
constexpr std::array<char16_t, 11> DEFAULT_MII_NAME = {'y', 'u', 'z', 'u', '\0'};
 | 
			
		||||
 | 
			
		||||
// This value was retrieved from HW test
 | 
			
		||||
constexpr MiiStoreData DEFAULT_MII = {
 | 
			
		||||
    {
 | 
			
		||||
        0x21, 0x40, 0x40, 0x01, 0x08, 0x01, 0x13, 0x08, 0x08, 0x02, 0x17, 0x8C, 0x06, 0x01,
 | 
			
		||||
        0x69, 0x6D, 0x8A, 0x6A, 0x82, 0x14, 0x00, 0x00, 0x00, 0x20, 0x64, 0x72, 0x44, 0x44,
 | 
			
		||||
    },
 | 
			
		||||
    {'y', 'u', 'z', 'u', '\0'},
 | 
			
		||||
    Common::UUID{1, 0},
 | 
			
		||||
    0,
 | 
			
		||||
    0,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
// Default values taken from multiple real databases
 | 
			
		||||
const MiiDatabase DEFAULT_MII_DATABASE{Common::MakeMagic('N', 'F', 'D', 'B'), {}, {1}, 0, 0};
 | 
			
		||||
 | 
			
		||||
template <typename T, std::size_t s1, std::size_t s2>
 | 
			
		||||
std::array<T, s2> ResizeArray(const std::array<T, s1>& in) {
 | 
			
		||||
    std::array<T, s2> out{};
 | 
			
		||||
    std::memcpy(out.data(), in.data(), sizeof(T) * std::min(s1, s2));
 | 
			
		||||
    return out;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiInfo ConvertStoreDataToInfo(const MiiStoreData& data) {
 | 
			
		||||
    MiiStoreBitFields bf{};
 | 
			
		||||
    std::memcpy(&bf, data.data.data(), sizeof(MiiStoreBitFields));
 | 
			
		||||
    return {
 | 
			
		||||
        data.uuid,
 | 
			
		||||
        ResizeArray<char16_t, 10, 11>(data.name),
 | 
			
		||||
        static_cast<u8>(bf.font_region.Value()),
 | 
			
		||||
        static_cast<u8>(bf.favorite_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.gender.Value()),
 | 
			
		||||
        static_cast<u8>(bf.height.Value()),
 | 
			
		||||
        static_cast<u8>(bf.weight.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mii_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mii_region.Value()),
 | 
			
		||||
        static_cast<u8>(bf.face_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.face_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.face_wrinkle.Value()),
 | 
			
		||||
        static_cast<u8>(bf.face_makeup.Value()),
 | 
			
		||||
        static_cast<u8>(bf.hair_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.hair_color.Value()),
 | 
			
		||||
        static_cast<bool>(bf.hair_flip.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_aspect.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_rotate.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_x.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eye_y.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_aspect.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_rotate.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_x.Value()),
 | 
			
		||||
        static_cast<u8>(bf.eyebrow_y.Value()),
 | 
			
		||||
        static_cast<u8>(bf.nose_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.nose_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.nose_y.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mouth_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mouth_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mouth_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mouth_aspect.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mouth_y.Value()),
 | 
			
		||||
        static_cast<u8>(bf.facial_hair_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.beard_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mustache_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mustache_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mustache_y.Value()),
 | 
			
		||||
        static_cast<u8>(bf.glasses_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.glasses_color.Value()),
 | 
			
		||||
        static_cast<u8>(bf.glasses_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.glasses_y.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mole_type.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mole_scale.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mole_x.Value()),
 | 
			
		||||
        static_cast<u8>(bf.mole_y.Value()),
 | 
			
		||||
        0x00,
 | 
			
		||||
    };
 | 
			
		||||
}
 | 
			
		||||
MiiStoreData ConvertInfoToStoreData(const MiiInfo& info) {
 | 
			
		||||
    MiiStoreData out{};
 | 
			
		||||
    out.name = ResizeArray<char16_t, 11, 10>(info.name);
 | 
			
		||||
    out.uuid = info.uuid;
 | 
			
		||||
 | 
			
		||||
    MiiStoreBitFields bf{};
 | 
			
		||||
 | 
			
		||||
    bf.hair_type.Assign(info.hair_type);
 | 
			
		||||
    bf.mole_type.Assign(info.mole_type);
 | 
			
		||||
    bf.height.Assign(info.height);
 | 
			
		||||
    bf.hair_flip.Assign(info.hair_flip);
 | 
			
		||||
    bf.weight.Assign(info.weight);
 | 
			
		||||
    bf.hair_color.Assign(info.hair_color);
 | 
			
		||||
 | 
			
		||||
    bf.gender.Assign(info.gender);
 | 
			
		||||
    bf.eye_color.Assign(info.eye_color);
 | 
			
		||||
    bf.eyebrow_color.Assign(info.eyebrow_color);
 | 
			
		||||
    bf.mouth_color.Assign(info.mouth_color);
 | 
			
		||||
    bf.facial_hair_color.Assign(info.facial_hair_color);
 | 
			
		||||
 | 
			
		||||
    bf.mii_type.Assign(info.mii_type);
 | 
			
		||||
    bf.glasses_color.Assign(info.glasses_color);
 | 
			
		||||
    bf.font_region.Assign(info.font_region);
 | 
			
		||||
    bf.eye_type.Assign(info.eye_type);
 | 
			
		||||
    bf.mii_region.Assign(info.mii_region);
 | 
			
		||||
    bf.mouth_type.Assign(info.mouth_type);
 | 
			
		||||
    bf.glasses_scale.Assign(info.glasses_scale);
 | 
			
		||||
    bf.eye_y.Assign(info.eye_y);
 | 
			
		||||
 | 
			
		||||
    bf.mustache_type.Assign(info.mustache_type);
 | 
			
		||||
    bf.eyebrow_type.Assign(info.eyebrow_type);
 | 
			
		||||
    bf.beard_type.Assign(info.beard_type);
 | 
			
		||||
    bf.nose_type.Assign(info.nose_type);
 | 
			
		||||
    bf.mouth_aspect.Assign(info.mouth_aspect_ratio);
 | 
			
		||||
    bf.nose_y.Assign(info.nose_y);
 | 
			
		||||
    bf.eyebrow_aspect.Assign(info.eyebrow_aspect_ratio);
 | 
			
		||||
    bf.mouth_y.Assign(info.mouth_y);
 | 
			
		||||
 | 
			
		||||
    bf.eye_rotate.Assign(info.eye_rotate);
 | 
			
		||||
    bf.mustache_y.Assign(info.mustache_y);
 | 
			
		||||
    bf.eye_aspect.Assign(info.eye_aspect_ratio);
 | 
			
		||||
    bf.glasses_y.Assign(info.glasses_y);
 | 
			
		||||
    bf.eye_scale.Assign(info.eye_scale);
 | 
			
		||||
    bf.mole_x.Assign(info.mole_x);
 | 
			
		||||
    bf.mole_y.Assign(info.mole_y);
 | 
			
		||||
 | 
			
		||||
    bf.glasses_type.Assign(info.glasses_type);
 | 
			
		||||
    bf.face_type.Assign(info.face_type);
 | 
			
		||||
    bf.favorite_color.Assign(info.favorite_color);
 | 
			
		||||
    bf.face_wrinkle.Assign(info.face_wrinkle);
 | 
			
		||||
    bf.face_color.Assign(info.face_color);
 | 
			
		||||
    bf.eye_x.Assign(info.eye_x);
 | 
			
		||||
    bf.face_makeup.Assign(info.face_makeup);
 | 
			
		||||
 | 
			
		||||
    bf.eyebrow_rotate.Assign(info.eyebrow_rotate);
 | 
			
		||||
    bf.eyebrow_scale.Assign(info.eyebrow_scale);
 | 
			
		||||
    bf.eyebrow_y.Assign(info.eyebrow_y);
 | 
			
		||||
    bf.eyebrow_x.Assign(info.eyebrow_x);
 | 
			
		||||
    bf.mouth_scale.Assign(info.mouth_scale);
 | 
			
		||||
    bf.nose_scale.Assign(info.nose_scale);
 | 
			
		||||
    bf.mole_scale.Assign(info.mole_scale);
 | 
			
		||||
    bf.mustache_scale.Assign(info.mustache_scale);
 | 
			
		||||
 | 
			
		||||
    std::memcpy(out.data.data(), &bf, sizeof(MiiStoreBitFields));
 | 
			
		||||
 | 
			
		||||
    return out;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::u16string MiiInfo::Name() const {
 | 
			
		||||
    return Common::UTF16StringFromFixedZeroTerminatedBuffer(name.data(), name.size());
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool operator==(const MiiInfo& lhs, const MiiInfo& rhs) {
 | 
			
		||||
    return std::memcmp(&lhs, &rhs, sizeof(MiiInfo));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool operator!=(const MiiInfo& lhs, const MiiInfo& rhs) {
 | 
			
		||||
    return !operator==(lhs, rhs);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
std::u16string MiiStoreData::Name() const {
 | 
			
		||||
    return Common::UTF16StringFromFixedZeroTerminatedBuffer(name.data(), name.size());
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiManager::MiiManager() = default;
 | 
			
		||||
 | 
			
		||||
MiiManager::~MiiManager() = default;
 | 
			
		||||
 | 
			
		||||
MiiInfo MiiManager::CreateRandom(RandomParameters params) {
 | 
			
		||||
    LOG_WARNING(Service_Mii,
 | 
			
		||||
                "(STUBBED) called with params={:08X}{:08X}{:08X}, returning default Mii",
 | 
			
		||||
                params.unknown_1, params.unknown_2, params.unknown_3);
 | 
			
		||||
 | 
			
		||||
    auto new_mii = DEFAULT_MII;
 | 
			
		||||
 | 
			
		||||
    do {
 | 
			
		||||
        new_mii.uuid = Common::UUID::Generate();
 | 
			
		||||
    } while (IndexOf(new_mii.uuid) == INVALID_INDEX);
 | 
			
		||||
 | 
			
		||||
    return ConvertStoreDataToInfo(new_mii);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiInfo MiiManager::CreateDefault(u32 index) {
 | 
			
		||||
    auto new_mii = DEFAULT_MII;
 | 
			
		||||
 | 
			
		||||
    do {
 | 
			
		||||
        new_mii.uuid = Common::UUID::Generate();
 | 
			
		||||
    } while (IndexOf(new_mii.uuid) == INVALID_INDEX);
 | 
			
		||||
 | 
			
		||||
    ASSERT(index < MAX_MIIS);
 | 
			
		||||
    database.miis[index] = new_mii;
 | 
			
		||||
    std::stable_partition(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                          [](const MiiStoreData& elem) { return elem.uuid; });
 | 
			
		||||
 | 
			
		||||
    return ConvertStoreDataToInfo(new_mii);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool MiiManager::Empty() const {
 | 
			
		||||
    return Size() == 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool MiiManager::Full() const {
 | 
			
		||||
    return Size() == MAX_MIIS;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void MiiManager::Clear() {
 | 
			
		||||
    std::fill(database.miis.begin(), database.miis.end(), MiiStoreData{});
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
u32 MiiManager::Size() const {
 | 
			
		||||
    return static_cast<u32>(std::count_if(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                                          [](const MiiStoreData& elem) { return elem.uuid; }));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiInfo MiiManager::GetInfo(u32 index) const {
 | 
			
		||||
    return ConvertStoreDataToInfo(GetStoreData(index));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiInfoElement MiiManager::GetInfoElement(u32 index) const {
 | 
			
		||||
    return {GetInfo(index), Source::Database};
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiStoreData MiiManager::GetStoreData(u32 index) const {
 | 
			
		||||
    return database.miis.at(index);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
MiiStoreDataElement MiiManager::GetStoreDataElement(u32 index) const {
 | 
			
		||||
    return {GetStoreData(index), Source::Database};
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool MiiManager::Remove(Common::UUID uuid) {
 | 
			
		||||
    const auto iter = std::find_if(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                                   [uuid](const MiiStoreData& elem) { return elem.uuid == uuid; });
 | 
			
		||||
 | 
			
		||||
    if (iter == database.miis.end())
 | 
			
		||||
        return false;
 | 
			
		||||
 | 
			
		||||
    *iter = MiiStoreData{};
 | 
			
		||||
    std::stable_partition(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                          [](const MiiStoreData& elem) { return elem.uuid; });
 | 
			
		||||
    return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
u32 MiiManager::IndexOf(Common::UUID uuid) const {
 | 
			
		||||
    const auto iter = std::find_if(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                                   [uuid](const MiiStoreData& elem) { return elem.uuid == uuid; });
 | 
			
		||||
 | 
			
		||||
    if (iter == database.miis.end())
 | 
			
		||||
        return INVALID_INDEX;
 | 
			
		||||
 | 
			
		||||
    return static_cast<u32>(std::distance(database.miis.begin(), iter));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
u32 MiiManager::IndexOf(MiiInfo info) const {
 | 
			
		||||
    const auto iter =
 | 
			
		||||
        std::find_if(database.miis.begin(), database.miis.end(), [info](const MiiStoreData& elem) {
 | 
			
		||||
            return ConvertStoreDataToInfo(elem) == info;
 | 
			
		||||
        });
 | 
			
		||||
 | 
			
		||||
    if (iter == database.miis.end())
 | 
			
		||||
        return INVALID_INDEX;
 | 
			
		||||
 | 
			
		||||
    return static_cast<u32>(std::distance(database.miis.begin(), iter));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool MiiManager::Move(Common::UUID uuid, u32 new_index) {
 | 
			
		||||
    const auto index = IndexOf(uuid);
 | 
			
		||||
 | 
			
		||||
    if (index == INVALID_INDEX || new_index >= MAX_MIIS)
 | 
			
		||||
        return false;
 | 
			
		||||
 | 
			
		||||
    const auto moving = database.miis[index];
 | 
			
		||||
    const auto replacing = database.miis[new_index];
 | 
			
		||||
    if (replacing.uuid) {
 | 
			
		||||
        database.miis[index] = replacing;
 | 
			
		||||
        database.miis[new_index] = moving;
 | 
			
		||||
    } else {
 | 
			
		||||
        database.miis[index] = MiiStoreData{};
 | 
			
		||||
        database.miis[new_index] = moving;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    std::stable_partition(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                          [](const MiiStoreData& elem) { return elem.uuid; });
 | 
			
		||||
    return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
bool MiiManager::AddOrReplace(MiiStoreData data) {
 | 
			
		||||
    const auto index = IndexOf(data.uuid);
 | 
			
		||||
 | 
			
		||||
    if (index == INVALID_INDEX) {
 | 
			
		||||
        const auto size = Size();
 | 
			
		||||
        if (size == MAX_MIIS)
 | 
			
		||||
            return false;
 | 
			
		||||
        database.miis[size] = data;
 | 
			
		||||
    } else {
 | 
			
		||||
        database.miis[index] = data;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return true;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void MiiManager::WriteToFile() {
 | 
			
		||||
    const auto raw_path =
 | 
			
		||||
        FileUtil::GetUserPath(FileUtil::UserPath::NANDDir) + "/system/save/8000000000000030";
 | 
			
		||||
    if (FileUtil::Exists(raw_path) && !FileUtil::IsDirectory(raw_path))
 | 
			
		||||
        FileUtil::Delete(raw_path);
 | 
			
		||||
 | 
			
		||||
    const auto path = FileUtil::GetUserPath(FileUtil::UserPath::NANDDir) + MII_SAVE_DATABASE_PATH;
 | 
			
		||||
 | 
			
		||||
    if (!FileUtil::CreateFullPath(path)) {
 | 
			
		||||
        LOG_WARNING(Service_Mii,
 | 
			
		||||
                    "Failed to create full path of MiiDatabase.dat. Create the directory "
 | 
			
		||||
                    "nand/system/save/8000000000000030 to mitigate this "
 | 
			
		||||
                    "issue.");
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    FileUtil::IOFile save(path, "wb");
 | 
			
		||||
 | 
			
		||||
    if (!save.IsOpen()) {
 | 
			
		||||
        LOG_WARNING(Service_Mii, "Failed to write save data to file... No changes to user data "
 | 
			
		||||
                                 "made in current session will be saved.");
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    save.Resize(sizeof(MiiDatabase));
 | 
			
		||||
    save.WriteBytes(&database, sizeof(MiiDatabase));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void MiiManager::ReadFromFile() {
 | 
			
		||||
    FileUtil::IOFile save(
 | 
			
		||||
        FileUtil::GetUserPath(FileUtil::UserPath::NANDDir) + MII_SAVE_DATABASE_PATH, "rb");
 | 
			
		||||
 | 
			
		||||
    if (!save.IsOpen()) {
 | 
			
		||||
        LOG_WARNING(Service_ACC, "Failed to load profile data from save data... Generating new "
 | 
			
		||||
                                 "blank Mii database with no Miis.");
 | 
			
		||||
        std::memcpy(&database, &DEFAULT_MII_DATABASE, sizeof(MiiDatabase));
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (save.ReadBytes(&database, sizeof(MiiDatabase)) != sizeof(MiiDatabase)) {
 | 
			
		||||
        LOG_WARNING(Service_ACC, "MiiDatabase.dat is smaller than expected... Generating new blank "
 | 
			
		||||
                                 "Mii database with no Miis.");
 | 
			
		||||
        std::memcpy(&database, &DEFAULT_MII_DATABASE, sizeof(MiiDatabase));
 | 
			
		||||
        return;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    std::stable_partition(database.miis.begin(), database.miis.end(),
 | 
			
		||||
                          [](const MiiStoreData& elem) { return elem.uuid; });
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
} // namespace Service::Mii
 | 
			
		||||
							
								
								
									
										253
									
								
								src/core/hle/service/mii/mii_manager.h
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										253
									
								
								src/core/hle/service/mii/mii_manager.h
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,253 @@
 | 
			
		||||
// Copyright 2018 yuzu emulator team
 | 
			
		||||
// Licensed under GPLv2 or any later version
 | 
			
		||||
// Refer to the license.txt file included.
 | 
			
		||||
 | 
			
		||||
#pragma once
 | 
			
		||||
 | 
			
		||||
#include "common/bit_field.h"
 | 
			
		||||
#include "common/common_funcs.h"
 | 
			
		||||
#include "common/uuid.h"
 | 
			
		||||
 | 
			
		||||
namespace Service::Mii {
 | 
			
		||||
 | 
			
		||||
constexpr std::size_t MAX_MIIS = 100;
 | 
			
		||||
constexpr u32 INVALID_INDEX = 0xFFFFFFFF;
 | 
			
		||||
 | 
			
		||||
struct RandomParameters {
 | 
			
		||||
    u32 unknown_1;
 | 
			
		||||
    u32 unknown_2;
 | 
			
		||||
    u32 unknown_3;
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(RandomParameters) == 0xC, "RandomParameters has incorrect size.");
 | 
			
		||||
 | 
			
		||||
enum class Source : u32 {
 | 
			
		||||
    Database = 0,
 | 
			
		||||
    Default = 1,
 | 
			
		||||
    Account = 2,
 | 
			
		||||
    Friend = 3,
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
struct MiiInfo {
 | 
			
		||||
    Common::UUID uuid;
 | 
			
		||||
    std::array<char16_t, 11> name;
 | 
			
		||||
    u8 font_region;
 | 
			
		||||
    u8 favorite_color;
 | 
			
		||||
    u8 gender;
 | 
			
		||||
    u8 height;
 | 
			
		||||
    u8 weight;
 | 
			
		||||
    u8 mii_type;
 | 
			
		||||
    u8 mii_region;
 | 
			
		||||
    u8 face_type;
 | 
			
		||||
    u8 face_color;
 | 
			
		||||
    u8 face_wrinkle;
 | 
			
		||||
    u8 face_makeup;
 | 
			
		||||
    u8 hair_type;
 | 
			
		||||
    u8 hair_color;
 | 
			
		||||
    bool hair_flip;
 | 
			
		||||
    u8 eye_type;
 | 
			
		||||
    u8 eye_color;
 | 
			
		||||
    u8 eye_scale;
 | 
			
		||||
    u8 eye_aspect_ratio;
 | 
			
		||||
    u8 eye_rotate;
 | 
			
		||||
    u8 eye_x;
 | 
			
		||||
    u8 eye_y;
 | 
			
		||||
    u8 eyebrow_type;
 | 
			
		||||
    u8 eyebrow_color;
 | 
			
		||||
    u8 eyebrow_scale;
 | 
			
		||||
    u8 eyebrow_aspect_ratio;
 | 
			
		||||
    u8 eyebrow_rotate;
 | 
			
		||||
    u8 eyebrow_x;
 | 
			
		||||
    u8 eyebrow_y;
 | 
			
		||||
    u8 nose_type;
 | 
			
		||||
    u8 nose_scale;
 | 
			
		||||
    u8 nose_y;
 | 
			
		||||
    u8 mouth_type;
 | 
			
		||||
    u8 mouth_color;
 | 
			
		||||
    u8 mouth_scale;
 | 
			
		||||
    u8 mouth_aspect_ratio;
 | 
			
		||||
    u8 mouth_y;
 | 
			
		||||
    u8 facial_hair_color;
 | 
			
		||||
    u8 beard_type;
 | 
			
		||||
    u8 mustache_type;
 | 
			
		||||
    u8 mustache_scale;
 | 
			
		||||
    u8 mustache_y;
 | 
			
		||||
    u8 glasses_type;
 | 
			
		||||
    u8 glasses_color;
 | 
			
		||||
    u8 glasses_scale;
 | 
			
		||||
    u8 glasses_y;
 | 
			
		||||
    u8 mole_type;
 | 
			
		||||
    u8 mole_scale;
 | 
			
		||||
    u8 mole_x;
 | 
			
		||||
    u8 mole_y;
 | 
			
		||||
    INSERT_PADDING_BYTES(1);
 | 
			
		||||
 | 
			
		||||
    std::u16string Name() const;
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(MiiInfo) == 0x58, "MiiInfo has incorrect size.");
 | 
			
		||||
 | 
			
		||||
bool operator==(const MiiInfo& lhs, const MiiInfo& rhs);
 | 
			
		||||
bool operator!=(const MiiInfo& lhs, const MiiInfo& rhs);
 | 
			
		||||
 | 
			
		||||
#pragma pack(push, 4)
 | 
			
		||||
struct MiiInfoElement {
 | 
			
		||||
    MiiInfo info;
 | 
			
		||||
    Source source;
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(MiiInfoElement) == 0x5C, "MiiInfoElement has incorrect size.");
 | 
			
		||||
 | 
			
		||||
struct MiiStoreBitFields {
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_0;
 | 
			
		||||
 | 
			
		||||
        BitField<24, 8, u32> hair_type;
 | 
			
		||||
        BitField<23, 1, u32> mole_type;
 | 
			
		||||
        BitField<16, 7, u32> height;
 | 
			
		||||
        BitField<15, 1, u32> hair_flip;
 | 
			
		||||
        BitField<8, 7, u32> weight;
 | 
			
		||||
        BitField<0, 7, u32> hair_color;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_1;
 | 
			
		||||
 | 
			
		||||
        BitField<31, 1, u32> gender;
 | 
			
		||||
        BitField<24, 7, u32> eye_color;
 | 
			
		||||
        BitField<16, 7, u32> eyebrow_color;
 | 
			
		||||
        BitField<8, 7, u32> mouth_color;
 | 
			
		||||
        BitField<0, 7, u32> facial_hair_color;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_2;
 | 
			
		||||
 | 
			
		||||
        BitField<31, 1, u32> mii_type;
 | 
			
		||||
        BitField<24, 7, u32> glasses_color;
 | 
			
		||||
        BitField<22, 2, u32> font_region;
 | 
			
		||||
        BitField<16, 6, u32> eye_type;
 | 
			
		||||
        BitField<14, 2, u32> mii_region;
 | 
			
		||||
        BitField<8, 6, u32> mouth_type;
 | 
			
		||||
        BitField<5, 3, u32> glasses_scale;
 | 
			
		||||
        BitField<0, 5, u32> eye_y;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_3;
 | 
			
		||||
 | 
			
		||||
        BitField<29, 3, u32> mustache_type;
 | 
			
		||||
        BitField<24, 5, u32> eyebrow_type;
 | 
			
		||||
        BitField<21, 3, u32> beard_type;
 | 
			
		||||
        BitField<16, 5, u32> nose_type;
 | 
			
		||||
        BitField<13, 3, u32> mouth_aspect;
 | 
			
		||||
        BitField<8, 5, u32> nose_y;
 | 
			
		||||
        BitField<5, 3, u32> eyebrow_aspect;
 | 
			
		||||
        BitField<0, 5, u32> mouth_y;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_4;
 | 
			
		||||
 | 
			
		||||
        BitField<29, 3, u32> eye_rotate;
 | 
			
		||||
        BitField<24, 5, u32> mustache_y;
 | 
			
		||||
        BitField<21, 3, u32> eye_aspect;
 | 
			
		||||
        BitField<16, 5, u32> glasses_y;
 | 
			
		||||
        BitField<13, 3, u32> eye_scale;
 | 
			
		||||
        BitField<8, 5, u32> mole_x;
 | 
			
		||||
        BitField<0, 5, u32> mole_y;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_5;
 | 
			
		||||
 | 
			
		||||
        BitField<24, 5, u32> glasses_type;
 | 
			
		||||
        BitField<20, 4, u32> face_type;
 | 
			
		||||
        BitField<16, 4, u32> favorite_color;
 | 
			
		||||
        BitField<12, 4, u32> face_wrinkle;
 | 
			
		||||
        BitField<8, 4, u32> face_color;
 | 
			
		||||
        BitField<4, 4, u32> eye_x;
 | 
			
		||||
        BitField<0, 4, u32> face_makeup;
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
    union {
 | 
			
		||||
        u32 word_6;
 | 
			
		||||
 | 
			
		||||
        BitField<28, 4, u32> eyebrow_rotate;
 | 
			
		||||
        BitField<24, 4, u32> eyebrow_scale;
 | 
			
		||||
        BitField<20, 4, u32> eyebrow_y;
 | 
			
		||||
        BitField<16, 4, u32> eyebrow_x;
 | 
			
		||||
        BitField<12, 4, u32> mouth_scale;
 | 
			
		||||
        BitField<8, 4, u32> nose_scale;
 | 
			
		||||
        BitField<4, 4, u32> mole_scale;
 | 
			
		||||
        BitField<0, 4, u32> mustache_scale;
 | 
			
		||||
    };
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(MiiStoreBitFields) == 0x1C, "MiiStoreBitFields has incorrect size.");
 | 
			
		||||
 | 
			
		||||
struct MiiStoreData {
 | 
			
		||||
    // This corresponds to the above structure MiiStoreBitFields. I did it like this because the
 | 
			
		||||
    // BitField<> type makes this (and any thing that contains it) not trivially copyable, which is
 | 
			
		||||
    // not suitable for our uses.
 | 
			
		||||
    std::array<u8, 0x1C> data;
 | 
			
		||||
    static_assert(sizeof(MiiStoreBitFields) == sizeof(data), "data field has incorrect size.");
 | 
			
		||||
 | 
			
		||||
    std::array<char16_t, 10> name;
 | 
			
		||||
    Common::UUID uuid;
 | 
			
		||||
    u16 crc_1;
 | 
			
		||||
    u16 crc_2;
 | 
			
		||||
 | 
			
		||||
    std::u16string Name() const;
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(MiiStoreData) == 0x44, "MiiStoreData has incorrect size.");
 | 
			
		||||
 | 
			
		||||
struct MiiStoreDataElement {
 | 
			
		||||
    MiiStoreData data;
 | 
			
		||||
    Source source;
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(MiiStoreDataElement) == 0x48, "MiiStoreDataElement has incorrect size.");
 | 
			
		||||
 | 
			
		||||
struct MiiDatabase {
 | 
			
		||||
    u32 magic; // 'NFDB'
 | 
			
		||||
    std::array<MiiStoreData, MAX_MIIS> miis;
 | 
			
		||||
    INSERT_PADDING_BYTES(1);
 | 
			
		||||
    u8 count;
 | 
			
		||||
    u16 crc;
 | 
			
		||||
};
 | 
			
		||||
static_assert(sizeof(MiiDatabase) == 0x1A98, "MiiDatabase has incorrect size.");
 | 
			
		||||
#pragma pack(pop)
 | 
			
		||||
 | 
			
		||||
// The Mii manager is responsible for loading and storing the Miis to the database in NAND along
 | 
			
		||||
// with providing an easy interface for HLE emulation of the mii service.
 | 
			
		||||
class MiiManager {
 | 
			
		||||
public:
 | 
			
		||||
    MiiManager();
 | 
			
		||||
    ~MiiManager();
 | 
			
		||||
 | 
			
		||||
    MiiInfo CreateRandom(RandomParameters params);
 | 
			
		||||
    MiiInfo CreateDefault(u32 index);
 | 
			
		||||
 | 
			
		||||
    bool Empty() const;
 | 
			
		||||
    bool Full() const;
 | 
			
		||||
 | 
			
		||||
    void Clear();
 | 
			
		||||
 | 
			
		||||
    u32 Size() const;
 | 
			
		||||
 | 
			
		||||
    MiiInfo GetInfo(u32 index) const;
 | 
			
		||||
    MiiInfoElement GetInfoElement(u32 index) const;
 | 
			
		||||
    MiiStoreData GetStoreData(u32 index) const;
 | 
			
		||||
    MiiStoreDataElement GetStoreDataElement(u32 index) const;
 | 
			
		||||
 | 
			
		||||
    bool Remove(Common::UUID uuid);
 | 
			
		||||
    u32 IndexOf(Common::UUID uuid) const;
 | 
			
		||||
    u32 IndexOf(MiiInfo info) const;
 | 
			
		||||
 | 
			
		||||
    bool Move(Common::UUID uuid, u32 new_index);
 | 
			
		||||
    bool AddOrReplace(MiiStoreData data);
 | 
			
		||||
 | 
			
		||||
private:
 | 
			
		||||
    void WriteToFile();
 | 
			
		||||
    void ReadFromFile();
 | 
			
		||||
 | 
			
		||||
    MiiDatabase database;
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
}; // namespace Service::Mii
 | 
			
		||||
		Reference in New Issue
	
	Block a user