third_party/blink/renderer/modules/xr/xr_object_space.h

// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_XR_XR_OBJECT_SPACE_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_XR_XR_OBJECT_SPACE_H_

#include "third_party/blink/renderer/modules/xr/xr_space.h"
#include "third_party/blink/renderer/platform/transforms/transformation_matrix.h"

namespace blink {

class XRSession;

// Helper class that returns an XRSpace that tracks the position of object of
// type T (for example XRPlane, XRAnchor). The type T has to have a poseMatrix()
// method.
template <typename T>
class XRObjectSpace : public XRSpace {
 public:
  explicit XRObjectSpace(XRSession* session, const T* object)
      : XRSpace(session), object_(object) {}

  std::unique_ptr<TransformationMatrix> MojoFromSpace() override {
    auto object_from_mojo = object_->poseMatrix();

    if (!object_from_mojo.IsInvertible()) {
      return nullptr;
    }

    return std::make_unique<TransformationMatrix>(object_from_mojo.Inverse());
  }

  base::Optional<XRNativeOriginInformation> NativeOrigin() const override {
    return XRNativeOriginInformation::Create(object_);
  }

  void Trace(blink::Visitor* visitor) override {
    visitor->Trace(object_);
    XRSpace::Trace(visitor);
  }

 private:
  Member<const T> object_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_XR_XR_OBJECT_SPACE_H_