Code/Sandbox/Plugins/VehicleEditor/VehicleHelperObject.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "VehicleHelperObject.h"
   5 #include "Viewport.h"
   6
   7 #include "VehicleData.h"
   8 #include "VehiclePrototype.h"
   9 #include "VehiclePart.h"
  10
  11 REGISTER_CLASS_DESC(CVehicleHelperClassDesc);
  12
  13 //////////////////////////////////////////////////////////////////////////
  14 // CBase implementation.
  15 //////////////////////////////////////////////////////////////////////////
  16 IMPLEMENT_DYNCREATE(CVehicleHelper, CBaseObject)
  17
  18 #define RADIUS                     0.05f
  19
  20 #define VEHICLE_HELPER_COLOR       (RGB(255, 255, 120))
  21 #define VEHICLE_ASSET_HELPER_COLOR (RGB(120, 255, 120))
  22
  23 //////////////////////////////////////////////////////////////////////////
  24 CVehicleHelper::CVehicleHelper()
  25 {
  26         m_pVar = 0;
  27         m_fromGeometry = false;
  28         m_pVehicle = 0;
  29
  30         ChangeColor(VEHICLE_HELPER_COLOR);
  31
  32         InitOnTransformCallback(this);
  33 }
  34
  35 CVehicleHelper::~CVehicleHelper()
  36 {
  37 }
  38
  39 //////////////////////////////////////////////////////////////////////////
  40 void CVehicleHelper::SetVariable(IVariable* pVar)
  41 {
  42         m_pVar = pVar;
  43
  44         DisableUpdateObjectOnVarChange("position");
  45         DisableUpdateObjectOnVarChange("direction");
  46
  47         if (IVariable* pVar = GetChildVar(m_pVar, "position"))
  48         {
  49                 pVar->SetLimits(-10, 10);
  50         }
  51
  52         EnableUpdateObjectOnVarChange("position");
  53         EnableUpdateObjectOnVarChange("direction");
  54 }
  55
  56 //////////////////////////////////////////////////////////////////////////
  57 void CVehicleHelper::UpdateVarFromObject()
  58 {
  59         if (IsFromGeometry())
  60         {
  61                 assert(!m_pVar);
  62                 return;
  63         }
  64
  65         assert(m_pVar);
  66         if (!m_pVar || !m_pVehicle)
  67                 return;
  68
  69         IEntity* pEnt = m_pVehicle->GetCEntity()->GetIEntity();
  70         if (!pEnt)
  71         {
  72                 CryWarning(VALIDATOR_MODULE_EDITOR, VALIDATOR_WARNING, "[CVehicleHelper::UpdateVariable] pEnt is null, returning");
  73                 return;
  74         }
  75
  76         if (IVariable* pVar = GetChildVar(m_pVar, "name"))
  77                 pVar->Set(GetName());
  78
  79         if (IVariable* pVar = GetChildVar(m_pVar, "position"))
  80         {
  81                 Vec3 pos = pEnt->GetWorldTM().GetInvertedFast().TransformPoint(GetWorldPos());
  82                 pVar->Set(pos);
  83         }
  84
  85         IVariable* pVar = GetChildVar(m_pVar, "direction");
  86         if (!pVar)
  87         {
  88                 pVar = new CVariable<Vec3>;
  89                 pVar->SetName("direction");
  90                 m_pVar->AddVariable(pVar);
  91         }
  92
  93         IVariable* pPartVar = GetChildVar(m_pVar, "part");
  94         if (!pPartVar)
  95         {
  96                 pPartVar = new CVariable<string>;
  97                 pPartVar->SetName("part");
  98                 m_pVar->AddVariable(pPartVar);
  99         }
 100
 101         Matrix33 relTM = Matrix33(pEnt->GetWorldTM().GetInvertedFast()) * Matrix33(GetWorldTM());
 102         Vec3 dir = relTM.TransformVector(FORWARD_DIRECTION);
 103         pVar->Set(dir);
 104
 105 }
 106
 107 //////////////////////////////////////////////////////////////////////////
 108 void CVehicleHelper::UpdateObjectFromVar()
 109 {
 110         if (!m_pVar || !m_pVehicle)
 111                 return;
 112
 113         if (IVariable* pVar = GetChildVar(m_pVar, "position"))
 114         {
 115                 Vec3 local(ZERO);
 116                 pVar->Get(local);
 117                 Matrix34 tm = GetWorldTM();
 118                 tm.SetTranslation(m_pVehicle->GetCEntity()->GetIEntity()->GetWorldTM().TransformPoint(local));
 119                 SetWorldTM(tm);
 120         }
 121 }
 122
 123 //////////////////////////////////////////////////////////////////////////
 124 void CVehicleHelper::Done()
 125 {
 126         VeedLog("[CVehicleHelper:Done] <%s>", GetName());
 127         CBaseObject::Done();
 128 }
 129
 130 //////////////////////////////////////////////////////////////////////////
 131 bool CVehicleHelper::Init(CBaseObject* prev, const string& file)
 132 {
 133         SetColor(RGB(255, 255, 0));
 134         bool res = CBaseObject::Init(prev, file);
 135
 136         return res;
 137 }
 138
 139 //////////////////////////////////////////////////////////////////////////
 140 void CVehicleHelper::PostClone(CBaseObject* pFromObject, CObjectCloneContext& ctx)
 141 {
 142         CBaseObject* pFromParent = pFromObject->GetParent();
 143         if (pFromParent)
 144         {
 145                 CBaseObject* pChildParent = ctx.FindClone(pFromParent);
 146                 if (pChildParent)
 147                         pChildParent->AttachChild(this, false);
 148                 else
 149                 {
 150                         //       // helper was cloned and attached to same parent
 151                         //       if (pFromParent->IsKindOf(RUNTIME_CLASS(CVehiclePart)))
 152                         //         ((CVehiclePart*)pFromParent)->AddHelper( this, 0 );
 153                         //       else
 154                         //         pFromParent->AttachChild(this, false);
 155                 }
 156         }
 157 }
 158
 159 //////////////////////////////////////////////////////////////////////////
 160 bool CVehicleHelper::HitTest(HitContext& hc)
 161 {
 162         Vec3 origin = GetWorldPos();
 163         float radius = RADIUS;
 164
 165         Vec3 w = origin - hc.raySrc;
 166         w = hc.rayDir.Cross(w);
 167         float d = w.GetLength();
 168
 169         if (d < radius + hc.distanceTolerance)
 170         {
 171                 Vec3 i0;
 172                 if (Intersect::Ray_SphereFirst(Ray(hc.raySrc, hc.rayDir), Sphere(origin, radius), i0))
 173                 {
 174                         hc.dist = hc.raySrc.GetDistance(i0);
 175                         return true;
 176                 }
 177                 hc.dist = hc.raySrc.GetDistance(origin);
 178                 return true;
 179         }
 180         return false;
 181 }
 182
 183 //////////////////////////////////////////////////////////////////////////
 184 void CVehicleHelper::Display(DisplayContext& dc)
 185 {
 186         COLORREF color = GetColor();
 187         float radius = RADIUS;
 188
 189         //dc.SetColor( color, 0.5f );
 190         //dc.DrawBall( GetPos(), radius );
 191
 192         if (IsSelected())
 193         {
 194                 dc.SetSelectedColor(0.6f);
 195         }
 196
 197         AABB box;
 198         GetLocalBounds(box);
 199         dc.PushMatrix(GetWorldTM());
 200
 201         if (!IsHighlighted())
 202         {
 203                 dc.SetColor(color, 0.8f);
 204                 dc.SetLineWidth(2);
 205                 dc.DrawWireBox(box.min, box.max);
 206         }
 207
 208         if (!IsSelected())
 209         {
 210                 // direction vector
 211                 Vec3 dirEndPos(0, 4 * box.max.y, 0);
 212                 dc.DrawArrow(Vec3(0, box.max.y, 0), dirEndPos, 0.15f);
 213         }
 214
 215         dc.PopMatrix();
 216
 217         // draw label
 218         if (dc.flags & DISPLAY_HIDENAMES)
 219         {
 220                 Vec3 p(GetWorldPos());
 221                 DrawLabel(dc, p, RGB(255, 255, 255));
 222         }
 223
 224         DrawDefault(dc);
 225 }
 226
 227 //////////////////////////////////////////////////////////////////////////
 228 void CVehicleHelper::GetBoundBox(AABB& box)
 229 {
 230         // Transform local bounding box into world space.
 231         GetLocalBounds(box);
 232         box.SetTransformedAABB(GetWorldTM(), box);
 233 }
 234
 235 //////////////////////////////////////////////////////////////////////////
 236 void CVehicleHelper::GetLocalBounds(AABB& box)
 237 {
 238         // return local bounds
 239         float r = RADIUS;
 240         box.min = -Vec3(r, r, r);
 241         box.max = Vec3(r, r, r);
 242 }
 243
 244 //////////////////////////////////////////////////////////////////////////
 245 void CVehicleHelper::GetBoundSphere(Vec3& pos, float& radius)
 246 {
 247         pos = GetPos();
 248         radius = RADIUS;
 249 }
 250
 251 //////////////////////////////////////////////////////////////////////////
 252 void CVehicleHelper::UpdateScale(float scale)
 253 {
 254         if (IVariable* pPos = GetChildVar(m_pVar, "position"))
 255         {
 256                 Vec3 pos;
 257                 pPos->Get(pos);
 258                 pPos->Set(pos *= scale);
 259                 UpdateObjectFromVar();
 260         }
 261 }
 262
 263 //////////////////////////////////////////////////////////////////////////
 264 void CVehicleHelper::SetIsFromGeometry(bool b)
 265 {
 266         m_fromGeometry = b;
 267         if (m_fromGeometry)
 268         {
 269                 ChangeColor(VEHICLE_ASSET_HELPER_COLOR);
 270                 SetFrozen(true);
 271         }
 272         else
 273         {
 274                 ChangeColor(VEHICLE_HELPER_COLOR);
 275                 SetFrozen(false);
 276         }
 277 }
 278
 279 //////////////////////////////////////////////////////////////////////////
 280 void CVehicleHelper::OnTransform()
 281 {
 282         UpdateVarFromObject();
 283 }