// ----------------------------------------------------------------------------
// -                        Open3D: www.open3d.org                            -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2023 www.open3d.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------

#include "tests/test_utility/Rand.h"

#include <iostream>

#include "tests/test_utility/Raw.h"

namespace open3d {
namespace tests {

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector3d.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(Eigen::Vector3d &v,
          const Eigen::Vector3d &vmin,
          const Eigen::Vector3d &vmax,
          const int &seed) {
    Raw raw(seed);

    Eigen::Vector3d factor;
    factor(0, 0) = vmax(0, 0) - vmin(0, 0);
    factor(1, 0) = vmax(1, 0) - vmin(1, 0);
    factor(2, 0) = vmax(2, 0) - vmin(2, 0);

    v(0, 0) = vmin(0, 0) + raw.Next<double>() * factor(0, 0);
    v(1, 0) = vmin(1, 0) + raw.Next<double>() * factor(1, 0);
    v(2, 0) = vmin(2, 0) + raw.Next<double>() * factor(2, 0);
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector3d.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(Eigen::Vector3d &v,
          const double &vmin,
          const double &vmax,
          const int &seed) {
    Raw raw(seed);

    double factor;
    factor = vmax - vmin;

    v(0, 0) = vmin + raw.Next<double>() * factor;
    v(1, 0) = vmin + raw.Next<double>() * factor;
    v(2, 0) = vmin + raw.Next<double>() * factor;
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector2i vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<Eigen::Vector2i> &v,
          const Eigen::Vector2i &vmin,
          const Eigen::Vector2i &vmax,
          const int &seed) {
    Raw raw(seed);

    Eigen::Vector2d factor;
    factor(0, 0) = (double)(vmax(0, 0) - vmin(0, 0)) / Raw::VMAX;
    factor(1, 0) = (double)(vmax(1, 0) - vmin(1, 0)) / Raw::VMAX;

    for (size_t i = 0; i < v.size(); i++) {
        v[i](0, 0) = vmin(0, 0) + (int)(raw.Next<int>() * factor(0, 0));
        v[i](1, 0) = vmin(1, 0) + (int)(raw.Next<int>() * factor(1, 0));
    }
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector3i vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<Eigen::Vector3i> &v,
          const Eigen::Vector3i &vmin,
          const Eigen::Vector3i &vmax,
          const int &seed) {
    Raw raw(seed);

    Eigen::Vector3d factor;
    factor(0, 0) = (double)(vmax(0, 0) - vmin(0, 0)) / Raw::VMAX;
    factor(1, 0) = (double)(vmax(1, 0) - vmin(1, 0)) / Raw::VMAX;
    factor(2, 0) = (double)(vmax(2, 0) - vmin(2, 0)) / Raw::VMAX;

    for (size_t i = 0; i < v.size(); i++) {
        v[i](0, 0) = vmin(0, 0) + (int)(raw.Next<int>() * factor(0, 0));
        v[i](1, 0) = vmin(1, 0) + (int)(raw.Next<int>() * factor(1, 0));
        v[i](2, 0) = vmin(2, 0) + (int)(raw.Next<int>() * factor(2, 0));
    }
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector2d vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<Eigen::Vector2d, open3d::utility::Vector2d_allocator> &v,
          const Eigen::Vector2d &vmin,
          const Eigen::Vector2d &vmax,
          const int &seed) {
    Raw raw(seed);

    Eigen::Vector2d factor;
    factor(0, 0) = vmax(0, 0) - vmin(0, 0);
    factor(1, 0) = vmax(1, 0) - vmin(1, 0);

    for (size_t i = 0; i < v.size(); i++) {
        v[i](0, 0) = vmin(0, 0) + raw.Next<double>() * factor(0, 0);
        v[i](1, 0) = vmin(1, 0) + raw.Next<double>() * factor(1, 0);
    }
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector3d vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<Eigen::Vector3d> &v,
          const Eigen::Vector3d &vmin,
          const Eigen::Vector3d &vmax,
          const int &seed) {
    Raw raw(seed);

    Eigen::Vector3d factor;
    factor(0, 0) = vmax(0, 0) - vmin(0, 0);
    factor(1, 0) = vmax(1, 0) - vmin(1, 0);
    factor(2, 0) = vmax(2, 0) - vmin(2, 0);

    for (size_t i = 0; i < v.size(); i++) {
        v[i](0, 0) = vmin(0, 0) + raw.Next<double>() * factor(0, 0);
        v[i](1, 0) = vmin(1, 0) + raw.Next<double>() * factor(1, 0);
        v[i](2, 0) = vmin(2, 0) + raw.Next<double>() * factor(2, 0);
    }
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector4i vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<Eigen::Vector4i, open3d::utility::Vector4i_allocator> &v,
          const int &vmin,
          const int &vmax,
          const int &seed) {
    Raw raw(seed);

    double factor = (double)(vmax - vmin) / Raw::VMAX;

    for (size_t i = 0; i < v.size(); i++) {
        v[i](0, 0) = vmin + (int)(raw.Next<int>() * factor);
        v[i](1, 0) = vmin + (int)(raw.Next<int>() * factor);
        v[i](2, 0) = vmin + (int)(raw.Next<int>() * factor);
        v[i](3, 0) = vmin + (int)(raw.Next<int>() * factor);
    }
}

// ----------------------------------------------------------------------------
// Initialize an Eigen::Vector4i vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<Eigen::Vector4i, open3d::utility::Vector4i_allocator> &v,
          const Eigen::Vector4i &vmin,
          const Eigen::Vector4i &vmax,
          const int &seed) {
    Raw raw(seed);

    Eigen::Vector4d factor;
    factor(0, 0) = (double)(vmax(0, 0) - vmin(0, 0)) / Raw::VMAX;
    factor(1, 0) = (double)(vmax(1, 0) - vmin(1, 0)) / Raw::VMAX;
    factor(2, 0) = (double)(vmax(2, 0) - vmin(2, 0)) / Raw::VMAX;
    factor(3, 0) = (double)(vmax(3, 0) - vmin(3, 0)) / Raw::VMAX;

    for (size_t i = 0; i < v.size(); i++) {
        v[i](0, 0) = vmin(0, 0) + (int)(raw.Next<int>() * factor(0, 0));
        v[i](1, 0) = vmin(1, 0) + (int)(raw.Next<int>() * factor(1, 0));
        v[i](2, 0) = vmin(2, 0) + (int)(raw.Next<int>() * factor(2, 0));
        v[i](3, 0) = vmin(3, 0) + (int)(raw.Next<int>() * factor(2, 0));
    }
}

// ----------------------------------------------------------------------------
// Initialize a uint8_t vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<uint8_t> &v,
          const uint8_t &vmin,
          const uint8_t &vmax,
          const int &seed) {
    Raw raw(seed);

    double factor = (double)(vmax - vmin) / Raw::VMAX;

    for (size_t i = 0; i < v.size(); i++)
        v[i] = vmin + (uint8_t)(raw.Next<uint8_t>() * factor);
}

// ----------------------------------------------------------------------------
// Initialize a int vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(int *const v,
          const size_t &size,
          const int &vmin,
          const int &vmax,
          const int &seed) {
    Raw raw(seed);

    double factor = (double)(vmax - vmin) / Raw::VMAX;

    for (size_t i = 0; i < size; i++)
        v[i] = vmin + (int)(raw.Next<int>() * factor);
}

