{

template<class T1, class T2>

bool operator()(const std::pair<T1, T2>& left, const std::pair<T1, T2>& right)

{

return P<T2>()(left.second, right.second);

}

std::vector<DFHack::t_matgloss*> &materials)

{

unsigned int total = 0;

MatSorter sorting_vector;

for (MatMap::const_iterator it = mat.begin(); it != mat.end(); ++it)

{

sorting_vector.push_back(*it);

}

std::sort(sorting_vector.begin(), sorting_vector.end(),

compare_pair_second<>());

for (MatSorter::const_iterator it = sorting_vector.begin();

it != sorting_vector.end(); ++it)

{

if(it->first >= materials.size())

{

con << "Bad index: " << it->first << " out of "

<< materials.size() << endl;

continue;

}

DFHack::t_matgloss* mat = materials[it->first];

con << std::setw(25) << mat->id << " : " << it->second << std::endl;

total += it->second;

}

con << ">>> TOTAL = " << total << std::endl << std::endl;

std::vector<DFHack::t_matgloss> &materials)

{

std::vector<DFHack::t_matgloss*> ptr_vec;

TO_PTR_VEC(materials, ptr_vec);

printMats(con, mat, ptr_vec);

DFHack::Materials* mats)

{

MatMap ores;

MatMap gems;

MatMap rest;

for (MatMap::const_iterator it = mat_map.begin(); it != mat_map.end(); ++it)

{

DFHack::t_matglossInorganic &gloss = mats->inorganic[it->first];

if (gloss.isGem())

gems[it->first] = it->second;

else if (gloss.isOre())

ores[it->first] = it->second;

else

rest[it->first] = it->second;

}

std::vector<DFHack::t_matgloss*> ptr_vec;

TO_PTR_VEC(mats->inorganic, ptr_vec);

con << "Ores:" << std::endl;

printMats(con, ores, ptr_vec);

con << "Gems:" << std::endl;

printMats(con, gems, ptr_vec);

con << "Other vein stone:" << std::endl;

printMats(con, rest, ptr_vec);

{

commands.clear();

commands.push_back(PluginCommand("prospect","Show stats of available raw resources. Use parameter 'all' to show hidden resources.",prospector));

return CR_OK;

{

bool showHidden = false;

bool showPlants = true;

bool showSlade = true;

bool showTemple = true;

Console & con = c->con;

if(parameters.size() && parameters[0] == "all")

{

showHidden = true;

}

uint32_t x_max = 0, y_max = 0, z_max = 0;

c->Suspend();

DFHack::Maps *maps = c->getMaps();

if (!maps->Start())

{

con << "Cannot get map info!" << std::endl;

c->Resume();

return CR_FAILURE;

}

maps->getSize(x_max, y_max, z_max);

MapExtras::MapCache map(maps);

DFHack::Materials *mats = c->getMaterials();

if (!mats->ReadInorganicMaterials())

{

con << "Unable to read inorganic material definitons!" << std::endl;

c->Resume();

return CR_FAILURE;

}

if (showPlants && !mats->ReadOrganicMaterials())

{

con << "Unable to read organic material definitons; plants won't be listed!" << std::endl;

showPlants = false;

}

FeatureList globalFeatures;

FeatureMap localFeatures;

DFHack::t_feature *blockFeatureGlobal = 0;

DFHack::t_feature *blockFeatureLocal = 0;

bool hasAquifer = false;

bool hasDemonTemple = false;

bool hasLair = false;

MatMap baseMats;

MatMap layerMats;

MatMap veinMats;

MatMap plantMats;

MatMap treeMats;

if (!(showSlade && maps->ReadGlobalFeatures(globalFeatures)))

{

con << "Unable to read global features; slade won't be listed!" << std::endl;

}

if (!maps->ReadLocalFeatures(localFeatures))

{

con << "Unable to read local features; adamantine "

<< (showTemple ? "and demon temples " : "")

<< "won't be listed!" << std::endl;

}

uint32_t vegCount = 0;

DFHack::Vegetation *veg = c->getVegetation();

if (showPlants && !veg->Start())

{

con << "Unable to read vegetation; plants won't be listed!" << std::endl;

}

for(uint32_t z = 0; z < z_max; z++)

{

for(uint32_t b_y = 0; b_y < y_max; b_y++)

{

for(uint32_t b_x = 0; b_x < x_max; b_x++)

{

// Get the map block

DFHack::DFCoord blockCoord(b_x, b_y);

MapExtras::Block *b = map.BlockAt(DFHack::DFCoord(b_x, b_y, z));

if (!b || !b->valid)

{

continue;

}

{ // Find features

uint16_t index = b->raw.global_feature;

if (index != -1 && index < globalFeatures.size())

{

blockFeatureGlobal = &globalFeatures[index];

}

index = b->raw.local_feature;

FeatureMap::const_iterator it = localFeatures.find(blockCoord);

if (it != localFeatures.end())

{

FeatureListPointer features = it->second;

if (index != -1 && index < features.size())

{

blockFeatureLocal = features[index];

}

}

}

// Iterate over all the tiles in the block

for(uint32_t y = 0; y < 16; y++)

{

for(uint32_t x = 0; x < 16; x++)

{

DFHack::DFCoord coord(x, y);

DFHack::t_designation des = b->DesignationAt(coord);

DFHack::t_occupancy occ = b->OccupancyAt(coord);

// Skip hidden tiles

if (!showHidden && des.bits.hidden)

{

continue;

}

// Check for aquifer

if (des.bits.water_table)

{

hasAquifer = true;

}

// Check for lairs

if (occ.bits.monster_lair)

{

hasLair = true;

}

uint16_t type = b->TileTypeAt(coord);

const DFHack::TileRow *info = DFHack::getTileRow(type);

if (!info)

{

con << "Bad type: " << type << std::endl;

continue;

}

// We only care about these types

switch (info->shape)

{

case DFHack::WALL:

case DFHack::PILLAR:

case DFHack::FORTIFICATION:

break;

default:

continue;

}

// Count the material type

baseMats[info->material]++;

// Find the type of the tile

switch (info->material)

{

case DFHack::SOIL:

case DFHack::STONE:

layerMats[b->baseMaterialAt(coord)]++;

break;

case DFHack::VEIN:

veinMats[b->veinMaterialAt(coord)]++;

break;

case DFHack::FEATSTONE:

if (blockFeatureLocal && des.bits.feature_local)

{

if (blockFeatureLocal->type == DFHack::feature_Adamantine_Tube

&& blockFeatureLocal->main_material == 0) // stone

{

veinMats[blockFeatureLocal->sub_material]++;

}

else if (showTemple

&& blockFeatureLocal->type == DFHack::feature_Hell_Temple)

{

hasDemonTemple = true;

}

}

if (showSlade && blockFeatureGlobal && des.bits.feature_global

&& blockFeatureGlobal->type == DFHack::feature_Underworld

&& blockFeatureGlobal->main_material == 0) // stone

{

layerMats[blockFeatureGlobal->sub_material]++;

}

break;

case DFHack::OBSIDIAN:

// TODO ?

break;

}

}

}

// Check plants this way, as the other way wasn't getting them all

// and we can check visibility more easily here

if (showPlants)

{

PlantList * plants;

if (maps->ReadVegetation(b_x, b_y, z, plants))

{

for (PlantList::const_iterator it = plants->begin(); it != plants->end(); it++)

{

const DFHack::df_plant & plant = *(*it);

DFHack::DFCoord loc(plant.x, plant.y);

loc = loc % 16;

if (showHidden || !b->DesignationAt(loc).bits.hidden)

{

if(plant.is_shrub)

plantMats[plant.material]++;

else

treeMats[plant.material]++;

}

}

}

}

// Block end

} // block x

// Clean uneeded memory

map.trash();

} // block y

} // z

MatMap::const_iterator it;

con << "Base materials:" << std::endl;

for (it = baseMats.begin(); it != baseMats.end(); ++it)

{

con << std::setw(25) << DFHack::TileMaterialString[it->first] << " : " << it->second << std::endl;

}

std::vector<t_matgloss*> ptr_vec;

TO_PTR_VEC(mats->inorganic, ptr_vec);

con << std::endl << "Layer materials:" << std::endl;

printMats(con, layerMats, ptr_vec);

printVeins(con, veinMats, mats);

if (showPlants)

{

con << "Shrubs:" << std::endl;

printMats(con, plantMats, mats->organic);

con << "Wood in trees:" << std::endl;

printMats(con, treeMats, mats->organic);

}

if (hasAquifer)

{

con << "Has aquifer" << std::endl;

}

if (hasDemonTemple)

{

con << "Has demon temple" << std::endl;

}

if (hasLair)

{

con << "Has lair" << std::endl;

}

// Cleanup

if (showPlants)

{

veg->Finish();

}

mats->Finish();

maps->Finish();

c->Resume();

con << std::endl;

return CR_OK;