Code/Sandbox/Plugins/VehicleEditor/VehicleSeat.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "VehicleSeat.h"
   5
   6 #include "VehiclePrototype.h"
   7 #include "VehicleData.h"
   8 #include "VehiclePart.h"
   9 #include "VehicleEditorDialog.h"
  10 #include "VehiclePartsPanel.h"
  11 #include "VehicleWeapon.h"
  12 #include "Objects/DisplayContext.h"
  13
  14 REGISTER_CLASS_DESC(CVehicleSeatClassDesc);
  15
  16 //////////////////////////////////////////////////////////////////////////
  17 // CBase implementation.
  18 //////////////////////////////////////////////////////////////////////////
  19 IMPLEMENT_DYNCREATE(CVehicleSeat, CBaseObject)
  20
  21 namespace
  22 {
  23 IVariable* CreateWeaponVar()
  24 {
  25         IVariable* pWeapon = new CVariableArray;
  26         pWeapon->SetName("Weapon");
  27
  28         IVariable* pClass = new CVariable<string>;
  29         pClass->SetName("class");
  30         pWeapon->AddVariable(pClass);
  31
  32         IVariable* pHelpers = new CVariableArray;
  33         pHelpers->SetName("Helpers");
  34         pWeapon->AddVariable(pHelpers);
  35
  36         return pWeapon;
  37 }
  38 }
  39
  40 //////////////////////////////////////////////////////////////////////////
  41 CVehicleSeat::CVehicleSeat()
  42 {
  43         m_pVehicle = 0;
  44         m_pPart = 0;
  45         m_pVar = 0;
  46 }
  47
  48 //////////////////////////////////////////////////////////////////////////
  49 bool CVehicleSeat::Init(CBaseObject* prev, const string& file)
  50 {
  51         bool res = CBaseObject::Init(prev, file);
  52         return res;
  53 }
  54
  55 //////////////////////////////////////////////////////////////////////////
  56 void CVehicleSeat::Display(DisplayContext& dc)
  57 {
  58         return;
  59
  60         COLORREF color = GetColor();
  61
  62         if (IsSelected())
  63         {
  64                 dc.SetSelectedColor(0.6f);
  65         }
  66
  67         //if (dc.flags & DISPLAY_2D)
  68         {
  69                 AABB box;
  70                 GetLocalBounds(box);
  71                 dc.SetColor(color, 0.8f);
  72                 dc.PushMatrix(GetWorldTM());
  73                 dc.SetLineWidth(2);
  74                 dc.DrawWireBox(box.min, box.max);
  75                 dc.SetLineWidth(0);
  76                 dc.PopMatrix();
  77         }
  78
  79         DrawDefault(dc);
  80 }
  81
  82 //////////////////////////////////////////////////////////////////////////
  83 bool CVehicleSeat::HitTest(HitContext& hc)
  84 {
  85         return false;
  86 }
  87
  88 //////////////////////////////////////////////////////////////////////////
  89 void CVehicleSeat::GetBoundBox(AABB& box)
  90 {
  91         box.Reset();
  92         // Transform local bounding box into world space.
  93         GetLocalBounds(box);
  94         box.SetTransformedAABB(GetWorldTM(), box);
  95 }
  96
  97 //////////////////////////////////////////////////////////////////////////
  98 void CVehicleSeat::GetLocalBounds(AABB& box)
  99 {
 100         // return local bounds
 101         // todo: take box around sit and eye helpers?
 102 }
 103
 104 //////////////////////////////////////////////////////////////////////////
 105 void CVehicleSeat::UpdateObjectFromVar()
 106 {
 107         if (!m_pVar)
 108         {
 109                 // create default seat here
 110                 m_pVar = CreateDefaultChildOf("Seats");
 111                 IVariable* pSeats = GetChildVar(m_pVehicle->GetVariable(), "Seats");
 112                 if (pSeats && m_pVar)
 113                         pSeats->AddVariable(m_pVar);
 114
 115                 if (IVariable* pPart = GetChildVar(m_pVar, "part"))
 116                 {
 117                         pPart->AddOnSetCallback(functor(*this, &CVehicleSeat::OnSetPart));
 118                 }
 119         }
 120
 121         if (IVariable* pName = GetChildVar(m_pVar, "name"))
 122         {
 123                 // update obj name
 124                 string name;
 125                 pName->Get(name);
 126                 SetName(name);
 127         }
 128
 129         if (IVariable* pPart = GetChildVar(m_pVar, "part"))
 130         {
 131                 string val;
 132                 pPart->Get(val);
 133
 134                 DetachThis();
 135                 CBaseObject* pPartObj = GetIEditor()->GetObjectManager()->FindObject(val);
 136
 137                 if (val == "" || !pPartObj || !pPartObj->IsKindOf(RUNTIME_CLASS(CVehiclePart)))
 138                 {
 139                         m_pVehicle->AttachChild(this);
 140                 }
 141                 else
 142                 {
 143                         pPartObj->AttachChild(this);
 144                 }
 145         }
 146
 147 }
 148
 149 //////////////////////////////////////////////////////////////////////////
 150 void CVehicleSeat::SetVariable(IVariable* pVar)
 151 {
 152         m_pVar = pVar;
 153         if (pVar)
 154                 UpdateObjectFromVar();
 155
 156         if (IVariable* pPart = GetChildVar(m_pVar, "part"))
 157         {
 158                 pPart->AddOnSetCallback(functor(*this, &CVehicleSeat::OnSetPart));
 159         }
 160 }
 161
 162 //////////////////////////////////////////////////////////////////////////
 163 void CVehicleSeat::OnSetPart(IVariable* pVar)
 164 {
 165         UpdateObjectFromVar();
 166 }
 167
 168 //////////////////////////////////////////////////////////////////////////
 169 void CVehicleSeat::Done()
 170 {
 171         VeedLog("[CVehicleSeat:Done] <%s>", GetName());
 172
 173         int nChildCount = GetChildCount();
 174         for (int i = nChildCount - 1; i >= 0; --i)
 175         {
 176                 CBaseObject* pChild = GetChild(i);
 177                 VeedLog("[CVehicleSeat]: deleting %s", pChild->GetName());
 178                 GetIEditor()->DeleteObject(GetChild(i));
 179         }
 180
 181         CBaseObject::Done();
 182 }
 183
 184 //////////////////////////////////////////////////////////////////////////
 185 void CVehicleSeat::UpdateVarFromObject()
 186 {
 187         assert(m_pVar);
 188         if (!m_pVar)
 189                 return;
 190
 191         IEntity* pEnt = m_pVehicle->GetCEntity()->GetIEntity();
 192         if (!pEnt)
 193         {
 194                 CryWarning(VALIDATOR_MODULE_EDITOR, VALIDATOR_WARNING, "[CVehicleSeat::UpdateVariable] pEnt is null, returning");
 195                 return;
 196         }
 197
 198         if (IVariable* pVar = GetChildVar(m_pVar, "name"))
 199                 pVar->Set(GetName());
 200
 201         // update part variable
 202         CBaseObject* pParent = GetParent();
 203         IVariable* pVar = GetChildVar(m_pVar, "part");
 204         if (pParent && pVar)
 205         {
 206                 if (pParent == m_pVehicle)
 207                         pVar->Set("");
 208                 else
 209                         pVar->Set(pParent->GetName());
 210         }
 211 }
 212
 213 //////////////////////////////////////////////////////////////////////////
 214 void CVehicleSeat::SetVehicle(CVehiclePrototype* pProt)
 215 {
 216         m_pVehicle = pProt;
 217         if (pProt)
 218                 SetWorldTM(pProt->GetWorldTM());
 219
 220 }
 221
 222 //////////////////////////////////////////////////////////////////////////
 223 void CVehicleSeat::AddWeapon(int weaponType, CVehicleWeapon* pWeap, IVariable* pVar /*=0*/)
 224 {
 225         AttachChild(pWeap, true);
 226         pWeap->SetVehicle(m_pVehicle);
 227
 228         IVariable* pWeaponVar = pVar;
 229
 230         // if no variable, create and add it
 231         if (!pWeaponVar)
 232         {
 233                 const char* category = (weaponType == WEAPON_PRIMARY) ? "Primary" : "Secondary";
 234
 235                 IVariable* pWeapons = GetChildVar(m_pVar, category, true);
 236                 if (pWeapons)
 237                 {
 238                         pWeaponVar = CreateWeaponVar();
 239                         pWeapons->AddVariable(pWeaponVar);
 240                 }
 241         }
 242
 243         pWeap->SetVariable(pWeaponVar);
 244         pWeap->UpdateObjectFromVar();
 245 }
 246
 247 //////////////////////////////////////////////////////////////////////////
 248 void CVehicleSeat::OnObjectEvent(CBaseObject* node, int event)
 249 {
 250         if (event == OBJECT_ON_DELETE)
 251         {
 252                 VeedLog("[CVehicleSeat]: ON_DELETE for %s", node->GetName());
 253                 // delete variable
 254                 if (IVeedObject* pVO = IVeedObject::GetVeedObject(node))
 255                 {
 256                         if (pVO->DeleteVar())
 257                         {
 258                                 if (m_pVar)
 259                                         m_pVar->DeleteVariable(pVO->GetVariable(), true);
 260                                 pVO->SetVariable(0);
 261                                 VeedLog("[CVehiclePart] deleting var for %s", node->GetName());
 262                         }
 263                 }
 264         }
 265 }
 266
 267 //////////////////////////////////////////////////////////////////////////
 268 void CVehicleSeat::AttachChild(CBaseObject* child, bool bKeepPos, bool bInvalidateTM)
 269 {
 270         child->AddEventListener(functor(*this, &CVehicleSeat::OnObjectEvent));
 271
 272         CBaseObject::AttachChild(child, bKeepPos, bInvalidateTM);
 273 }