edg-l/ddnet
1
0
Fork 0
mirror of https://github.com/ddnet/ddnet.git synced 2024-11-19 14:38:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tool to extract images and sounds from maps

Browse source
This commit is contained in:
def 2017-06-12 21:00:13 +02:00
parent 49e4501cc7
commit c943efbec0
1 changed files with 114 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

114
src/tools/map_extract.cpp Normal file
View file

@ -0,0 +1,114 @@
// Adapted from TWMapImagesRecovery by Tardo: https://github.com/Tardo/TWMapImagesRecovery
#include <game/mapitems.h>
#include <base/system.h>
#include <engine/external/pnglite/pnglite.h>
#include <engine/shared/datafile.h>
#include <engine/storage.h>
bool Process(IStorage *pStorage, const char *pMapName, const char *pPathSave)
{
CDataFileReader Map;
if(!Map.Open(pStorage, pMapName, IStorage::TYPE_ALL))
{
dbg_msg("map_extract", "error opening map '%s'", pMapName);
return false;
}
// check version
CMapItemVersion *pVersion = (CMapItemVersion *)Map.FindItem(MAPITEMTYPE_VERSION, 0);
if(pVersion && pVersion->m_Version != 1)
return false;
dbg_msg("map_extract", "Make sure you have the permission to use these images and sounds in your own maps");
CMapItemInfo *pInfo = (CMapItemInfo *)Map.FindItem(MAPITEMTYPE_INFO, 0);
dbg_msg("map_extract", "author: %s", Map.GetData(pInfo->m_Author));
dbg_msg("map_extract", "version: %s", Map.GetData(pInfo->m_MapVersion));
dbg_msg("map_extract", "credits: %s", Map.GetData(pInfo->m_Credits));
dbg_msg("map_extract", "license: %s", Map.GetData(pInfo->m_License));
int Start, Num;
// load images
Map.GetType(MAPITEMTYPE_IMAGE, &Start, &Num);
for(int i = 0; i < Num; i++)
{
CMapItemImage *pItem = (CMapItemImage *)Map.GetItem(Start+i, 0, 0);
char *pName = (char *)Map.GetData(pItem->m_ImageName);
if(pItem->m_External)
continue;
char aBuf[512];
str_format(aBuf, sizeof(aBuf), "%s/%s.png", pPathSave, pName);
dbg_msg("map_extract", "writing image: %s (%dx%d)", aBuf, pItem->m_Width, pItem->m_Height);
// copy image data
png_t Png;
png_open_file_write(&Png, aBuf);
png_set_data(&Png, pItem->m_Width, pItem->m_Height, 8, PNG_TRUECOLOR_ALPHA, (unsigned char*)Map.GetData(pItem->m_ImageData));
png_close_file(&Png);
}
// load sounds
Map.GetType(MAPITEMTYPE_SOUND, &Start, &Num);
for(int i = 0; i < Num; i++)
{
CMapItemSound *pItem = (CMapItemSound *)Map.GetItem(Start+i, 0, 0);
char *pName = (char *)Map.GetData(pItem->m_SoundName);
if(pItem->m_External)
continue;
char aBuf[512];
str_format(aBuf, sizeof(aBuf), "%s/%s.opus", pPathSave, pName);
dbg_msg("map_extract", "writing sound: %s (%d B)", aBuf, pItem->m_SoundDataSize);
IOHANDLE Opus = io_open(aBuf, IOFLAG_WRITE);
io_write(Opus, (unsigned char*)Map.GetData(pItem->m_SoundData), pItem->m_SoundDataSize);
io_close(Opus);
}
return Map.Close();
}
int main(int argc, char* argv[])
{
dbg_logger_stdout();
char aMap[512];
char aDir[512];
IStorage *pStorage = CreateLocalStorage();
if (argc == 2)
{
str_copy(aMap, argv[1], sizeof(aMap));
str_copy(aDir, ".", sizeof(aMap));
}
else if (argc == 3)
{
str_copy(aMap, argv[1], sizeof(aMap));
str_copy(aDir, argv[2], sizeof(aDir));
}
else
{
dbg_msg("usage", "%s <map> <destination folder>", argv[0]);
return -1;
}
if (!fs_is_dir(aDir))
{
dbg_msg("usage" "directory '%s' does not exist", aDir);
return -1;
}
png_init(0, 0);
return Process(pStorage, aMap, aDir) ? 0 : 1;
}