collective_config.cpp

#include "stdafx.h"
#include "util.h"
#include "creature.h"
#include "collective_config.h"
#include "tribe.h"
#include "game.h"
#include "technology.h"
#include "collective_warning.h"
#include "item_type.h"
#include "resource_id.h"
#include "inventory.h"
#include "workshops.h"
#include "lasting_effect.h"
#include "item.h"
#include "view_id.h"
#include "furniture_type.h"
#include "minion_activity.h"
#include "furniture_usage.h"
#include "creature_attributes.h"
#include "collective.h"
#include "zones.h"
#include "construction_map.h"
#include "item_class.h"
#include "villain_type.h"
#include "furniture.h"
#include "tutorial_highlight.h"
#include "spell.h"
#include "creature_factory.h"
#include "resource_info.h"
#include "workshop_item.h"
#include "body.h"
#include "view_id.h"
#include "view_object.h"
#include "territory.h"
#include "furniture_factory.h"
#include "storage_id.h"
#include "immigrant_info.h"
#include "conquer_condition.h"
#include "content_factory.h"
#include "content_factory.h"
#include "bed_type.h"
#include "item_types.h"
#include "furnace.h"

template <class Archive>
void CollectiveConfig::serialize(Archive& ar, const unsigned int version) {
  ar(OPTION(immigrantInterval), OPTION(maxPopulation), OPTION(conquerCondition), OPTION(canEnemyRetire));
  ar(SKIP(type), OPTION(leaderAsFighter), OPTION(spawnGhosts), OPTION(ghostProb), OPTION(guardianInfo));
  ar(SKIP(populationString), OPTION(prisoners), OPTION(alwaysMount), OPTION(discoverAchievement));
  ar(SKIP(requireQuartersForExp));
}

SERIALIZABLE(CollectiveConfig);
SERIALIZATION_CONSTRUCTOR_IMPL(CollectiveConfig);

template <class Archive>
void GuardianInfo::serialize(Archive& ar, const unsigned int version) {
  ar(creature, probability, minEnemies, minVictims);
}

SERIALIZABLE(GuardianInfo);

BedType CollectiveConfig::getPrisonBedType(const Creature* c) {
  return c->getBody().isHumanoid() ? BedType::PRISON : BedType::CAGE;
}

static optional<BedType> getBedType(const Creature* c, const ContentFactory* factory) {
  if (c->getBody().isImmuneTo(LastingEffect::SLEEP, factory))
    return none;
  if (c->isAffected(LastingEffect::STEED))
    return BedType::STABLE;
  if (c->getStatus().contains(CreatureStatus::PRISONER))
    return CollectiveConfig::getPrisonBedType(c);
  if (c->getBody().isUndead(factory))
    return BedType::COFFIN;
  if (c->getBody().isHumanoid())
    return BedType::BED;
  else
    return BedType::CAGE;
}

void CollectiveConfig::addBedRequirementToImmigrants(vector<ImmigrantInfo>& immigrantInfo, ContentFactory* factory) {
  for (auto& info : immigrantInfo) {
    PCreature c = factory->getCreatures().fromId(info.getId(0), TribeId::getDarkKeeper());
    if (info.getInitialRecruitment() == 0)
      if (auto bedType = getBedType(c.get(), factory)) {
        bool hasBed = false;
        info.visitRequirements(makeVisitor(
            [&](const AttractionInfo& attraction) -> void {
              for (auto& type : attraction.types)
                if (auto fType = type.getValueMaybe<FurnitureType>())
                  if (factory->furniture.getData(*fType).getBedType() == *bedType)
                    hasBed = true;
            },
            [&](const auto&) {}
        ));
        if (!hasBed) {
          info.addRequirement(AttractionInfo(1, factory->furniture.getBedFurniture(*bedType)[0]));
        }
      }
  }
}

CollectiveConfig::CollectiveConfig(TimeInterval interval, CollectiveType t, int maxPop, ConquerCondition conquerCondition)
    : immigrantInterval(interval), maxPopulation(maxPop), type(t),
      conquerCondition(conquerCondition) {
}

CollectiveConfig CollectiveConfig::keeper(TimeInterval immigrantInterval, int maxPopulation,
    string populationString, bool prisoners, ConquerCondition conquerCondition, bool requireQuartersForExp) {
  auto ret = CollectiveConfig(immigrantInterval, KEEPER, maxPopulation, conquerCondition);
  ret.populationString = populationString;
  ret.prisoners = prisoners;
  ret.requireQuartersForExp = requireQuartersForExp;
  return ret;
}

CollectiveConfig CollectiveConfig::noImmigrants() {
  return CollectiveConfig(TimeInterval {}, VILLAGE, 10000, ConquerCondition::KILL_FIGHTERS_AND_LEADER);
}

