File: TruncatedEllipsoidNet.cpp

package info (click to toggle)
bornagain 23.0-4
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 103,936 kB
  • sloc: cpp: 423,131; python: 40,997; javascript: 11,167; awk: 630; sh: 318; ruby: 173; xml: 130; makefile: 51; ansic: 24
file content (45 lines) | stat: -rw-r--r-- 2,006 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
//  ************************************************************************************************
//
//  BornAgain: simulate and fit reflection and scattering
//
//! @file      Sample/Shape/TruncatedEllipsoidNet.cpp
//! @brief     Implements class TruncatedEllipsoidNet.
//!
//! @homepage  http://www.bornagainproject.org
//! @license   GNU General Public License v3 or higher (see COPYING)
//! @copyright Forschungszentrum Jülich GmbH 2018
//! @authors   Scientific Computing Group at MLZ (see CITATION, AUTHORS)
//
//  ************************************************************************************************

#include "Sample/Shape/TruncatedEllipsoidNet.h"
#include "Base/Util/Assert.h"
#include <algorithm>
#include <cmath>

TruncatedEllipsoidNet::TruncatedEllipsoidNet(double r_x, double r_y, double r_z, double height,
                                             double dh)
{
    ASSERT(std::isfinite(r_z));
    static const int n_heights =
        std::max(2, static_cast<int>(std::round(
                        static_cast<double>(IShape3D::N_Circle) * height / 2.0 / r_z + 0.5)));
    double h_step = (height - dh) / (n_heights - 1);
    ASSERT(std::isfinite(h_step));
    m_vertices.resize(n_heights * IShape3D::N_Circle);
    auto it = m_vertices.begin();
    for (int i = 0; i < n_heights; ++i) {
        double z = i * h_step;
        // for r_z=height, dh=0, i=n_heights-1, the argument of the following sqrt should be 0,
        // but can be slightly smaller because of floating-point errors, therefore needs
        // to be overwritten by 0.
        double radius_factor = std::sqrt(std::max(0., 1.0 - std::pow((z + r_z - height) / r_z, 2)));
        ASSERT(std::isfinite(z));
        ASSERT(std::isfinite(radius_factor));
        auto ellipse = EllipseVerticesZ(radius_factor * r_x, radius_factor * r_y, i * h_step);
        std::move(ellipse.begin(), ellipse.end(), it);
        it = it + ellipse.size();
    }
}

TruncatedEllipsoidNet::~TruncatedEllipsoidNet() = default;