dom/vr/XRRigidTransform.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this file,
   5  * You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/dom/XRRigidTransform.h"
   8 #include "mozilla/dom/DOMPoint.h"
   9 #include "mozilla/dom/Pose.h"
  10 #include "mozilla/dom/DOMPointBinding.h"
  11 #include "mozilla/HoldDropJSObjects.h"
  12
  13 namespace mozilla {
  14 namespace dom {
  15
  16 NS_IMPL_CYCLE_COLLECTION_CLASS(XRRigidTransform)
  17
  18 NS_IMPL_CYCLE_COLLECTION_UNLINK_BEGIN(XRRigidTransform)
  19   NS_IMPL_CYCLE_COLLECTION_UNLINK(mParent, mPosition, mOrientation, mInverse)
  20   NS_IMPL_CYCLE_COLLECTION_UNLINK_PRESERVED_WRAPPER
  21   tmp->mMatrixArray = nullptr;
  22 NS_IMPL_CYCLE_COLLECTION_UNLINK_END
  23
  24 NS_IMPL_CYCLE_COLLECTION_TRAVERSE_BEGIN(XRRigidTransform)
  25   NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mParent, mPosition, mOrientation, mInverse)
  26 NS_IMPL_CYCLE_COLLECTION_TRAVERSE_END
  27
  28 NS_IMPL_CYCLE_COLLECTION_TRACE_BEGIN(XRRigidTransform)
  29   NS_IMPL_CYCLE_COLLECTION_TRACE_PRESERVED_WRAPPER
  30   NS_IMPL_CYCLE_COLLECTION_TRACE_JS_MEMBER_CALLBACK(mMatrixArray)
  31 NS_IMPL_CYCLE_COLLECTION_TRACE_END
  32
  33 NS_IMPL_CYCLE_COLLECTION_ROOT_NATIVE(XRRigidTransform, AddRef)
  34 NS_IMPL_CYCLE_COLLECTION_UNROOT_NATIVE(XRRigidTransform, Release)
  35
  36 XRRigidTransform::XRRigidTransform(nsISupports* aParent,
  37                                    const gfx::PointDouble3D& aPosition,
  38                                    const gfx::QuaternionDouble& aOrientation)
  39     : mParent(aParent),
  40       mMatrixArray(nullptr),
  41       mPosition(nullptr),
  42       mOrientation(nullptr),
  43       mInverse(nullptr),
  44       mRawPosition(aPosition),
  45       mRawOrientation(aOrientation),
  46       mNeedsUpdate(true) {
  47   mozilla::HoldJSObjects(this);
  48   mRawTransformMatrix.SetRotationFromQuaternion(mRawOrientation);
  49   mRawTransformMatrix.PostTranslate(mRawPosition);
  50 }
  51
  52 XRRigidTransform::XRRigidTransform(nsISupports* aParent,
  53                                    const gfx::Matrix4x4Double& aTransform)
  54     : mParent(aParent),
  55       mMatrixArray(nullptr),
  56       mPosition(nullptr),
  57       mOrientation(nullptr),
  58       mInverse(nullptr),
  59       mNeedsUpdate(true) {
  60   mozilla::HoldJSObjects(this);
  61   gfx::PointDouble3D scale;
  62   mRawTransformMatrix = aTransform;
  63   mRawTransformMatrix.Decompose(mRawPosition, mRawOrientation, scale);
  64 }
  65
  66 XRRigidTransform::~XRRigidTransform() { mozilla::DropJSObjects(this); }
  67
  68 /* static */ already_AddRefed<XRRigidTransform> XRRigidTransform::Constructor(
  69     const GlobalObject& aGlobal, const DOMPointInit& aOrigin,
  70     const DOMPointInit& aDirection, ErrorResult& aRv) {
  71   gfx::PointDouble3D position(aOrigin.mX, aOrigin.mY, aOrigin.mZ);
  72   gfx::QuaternionDouble orientation(aDirection.mX, aDirection.mY, aDirection.mZ,
  73                                     aDirection.mW);
  74   orientation.Normalize();
  75   RefPtr<XRRigidTransform> obj =
  76       new XRRigidTransform(aGlobal.GetAsSupports(), position, orientation);
  77   return obj.forget();
  78 }
  79
  80 JSObject* XRRigidTransform::WrapObject(JSContext* aCx,
  81                                        JS::Handle<JSObject*> aGivenProto) {
  82   return XRRigidTransform_Binding::Wrap(aCx, this, aGivenProto);
  83 }
  84
  85 DOMPoint* XRRigidTransform::Position() {
  86   if (!mPosition) {
  87     mPosition = new DOMPoint(mParent, mRawPosition.x, mRawPosition.y,
  88                              mRawPosition.z, 1.0f);
  89   }
  90   return mPosition;
  91 }
  92
  93 DOMPoint* XRRigidTransform::Orientation() {
  94   if (!mOrientation) {
  95     mOrientation = new DOMPoint(mParent, mRawOrientation.x, mRawOrientation.y,
  96                                 mRawOrientation.z, mRawOrientation.w);
  97   }
  98   return mOrientation;
  99 }
 100
 101 XRRigidTransform& XRRigidTransform::operator=(const XRRigidTransform& aOther) {
 102   Update(aOther.mRawPosition, aOther.mRawOrientation);
 103   return *this;
 104 }
 105
 106 gfx::QuaternionDouble XRRigidTransform::RawOrientation() const {
 107   return mRawOrientation;
 108 }
 109 gfx::PointDouble3D XRRigidTransform::RawPosition() const {
 110   return mRawPosition;
 111 }
 112
 113 void XRRigidTransform::Update(const gfx::PointDouble3D& aPosition,
 114                               const gfx::QuaternionDouble& aOrientation) {
 115   mNeedsUpdate = true;
 116   mRawPosition = aPosition;
 117   mRawOrientation = aOrientation;
 118   mRawTransformMatrix.SetRotationFromQuaternion(mRawOrientation);
 119   mRawTransformMatrix.PostTranslate(mRawPosition);
 120   UpdateInternal();
 121 }
 122
 123 void XRRigidTransform::Update(const gfx::Matrix4x4Double& aTransform) {
 124   mNeedsUpdate = true;
 125   mRawTransformMatrix = aTransform;
 126   gfx::PointDouble3D scale;
 127   mRawTransformMatrix.Decompose(mRawPosition, mRawOrientation, scale);
 128   UpdateInternal();
 129 }
 130
 131 void XRRigidTransform::UpdateInternal() {
 132   if (mPosition) {
 133     mPosition->SetX(mRawPosition.x);
 134     mPosition->SetY(mRawPosition.y);
 135     mPosition->SetZ(mRawPosition.z);
 136   }
 137   if (mOrientation) {
 138     mOrientation->SetX(mRawOrientation.x);
 139     mOrientation->SetY(mRawOrientation.y);
 140     mOrientation->SetZ(mRawOrientation.z);
 141     mOrientation->SetW(mRawOrientation.w);
 142   }
 143   if (mInverse) {
 144     gfx::Matrix4x4Double inverseMatrix = mRawTransformMatrix;
 145     Unused << inverseMatrix.Invert();
 146     mInverse->Update(inverseMatrix);
 147   }
 148 }
 149
 150 void XRRigidTransform::GetMatrix(JSContext* aCx,
 151                                  JS::MutableHandle<JSObject*> aRetval,
 152                                  ErrorResult& aRv) {
 153   if (!mMatrixArray || mNeedsUpdate) {
 154     mNeedsUpdate = false;
 155
 156     const uint32_t size = 16;
 157     float components[size] = {};
 158     // In order to avoid some platforms which only copy
 159     // the first or last two bytes of a Float64 to a Float32.
 160     for (uint32_t i = 0; i < size; ++i) {
 161       components[i] = mRawTransformMatrix.components[i];
 162     }
 163     Pose::SetFloat32Array(aCx, this, aRetval, mMatrixArray, components, 16,
 164                           aRv);
 165     if (!mMatrixArray) {
 166       return;
 167     }
 168   }
 169   if (mMatrixArray) {
 170     JS::ExposeObjectToActiveJS(mMatrixArray);
 171   }
 172   aRetval.set(mMatrixArray);
 173 }
 174
 175 already_AddRefed<XRRigidTransform> XRRigidTransform::Inverse() {
 176   if (!mInverse) {
 177     gfx::Matrix4x4Double inverseMatrix = mRawTransformMatrix;
 178     Unused << inverseMatrix.Invert();
 179     mInverse = new XRRigidTransform(mParent, inverseMatrix);
 180   }
 181
 182   RefPtr<XRRigidTransform> inverse = mInverse;
 183   return inverse.forget();
 184 }
 185
 186 }  // namespace dom
 187 }  // namespace mozilla