Code/Sandbox/EditorQt/AI/CoverSurfaceManager.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "CoverSurfaceManager.h"
   5 #include "Objects/AICoverSurface.h"
   6
   7 const uint32 BAI_COVER_FILE_VERSION_WRITE = 2;
   8
   9 CCoverSurfaceManager::CCoverSurfaceManager()
  10 {
  11 }
  12
  13 CCoverSurfaceManager::~CCoverSurfaceManager()
  14 {
  15 }
  16
  17 void CCoverSurfaceManager::ClearGameSurfaces()
  18 {
  19         gEnv->pAISystem->GetCoverSystem()->Clear();
  20
  21         SurfaceObjects::iterator it = m_surfaceObjects.begin();
  22         SurfaceObjects::iterator end = m_surfaceObjects.end();
  23
  24         for (; it != end; ++it)
  25         {
  26                 CAICoverSurface* coverSurfaceObject = *it;
  27                 coverSurfaceObject->SetSurfaceID(CoverSurfaceID(0));
  28         }
  29 }
  30
  31 // This also compresses CoverSurfaceIDs down to a sequential range
  32 bool CCoverSurfaceManager::WriteToFile(const char* fileName)
  33 {
  34         CCryFile file;
  35
  36         if (!file.Open(fileName, "wb"))
  37         {
  38                 Warning("Could not save AI Cover Surfaces. [%s]", fileName);
  39
  40                 return false;
  41         }
  42
  43         ICoverSystem* coverSystem = gEnv->pAISystem->GetCoverSystem();
  44
  45         uint32 fileVersion = BAI_COVER_FILE_VERSION_WRITE;
  46         uint32 surfaceCount = coverSystem->GetSurfaceCount();
  47
  48         file.Write(&fileVersion, sizeof(fileVersion));
  49         file.Write(&surfaceCount, sizeof(surfaceCount));
  50
  51         SurfaceObjects::iterator it = m_surfaceObjects.begin();
  52         SurfaceObjects::iterator end = m_surfaceObjects.end();
  53
  54         // How this works:
  55         // 1. Save all valid surfaces in sequence to the file
  56         // 2. Assign a sequenced ID to all the CAICoverSurface objects
  57         //              Assign 0 to any invalid CAICoverSurface objects
  58         // 3. Load the valid surfaces again in sequence
  59         uint32 runningID = 0;
  60
  61         for (; it != end; ++it)
  62         {
  63                 CAICoverSurface* coverSurfaceObject = *it;
  64                 CoverSurfaceID surfaceID = coverSurfaceObject->GetSurfaceID();
  65                 coverSurfaceObject->SetSurfaceID(CoverSurfaceID(0));
  66
  67                 if (!surfaceID)
  68                         continue;
  69
  70                 ICoverSystem::SurfaceInfo surfaceInfo;
  71                 if (!coverSystem->GetSurfaceInfo(surfaceID, &surfaceInfo))
  72                         continue;
  73
  74                 file.Write(&surfaceInfo.sampleCount, sizeof(surfaceInfo.sampleCount));
  75                 file.Write(&surfaceInfo.flags, sizeof(surfaceInfo.flags));
  76
  77                 for (uint i = 0; i < surfaceInfo.sampleCount; ++i)
  78                 {
  79                         const ICoverSampler::Sample& sample = surfaceInfo.samples[i];
  80
  81                         file.Write(&sample.position, sizeof(sample.position));
  82                         file.Write(&sample.height, sizeof(sample.height));
  83                         file.Write(&sample.flags, sizeof(sample.flags));
  84                 }
  85
  86                 ++runningID;
  87                 coverSurfaceObject->SetSurfaceID(CoverSurfaceID(runningID));
  88         }
  89
  90         file.Close();
  91
  92         coverSystem->Clear();
  93         return coverSystem->ReadSurfacesFromFile(fileName);
  94 }
  95
  96 bool CCoverSurfaceManager::ReadFromFile(const char* fileName)
  97 {
  98         return gEnv->pAISystem->GetCoverSystem()->ReadSurfacesFromFile(fileName);
  99 }
 100
 101 void CCoverSurfaceManager::AddSurfaceObject(CAICoverSurface* surface)
 102 {
 103         m_surfaceObjects.insert(surface);
 104 }
 105
 106 void CCoverSurfaceManager::RemoveSurfaceObject(CAICoverSurface* surface)
 107 {
 108         m_surfaceObjects.erase(surface);
 109 }
 110
 111 const CCoverSurfaceManager::SurfaceObjects& CCoverSurfaceManager::GetSurfaceObjects() const
 112 {
 113         return m_surfaceObjects;
 114 }