int CollectiveConfig::getNumGhostSpawns() const {
  return spawnGhosts;
}

TimeInterval CollectiveConfig::getImmigrantTimeout() const {
  return 500_visible;
}

double CollectiveConfig::getGhostProb() const {
  return ghostProb;
}

bool CollectiveConfig::isLeaderFighter() const {
  return leaderAsFighter;
}

bool CollectiveConfig::getManageEquipment() const {
  return type == KEEPER;
}

bool CollectiveConfig::getFollowLeaderIfNoTerritory() const {
  return type == KEEPER;
}

bool CollectiveConfig::stayInTerritory() const {
  return type != KEEPER;
}

bool CollectiveConfig::hasVillainSleepingTask() const {
  return type != KEEPER;
}

bool CollectiveConfig::allowHealingTaskOutsideTerritory() const {
  return type == KEEPER;
}

bool CollectiveConfig::hasImmigrantion(bool currentlyActiveModel) const {
  return type != KEEPER || currentlyActiveModel;
}

TimeInterval CollectiveConfig::getImmigrantInterval() const {
  return immigrantInterval;
}

bool CollectiveConfig::getStripSpawns() const {
  return type == KEEPER;
}

bool CollectiveConfig::getEnemyPositions() const {
  return type == KEEPER;
}

bool CollectiveConfig::getWarnings() const {
  return type == KEEPER;
}

bool CollectiveConfig::getConstructions() const {
  return type == KEEPER;
}

int CollectiveConfig::getMaxPopulation() const {
  return maxPopulation;
}

const string& CollectiveConfig::getPopulationString() const {
  return populationString;
}

const optional<GuardianInfo>& CollectiveConfig::getGuardianInfo() const {
  return guardianInfo;
}

bool CollectiveConfig::isConquered(const Collective* collective) const {
  switch (conquerCondition) {
    case ConquerCondition::KILL_LEADER:
      return collective->getLeaders().empty();
    case ConquerCondition::KILL_FIGHTERS_AND_LEADER:
      return collective->getCreatures(MinionTrait::FIGHTER).empty() && collective->getLeaders().empty();
    case ConquerCondition::DESTROY_BUILDINGS:
      for (auto& elem : collective->getConstructions().getAllFurniture())
        if (auto f = elem.first.getFurniture(elem.second))
          if (f->isWall())
            return false;
      return true;
    case ConquerCondition::NEVER:
      return false;
  }
}

bool CollectiveConfig::xCanEnemyRetire() const {
  return canEnemyRetire && type != KEEPER;
}

CollectiveConfig& CollectiveConfig::setConquerCondition(ConquerCondition c) {
  conquerCondition = c;
  return *this;
}

bool CollectiveConfig::canCapturePrisoners() const {
  return prisoners;
}

bool CollectiveConfig::alwaysMountSteeds() const {
  return alwaysMount;
}

bool CollectiveConfig::minionsRequireQuarters() const {
  return type == KEEPER && requireQuartersForExp;
}

MinionActivityInfo::MinionActivityInfo(Type t) : type(t) {
  CHECK(type != FURNITURE);
}

MinionActivityInfo::MinionActivityInfo() {}

MinionActivityInfo::MinionActivityInfo(FurnitureType type)
    : type(FURNITURE), furniturePredicate(
        [type](const ContentFactory*, const Collective*, const Creature*, FurnitureType t) { return t == type;})
       {
}

MinionActivityInfo::MinionActivityInfo(BuiltinUsageId usage)
  : type(FURNITURE), furniturePredicate(
      [usage](const ContentFactory* f, const Collective*, const Creature*, FurnitureType t) {
        return f->furniture.getData(t).hasUsageType(usage);
      })
     {
}

MinionActivityInfo::MinionActivityInfo(UsagePredicate pred, SecondaryPredicate pred2)
    : type(FURNITURE), furniturePredicate(std::move(pred)), secondaryPredicate(std::move(pred2)) {
}

CollectiveConfig::CollectiveConfig(const CollectiveConfig&) = default;
CollectiveConfig::CollectiveConfig(CollectiveConfig&&) noexcept = default;

CollectiveConfig::~CollectiveConfig() {
}

static auto getTrainingPredicate(AttrType experienceType) {
  return [experienceType] (const ContentFactory* contentFactory, const Collective*, const Creature* c, FurnitureType t) {
      if (auto maxIncrease = contentFactory->furniture.getData(t).getMaxTraining(experienceType))
        return !c || (c->getAttributes().getExpLevel(experienceType) < maxIncrease &&
            !c->getAttributes().isTrainingMaxedOut(experienceType));
      else
        return false;
    };
}

