File: constellation_s2.cpp

package info (click to toggle)
satdump 1.2.2%2Bgb79af48-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 81,648 kB
  • sloc: cpp: 276,768; ansic: 164,598; lisp: 1,219; sh: 283; xml: 106; makefile: 7
file content (92 lines) | stat: -rw-r--r-- 4,483 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#include "constellation_s2.h"
#include <cstring>
#include "core/style.h"
#include "common/dsp/block.h"

namespace widgets
{
    ConstellationViewerDVBS2::ConstellationViewerDVBS2(float hscale, float vscale, int constellation_size) : d_hscale(hscale), d_vscale(vscale), d_constellation_size(constellation_size)
    {
    }

    ConstellationViewerDVBS2::~ConstellationViewerDVBS2()
    {
    }

    void ConstellationViewerDVBS2::pushComplexPL(complex_t *buffer, int size)
    {
        int to_copy = std::min<int>(CONST_SIZE, size);
        int to_move = size >= CONST_SIZE ? 0 : CONST_SIZE - size;

        if (to_move > 0)
            std::memmove(&sample_buffer_complex_float_plheader[size], sample_buffer_complex_float_plheader, to_move * sizeof(complex_t));

        std::memcpy(sample_buffer_complex_float_plheader, buffer, to_copy * sizeof(complex_t));
    }

    void ConstellationViewerDVBS2::pushComplexSlots(complex_t *buffer, int size)
    {
        int to_copy = std::min<int>(CONST_SIZE, size);
        int to_move = size >= CONST_SIZE ? 0 : CONST_SIZE - size;

        if (to_move > 0)
            std::memmove(&sample_buffer_complex_float_slots[size], sample_buffer_complex_float_slots, to_move * sizeof(complex_t));

        std::memcpy(sample_buffer_complex_float_slots, buffer, to_copy * sizeof(complex_t));
    }

    void ConstellationViewerDVBS2::pushComplexPilots(complex_t *buffer, int size)
    {
        has_pilots = true;

        int to_copy = std::min<int>(CONST_SIZE, size);
        int to_move = size >= CONST_SIZE ? 0 : CONST_SIZE - size;

        if (to_move > 0)
            std::memmove(&sample_buffer_complex_float_pilots[size], sample_buffer_complex_float_pilots, to_move * sizeof(complex_t));

        std::memcpy(sample_buffer_complex_float_pilots, buffer, to_copy * sizeof(complex_t));
    }

    void ConstellationViewerDVBS2::draw()
    {
        ImDrawList *draw_list = ImGui::GetWindowDrawList();
        ImVec2 rect_min = ImGui::GetCursorScreenPos();
        ImVec2 rect_max = { rect_min.x + d_constellation_size * ui_scale, rect_min.y + d_constellation_size * ui_scale };
        draw_list->AddRectFilled(rect_min, rect_max, style::theme.widget_bg);
        draw_list->PushClipRect(rect_min, rect_max);

        // Draw PLHeader
        for (int i = 0; i < CONST_SIZE / 4; i++)
        {
            draw_list->AddCircleFilled(ImVec2(ImGui::GetCursorScreenPos().x + ((d_constellation_size / 2) * ui_scale + sample_buffer_complex_float_plheader[i].real * (d_constellation_size / 2) * d_hscale * ui_scale),
                                              ImGui::GetCursorScreenPos().y + ((d_constellation_size / 2) * ui_scale + sample_buffer_complex_float_plheader[i].imag * (d_constellation_size / 2) * d_vscale * ui_scale)),
                                       2 * ui_scale * (d_constellation_size / 200.0f),
                                       style::theme.red);
        }

        // Draw Slots
        for (int i = 0; i < CONST_SIZE; i++)
        {
            draw_list->AddCircleFilled(ImVec2(ImGui::GetCursorScreenPos().x + ((d_constellation_size / 2) * ui_scale + sample_buffer_complex_float_slots[i].real * (d_constellation_size / 2) * d_hscale * ui_scale),
                                              ImGui::GetCursorScreenPos().y + ((d_constellation_size / 2) * ui_scale + sample_buffer_complex_float_slots[i].imag * (d_constellation_size / 2) * d_vscale * ui_scale)),
                                       2 * ui_scale * (d_constellation_size / 200.0f),
                                       style::theme.constellation);
        }

        if (has_pilots)
        {
            // Draw Pilots
            for (int i = 0; i < CONST_SIZE; i++)
            {
                draw_list->AddCircleFilled(ImVec2(ImGui::GetCursorScreenPos().x + ((d_constellation_size / 2) * ui_scale + sample_buffer_complex_float_pilots[i].real * (d_constellation_size / 2) * d_hscale * ui_scale),
                                                  ImGui::GetCursorScreenPos().y + ((d_constellation_size / 2) * ui_scale + sample_buffer_complex_float_pilots[i].imag * (d_constellation_size / 2) * d_vscale * ui_scale)),
                                           2 * ui_scale * (d_constellation_size / 200.0f),
                                           style::theme.red);
            }
        }

        draw_list->PopClipRect();
        ImGui::Dummy(ImVec2(d_constellation_size * ui_scale + 3, d_constellation_size * ui_scale + 3));
    }
}