From b85fe2becf57f7c377a0ea66438bfe4bb2c9693c Mon Sep 17 00:00:00 2001
From: Marek Kochanowicz <sirherrbatka@gmail.com>
Date: Mon, 6 Jan 2014 23:37:18 +0100
Subject: [PATCH] Changes according to the comment.

---
 apps/opencs/model/tools/referenceablecheck.cpp | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/apps/opencs/model/tools/referenceablecheck.cpp b/apps/opencs/model/tools/referenceablecheck.cpp
index 12f347afb6..6132b34076 100644
--- a/apps/opencs/model/tools/referenceablecheck.cpp
+++ b/apps/opencs/model/tools/referenceablecheck.cpp
@@ -1,6 +1,7 @@
 #include "referenceablecheck.hpp"
 #include "../world/record.hpp"
 #include "../world/universalid.hpp"
+#include <iostream>
 
 CSMTools::ReferenceableCheckStage::ReferenceableCheckStage(
     const CSMWorld::RefIdData& referenceable, const CSMWorld::IdCollection<ESM::Race >& races,
@@ -17,11 +18,6 @@ CSMTools::ReferenceableCheckStage::ReferenceableCheckStage(
 
 void CSMTools::ReferenceableCheckStage::perform(int stage, std::vector< std::string >& messages)
 {
-    if (stage == mReferencables.getSize() - 1)
-    {
-        finalCheck(messages);
-    }
-
     //Checks for books, than, when stage is above mBooksSize goes to other checks, with (stage - PrevSum) as stage.
     const int bookSize(mReferencables.getBooks().getSize());
 
@@ -102,6 +98,7 @@ void CSMTools::ReferenceableCheckStage::perform(int stage, std::vector< std::str
     }
 
     stage -= doorSize;
+
     const int ingredientSize(mReferencables.getIngredients().getSize());
 
     if (stage < ingredientSize)
@@ -209,11 +206,15 @@ void CSMTools::ReferenceableCheckStage::perform(int stage, std::vector< std::str
         staticCheck(stage, mReferencables.getStatics(), messages);
         return;
     }
+// if we come that far, we are about to perform our last, final check.
+    finalCheck(messages);
+    return;
 }
 
 int CSMTools::ReferenceableCheckStage::setup()
 {
-    return mReferencables.getSize();
+    mPlayerPresent = false;
+    return mReferencables.getSize() + 1;
 }
 
 void CSMTools::ReferenceableCheckStage::bookCheck(
@@ -965,8 +966,6 @@ void CSMTools::ReferenceableCheckStage::finalCheck(std::vector< std::string >& m
         CSMWorld::UniversalId id(CSMWorld::UniversalId::Type_Npc);
         messages.push_back(id.toString() + "| There is no player record");
     }
-
-    mPlayerPresent = false;
 }