const MinionActivityInfo& CollectiveConfig::getActivityInfo(MinionActivity task) {
  static EnumMap<MinionActivity, MinionActivityInfo> map([](MinionActivity task) -> MinionActivityInfo {
    switch (task) {
      case MinionActivity::IDLE: return {MinionActivityInfo::IDLE};
      case MinionActivity::CONSTRUCTION: return {MinionActivityInfo::WORKER};
      case MinionActivity::HAULING: return {MinionActivityInfo::WORKER};
      case MinionActivity::WORKING: return {MinionActivityInfo::WORKER};
      case MinionActivity::DIGGING: return {MinionActivityInfo::WORKER};
      case MinionActivity::TRAIN: return {getTrainingPredicate(AttrType("DAMAGE"))};
      case MinionActivity::SLEEP:
        return {[](const ContentFactory* f, const Collective*, const Creature* c, FurnitureType t) {
            if (!c)
              return !!f->furniture.getData(t).getBedType();
            return getBedType(c, f) == f->furniture.getData(t).getBedType();
          }};
      case MinionActivity::EAT:
        return {[](const ContentFactory* factory, const Collective* col, const Creature* c, FurnitureType t) {
            auto f = factory->furniture.getData(t);
            if (!c)
              return !!f.getDiningFurnitureType();
            return f.getDiningFurnitureType() == c->getBody().getDiningFurniture();
          }};
      case MinionActivity::PHYLACTERY: return {FurnitureType("PHYLACTERY")};
      case MinionActivity::GUARDING1: return {MinionActivityInfo::GUARD1};
      case MinionActivity::GUARDING2: return {MinionActivityInfo::GUARD2};
      case MinionActivity::GUARDING3: return {MinionActivityInfo::GUARD3};
      case MinionActivity::POETRY:
        return {[](const ContentFactory* f, const Collective*, const Creature* c, FurnitureType t) {
          return t == FurnitureType("POETRY_TABLE")
              || t == FurnitureType("PAINTING_N")
              || t == FurnitureType("PAINTING_S")
              || t == FurnitureType("PAINTING_E")
              || t == FurnitureType("PAINTING_W");
          }, [](const Furniture* f, const Collective* col) {
            CHECK(f);
            auto id = f->getViewObject()->id().data();
            return !startsWith(id, "painting") && (!startsWith(id, "canvas") || !NOTNULL(col)->getRecordedEvents().empty());
          }};
      case MinionActivity::STUDY: return {getTrainingPredicate(AttrType("SPELL_DAMAGE"))};
      case MinionActivity::DISTILLATION: return {FurnitureType("DISTILLERY")};
      case MinionActivity::CROPS: return {FurnitureType("CROPS")};
      case MinionActivity::RITUAL: return {BuiltinUsageId::DEMON_RITUAL};
      case MinionActivity::ARCHERY: return {MinionActivityInfo::ARCHERY};
      case MinionActivity::COPULATE: return {MinionActivityInfo::COPULATE};
      case MinionActivity::EXPLORE: return {MinionActivityInfo::EXPLORE};
      case MinionActivity::SPIDER: return {MinionActivityInfo::SPIDER};
      case MinionActivity::MINION_ABUSE: return {MinionActivityInfo::MINION_ABUSE};
      case MinionActivity::EXPLORE_NOCTURNAL: return {MinionActivityInfo::EXPLORE};
      case MinionActivity::EXPLORE_CAVES: return {MinionActivityInfo::EXPLORE};
      case MinionActivity::BE_WHIPPED: return {FurnitureType("WHIPPING_POST")};
      case MinionActivity::BE_TORTURED: return {FurnitureType("TORTURE_TABLE")};
      case MinionActivity::BE_EXECUTED: return {FurnitureType("GALLOWS")};
      case MinionActivity::CRAFT:
        return {[](const ContentFactory* f, const Collective* col, const Creature* c, FurnitureType t) {
            PROFILE_BLOCK("Crafting predicate");
            if (t == FurnitureType("FURNACE")) {
              if (!c || !col)
                return true;
              auto skill = c->getAttr(AttrType("FURNACE"));
              return skill > 0 && !col->getFurnace().isIdle();
            } else
            if (auto type = f->getWorkshopType(t)) {
              if (!c || !col)
                return true;
              auto workshop = getReferenceMaybe(col->getWorkshops().types, *type);
              auto& info = f->workshopInfo.at(*type);
              return !!workshop && !workshop->isIdle(col, c->getAttr(info.attr));
            } else
              return false;
          }};
    }
  });
  return map[task];
}

#include "pretty_archive.h"
template
void CollectiveConfig::serialize(PrettyInputArchive& ar1, unsigned);

template
void GuardianInfo::serialize(PrettyInputArchive& ar1, unsigned);