Code/Sandbox/Plugins/EditorCommon/QPropertyTree/CharacterAnimationSelector.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #pragma once
   4
   5 #include "StdAfx.h"
   6
   7 #include <Models/CharacterAnimationsModel.h>
   8 #include <Dialogs/TreeViewDialog.h>
   9 #include "IResourceSelectorHost.h"
  10
  11 #include <CryAnimation/ICryAnimation.h>
  12
  13 dll_string CharacterAnimationSelector(const SResourceSelectorContext& x, const char* previousValue, IEntity* pEntity)
  14 {
  15         if (!pEntity || !pEntity->GetCharacter(0) || !pEntity->GetCharacter(0)->GetIAnimationSet())
  16                 return previousValue;
  17
  18         IAnimationSet* pAnimSet = pEntity->GetCharacter(0)->GetIAnimationSet();
  19         CCharacterAnimationsModel* pModel = new CCharacterAnimationsModel(*pAnimSet);
  20         CTreeViewDialog dialog(x.parentWidget);
  21         QString selectedValue(previousValue);
  22
  23         dialog.Initialize(pModel, 0);
  24
  25         for (int32 row = pModel->rowCount(); row--; )
  26         {
  27                 const QModelIndex index = pModel->index(row, 0);
  28                 if (pModel->data(index, Qt::DisplayRole).value<QString>() == selectedValue)
  29                 {
  30                         dialog.SetSelectedValue(index, false);
  31                 }
  32         }
  33
  34         if (dialog.exec())
  35         {
  36                 QModelIndex index = dialog.GetSelected();
  37                 if (index.isValid())
  38                 {
  39                         return index.data().value<QString>().toLocal8Bit().data();
  40                 }
  41         }
  42
  43         return previousValue;
  44 }
  45
  46 dll_string ValidateCharacterAnimation(const SResourceSelectorContext& x, const char* newValue, const char* previousValue, IEntity* pEntity)
  47 {
  48         if (!newValue || !*newValue)
  49                 return dll_string();
  50
  51         if (!pEntity || !pEntity->GetCharacter(0) || !pEntity->GetCharacter(0)->GetIAnimationSet())
  52                 return previousValue;
  53
  54         IAnimationSet* pAnimSet = pEntity->GetCharacter(0)->GetIAnimationSet();
  55         CCharacterAnimationsModel* pModel = new CCharacterAnimationsModel(*pAnimSet);
  56         int itemCount = pModel->rowCount();
  57         for (int i = 0; i < itemCount; ++i)
  58         {
  59                 QModelIndex index = pModel->index(i, 0);
  60                 if (pModel->data(index, Qt::DisplayRole).value<QString>() == newValue)
  61                         return newValue;
  62         }
  63
  64         return previousValue;
  65 }
  66
  67 dll_string ValidateTrackCharacterAnimation(const SResourceSelectorContext& x, const char* newValue, const char* previousValue, IEntity* pEntity)
  68 {
  69         return newValue;
  70 }
  71
  72 REGISTER_RESOURCE_VALIDATING_SELECTOR("TrackCharacterAnimation", CharacterAnimationSelector, ValidateTrackCharacterAnimation, "")
  73 REGISTER_RESOURCE_VALIDATING_SELECTOR("CharacterAnimation", CharacterAnimationSelector, ValidateCharacterAnimation, "")