//-----------------------------------------------------------------------------
// Copyright (c) 2013 GarageGames, LLC
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.
//-----------------------------------------------------------------------------

#ifndef LEVEL_ASSET_H
#include "LevelAsset.h"
#endif

#ifndef _ASSET_MANAGER_H_
#include "assets/assetManager.h"
#endif

#ifndef _CONSOLETYPES_H_
#include "console/consoleTypes.h"
#endif

#ifndef _TAML_
#include "persistence/taml/taml.h"
#endif

#ifndef _ASSET_PTR_H_
#include "assets/assetPtr.h"
#endif

// Debug Profiling.
#include "platform/profiler.h"
#include "gfx/gfxDrawUtil.h"
#include "T3D/SubScene.h"


//-----------------------------------------------------------------------------

IMPLEMENT_CONOBJECT(LevelAsset);

ConsoleType(LevelAssetPtr, TypeLevelAssetPtr, const char*, "")

//-----------------------------------------------------------------------------

ConsoleGetType(TypeLevelAssetPtr)
{
   // Fetch asset Id.
   return *((const char**)(dptr));
}

//-----------------------------------------------------------------------------

ConsoleSetType(TypeLevelAssetPtr)
{
   // Was a single argument specified?
   if (argc == 1)
   {
      // Yes, so fetch field value.
      *((const char**)dptr) = StringTable->insert(argv[0]);

      return;
   }

   // Warn.
   Con::warnf("(TypeLevelAssetPtr) - Cannot set multiple args to a single asset.");
}

//-----------------------------------------------------------------------------
ConsoleType(assetIdString, TypeLevelAssetId, const char*, "")

ConsoleGetType(TypeLevelAssetId)
{
   // Fetch asset Id.
   return *((const char**)(dptr));
}

ConsoleSetType(TypeLevelAssetId)
{
   // Was a single argument specified?
   if (argc == 1)
   {
      *((const char**)dptr) = StringTable->insert(argv[0]);

      return;
   }

   // Warn.
   Con::warnf("(TypeLevelAssetId) - Cannot set multiple args to a single asset.");
}
//-----------------------------------------------------------------------------

LevelAsset::LevelAsset() : AssetBase()
{
   mLevelName = StringTable->EmptyString();
   mLevelFile = StringTable->EmptyString();
   mPostFXPresetFile = StringTable->EmptyString();
   mDecalsFile = StringTable->EmptyString();
   mForestFile = StringTable->EmptyString();
   mNavmeshFile = StringTable->EmptyString();

   mLevelPath = StringTable->EmptyString();
   mPostFXPresetPath = StringTable->EmptyString();
   mDecalsPath = StringTable->EmptyString();
   mForestPath = StringTable->EmptyString();
   mNavmeshPath = StringTable->EmptyString();

   mGameModesNames = StringTable->EmptyString();

   mEditorFile = StringTable->EmptyString();
   mBakedSceneFile = StringTable->EmptyString();

   mPreviewImageAssetId = StringTable->EmptyString();
   mPreviewImageAsset = StringTable->EmptyString();
}

//-----------------------------------------------------------------------------

LevelAsset::~LevelAsset()
{
}

//-----------------------------------------------------------------------------

void LevelAsset::initPersistFields()
{
   docsURL;
   // Call parent.
   Parent::initPersistFields();

   addProtectedField("LevelFile", TypeAssetLooseFilePath, Offset(mLevelFile, LevelAsset),
      &setLevelFile, &getLevelFile, "Path to the actual level file.");
   addField("LevelName", TypeString, Offset(mLevelName, LevelAsset), "Human-friendly name for the level.");

   addProtectedField("PostFXPresetFile", TypeAssetLooseFilePath, Offset(mPostFXPresetFile, LevelAsset),
      &setPostFXPresetFile, &getPostFXPresetFile, "Path to the level's postFXPreset.");
   addProtectedField("DecalsFile", TypeAssetLooseFilePath, Offset(mDecalsFile, LevelAsset),
      &setDecalsFile, &getDecalsFile, "Path to the decals cache file.");
   addProtectedField("ForestFile", TypeAssetLooseFilePath, Offset(mForestFile, LevelAsset),
      &setForestFile, &getForestFile, "Path to the Forest cache file.");
   addProtectedField("NavmeshFile", TypeAssetLooseFilePath, Offset(mNavmeshFile, LevelAsset),
      &setNavmeshFile, &getNavmeshFile, "Path to the navmesh file.");

   addProtectedField("EditorFile", TypeAssetLooseFilePath, Offset(mEditorFile, LevelAsset),
      &setEditorFile, &getEditorFile, "Path to the level file with objects that were removed as part of the baking process. Loaded when the editor is loaded for ease of editing.");
   addProtectedField("BakedSceneFile", TypeAssetLooseFilePath, Offset(mBakedSceneFile, LevelAsset),
      &setBakedSceneFile, &getBakedSceneFile, "Path to the level file with the objects generated as part of the baking process");

   addField("gameModesNames", TypeString, Offset(mGameModesNames, LevelAsset), "Name of the Game Mode to be used with this level");
}

