diff options
Diffstat (limited to '')
| -rw-r--r-- | dom/vr/XRRigidTransform.cpp | 171 |
1 files changed, 171 insertions, 0 deletions
diff --git a/dom/vr/XRRigidTransform.cpp b/dom/vr/XRRigidTransform.cpp new file mode 100644 index 0000000000..d29937c56a --- /dev/null +++ b/dom/vr/XRRigidTransform.cpp @@ -0,0 +1,171 @@ +/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ +/* vim: set ts=8 sts=2 et sw=2 tw=80: */ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this file, + * You can obtain one at http://mozilla.org/MPL/2.0/. */ + +#include "mozilla/dom/XRRigidTransform.h" +#include "mozilla/dom/DOMPoint.h" +#include "mozilla/dom/Pose.h" +#include "mozilla/dom/DOMPointBinding.h" +#include "mozilla/HoldDropJSObjects.h" +#include "nsWrapperCache.h" + +namespace mozilla::dom { + +NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE_WITH_JS_MEMBERS(XRRigidTransform, + (mParent, mPosition, + mOrientation, mInverse), + (mMatrixArray)) + +XRRigidTransform::XRRigidTransform(nsISupports* aParent, + const gfx::PointDouble3D& aPosition, + const gfx::QuaternionDouble& aOrientation) + : mParent(aParent), + mMatrixArray(nullptr), + mPosition(nullptr), + mOrientation(nullptr), + mInverse(nullptr), + mRawPosition(aPosition), + mRawOrientation(aOrientation), + mNeedsUpdate(true) { + mozilla::HoldJSObjects(this); + mRawTransformMatrix.SetRotationFromQuaternion(mRawOrientation); + mRawTransformMatrix.PostTranslate(mRawPosition); +} + +XRRigidTransform::XRRigidTransform(nsISupports* aParent, + const gfx::Matrix4x4Double& aTransform) + : mParent(aParent), + mMatrixArray(nullptr), + mPosition(nullptr), + mOrientation(nullptr), + mInverse(nullptr), + mNeedsUpdate(true) { + mozilla::HoldJSObjects(this); + gfx::PointDouble3D scale; + mRawTransformMatrix = aTransform; + mRawTransformMatrix.Decompose(mRawPosition, mRawOrientation, scale); +} + +XRRigidTransform::~XRRigidTransform() { mozilla::DropJSObjects(this); } + +/* static */ already_AddRefed<XRRigidTransform> XRRigidTransform::Constructor( + const GlobalObject& aGlobal, const DOMPointInit& aOrigin, + const DOMPointInit& aDirection, ErrorResult& aRv) { + gfx::PointDouble3D position(aOrigin.mX, aOrigin.mY, aOrigin.mZ); + gfx::QuaternionDouble orientation(aDirection.mX, aDirection.mY, aDirection.mZ, + aDirection.mW); + orientation.Normalize(); + RefPtr<XRRigidTransform> obj = + new XRRigidTransform(aGlobal.GetAsSupports(), position, orientation); + return obj.forget(); +} + +JSObject* XRRigidTransform::WrapObject(JSContext* aCx, + JS::Handle<JSObject*> aGivenProto) { + return XRRigidTransform_Binding::Wrap(aCx, this, aGivenProto); +} + +DOMPoint* XRRigidTransform::Position() { + if (!mPosition) { + mPosition = new DOMPoint(mParent, mRawPosition.x, mRawPosition.y, + mRawPosition.z, 1.0f); + } + return mPosition; +} + +DOMPoint* XRRigidTransform::Orientation() { + if (!mOrientation) { + mOrientation = new DOMPoint(mParent, mRawOrientation.x, mRawOrientation.y, + mRawOrientation.z, mRawOrientation.w); + } + return mOrientation; +} + +XRRigidTransform& XRRigidTransform::operator=(const XRRigidTransform& aOther) { + Update(aOther.mRawPosition, aOther.mRawOrientation); + return *this; +} + +gfx::QuaternionDouble XRRigidTransform::RawOrientation() const { + return mRawOrientation; +} +gfx::PointDouble3D XRRigidTransform::RawPosition() const { + return mRawPosition; +} + +void XRRigidTransform::Update(const gfx::PointDouble3D& aPosition, + const gfx::QuaternionDouble& aOrientation) { + mNeedsUpdate = true; + mRawPosition = aPosition; + mRawOrientation = aOrientation; + mRawTransformMatrix.SetRotationFromQuaternion(mRawOrientation); + mRawTransformMatrix.PostTranslate(mRawPosition); + UpdateInternal(); +} + +void XRRigidTransform::Update(const gfx::Matrix4x4Double& aTransform) { + mNeedsUpdate = true; + mRawTransformMatrix = aTransform; + gfx::PointDouble3D scale; + mRawTransformMatrix.Decompose(mRawPosition, mRawOrientation, scale); + UpdateInternal(); +} + +void XRRigidTransform::UpdateInternal() { + if (mPosition) { + mPosition->SetX(mRawPosition.x); + mPosition->SetY(mRawPosition.y); + mPosition->SetZ(mRawPosition.z); + } + if (mOrientation) { + mOrientation->SetX(mRawOrientation.x); + mOrientation->SetY(mRawOrientation.y); + mOrientation->SetZ(mRawOrientation.z); + mOrientation->SetW(mRawOrientation.w); + } + if (mInverse) { + gfx::Matrix4x4Double inverseMatrix = mRawTransformMatrix; + Unused << inverseMatrix.Invert(); + mInverse->Update(inverseMatrix); + } +} + +void XRRigidTransform::GetMatrix(JSContext* aCx, + JS::MutableHandle<JSObject*> aRetval, + ErrorResult& aRv) { + if (!mMatrixArray || mNeedsUpdate) { + mNeedsUpdate = false; + + const uint32_t size = 16; + float components[size] = {}; + // In order to avoid some platforms which only copy + // the first or last two bytes of a Float64 to a Float32. + for (uint32_t i = 0; i < size; ++i) { + components[i] = mRawTransformMatrix.components[i]; + } + Pose::SetFloat32Array(aCx, this, aRetval, mMatrixArray, components, 16, + aRv); + if (!mMatrixArray) { + return; + } + } + if (mMatrixArray) { + JS::ExposeObjectToActiveJS(mMatrixArray); + } + aRetval.set(mMatrixArray); +} + +already_AddRefed<XRRigidTransform> XRRigidTransform::Inverse() { + if (!mInverse) { + gfx::Matrix4x4Double inverseMatrix = mRawTransformMatrix; + Unused << inverseMatrix.Invert(); + mInverse = new XRRigidTransform(mParent, inverseMatrix); + } + + RefPtr<XRRigidTransform> inverse = mInverse; + return inverse.forget(); +} + +} // namespace mozilla::dom |