// ----------------------------------------------------------------------------
// Initialize an int vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<int> &v,
          const int &vmin,
          const int &vmax,
          const int &seed) {
    Rand(&v[0], v.size(), vmin, vmax, seed);
}

// ----------------------------------------------------------------------------
// Initialize a size_t vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<size_t> &v,
          const size_t &vmin,
          const size_t &vmax,
          const int &seed) {
    Raw raw(seed);

    double factor = (double)(vmax - vmin) / Raw::VMAX;

    for (size_t i = 0; i < v.size(); i++)
        v[i] = vmin + (size_t)(raw.Next<size_t>() * factor);
}

// ----------------------------------------------------------------------------
// Initialize an array of float.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(float *const v,
          const size_t &size,
          const float &vmin,
          const float &vmax,
          const int &seed) {
    Raw raw(seed);

    float factor = vmax - vmin;

    for (size_t i = 0; i < size; i++) v[i] = vmin + raw.Next<float>() * factor;
}

// ----------------------------------------------------------------------------
// Initialize a float vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<float> &v,
          const float &vmin,
          const float &vmax,
          const int &seed) {
    Rand(&v[0], v.size(), vmin, vmax, seed);
}

// ----------------------------------------------------------------------------
// Initialize an array of double.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(double *const v,
          const size_t &size,
          const double &vmin,
          const double &vmax,
          const int &seed) {
    Raw raw(seed);

    double factor = vmax - vmin;

    for (size_t i = 0; i < size; i++) v[i] = vmin + raw.Next<double>() * factor;
}

// ----------------------------------------------------------------------------
// Initialize a double vector.
// Output range: [vmin:vmax].
// ----------------------------------------------------------------------------
void Rand(std::vector<double> &v,
          const double &vmin,
          const double &vmax,
          const int &seed) {
    Rand(&v[0], v.size(), vmin, vmax, seed);
}

}  // namespace tests
}  // namespace open3d