Move item reader to a separate file.
Japa
parent
ea6757377e
commit
a63347cf7a
@ -0,0 +1,395 @@
|
|||||||
|
#include "item_reader.h"
|
||||||
|
|
||||||
|
#include "df/descriptor_shape.h"
|
||||||
|
#include "df/item_type.h"
|
||||||
|
#include "df/item_constructed.h"
|
||||||
|
#include "df/item_gemst.h"
|
||||||
|
#include "df/item_threadst.h"
|
||||||
|
#include "df/item_toolst.h"
|
||||||
|
#include "df/item_smallgemst.h"
|
||||||
|
#include "df/itemimprovement.h"
|
||||||
|
#include "df/itemimprovement_threadst.h"
|
||||||
|
#include "df/itemdef.h"
|
||||||
|
#include "df/map_block.h"
|
||||||
|
#include "df/vehicle.h"
|
||||||
|
#include "df/world.h"
|
||||||
|
|
||||||
|
#include "modules/Items.h"
|
||||||
|
#include "modules/MapCache.h"
|
||||||
|
#include "modules/Materials.h"
|
||||||
|
#include "MiscUtils.h"
|
||||||
|
|
||||||
|
|
||||||
|
using namespace DFHack;
|
||||||
|
using namespace df::enums;
|
||||||
|
using namespace RemoteFortressReader;
|
||||||
|
using namespace std;
|
||||||
|
using namespace df::global;
|
||||||
|
|
||||||
|
|
||||||
|
void CopyItem(RemoteFortressReader::Item * NetItem, df::item * DfItem)
|
||||||
|
{
|
||||||
|
NetItem->set_id(DfItem->id);
|
||||||
|
NetItem->set_flags1(DfItem->flags.whole);
|
||||||
|
NetItem->set_flags2(DfItem->flags2.whole);
|
||||||
|
auto pos = NetItem->mutable_pos();
|
||||||
|
pos->set_x(DfItem->pos.x);
|
||||||
|
pos->set_y(DfItem->pos.y);
|
||||||
|
pos->set_z(DfItem->pos.z);
|
||||||
|
auto mat = NetItem->mutable_material();
|
||||||
|
mat->set_mat_index(DfItem->getMaterialIndex());
|
||||||
|
mat->set_mat_type(DfItem->getMaterial());
|
||||||
|
auto type = NetItem->mutable_type();
|
||||||
|
type->set_mat_type(DfItem->getType());
|
||||||
|
type->set_mat_index(DfItem->getSubtype());
|
||||||
|
|
||||||
|
bool isProjectile = false;
|
||||||
|
|
||||||
|
item_type::item_type itemType = DfItem->getType();
|
||||||
|
|
||||||
|
switch (itemType)
|
||||||
|
{
|
||||||
|
case df::enums::item_type::NONE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BAR:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SMALLGEM:
|
||||||
|
{
|
||||||
|
VIRTUAL_CAST_VAR(smallgem_item, df::item_smallgemst, DfItem);
|
||||||
|
type->set_mat_index(smallgem_item->shape);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case df::enums::item_type::BLOCKS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::ROUGH:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BOULDER:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::WOOD:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::DOOR:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::FLOODGATE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BED:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CHAIR:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CHAIN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::FLASK:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::GOBLET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::INSTRUMENT:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TOY:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::WINDOW:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CAGE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BARREL:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BUCKET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::ANIMALTRAP:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TABLE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::COFFIN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::STATUE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CORPSE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::WEAPON:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::ARMOR:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SHOES:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SHIELD:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::HELM:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::GLOVES:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BOX:
|
||||||
|
type->set_mat_index(DfItem->isBag());
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BIN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::ARMORSTAND:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::WEAPONRACK:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CABINET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::FIGURINE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::AMULET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SCEPTER:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::AMMO:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CROWN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::RING:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::EARRING:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BRACELET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::GEM:
|
||||||
|
{
|
||||||
|
VIRTUAL_CAST_VAR(gem_item, df::item_gemst, DfItem);
|
||||||
|
type->set_mat_index(gem_item->shape);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case df::enums::item_type::ANVIL:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CORPSEPIECE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::REMAINS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::MEAT:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::FISH:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::FISH_RAW:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::VERMIN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::PET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SEEDS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::PLANT:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SKIN_TANNED:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::PLANT_GROWTH:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::THREAD:
|
||||||
|
{
|
||||||
|
VIRTUAL_CAST_VAR(thread, df::item_threadst, DfItem);
|
||||||
|
if (thread && thread->dye_mat_type >= 0)
|
||||||
|
{
|
||||||
|
DFHack::MaterialInfo info;
|
||||||
|
if (info.decode(thread->dye_mat_type, thread->dye_mat_index))
|
||||||
|
ConvertDFColorDescriptor(info.material->powder_dye, NetItem->mutable_dye());
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case df::enums::item_type::CLOTH:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TOTEM:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::PANTS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BACKPACK:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::QUIVER:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CATAPULTPARTS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BALLISTAPARTS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SIEGEAMMO:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BALLISTAARROWHEAD:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TRAPPARTS:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TRAPCOMP:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::DRINK:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::POWDER_MISC:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CHEESE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::FOOD:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::LIQUID_MISC:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::COIN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::GLOB:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::ROCK:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::PIPE_SECTION:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::HATCH_COVER:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::GRATE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::QUERN:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::MILLSTONE:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SPLINT:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::CRUTCH:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TRACTION_BENCH:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::ORTHOPEDIC_CAST:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::TOOL:
|
||||||
|
if (!isProjectile)
|
||||||
|
{
|
||||||
|
VIRTUAL_CAST_VAR(tool, df::item_toolst, DfItem);
|
||||||
|
if (tool)
|
||||||
|
{
|
||||||
|
auto vehicle = binsearch_in_vector(world->vehicles.active, tool->vehicle_id);
|
||||||
|
if (vehicle)
|
||||||
|
{
|
||||||
|
NetItem->set_subpos_x(vehicle->offset_x / 100000.0);
|
||||||
|
NetItem->set_subpos_y(vehicle->offset_y / 100000.0);
|
||||||
|
NetItem->set_subpos_z(vehicle->offset_z / 140000.0);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SLAB:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::EGG:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BOOK:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::SHEET:
|
||||||
|
break;
|
||||||
|
case df::enums::item_type::BRANCH:
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
VIRTUAL_CAST_VAR(actual_item, df::item_actual, DfItem);
|
||||||
|
if (actual_item)
|
||||||
|
{
|
||||||
|
NetItem->set_stack_size(actual_item->stack_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
VIRTUAL_CAST_VAR(constructed_item, df::item_constructed, DfItem);
|
||||||
|
if (constructed_item)
|
||||||
|
{
|
||||||
|
for (int i = 0; i < constructed_item->improvements.size(); i++)
|
||||||
|
{
|
||||||
|
auto improvement = constructed_item->improvements[i];
|
||||||
|
|
||||||
|
if (!improvement)
|
||||||
|
continue;
|
||||||
|
|
||||||
|
improvement_type::improvement_type impType = improvement->getType();
|
||||||
|
|
||||||
|
switch (impType)
|
||||||
|
{
|
||||||
|
case df::enums::improvement_type::ART_IMAGE:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::COVERED:
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case df::enums::improvement_type::RINGS_HANGING:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::BANDS:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::SPIKES:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::ITEMSPECIFIC:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::THREAD:
|
||||||
|
{
|
||||||
|
VIRTUAL_CAST_VAR(improvement_thread, df::itemimprovement_threadst, improvement);
|
||||||
|
if (improvement_thread->dye.mat_type >= 0)
|
||||||
|
{
|
||||||
|
DFHack::MaterialInfo info;
|
||||||
|
if (!info.decode(improvement_thread->dye.mat_type, improvement_thread->dye.mat_index))
|
||||||
|
continue;
|
||||||
|
ConvertDFColorDescriptor(info.material->powder_dye, NetItem->mutable_dye());
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case df::enums::improvement_type::CLOTH:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::SEWN_IMAGE:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::PAGES:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::ILLUSTRATION:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::INSTRUMENT_PIECE:
|
||||||
|
break;
|
||||||
|
case df::enums::improvement_type::WRITING:
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
NetItem->set_volume(DfItem->getVolume());
|
||||||
|
}
|
||||||
|
|
||||||
|
DFHack::command_result GetItemList(DFHack::color_ostream &stream, const DFHack::EmptyMessage *in, RemoteFortressReader::MaterialList *out)
|
||||||
|
{
|
||||||
|
if (!Core::getInstance().isWorldLoaded()) {
|
||||||
|
//out->set_available(false);
|
||||||
|
return CR_OK;
|
||||||
|
}
|
||||||
|
FOR_ENUM_ITEMS(item_type, it)
|
||||||
|
{
|
||||||
|
MaterialDefinition *mat_def = out->add_material_list();
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_type((int)it);
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_index(-1);
|
||||||
|
mat_def->set_id(ENUM_KEY_STR(item_type, it));
|
||||||
|
switch (it)
|
||||||
|
{
|
||||||
|
case df::enums::item_type::GEM:
|
||||||
|
case df::enums::item_type::SMALLGEM:
|
||||||
|
{
|
||||||
|
for (int i = 0; i < world->raws.language.shapes.size(); i++)
|
||||||
|
{
|
||||||
|
auto shape = world->raws.language.shapes[i];
|
||||||
|
if (shape->gems_use.whole == 0)
|
||||||
|
continue;
|
||||||
|
mat_def = out->add_material_list();
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_type((int)it);
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_index(i);
|
||||||
|
mat_def->set_id(ENUM_KEY_STR(item_type, it) + "/" + shape->id);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
case df::enums::item_type::BOX:
|
||||||
|
{
|
||||||
|
mat_def = out->add_material_list();
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_type((int)it);
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_index(0);
|
||||||
|
mat_def->set_id("BOX_CHEST");
|
||||||
|
mat_def = out->add_material_list();
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_type((int)it);
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_index(1);
|
||||||
|
mat_def->set_id("BOX_BAG");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
int subtypes = Items::getSubtypeCount(it);
|
||||||
|
if (subtypes >= 0)
|
||||||
|
{
|
||||||
|
for (int i = 0; i < subtypes; i++)
|
||||||
|
{
|
||||||
|
mat_def = out->add_material_list();
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_type((int)it);
|
||||||
|
mat_def->mutable_mat_pair()->set_mat_index(i);
|
||||||
|
df::itemdef * item = Items::getSubtypeDef(it, i);
|
||||||
|
mat_def->set_id(ENUM_KEY_STR(item_type, it) + "/" + item->id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return CR_OK;
|
||||||
|
}
|
||||||
@ -0,0 +1,25 @@
|
|||||||
|
#ifndef ITEM_READER_H
|
||||||
|
#define ITEM_READER_H
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
#include "RemoteClient.h"
|
||||||
|
#include "RemoteFortressReader.pb.h"
|
||||||
|
|
||||||
|
#include "DataDefs.h"
|
||||||
|
|
||||||
|
namespace df
|
||||||
|
{
|
||||||
|
struct item;
|
||||||
|
struct map_block;
|
||||||
|
}
|
||||||
|
|
||||||
|
namespace MapExtras
|
||||||
|
{
|
||||||
|
struct MapCache;
|
||||||
|
}
|
||||||
|
|
||||||
|
DFHack::command_result GetItemList(DFHack::color_ostream &stream, const DFHack::EmptyMessage *in, RemoteFortressReader::MaterialList *out);
|
||||||
|
void CopyItem(RemoteFortressReader::Item * NetItem, df::item * DfItem);
|
||||||
|
void ConvertDFColorDescriptor(int16_t index, RemoteFortressReader::ColorDefinition * out);
|
||||||
|
|
||||||
|
#endif // !ITEM_READER_H
|
||||||
Loading…
Reference in New Issue