/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

#include <filesystem>
#include <unordered_map>
#include <vector>

#include <yoga/Yoga.h>

namespace facebook::yoga {

struct SerializedMeasureFunc {
  float inputWidth{0.0f};
  YGMeasureMode widthMode{YGMeasureModeUndefined};
  float inputHeight{0.0};
  YGMeasureMode heightMode{YGMeasureModeUndefined};
  float outputWidth{0.0f};
  float outputHeight{0.0f};
  std::chrono::steady_clock::duration::rep durationNs{};
};

using SerializedMeasureFuncMap =
    std::unordered_map<YGNodeConstRef, std::vector<SerializedMeasureFunc>>;

void YGNodeCalculateLayoutWithCapture(
    YGNodeRef node,
    float availableWidth,
    float availableHeight,
    YGDirection ownerDirection,
    const std::filesystem::path& path);

void captureMeasureFunc(
    YGNodeConstRef node,
    float width,
    YGMeasureMode widthMode,
    float height,
    YGMeasureMode heightMode,
    YGSize output,
    std::chrono::steady_clock::duration durationNs);

} // namespace facebook::yoga