generate_rounded_cone.h

/*
 * Copyright 2019 Adobe. All rights reserved.
 * This file is licensed to you 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 REPRESENTATIONS
 * OF ANY KIND, either express or implied. See the License for the specific language
 * governing permissions and limitations under the License.
 */
#pragma once
 
#ifdef LAGRANGE_ENABLE_LEGACY_FUNCTIONS
    #include <lagrange/primitive/legacy/generate_rounded_cone.h>
#endif
 
#include <lagrange/SurfaceMesh.h>
#include <lagrange/internal/constants.h>
#include <lagrange/primitive/PrimitiveOptions.h>
 
namespace lagrange::primitive {


struct RoundedConeOptions : public PrimitiveOptions
{
    using Scalar = PrimitiveOptions::Scalar;
    using Index = size_t;

    Scalar radius_top = 0;

    Scalar radius_bottom = 1;

    Scalar height = 1;

    Scalar bevel_radius_top = 0;

    Scalar bevel_radius_bottom = 0;

    Index radial_sections = 32;

    Index bevel_segments_top = 1;

    Index bevel_segments_bottom = 1;

    Index side_segments = 1;

    Index top_segments = 1;

    Index bottom_segments = 1;

    Scalar start_sweep_angle = 0;

    Scalar end_sweep_angle = static_cast<Scalar>(2 * lagrange::internal::pi);

    void project_to_valid_range()
    {
        radius_top = std::max(radius_top, Scalar(0));
        radius_bottom = std::max(radius_bottom, Scalar(0));
        height = std::max(height, Scalar(0));
 
        auto [max_bevel_top, max_bevel_bottom] = get_max_cone_bevel();
        bevel_radius_top = std::clamp(bevel_radius_top, Scalar(0), max_bevel_top);
        bevel_radius_bottom = std::clamp(bevel_radius_bottom, Scalar(0), max_bevel_bottom);
 
        radial_sections = std::max(radial_sections, Index(1));
        bevel_segments_top = std::max(bevel_segments_top, Index(1));
        bevel_segments_bottom = std::max(bevel_segments_bottom, Index(1));
 
        side_segments = std::max(side_segments, Index(1));
        top_segments = std::max(top_segments, Index(1));
        bottom_segments = std::max(bottom_segments, Index(1));
    }

    std::pair<Scalar, Scalar> get_max_cone_bevel() const
    {
        // angle between the cone slope and the vertical line (0 for cylinders)
        Scalar psi = std::atan2((radius_top - radius_bottom), height);
        Scalar a1 = (Scalar)(lagrange::internal::pi_2 + psi) * .5f;
        Scalar a2 = (Scalar)(lagrange::internal::pi_2 - psi) * .5f;
 
        Scalar max_bevel_bottom =
            radius_bottom * std::tan(a1); // max bevel against radius and slope
        max_bevel_bottom = std::min(height * .5f, max_bevel_bottom); // also check against height
        max_bevel_bottom = std::max(max_bevel_bottom, 0.f);
 
 
        Scalar max_bevel_top =
            radius_top * std::tan(a2); // max bevel against other radius and inverse slope
        max_bevel_top = std::min(height * .5f, max_bevel_top); // also check against height
        max_bevel_top = std::max(max_bevel_top, 0.f);
 
        return std::pair<Scalar, Scalar>{max_bevel_top, max_bevel_bottom};
    }
};

template <typename Scalar, typename Index>
SurfaceMesh<Scalar, Index> generate_rounded_cone(RoundedConeOptions setting);

 
} // namespace lagrange::primitive