//------------------------------------------------------------------------------

void LevelAsset::copyTo(SimObject* object)
{
   // Call to parent.
   Parent::copyTo(object);
}

//
void LevelAsset::initializeAsset()
{
   // Call parent.
   Parent::initializeAsset();

   loadAsset();
}

void LevelAsset::onAssetRefresh(void)
{
   loadAsset();
}

void LevelAsset::loadAsset()
{
   // Ensure the image-file is expanded.
   mLevelPath = getOwned() ? expandAssetFilePath(mLevelFile) : mLevelPath;
   mPostFXPresetPath = getOwned() ? expandAssetFilePath(mPostFXPresetFile) : mPostFXPresetPath;
   mDecalsPath = getOwned() ? expandAssetFilePath(mDecalsFile) : mDecalsPath;
   mForestPath = getOwned() ? expandAssetFilePath(mForestFile) : mForestPath;
   mNavmeshPath = getOwned() ? expandAssetFilePath(mNavmeshFile) : mNavmeshPath;

   StringTableEntry previewImageAssetId = getAssetDependencyField("previewImageAsset");

   if (previewImageAssetId != StringTable->EmptyString())
   {
      mPreviewImageAssetId = previewImageAssetId;
      mPreviewImageAsset = mPreviewImageAssetId;
   }
}

//
void LevelAsset::setLevelFile(const char* pLevelFile)
{
   // Sanity!
   AssertFatal(pLevelFile != NULL, "Cannot use a NULL level file.");

   // Fetch image file.
   pLevelFile = StringTable->insert(pLevelFile, true);

   // Ignore no change,
   if (pLevelFile == mLevelFile)
      return;

   // Update.
   mLevelFile = getOwned() ? expandAssetFilePath(pLevelFile) : pLevelFile;

   // Refresh the asset.
   refreshAsset();
}

StringTableEntry LevelAsset::getPreviewImageAsset() const
{
   return mPreviewImageAssetId;
}

StringTableEntry LevelAsset::getPreviewImagePath(void) const
{
   if (mPreviewImageAsset.notNull())
   {
      return mPreviewImageAsset->getImageFile();
   }

   return StringTable->EmptyString();
}

void LevelAsset::setEditorFile(const char* pEditorFile)
{
   // Sanity!
   AssertFatal(pEditorFile != NULL, "Cannot use a NULL level file.");

   // Fetch image file.
   pEditorFile = StringTable->insert(pEditorFile, true);

   // Ignore no change,
   if (pEditorFile == mEditorFile)
      return;

   // Update.
   mEditorFile = getOwned() ? expandAssetFilePath(pEditorFile) : pEditorFile;

   // Refresh the asset.
   refreshAsset();
}

void LevelAsset::setBakedSceneFile(const char* pBakedSceneFile)
{
   // Sanity!
   AssertFatal(pBakedSceneFile != NULL, "Cannot use a NULL level file.");

   // Fetch image file.
   pBakedSceneFile = StringTable->insert(pBakedSceneFile, true);

   // Ignore no change,
   if (pBakedSceneFile == mBakedSceneFile)
      return;

   // Update.
   mBakedSceneFile = getOwned() ? expandAssetFilePath(pBakedSceneFile) : pBakedSceneFile;

   // Refresh the asset.
   refreshAsset();
}

void LevelAsset::setPostFXPresetFile(const char* pPostFXPresetFile)
{
   // Sanity!
   AssertFatal(pPostFXPresetFile != NULL, "Cannot use a NULL postFX preset file.");

   // Fetch file.
   pPostFXPresetFile = StringTable->insert(pPostFXPresetFile, true);

   // Ignore no change,
   if (pPostFXPresetFile == mPostFXPresetFile)
      return;

   // Update.
   mPostFXPresetFile = getOwned() ? expandAssetFilePath(pPostFXPresetFile) : pPostFXPresetFile;

   // Refresh the asset.
   refreshAsset();
}

