/*
 * Copyright 2022 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef ANDROID_HWC_VSYNCTHREAD_H
#define ANDROID_HWC_VSYNCTHREAD_H

#include <aidl/android/hardware/graphics/composer3/VsyncPeriodChangeConstraints.h>
#include <aidl/android/hardware/graphics/composer3/VsyncPeriodChangeTimeline.h>
#include <android/hardware/graphics/common/1.0/types.h>

#include <chrono>
#include <mutex>
#include <optional>
#include <thread>

#include "Common.h"

namespace aidl::android::hardware::graphics::composer3::impl {

// Generates Vsync signals in software.
class VsyncThread {
 public:
  VsyncThread(int64_t id);
  virtual ~VsyncThread();

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

  VsyncThread(VsyncThread&&) = delete;
  VsyncThread& operator=(VsyncThread&&) = delete;

  HWC3::Error start(int32_t periodNanos);

  HWC3::Error setCallbacks(const std::shared_ptr<IComposerCallback>& callback);

  HWC3::Error setVsyncEnabled(bool enabled);

  HWC3::Error scheduleVsyncUpdate(
      int32_t newVsyncPeriod,
      const VsyncPeriodChangeConstraints& newVsyncPeriodChangeConstraints,
      VsyncPeriodChangeTimeline* timeline);

 private:
  HWC3::Error stop();

  void threadLoop();

  std::chrono::nanoseconds updateVsyncPeriodLocked(
      std::chrono::time_point<std::chrono::steady_clock> now);

  const int64_t mDisplayId;

  std::thread mThread;

  std::mutex mStateMutex;

  std::atomic<bool> mShuttingDown{false};

  std::shared_ptr<IComposerCallback> mCallbacks;

  bool mVsyncEnabled = false;
  std::chrono::nanoseconds mVsyncPeriod;
  std::chrono::time_point<std::chrono::steady_clock> mPreviousVsync;

  struct PendingUpdate {
    std::chrono::nanoseconds period;
    std::chrono::time_point<std::chrono::steady_clock> updateAfter;
  };
  std::optional<PendingUpdate> mPendingUpdate;
};

}  // namespace aidl::android::hardware::graphics::composer3::impl

#endif