set record count in TES3 header

apps/opencs/model/doc/savingstages.cpp

@@ -48,7 +48,10 @@ void CSMDoc::WriteHeaderStage::perform (int stage, std::vector<std::string>& mes
     /// \todo fill in missing header information
     mState.getWriter().setAuthor ("");
     mState.getWriter().setDescription ("");
-    mState.getWriter().setRecordCount (0);
+    mState.getWriter().setRecordCount (
+        mDocument.getData().count (CSMWorld::RecordBase::State_Modified) +
+        mDocument.getData().count (CSMWorld::RecordBase::State_ModifiedOnly) +
+        mDocument.getData().count (CSMWorld::RecordBase::State_Deleted));
 
     /// \todo refine dependency list (at least remove redundant dependencies)
     std::vector<boost::filesystem::path> dependencies = mDocument.getContentFiles();

apps/opencs/model/world/data.cpp

@@ -43,6 +43,17 @@ void CSMWorld::Data::appendIds (std::vector<std::string>& ids, const CollectionB
     ids.insert (ids.end(), ids2.begin(), ids2.end());
 }
 
+int CSMWorld::Data::count (RecordBase::State state, const CollectionBase& collection)
+{
+    int number = 0;
+
+    for (int i=0; i<collection.getSize(); ++i)
+        if (collection.getRecord (i).mState==state)
+            ++number;
+
+    return number;
+}
+
 CSMWorld::Data::Data() : mRefs (mCells)
 {
     mGlobals.addColumn (new StringIdColumn<ESM::Global>);
@@ -460,6 +471,24 @@ bool CSMWorld::Data::hasId (const std::string& id) const
         getReferenceables().searchId (id)!=-1;
 }
 
+int CSMWorld::Data::count (RecordBase::State state) const
+{
+    return
+        count (state, mGlobals) +
+        count (state, mGmsts) +
+        count (state, mSkills) +
+        count (state, mClasses) +
+        count (state, mFactions) +
+        count (state, mRaces) +
+        count (state, mSounds) +
+        count (state, mScripts) +
+        count (state, mRegions) +
+        count (state, mBirthsigns) +
+        count (state, mSpells) +
+        count (state, mCells) +
+        count (state, mReferenceables);
+}
+
 std::vector<std::string> CSMWorld::Data::getIds (bool listDeleted) const
 {
     std::vector<std::string> ids;

apps/opencs/model/world/data.hpp

@@ -66,6 +66,8 @@ namespace CSMWorld
                 bool listDeleted);
             ///< Append all IDs from collection to \a ids.
 
+            static int count (RecordBase::State state, const CollectionBase& collection);
+
         public:
 
             Data();
@@ -151,6 +153,9 @@ namespace CSMWorld
             ///
             /// \param listDeleted include deleted record in the list
 
+            int count (RecordBase::State state) const;
+            ///< Return number of top-level records with the given \a state.
+
         signals:
 
             void idListChanged();