mirror of
https://github.com/GPUOpen-LibrariesAndSDKs/VulkanMemoryAllocator.git
synced 2024-09-20 22:52:16 +00:00
181 lines
5.0 KiB
C++
181 lines
5.0 KiB
C++
//
|
|
// Copyright (c) 2017-2019 Advanced Micro Devices, Inc. All rights reserved.
|
|
//
|
|
// Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
// of this software and associated documentation files (the "Software"), to deal
|
|
// in the Software without restriction, including without limitation the rights
|
|
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
// copies of the Software, and to permit persons to whom the Software is
|
|
// furnished to do so, subject to the following conditions:
|
|
//
|
|
// The above copyright notice and this permission notice shall be included in
|
|
// all copies or substantial portions of the Software.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
// THE SOFTWARE.
|
|
//
|
|
|
|
#include "Common.h"
|
|
|
|
#ifdef _WIN32
|
|
|
|
void ReadFile(std::vector<char>& out, const char* fileName)
|
|
{
|
|
std::ifstream file(fileName, std::ios::ate | std::ios::binary);
|
|
assert(file.is_open());
|
|
size_t fileSize = (size_t)file.tellg();
|
|
if(fileSize > 0)
|
|
{
|
|
out.resize(fileSize);
|
|
file.seekg(0);
|
|
file.read(out.data(), fileSize);
|
|
}
|
|
else
|
|
out.clear();
|
|
}
|
|
|
|
void SetConsoleColor(CONSOLE_COLOR color)
|
|
{
|
|
WORD attr = 0;
|
|
switch(color)
|
|
{
|
|
case CONSOLE_COLOR::INFO:
|
|
attr = FOREGROUND_INTENSITY;;
|
|
break;
|
|
case CONSOLE_COLOR::NORMAL:
|
|
attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE;
|
|
break;
|
|
case CONSOLE_COLOR::WARNING:
|
|
attr = FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_INTENSITY;
|
|
break;
|
|
case CONSOLE_COLOR::ERROR_:
|
|
attr = FOREGROUND_RED | FOREGROUND_INTENSITY;
|
|
break;
|
|
default:
|
|
assert(0);
|
|
}
|
|
|
|
HANDLE out = GetStdHandle(STD_OUTPUT_HANDLE);
|
|
SetConsoleTextAttribute(out, attr);
|
|
}
|
|
|
|
void PrintMessage(CONSOLE_COLOR color, const char* msg)
|
|
{
|
|
if(color != CONSOLE_COLOR::NORMAL)
|
|
SetConsoleColor(color);
|
|
|
|
printf("%s\n", msg);
|
|
|
|
if (color != CONSOLE_COLOR::NORMAL)
|
|
SetConsoleColor(CONSOLE_COLOR::NORMAL);
|
|
}
|
|
|
|
void PrintMessage(CONSOLE_COLOR color, const wchar_t* msg)
|
|
{
|
|
if(color != CONSOLE_COLOR::NORMAL)
|
|
SetConsoleColor(color);
|
|
|
|
wprintf(L"%s\n", msg);
|
|
|
|
if (color != CONSOLE_COLOR::NORMAL)
|
|
SetConsoleColor(CONSOLE_COLOR::NORMAL);
|
|
}
|
|
|
|
static const size_t CONSOLE_SMALL_BUF_SIZE = 256;
|
|
|
|
void PrintMessageV(CONSOLE_COLOR color, const char* format, va_list argList)
|
|
{
|
|
size_t dstLen = (size_t)::_vscprintf(format, argList);
|
|
if(dstLen)
|
|
{
|
|
bool useSmallBuf = dstLen < CONSOLE_SMALL_BUF_SIZE;
|
|
char smallBuf[CONSOLE_SMALL_BUF_SIZE];
|
|
std::vector<char> bigBuf(useSmallBuf ? 0 : dstLen + 1);
|
|
char* bufPtr = useSmallBuf ? smallBuf : bigBuf.data();
|
|
::vsprintf_s(bufPtr, dstLen + 1, format, argList);
|
|
PrintMessage(color, bufPtr);
|
|
}
|
|
}
|
|
|
|
void PrintMessageV(CONSOLE_COLOR color, const wchar_t* format, va_list argList)
|
|
{
|
|
size_t dstLen = (size_t)::_vcwprintf(format, argList);
|
|
if(dstLen)
|
|
{
|
|
bool useSmallBuf = dstLen < CONSOLE_SMALL_BUF_SIZE;
|
|
wchar_t smallBuf[CONSOLE_SMALL_BUF_SIZE];
|
|
std::vector<wchar_t> bigBuf(useSmallBuf ? 0 : dstLen + 1);
|
|
wchar_t* bufPtr = useSmallBuf ? smallBuf : bigBuf.data();
|
|
::vswprintf_s(bufPtr, dstLen + 1, format, argList);
|
|
PrintMessage(color, bufPtr);
|
|
}
|
|
}
|
|
|
|
void PrintMessageF(CONSOLE_COLOR color, const char* format, ...)
|
|
{
|
|
va_list argList;
|
|
va_start(argList, format);
|
|
PrintMessageV(color, format, argList);
|
|
va_end(argList);
|
|
}
|
|
|
|
void PrintMessageF(CONSOLE_COLOR color, const wchar_t* format, ...)
|
|
{
|
|
va_list argList;
|
|
va_start(argList, format);
|
|
PrintMessageV(color, format, argList);
|
|
va_end(argList);
|
|
}
|
|
|
|
void PrintWarningF(const char* format, ...)
|
|
{
|
|
va_list argList;
|
|
va_start(argList, format);
|
|
PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
|
|
va_end(argList);
|
|
}
|
|
|
|
void PrintWarningF(const wchar_t* format, ...)
|
|
{
|
|
va_list argList;
|
|
va_start(argList, format);
|
|
PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
|
|
va_end(argList);
|
|
}
|
|
|
|
void PrintErrorF(const char* format, ...)
|
|
{
|
|
va_list argList;
|
|
va_start(argList, format);
|
|
PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
|
|
va_end(argList);
|
|
}
|
|
|
|
void PrintErrorF(const wchar_t* format, ...)
|
|
{
|
|
va_list argList;
|
|
va_start(argList, format);
|
|
PrintMessageV(CONSOLE_COLOR::WARNING, format, argList);
|
|
va_end(argList);
|
|
}
|
|
|
|
void SaveFile(const wchar_t* filePath, const void* data, size_t dataSize)
|
|
{
|
|
FILE* f = nullptr;
|
|
_wfopen_s(&f, filePath, L"wb");
|
|
if(f)
|
|
{
|
|
fwrite(data, 1, dataSize, f);
|
|
fclose(f);
|
|
}
|
|
else
|
|
assert(0);
|
|
}
|
|
|
|
#endif // #ifdef _WIN32
|