浏览代码

Use switch for CArtHandler::isTradableArtifact

ArseniyShestakov 10 年之前
父节点
当前提交
04d15174e7
共有 1 个文件被更改,包括 11 次插入3 次删除
  1. 11 3
      lib/CArtHandler.cpp

+ 11 - 3
lib/CArtHandler.cpp

@@ -585,10 +585,18 @@ bool CArtHandler::legalArtifact(ArtifactID id)
 
 bool CArtHandler::isTradableArtifact(ArtifactID id) const
 {
-	if (id < 7 && id != ArtifactID::SPELL_SCROLL)
+	switch (id)
+	{
+	case ArtifactID::SPELLBOOK:
+	case ArtifactID::GRAIL:
+	case ArtifactID::CATAPULT:
+	case ArtifactID::BALLISTA:
+	case ArtifactID::AMMO_CART:
+	case ArtifactID::FIRST_AID_TENT:
 		return false;
-
-	return true;
+	default:
+		return true;
+	}
 }
 
 void CArtHandler::initAllowedArtifactsList(const std::vector<bool> &allowed)