void LevelAsset::setDecalsFile(const char* pDecalsFile)
{
   // Sanity!
   AssertFatal(pDecalsFile != NULL, "Cannot use a NULL decals file.");

   // Fetch file.
   pDecalsFile = StringTable->insert(pDecalsFile, true);

   // Ignore no change,
   if (pDecalsFile == mDecalsFile)
      return;

   // Update.
   mDecalsFile = getOwned() ? expandAssetFilePath(pDecalsFile) : pDecalsFile;

   // Refresh the asset.
   refreshAsset();
}

void LevelAsset::setForestFile(const char* pForestFile)
{
   // Sanity!
   AssertFatal(pForestFile != NULL, "Cannot use a NULL decals file.");

   // Fetch file.
   pForestFile = StringTable->insert(pForestFile, true);

   // Ignore no change,
   if (pForestFile == mForestFile)
      return;

   // Update.
   mForestFile = getOwned() ? expandAssetFilePath(pForestFile) : pForestFile;

   // Refresh the asset.
   refreshAsset();
}

void LevelAsset::setNavmeshFile(const char* pNavmeshFile)
{
   // Sanity!
   AssertFatal(pNavmeshFile != NULL, "Cannot use a NULL Navmesh file.");

   // Fetch file.
   pNavmeshFile = StringTable->insert(pNavmeshFile, true);

   // Ignore no change,
   if (pNavmeshFile == mNavmeshFile)
      return;

   // Update.
   mNavmeshFile = getOwned() ? expandAssetFilePath(pNavmeshFile) : pNavmeshFile;

   // Refresh the asset.
   refreshAsset();
}

void LevelAsset::loadDependencies()
{
   //First, load any material, animation, etc assets we may be referencing in our asset
   // Find any asset dependencies.
   AssetManager::typeAssetDependsOnHash::Iterator assetDependenciesItr = mpOwningAssetManager->getDependedOnAssets()->find(mpAssetDefinition->mAssetId);

   // Does the asset have any dependencies?
   if (assetDependenciesItr != mpOwningAssetManager->getDependedOnAssets()->end())
   {
      // Iterate all dependencies.
      while (assetDependenciesItr != mpOwningAssetManager->getDependedOnAssets()->end() && assetDependenciesItr->key == mpAssetDefinition->mAssetId)
      {
         //Force it to be loaded by acquiring it
         StringTableEntry assetId = assetDependenciesItr->value;
         mAssetDependencies.push_back(AssetDatabase.acquireAsset<AssetBase>(assetId));

         // Next dependency.
         assetDependenciesItr++;
      }
   }
}

void LevelAsset::unloadDependencies()
{
   for (U32 i = 0; i < mAssetDependencies.size(); i++)
   {
      AssetBase* assetDef = mAssetDependencies[i];
      AssetDatabase.releaseAsset(assetDef->getAssetId());
   }
}

DefineEngineMethod(LevelAsset, getLevelPath, const char*, (), ,
   "Gets the full path of the asset's defined level file.\n"
   "@return The string result of the level path")
{
   return object->getLevelPath();
}

DefineEngineMethod(LevelAsset, getPreviewImageAsset, const char*, (), ,
   "Gets the full path of the asset's defined preview image file.\n"
   "@return The string result of the level preview image path")
{
   return object->getPreviewImageAsset();
}

DefineEngineMethod(LevelAsset, getPreviewImagePath, const char*, (), ,
   "Gets the full path of the asset's defined preview image file.\n"
   "@return The string result of the level preview image path")
{
   return object->getPreviewImagePath();
}

DefineEngineMethod(LevelAsset, getPostFXPresetPath, const char*, (), ,
   "Gets the full path of the asset's defined postFX preset file.\n"
   "@return The string result of the postFX preset path")
{
   return object->getPostFXPresetPath();
}

DefineEngineMethod(LevelAsset, getDecalsPath, const char*, (), ,
   "Gets the full path of the asset's defined decal file.\n"
   "@return The string result of the decal path")
{
   return object->getDecalsPath();
}

DefineEngineMethod(LevelAsset, getForestPath, const char*, (), ,
   "Gets the full path of the asset's defined forest file.\n"
   "@return The string result of the forest path")
{
   return object->getForestPath();
}

DefineEngineMethod(LevelAsset, getNavmeshPath, const char*, (), ,
   "Gets the full path of the asset's defined navmesh file.\n"
   "@return The string result of the navmesh path")
{
   return object->getNavmeshPath();
}

