{

commands.push_back(PluginCommand(

"changeitem",

"Change item attributes (material, quality).",

df_changeitem));

return CR_OK;

{

switch(q)

{

case 0:

return "Basic";

case 1:

return "-Well-crafted-";

case 2:

return "+Finely-crafted+";

case 3:

return "*Superior quality*";

case 4:

return "#Exceptional#";

case 5:

return "$Masterful$";

default:

return "!INVALID!";

}

color_ostream &out, df::item * item,

bool info, bool force,

bool change_material, string new_material,

bool change_quality, int new_quality,

{

CoreSuspender suspend;

bool here = false;

bool info = false;

bool force = false;

bool change_material = false;

string new_material = "none";

bool change_subtype = false;

string new_subtype = "NONE";

bool change_quality = false;

int new_quality = 0;

for (size_t i = 0; i < parameters.size(); i++)

{

string & p = parameters[i];

if (p == "help" || p == "?")

{

return CR_WRONG_USAGE;

}

else if (p == "here")

{

here = true;

}

else if (p == "info")

{

info = true;

}

else if (p == "force")

{

force = true;

}

else if (p == "material" || p == "m" )

{

// must be followed by material RAW id

// (string like 'INORGANIC:GRANITE', 'PLANT:MAPLE:WOOD', ...)

if(i == parameters.size()-1)

{

out.printerr("no material specified!\n");

return CR_WRONG_USAGE;

}

change_material = true;

new_material = parameters[i+1];

i++;

}

else if (p == "quality" || p == "q")

{

// must be followed by numeric quality (allowed: 0-5)

if(i == parameters.size()-1)

{

out.printerr("no quality specified!\n");

return CR_WRONG_USAGE;

}

string & q = parameters[i+1];

// meh. should use a stringstream instead. but it's only 6 numbers

if(q == "0")

new_quality = 0;

else if(q == "1")

new_quality = 1;

else if(q == "2")

new_quality = 2;

else if(q == "3")

new_quality = 3;

else if(q == "4")

new_quality = 4;

else if(q == "5")

new_quality = 5;

else

{

out << "Invalid quality specified!" << endl;

return CR_WRONG_USAGE;

}

out << "change to quality: " << describeQuality(new_quality) << endl;

change_quality = true;

i++;

}

else if (p == "subtype" || p == "s" )

{

// must be followed by subtype RAW id

// (string like 'ITEM_GLOVES_GAUNTLETS', 'ITEM_WEAPON_DAGGER_LARGE', 'ITEM_TOOL_KNIFE_MEAT_CLEAVER', ...)

if(i == parameters.size()-1)

{

out.printerr("no subtype specified!\n");

return CR_WRONG_USAGE;

}

change_subtype = true;

new_subtype = parameters[i+1];

i++;

}

else

{

out << p << ": Unknown command!" << endl;

return CR_WRONG_USAGE;

}

}

if (!Maps::IsValid())

{

out.printerr("Map is not available!\n");

return CR_FAILURE;

}

MaterialInfo mat_new;

if (change_material && !mat_new.find(new_material))

{

out.printerr("No such material!\n");

return CR_FAILURE;

}

if (here)

{

int processed_total = 0;

int cx, cy, cz;

DFCoord pos_cursor;

// needs a cursor

if (!Gui::getCursorCoords(cx,cy,cz))

{

out.printerr("Cursor position not found. Please enable the cursor.\n");

return CR_FAILURE;

}

pos_cursor = DFCoord(cx,cy,cz);

// uh. is this check necessary?

// changeitem doesn't do stuff with map blocks...

{

MapCache MC;

Block * b = MC.BlockAt(pos_cursor / 16);

if(!b)

{

out.printerr("Cursor is in an invalid/uninitialized area. Place it over a floor.\n");

return CR_FAILURE;

}

// when only changing material it doesn't matter if cursor is over a tile

//df::tiletype ttype = MC.tiletypeAt(pos_cursor);

//if(!DFHack::isFloorTerrain(ttype))

//{

// out.printerr("Cursor should be placed over a floor.\n");

// return CR_FAILURE;

//}

}

// iterate over all items, process those where pos = pos_cursor

size_t numItems = world->items.all.size();

for(size_t i=0; i< numItems; i++)

{

df::item * item = world->items.all[i];

DFCoord pos_item(item->pos.x, item->pos.y, item->pos.z);

if (pos_item != pos_cursor)

continue;

changeitem_execute(out, item, info, force, change_material, new_material, change_quality, new_quality, change_subtype, new_subtype);

processed_total++;

}

out.print("Done. %d items processed.\n", processed_total);

}

else

{

// needs a selected item

df::item *item = Gui::getSelectedItem(out);

if (!item)

{

out.printerr("No item selected.\n");

return CR_FAILURE;

}

changeitem_execute(out, item, info, force, change_material, new_material, change_quality, new_quality, change_subtype, new_subtype);

}

return CR_OK;

color_ostream &out, df::item * item,

bool info, bool force,

bool change_material, string new_material,

bool change_quality, int new_quality,

bool change_subtype, string new_subtype )

{

MaterialInfo mat_new;

MaterialInfo mat_old;

ItemTypeInfo sub_old;

ItemTypeInfo sub_new;

int new_subtype_id = -1;

if(change_material)

mat_new.find(new_material);

if(change_material || info)

mat_old.decode(item);

if(change_subtype || info)

sub_old.decode(item);

if(change_subtype)

{

string new_type = ENUM_KEY_STR(item_type, item->getType()) + ":" + new_subtype;

if (new_subtype == "NONE")

new_subtype_id = -1;

else if (sub_new.find(new_type))

new_subtype_id = sub_new.subtype;

else

{

return CR_FAILURE;

}

}

// print some info, don't change stuff

if(info)

{

out << "Item info: " << endl;

out << " type: " << ENUM_KEY_STR(item_type, item->getType()) << endl;

out << " subtype: " << (sub_old.custom ? sub_old.custom->id : "NONE") << endl;

out << " quality: " << describeQuality(item->getQuality()) << endl;

//if(item->isImproved())

// out << " imp.quality: " << describeQuality(item->getImprovementQuality()) << endl;

out << " material: " << mat_old.getToken() << endl;

return CR_OK;

}

if(change_quality)

{

item->setQuality(new_quality);

// it would be nice to be able to change the improved quality, too

// (only allowed if the item is already improved)

// but there is no method in item.h which supports that

// ok: hints from _Q/angavrilov: improvent is a vector, an item can have more than one improvement

// -> virtual_cast to item_constructedst

}

if(change_material)

{

// subtype and mode should match to avoid doing dumb stuff like changing boulders into meat whatever

// changing a stone cabinet to wood is fine, though. as well as lots of other combinations.

// still, it's better to make the user activate 'force' if he really wants to.

// fixme: changing material of cloth items needs more work...

// <_Q> cloth items have a "CLOTH" improvement which tells you about the cloth that was used to make it

if(force||(mat_old.subtype == mat_new.subtype && mat_old.mode==mat_new.mode))

{

item->setMaterial(mat_new.type);

item->setMaterialIndex(mat_new.index);

}

else

{

out.printerr("change denied: subtype doesn't match. use 'force' to override.\n");

}

item->flags.bits.temps_computed = 0; // recalc temperatures next time touched

item->flags.bits.weight_computed = 0; // recalc weight next time touched

}

if(change_subtype)

{

item->setSubtype(new_subtype_id);

item->flags.bits.weight_computed = 0; // recalc weight next time touched

}

return CR_OK;