Merge branch 'export-plugin' of https://github.com/espenhw/dfhack
Petr Mrázek
commit
7d8a176e59
@ -0,0 +1,31 @@
|
||||
PROJECT (export)
|
||||
# A list of source files
|
||||
SET(PROJECT_SRCS
|
||||
export.cpp
|
||||
)
|
||||
# A list of headers
|
||||
SET(PROJECT_HDRS
|
||||
export.h
|
||||
)
|
||||
SET_SOURCE_FILES_PROPERTIES( ${PROJECT_HDRS} PROPERTIES HEADER_FILE_ONLY TRUE)
|
||||
|
||||
# mash them together (headers are marked as headers and nothing will try to compile them)
|
||||
LIST(APPEND PROJECT_SRCS ${PROJECT_HDRS})
|
||||
|
||||
#linux
|
||||
IF(UNIX)
|
||||
add_definitions(-DLINUX_BUILD)
|
||||
SET(PROJECT_LIBS
|
||||
# add any extra linux libs here
|
||||
${PROJECT_LIBS}
|
||||
)
|
||||
# windows
|
||||
ELSE(UNIX)
|
||||
SET(PROJECT_LIBS
|
||||
# add any extra linux libs here
|
||||
${PROJECT_LIBS}
|
||||
$(NOINHERIT)
|
||||
)
|
||||
ENDIF(UNIX)
|
||||
# this makes sure all the stuff is put in proper places and linked to dfhack
|
||||
DFHACK_PLUGIN(export ${PROJECT_SRCS} LINK_LIBRARIES ${PROJECT_LIBS})
|
||||
@ -0,0 +1,183 @@
|
||||
// some headers required for a plugin. Nothing special, just the basics.
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <fstream>
|
||||
#include <iostream>
|
||||
using namespace std;
|
||||
|
||||
#define DFHACK_WANT_MISCUTILS
|
||||
#include <Core.h>
|
||||
#include <VersionInfo.h>
|
||||
#include <Console.h>
|
||||
#include <Export.h>
|
||||
#include <PluginManager.h>
|
||||
#include <modules/Units.h>
|
||||
#include <modules/Translation.h>
|
||||
|
||||
#include <df/ui.h>
|
||||
#include <df/world.h>
|
||||
#include <df/unit.h>
|
||||
#include <df/unit_soul.h>
|
||||
|
||||
using namespace DFHack;
|
||||
using namespace DFHack::Simple;
|
||||
using namespace DFHack::Simple;
|
||||
|
||||
using df::global::ui;
|
||||
using df::global::world;
|
||||
|
||||
// our own, empty header.
|
||||
#include "export.h"
|
||||
|
||||
|
||||
// Here go all the command declarations...
|
||||
// mostly to allow having the mandatory stuff on top of the file and commands on the bottom
|
||||
DFhackCExport command_result export_dwarves (Core * c, std::vector <std::string> & parameters);
|
||||
|
||||
// A plugins must be able to return its name. This must correspond to the filename - export.plug.so or export.plug.dll
|
||||
DFhackCExport const char * plugin_name ( void )
|
||||
{
|
||||
return "export";
|
||||
}
|
||||
|
||||
// Mandatory init function. If you have some global state, create it here.
|
||||
DFhackCExport command_result plugin_init ( Core * c, std::vector <PluginCommand> &commands)
|
||||
{
|
||||
// Fill the command list with your commands.
|
||||
commands.clear();
|
||||
commands.push_back(PluginCommand("export",
|
||||
"Export dwarves to RuneSmith-compatible XML.",
|
||||
export_dwarves /*,
|
||||
true or false - true means that the command can't be used from non-interactive user interface'*/));
|
||||
return CR_OK;
|
||||
}
|
||||
|
||||
// This is called right before the plugin library is removed from memory.
|
||||
DFhackCExport command_result plugin_shutdown ( Core * c )
|
||||
{
|
||||
return CR_OK;
|
||||
}
|
||||
|
||||
static char* physicals[] = {
|
||||
"Strength",
|
||||
"Agility",
|
||||
"Toughness",
|
||||
"Endurance",
|
||||
"Recuperation",
|
||||
"DiseaseResistance",
|
||||
};
|
||||
|
||||
static char* mentals[] = {
|
||||
"Willpower",
|
||||
"Memory",
|
||||
"Focus",
|
||||
"Intuition",
|
||||
"Patience",
|
||||
"Empathy",
|
||||
"SocialAwareness",
|
||||
"Creatvity", //Speeling deliberate
|
||||
"Musicality",
|
||||
"AnalyticalAbility",
|
||||
"LinguisticAbility",
|
||||
"SpatialSense",
|
||||
"KinaestheticSense",
|
||||
};
|
||||
|
||||
static void element(const char* name, const char* content, ostream& out, const char* extra_indent="") {
|
||||
out << extra_indent << " <" << name << ">" << content << "</" << name << ">" << endl;
|
||||
}
|
||||
|
||||
static void element(const char* name, const uint32_t content, ostream& out, const char* extra_indent="") {
|
||||
out << extra_indent << " <" << name << ">" << content << "</" << name << ">" << endl;
|
||||
}
|
||||
|
||||
static void printAttributes(Core* c, df::unit* cre, ostream& out) {
|
||||
out << " <Attributes>" << endl;
|
||||
for (int i = 0; i < NUM_CREATURE_PHYSICAL_ATTRIBUTES; i++) {
|
||||
element(physicals[i], cre->body.physical_attrs[i].unk1, out, " ");
|
||||
}
|
||||
|
||||
df::unit_soul * s = cre->status.current_soul;
|
||||
if (s) {
|
||||
for (int i = 0; i < NUM_CREATURE_MENTAL_ATTRIBUTES; i++) {
|
||||
element(mentals[i], s->mental_attrs[i].unk1, out, " ");
|
||||
}
|
||||
}
|
||||
out << " </Attributes>" << endl;
|
||||
}
|
||||
|
||||
static void printTraits(Core* c, df::unit* cre, ostream& out) {
|
||||
out << " <Traits>" << endl;
|
||||
df::unit_soul * s = cre->status.current_soul;
|
||||
if (s) {
|
||||
for (int i = 0; i < NUM_CREATURE_TRAITS; i++) {
|
||||
out << " <Trait name='" << c->vinfo->getTraitName(i) <<
|
||||
"' value='" << s->traits[i] << "'>";
|
||||
string trait = c->vinfo->getTrait(i, s->traits[i]);
|
||||
if (!trait.empty()) {
|
||||
out << trait.c_str();
|
||||
}
|
||||
out << "</Trait>" << endl;
|
||||
}
|
||||
}
|
||||
out << " </Traits>" << endl;
|
||||
}
|
||||
|
||||
// GDC needs:
|
||||
// Name
|
||||
// Nickname
|
||||
// Sex
|
||||
// Attributes
|
||||
// Traits
|
||||
static void export_dwarf(Core* c, df::unit* cre, ostream& out) {
|
||||
string info = cre->name.first_name;
|
||||
info += " ";
|
||||
info += Translation::TranslateName(&cre->name, false);
|
||||
info[0] = toupper(info[0]);
|
||||
c->con.print("Exporting %s\n", info.c_str());
|
||||
|
||||
out << " <Creature>" << endl;
|
||||
element("Name", info.c_str(), out);
|
||||
element("Nickname", cre->name.nickname.c_str(), out);
|
||||
element("Sex", cre->sex == 0 ? "Female" : "Male", out);
|
||||
printAttributes(c, cre, out);
|
||||
printTraits(c, cre, out);
|
||||
|
||||
out << " </Creature>" << endl;
|
||||
}
|
||||
|
||||
DFhackCExport command_result export_dwarves (Core * c, std::vector <std::string> & parameters)
|
||||
{
|
||||
string filename;
|
||||
if (parameters.size() == 1) {
|
||||
filename = parameters[0];
|
||||
} else {
|
||||
c->con.print("export <filename>\n");
|
||||
return CR_OK;
|
||||
}
|
||||
|
||||
ofstream outf(filename);
|
||||
if (!outf) {
|
||||
c->con.printerr("Failed to open file %s\n", filename.c_str());
|
||||
return CR_FAILURE;
|
||||
}
|
||||
|
||||
c->Suspend();
|
||||
|
||||
uint32_t race = ui->race_id;
|
||||
uint32_t civ = ui->civ_id;
|
||||
|
||||
outf << "<?xml version='1.0' encoding='ibm850'?>" << endl << "<Creatures>" << endl;
|
||||
|
||||
for (int i = 0; i < world->units.all.size(); ++i)
|
||||
{
|
||||
df::unit* cre = world->units.all[i];
|
||||
if (cre->race == race && cre->civ_id == civ) {
|
||||
export_dwarf(c, cre, outf);
|
||||
}
|
||||
}
|
||||
outf << "</Creatures>" << endl;
|
||||
|
||||
c->Resume();
|
||||
return CR_OK;
|
||||
}
|
||||
@ -0,0 +1 @@
|
||||
#pragma once
|
||||
Loading…
Reference in New Issue