PulseConvolution/main.cpp at main · IlyaKodua/PulseConvolution · 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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
#include <iostream>
#include <utility>
#include <memory>
#include <cassert>
#include <cmath>
#include <type_traits>
#include <fstream>

#include "conv.hpp"
#include "conv_opencl.hpp"

using namespace std;


template<typename T_float>
std::pair<std::vector<T_float>, std::vector<uint>> signal_prepare(const std::vector<T_float> &sig)
{
    std::vector<T_float> sig_out;
    std::vector<uint> id;
    sig_out.reserve(sig.size()/2);
    id.reserve(sig.size()/2);

    for(size_t i = 0; i < sig.size(); i++)
    {
        if(std::abs(sig[i]) != 0)
        {
            sig_out.push_back(sig[i]);
            id.push_back(i);
        }
    }
    sig_out.shrink_to_fit();
    id.shrink_to_fit();
    return pair<std::vector<T_float>, std::vector<uint>>(std::move(sig_out),
                                                     std::move(id));
}


std::vector<std::complex<float>> sig_gen()
{
    size_t n = 1e3;
    std::vector<std::complex<float>> sig(n);

    int cnt = 0;
    for(size_t i = 0; i < sig.size();i++)
    {
        sig[i] = (cnt / 10) % 10;
        cnt++;
    }
    return sig;
}

void check_signals()
{
        auto sig1 = sig_gen();
        auto sig2 = sig1;
        std::vector<std::complex<float>> conv1(201);
        std::vector<std::complex<float>> conv2(201);

        conv_classic(sig1, sig2, conv1);

        auto pair1 = signal_prepare(sig1);
        auto pair2 = signal_prepare(sig2);

        conv_pulse(pair1.first, pair1.second,
                   pair2.first, pair2.second,
                   conv2);

        assert(conv1 == conv2);
}

void save_bin_conv(std::vector<std::complex<float>> &conv)
{
    std::vector<float> convf;
    convf.resize(conv.size());
    for(size_t i = 0; i < conv.size(); i++)
    {
        convf.at(i) = std::abs(conv.at(i));
//        std::cout<<convf.at(i)<<std::endl;
    }

    std::ofstream out;
    out.open("conv.bin", std::ios::out | std::ios::binary);
    out.write(reinterpret_cast<const char*>(convf.data()), sizeof(float) * convf.size());
    out.close();
}


int main()
{
    auto sig1 = sig_gen();
    auto sig2 = sig1;


    params param;

    param.sig_n = sig1.size();
    param.n_dist = 2*sig1.size() + 1;
    param.n_dopp = 101;
    param.dist_0_idx = sig1.size();
    param.dopp_0_idx= (param.n_dopp - 1)/2;

    std::vector<std::complex<float>> conv(param.n_dist * param.n_dopp);
    auto pair1 = signal_prepare(sig1);
    auto pair2 = signal_prepare(sig2);

    param.size_id1 = pair1.second.size();
    param.size_id2 = pair2.second.size();
    Timer t;
    conv_calculator calc;

    calc.LazyAllocate(pair1.first, pair1.second,
                      pair2.first, pair2.second, param.n_dist * param.n_dopp);

    calc.start_conv(param, -50);

    int ret = calc.get_conv(conv);
    assert(ret == 0);
    std::cout<<"Time gpu: "<<t.elapsed()<<std::endl;

    t.reset();
    std::vector<std::complex<float>> conv2(2*sig1.size()+1);
    conv_pulse(pair1.first, pair1.second,
               pair2.first, pair2.second, conv2);
    std::cout<<"time cpu: "<<t.elapsed()<<std::endl;
//    save_bin_conv(conv);
    return 0;
}