DefineEngineMethod(LevelAsset, loadDependencies, void, (), ,
   "Initiates the loading of asset dependencies for this level.")
{
   return object->loadDependencies();
}

DefineEngineMethod(LevelAsset, unloadDependencies, void, (), ,
   "Initiates the unloading of previously loaded asset dependencies for this level.")
{
   return object->unloadDependencies();
}

#ifdef TORQUE_TOOLS
//-----------------------------------------------------------------------------
// GuiInspectorTypeAssetId
//-----------------------------------------------------------------------------

IMPLEMENT_CONOBJECT(GuiInspectorTypeLevelAssetPtr);

ConsoleDocClass(GuiInspectorTypeLevelAssetPtr,
   "@brief Inspector field type for Shapes\n\n"
   "Editor use only.\n\n"
   "@internal"
);

void GuiInspectorTypeLevelAssetPtr::consoleInit()
{
   Parent::consoleInit();

   ConsoleBaseType::getType(TypeLevelAssetPtr)->setInspectorFieldType("GuiInspectorTypeLevelAssetPtr");
}

GuiControl* GuiInspectorTypeLevelAssetPtr::constructEditControl()
{
   // Create base filename edit controls
   GuiControl* retCtrl = Parent::constructEditControl();
   if (retCtrl == NULL)
      return retCtrl;

   // Change filespec
   char szBuffer[512];
   dSprintf(szBuffer, sizeof(szBuffer), "AssetBrowser.showDialog(\"LevelAsset\", \"AssetBrowser.changeAsset\", %s, \"\");",
      getIdString());
   mBrowseButton->setField("Command", szBuffer);

   setDataField(StringTable->insert("targetObject"), NULL, mInspector->getInspectObject()->getIdString());

   // Create "Open in Editor" button
   mEditButton = new GuiBitmapButtonCtrl();

   dSprintf(szBuffer, sizeof(szBuffer), "$createAndAssignField = %s; AssetBrowser.setupCreateNewAsset(\"LevelAsset\", AssetBrowser.selectedModule, \"createAndAssignLevelAsset\");",
      getIdString());
   mEditButton->setField("Command", szBuffer);

   char bitmapName[512] = "ToolsModule:iconAdd_image";
   mEditButton->setBitmap(StringTable->insert(bitmapName));

   mEditButton->setDataField(StringTable->insert("Profile"), NULL, "GuiButtonProfile");
   mEditButton->setDataField(StringTable->insert("tooltipprofile"), NULL, "GuiToolTipProfile");
   mEditButton->setDataField(StringTable->insert("hovertime"), NULL, "1000");
   mEditButton->setDataField(StringTable->insert("tooltip"), NULL, "Test play this sound");

   mEditButton->registerObject();
   addObject(mEditButton);

   return retCtrl;
}

bool GuiInspectorTypeLevelAssetPtr::updateRects()
{
   S32 dividerPos, dividerMargin;
   mInspector->getDivider(dividerPos, dividerMargin);
   Point2I fieldExtent = getExtent();
   Point2I fieldPos = getPosition();

   mCaptionRect.set(0, 0, fieldExtent.x - dividerPos - dividerMargin, fieldExtent.y);
   mEditCtrlRect.set(fieldExtent.x - dividerPos + dividerMargin, 1, dividerPos - dividerMargin - 34, fieldExtent.y);

   bool resized = mEdit->resize(mEditCtrlRect.point, mEditCtrlRect.extent);
   if (mBrowseButton != NULL)
   {
      mBrowseRect.set(fieldExtent.x - 32, 2, 14, fieldExtent.y - 4);
      resized |= mBrowseButton->resize(mBrowseRect.point, mBrowseRect.extent);
   }

   if (mEditButton != NULL)
   {
      RectI shapeEdRect(fieldExtent.x - 16, 2, 14, fieldExtent.y - 4);
      resized |= mEditButton->resize(shapeEdRect.point, shapeEdRect.extent);
   }

   return resized;
}

IMPLEMENT_CONOBJECT(GuiInspectorTypeLevelAssetId);

ConsoleDocClass(GuiInspectorTypeLevelAssetId,
   "@brief Inspector field type for Levels\n\n"
   "Editor use only.\n\n"
   "@internal"
);

void GuiInspectorTypeLevelAssetId::consoleInit()
{
   Parent::consoleInit();

   ConsoleBaseType::getType(TypeLevelAssetId)->setInspectorFieldType("GuiInspectorTypeLevelAssetId");
}

#endif