device/vr/openxr/openxr_input_helper.h

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

#ifndef DEVICE_VR_OPENXR_OPENXR_INPUT_HELPER_H_
#define DEVICE_VR_OPENXR_OPENXR_INPUT_HELPER_H_

#include <array>
#include <memory>
#include <optional>
#include <vector>

#include "device/vr/openxr/openxr_controller.h"
#include "device/vr/openxr/openxr_interaction_profiles.h"
#include "device/vr/public/mojom/vr_service.mojom-forward.h"
#include "third_party/openxr/src/include/openxr/openxr.h"

namespace device {

class OpenXrExtensionHelper;
struct XrLocation;

class OpenXRInputHelper {
 public:
  static XrResult CreateOpenXRInputHelper(
      XrInstance instance,
      XrSystemId system,
      const OpenXrExtensionHelper& extension_helper,
      XrSession session,
      XrSpace local_space,
      bool hand_input_enabled,
      std::unique_ptr<OpenXRInputHelper>* helper);

  OpenXRInputHelper(XrSession session,
                    XrSpace local_space,
                    bool hand_input_enabled);

  OpenXRInputHelper(const OpenXRInputHelper&) = delete;
  OpenXRInputHelper& operator=(const OpenXRInputHelper&) = delete;

  ~OpenXRInputHelper();

  bool IsHandTrackingEnabled() const;

  std::vector<mojom::XRInputSourceStatePtr> GetInputState(
      XrTime predicted_display_time);

  XrResult OnInteractionProfileChanged();

  bool ReceivedExitGesture() { return received_exit_gesture_; }

  // Called when the device is going to hide input sources from the page.
  void OnHideInputSources();

  std::optional<XrLocation> GetXrLocationFromHandJoint(
      XrSpace mojo_space,
      const mojom::XRHandJointSpaceInfo& hand_joint_space_info,
      const gfx::Transform& joint_from_object) const;

  std::optional<XrLocation> GetXrLocationFromInputSource(
      const mojom::XRInputSourceSpaceInfo& input_source_space_info,
      const gfx::Transform& input_space_from_object) const;

 private:
  XrResult Initialize(XrInstance instance,
                      XrSystemId system,
                      const OpenXrExtensionHelper& extension_helper);

  XrResult SyncActions(XrTime predicted_display_time);

  XrSpace GetMojomSpace() const {
    return local_space_;  // Mojom space is currently defined as local space
  }

  void OnExitGesture() { received_exit_gesture_ = true; }

  void ResetControllerButtonState();

  XrSession session_;
  XrSpace local_space_;

  struct OpenXrControllerState {
    OpenXrController controller;
    bool primary_button_pressed;
    bool squeeze_button_pressed;
  };
  std::array<OpenXrControllerState,
             static_cast<size_t>(OpenXrHandednessType::kCount)>
      controller_states_;

  std::unique_ptr<OpenXRPathHelper> path_helper_;
  bool received_exit_gesture_ = false;
  bool hand_input_enabled_ = false;
};

}  // namespace device

#endif  // DEVICE_VR_OPENXR_OPENXR_INPUT_HELPER_H_