vortex-processing/gauss_beam.cpp at master · Laufinsconsca/vortex-processing · GitHub

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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
#include "gauss_beam.h"

/**
 * Default constructor.
 **************************************************************************************************/

gauss_beam::gauss_beam() : sigma(0), shift(0), shift_angle(0) {}

gauss_beam::gauss_beam(double sigma, double shift, double shift_angle)
    : sigma(sigma), shift(shift), shift_angle(shift_angle) {}

std::vector<std::vector<double>>& gauss_beam::gauss_beam_to_vector(const gauss_beam& gauss_beam, std::vector<std::vector<double>>& ref_beam, const QSize& size) {
    ref_beam.clear();
    double hx, hy, x, y;
    hx = 2. / size.width();
    hy = 2. / size.height();
    ref_beam.reserve(size.height());
    for (int i = 0; i < size.height(); i++) {
        y = i * hy - 1;
        ref_beam.push_back(std::vector<double>());
        ref_beam.at(i).reserve(size.width());
        for (int j = 0; j < size.width(); j++) {
            x = j * hx - 1;
            ref_beam.at(i).push_back((exp(-(pow(x - gauss_beam.shift, 2) + y * y) / (2 * gauss_beam.sigma * gauss_beam.sigma))));
        }
    }
    return transformation<std::vector<double>>::rotate(ref_beam, gauss_beam.shift_angle*M_PI/180);
}

std::vector<std::vector<double>>& gauss_beam::gauss_beam_to_vector(const gauss_beam& gauss_beam, std::vector<std::vector<double>>& ref_beam, const QSize& size, const hole &hole) {
    ref_beam = gauss_beam_to_vector(gauss_beam, ref_beam, size);
    hole::create_hole(ref_beam, hole, 1);
    return ref_beam;
}

bool gauss_beam::gauss_param_preprocessing(QLineEdit* sigma_line, QLineEdit* shift_line, QLineEdit* shift_angle_line, gauss_beam &gauss_beam) {
    if (sigma_line->text().isEmpty() || shift_line->text().isEmpty() || shift_angle_line->text().isEmpty()) {
        return false;
    }
    bool sigma_line_to_string_ok, shift_line_to_string_ok, shift_angle_line_to_string_ok;
    double sigma = sigma_line->text().toDouble(&sigma_line_to_string_ok);
    double shift = shift_line->text().toDouble(&shift_line_to_string_ok);
    double shift_angle = shift_angle_line->text().toDouble(&shift_angle_line_to_string_ok);
    if (shift_line_to_string_ok && sigma_line_to_string_ok && shift_angle_line_to_string_ok) {
        class gauss_beam result_gauss(sigma, shift, shift_angle);
        gauss_beam = result_gauss;
        return true;
    } else {
        return false;
    }
}

/**
 * Assignment operator.
 *
 * @param	obj	The object to copy.
 *
 * @returns	A shallow copy of this object.
 **************************************************************************************************/

class gauss_beam& gauss_beam::operator=(const gauss_beam &obj) {
    if (this == &obj) {
        return *this;
    }
    sigma = obj.sigma;
    shift = obj.shift;
    shift_angle = obj.shift_angle;
    return *this;
}