521 lines
15 KiB
C++
521 lines
15 KiB
C++
#include <hex/api/imhex_api.hpp>
|
|
#include <hex/api/content_registry.hpp>
|
|
|
|
#include <hex/api/event.hpp>
|
|
#include <hex/api/task.hpp>
|
|
#include <hex/providers/provider.hpp>
|
|
|
|
#include <utility>
|
|
#include <unistd.h>
|
|
|
|
#include <imgui.h>
|
|
|
|
#include <nlohmann/json.hpp>
|
|
|
|
#if defined(OS_WINDOWS)
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#include <windows.h>
|
|
#include <shellapi.h>
|
|
#endif
|
|
|
|
namespace hex {
|
|
|
|
|
|
namespace ImHexApi::HexEditor {
|
|
|
|
Highlighting::Highlighting(Region region, color_t color)
|
|
: m_region(region), m_color(color) {
|
|
}
|
|
|
|
Tooltip::Tooltip(Region region, std::string value, color_t color) : m_region(region), m_value(std::move(value)), m_color(color) {
|
|
|
|
}
|
|
|
|
namespace impl {
|
|
|
|
static std::map<u32, Highlighting> s_backgroundHighlights;
|
|
std::map<u32, Highlighting> &getBackgroundHighlights() {
|
|
return s_backgroundHighlights;
|
|
}
|
|
|
|
static std::map<u32, HighlightingFunction> s_backgroundHighlightingFunctions;
|
|
std::map<u32, HighlightingFunction> &getBackgroundHighlightingFunctions() {
|
|
return s_backgroundHighlightingFunctions;
|
|
}
|
|
|
|
static std::map<u32, Highlighting> s_foregroundHighlights;
|
|
std::map<u32, Highlighting> &getForegroundHighlights() {
|
|
return s_foregroundHighlights;
|
|
}
|
|
|
|
static std::map<u32, HighlightingFunction> s_foregroundHighlightingFunctions;
|
|
std::map<u32, HighlightingFunction> &getForegroundHighlightingFunctions() {
|
|
return s_foregroundHighlightingFunctions;
|
|
}
|
|
|
|
static std::map<u32, Tooltip> s_tooltips;
|
|
std::map<u32, Tooltip> &getTooltips() {
|
|
return s_tooltips;
|
|
}
|
|
|
|
static std::map<u32, TooltipFunction> s_tooltipFunctions;
|
|
std::map<u32, TooltipFunction> &getTooltipFunctions() {
|
|
return s_tooltipFunctions;
|
|
}
|
|
|
|
static std::optional<ProviderRegion> s_currentSelection;
|
|
void setCurrentSelection(std::optional<ProviderRegion> region) {
|
|
s_currentSelection = region;
|
|
}
|
|
|
|
}
|
|
|
|
u32 addBackgroundHighlight(const Region ®ion, color_t color) {
|
|
static u32 id = 0;
|
|
|
|
id++;
|
|
|
|
impl::getBackgroundHighlights().insert({
|
|
id, Highlighting {region, color}
|
|
});
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
|
|
return id;
|
|
}
|
|
|
|
void removeBackgroundHighlight(u32 id) {
|
|
impl::getBackgroundHighlights().erase(id);
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
}
|
|
|
|
u32 addBackgroundHighlightingProvider(const impl::HighlightingFunction &function) {
|
|
static u32 id = 0;
|
|
|
|
id++;
|
|
|
|
impl::getBackgroundHighlightingFunctions().insert({ id, function });
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
|
|
return id;
|
|
}
|
|
|
|
void removeBackgroundHighlightingProvider(u32 id) {
|
|
impl::getBackgroundHighlightingFunctions().erase(id);
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
}
|
|
|
|
u32 addForegroundHighlight(const Region ®ion, color_t color) {
|
|
static u32 id = 0;
|
|
|
|
id++;
|
|
|
|
impl::getForegroundHighlights().insert({
|
|
id, Highlighting {region, color}
|
|
});
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
|
|
return id;
|
|
}
|
|
|
|
void removeForegroundHighlight(u32 id) {
|
|
impl::getForegroundHighlights().erase(id);
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
}
|
|
|
|
u32 addForegroundHighlightingProvider(const impl::HighlightingFunction &function) {
|
|
static u32 id = 0;
|
|
|
|
id++;
|
|
|
|
impl::getForegroundHighlightingFunctions().insert({ id, function });
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
|
|
return id;
|
|
}
|
|
|
|
void removeForegroundHighlightingProvider(u32 id) {
|
|
impl::getForegroundHighlightingFunctions().erase(id);
|
|
|
|
EventManager::post<EventHighlightingChanged>();
|
|
}
|
|
|
|
static u32 tooltipId = 0;
|
|
u32 addTooltip(Region region, std::string value, color_t color) {
|
|
tooltipId++;
|
|
impl::getTooltips().insert({ tooltipId, { region, std::move(value), color } });
|
|
|
|
return tooltipId;
|
|
}
|
|
|
|
void removeTooltip(u32 id) {
|
|
impl::getTooltips().erase(id);
|
|
}
|
|
|
|
static u32 tooltipFunctionId;
|
|
u32 addTooltipProvider(TooltipFunction function) {
|
|
tooltipFunctionId++;
|
|
impl::getTooltipFunctions().insert({ tooltipFunctionId, std::move(function) });
|
|
|
|
return tooltipFunctionId;
|
|
}
|
|
|
|
void removeTooltipProvider(u32 id) {
|
|
impl::getTooltipFunctions().erase(id);
|
|
}
|
|
|
|
bool isSelectionValid() {
|
|
return getSelection().has_value();
|
|
}
|
|
|
|
std::optional<ProviderRegion> getSelection() {
|
|
return impl::s_currentSelection;
|
|
}
|
|
|
|
void clearSelection() {
|
|
impl::s_currentSelection.reset();
|
|
}
|
|
|
|
void setSelection(const Region ®ion, prv::Provider *provider) {
|
|
setSelection(ProviderRegion { region, provider == nullptr ? Provider::get() : provider });
|
|
}
|
|
|
|
void setSelection(const ProviderRegion ®ion) {
|
|
EventManager::post<RequestSelectionChange>(region);
|
|
}
|
|
|
|
void setSelection(u64 address, size_t size, prv::Provider *provider) {
|
|
setSelection({ { address, size }, provider == nullptr ? Provider::get() : provider });
|
|
}
|
|
|
|
}
|
|
|
|
|
|
namespace ImHexApi::Bookmarks {
|
|
|
|
void add(Region region, const std::string &name, const std::string &comment, u32 color) {
|
|
EventManager::post<RequestAddBookmark>(region, name, comment, color);
|
|
}
|
|
|
|
void add(u64 address, size_t size, const std::string &name, const std::string &comment, u32 color) {
|
|
add(Region { address, size }, name, comment, color);
|
|
}
|
|
|
|
}
|
|
|
|
|
|
namespace ImHexApi::Provider {
|
|
|
|
static i64 s_currentProvider = -1;
|
|
static std::vector<prv::Provider *> s_providers;
|
|
|
|
namespace impl {
|
|
|
|
static prv::Provider *s_closingProvider = nullptr;
|
|
void resetClosingProvider() {
|
|
s_closingProvider = nullptr;
|
|
}
|
|
|
|
prv::Provider* getClosingProvider() {
|
|
return s_closingProvider;
|
|
}
|
|
|
|
}
|
|
|
|
prv::Provider *get() {
|
|
if (!ImHexApi::Provider::isValid())
|
|
return nullptr;
|
|
|
|
return s_providers[s_currentProvider];
|
|
}
|
|
|
|
const std::vector<prv::Provider *> &getProviders() {
|
|
return s_providers;
|
|
}
|
|
|
|
void setCurrentProvider(u32 index) {
|
|
if (TaskManager::getRunningTaskCount() > 0)
|
|
return;
|
|
|
|
if (index < s_providers.size() && s_currentProvider != index) {
|
|
auto oldProvider = get();
|
|
s_currentProvider = index;
|
|
EventManager::post<EventProviderChanged>(oldProvider, get());
|
|
}
|
|
}
|
|
|
|
bool isValid() {
|
|
return !s_providers.empty() && s_currentProvider >= 0 && s_currentProvider < i64(s_providers.size());
|
|
}
|
|
|
|
void markDirty() {
|
|
get()->markDirty();
|
|
}
|
|
|
|
void resetDirty() {
|
|
for (auto &provider : s_providers)
|
|
provider->markDirty(false);
|
|
}
|
|
|
|
bool isDirty() {
|
|
return std::ranges::any_of(s_providers, [](const auto &provider) {
|
|
return provider->isDirty();
|
|
});
|
|
}
|
|
|
|
void add(prv::Provider *provider, bool skipLoadInterface) {
|
|
if (TaskManager::getRunningTaskCount() > 0)
|
|
return;
|
|
|
|
if (skipLoadInterface)
|
|
provider->skipLoadInterface();
|
|
|
|
s_providers.push_back(provider);
|
|
EventManager::post<EventProviderCreated>(provider);
|
|
|
|
setCurrentProvider(s_providers.size() - 1);
|
|
}
|
|
|
|
void remove(prv::Provider *provider, bool noQuestions) {
|
|
if (provider == nullptr)
|
|
return;
|
|
|
|
if (TaskManager::getRunningTaskCount() > 0)
|
|
return;
|
|
|
|
if (!noQuestions) {
|
|
impl::s_closingProvider = provider;
|
|
|
|
bool shouldClose = true;
|
|
EventManager::post<EventProviderClosing>(provider, &shouldClose);
|
|
if (!shouldClose)
|
|
return;
|
|
}
|
|
|
|
auto it = std::find(s_providers.begin(), s_providers.end(), provider);
|
|
if (it == s_providers.end())
|
|
return;
|
|
|
|
if (!s_providers.empty() && it - s_providers.begin() == s_currentProvider)
|
|
setCurrentProvider(0);
|
|
|
|
s_providers.erase(it);
|
|
if (s_providers.empty())
|
|
EventManager::post<EventProviderChanged>(provider, nullptr);
|
|
|
|
provider->close();
|
|
EventManager::post<EventProviderClosed>(provider);
|
|
|
|
TaskManager::runWhenTasksFinished([provider] {
|
|
EventManager::post<EventProviderDeleted>(provider);
|
|
delete provider;
|
|
});
|
|
}
|
|
|
|
prv::Provider* createProvider(const std::string &unlocalizedName, bool skipLoadInterface) {
|
|
prv::Provider* result = nullptr;
|
|
EventManager::post<RequestCreateProvider>(unlocalizedName, skipLoadInterface, &result);
|
|
|
|
return result;
|
|
}
|
|
|
|
}
|
|
|
|
namespace ImHexApi::System {
|
|
|
|
namespace impl {
|
|
|
|
static ImVec2 s_mainWindowPos;
|
|
static ImVec2 s_mainWindowSize;
|
|
void setMainWindowPosition(u32 x, u32 y) {
|
|
s_mainWindowPos = ImVec2(x, y);
|
|
}
|
|
|
|
void setMainWindowSize(u32 width, u32 height) {
|
|
s_mainWindowSize = ImVec2(width, height);
|
|
}
|
|
|
|
static ImGuiID s_mainDockSpaceId;
|
|
void setMainDockSpaceId(ImGuiID id) {
|
|
s_mainDockSpaceId = id;
|
|
}
|
|
|
|
|
|
static float s_globalScale = 1.0;
|
|
void setGlobalScale(float scale) {
|
|
s_globalScale = scale;
|
|
}
|
|
|
|
static float s_nativeScale = 1.0;
|
|
void setNativeScale(float scale) {
|
|
s_nativeScale = scale;
|
|
}
|
|
|
|
|
|
static ProgramArguments s_programArguments;
|
|
void setProgramArguments(int argc, char **argv, char **envp) {
|
|
s_programArguments.argc = argc;
|
|
s_programArguments.argv = argv;
|
|
s_programArguments.envp = envp;
|
|
}
|
|
|
|
static bool s_borderlessWindowMode;
|
|
void setBorderlessWindowMode(bool enabled) {
|
|
s_borderlessWindowMode = enabled;
|
|
}
|
|
|
|
static std::fs::path s_customFontPath;
|
|
void setCustomFontPath(const std::fs::path &path) {
|
|
s_customFontPath = path;
|
|
}
|
|
|
|
static float s_fontSize = DefaultFontSize;
|
|
void setFontSize(float size) {
|
|
s_fontSize = size;
|
|
}
|
|
|
|
static std::string s_gpuVendor;
|
|
void setGPUVendor(const std::string &vendor) {
|
|
s_gpuVendor = vendor;
|
|
}
|
|
|
|
static bool s_portableVersion = false;
|
|
void setPortableVersion(bool enabled) {
|
|
s_portableVersion = enabled;
|
|
}
|
|
|
|
void addInitArgument(const std::string &key, const std::string &value) {
|
|
static std::mutex initArgumentsMutex;
|
|
std::scoped_lock lock(initArgumentsMutex);
|
|
|
|
getInitArguments()[key] = value;
|
|
}
|
|
|
|
}
|
|
|
|
void closeImHex(bool noQuestions) {
|
|
EventManager::post<RequestCloseImHex>(noQuestions);
|
|
}
|
|
|
|
void restartImHex() {
|
|
EventManager::post<RequestRestartImHex>();
|
|
EventManager::post<RequestCloseImHex>(false);
|
|
}
|
|
|
|
void setTaskBarProgress(TaskProgressState state, TaskProgressType type, u32 progress) {
|
|
EventManager::post<EventSetTaskBarIconState>(u32(state), u32(type), progress);
|
|
}
|
|
|
|
const ProgramArguments &getProgramArguments() {
|
|
return impl::s_programArguments;
|
|
}
|
|
|
|
std::optional<std::u8string> getProgramArgument(int index) {
|
|
if (index >= impl::s_programArguments.argc) {
|
|
return std::nullopt;
|
|
}
|
|
|
|
#if defined(OS_WINDOWS)
|
|
std::wstring wideArg = ::CommandLineToArgvW(::GetCommandLineW(), &impl::s_programArguments.argc)[index];
|
|
std::string byteArg = std::wstring_convert<std::codecvt_utf8_utf16<wchar_t>, wchar_t>().to_bytes(wideArg);
|
|
|
|
return std::u8string(byteArg.begin(), byteArg.end());
|
|
#else
|
|
return std::u8string(reinterpret_cast<const char8_t *>(impl::s_programArguments.argv[index]));
|
|
#endif
|
|
}
|
|
|
|
|
|
static float s_targetFPS = 14.0F;
|
|
|
|
float getTargetFPS() {
|
|
return s_targetFPS;
|
|
}
|
|
|
|
void setTargetFPS(float fps) {
|
|
s_targetFPS = fps;
|
|
}
|
|
|
|
float getGlobalScale() {
|
|
return impl::s_globalScale;
|
|
}
|
|
|
|
float getNativeScale() {
|
|
return impl::s_nativeScale;
|
|
}
|
|
|
|
|
|
ImVec2 getMainWindowPosition() {
|
|
return impl::s_mainWindowPos;
|
|
}
|
|
|
|
ImVec2 getMainWindowSize() {
|
|
return impl::s_mainWindowSize;
|
|
}
|
|
|
|
|
|
ImGuiID getMainDockSpaceId() {
|
|
return impl::s_mainDockSpaceId;
|
|
}
|
|
|
|
bool isBorderlessWindowModeEnabled() {
|
|
return impl::s_borderlessWindowMode;
|
|
}
|
|
|
|
std::map<std::string, std::string> &getInitArguments() {
|
|
static std::map<std::string, std::string> initArgs;
|
|
|
|
return initArgs;
|
|
}
|
|
|
|
const std::fs::path &getCustomFontPath() {
|
|
return impl::s_customFontPath;
|
|
}
|
|
|
|
float getFontSize() {
|
|
return impl::s_fontSize;
|
|
}
|
|
|
|
|
|
static bool s_systemThemeDetection;
|
|
|
|
|
|
void enableSystemThemeDetection(bool enabled) {
|
|
s_systemThemeDetection = enabled;
|
|
|
|
EventManager::post<EventSettingsChanged>();
|
|
EventManager::post<EventOSThemeChanged>();
|
|
}
|
|
|
|
bool usesSystemThemeDetection() {
|
|
return s_systemThemeDetection;
|
|
}
|
|
|
|
|
|
static std::vector<std::fs::path> s_additionalFolderPaths;
|
|
const std::vector<std::fs::path> &getAdditionalFolderPaths() {
|
|
return s_additionalFolderPaths;
|
|
}
|
|
|
|
void setAdditionalFolderPaths(const std::vector<std::fs::path> &paths) {
|
|
s_additionalFolderPaths = paths;
|
|
}
|
|
|
|
|
|
const std::string &getGPUVendor() {
|
|
return impl::s_gpuVendor;
|
|
}
|
|
|
|
bool isPortableVersion() {
|
|
return impl::s_portableVersion;
|
|
}
|
|
}
|
|
|
|
}
|