File: benchmark_dataset.cpp

package info (click to toggle)
ismrmrd 1.14.3-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 1,564 kB
  • sloc: cpp: 6,439; ansic: 2,276; xml: 1,025; sh: 187; python: 72; makefile: 42
file content (52 lines) | stat: -rw-r--r-- 1,820 bytes parent folder | download
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
#include <boost/filesystem.hpp>
#include <boost/random.hpp>
#include <chrono>
#include <iostream>
#include <ismrmrd/dataset.h>
#include <ismrmrd/ismrmrd.h>

using namespace ISMRMRD;

static boost::random::mt19937 rng;
static boost::random::uniform_real_distribution<float> dist = boost::random::uniform_real_distribution<float>();

float create_random_float() {
    return dist(rng);
}

int main(int argc, char **argv) {

    boost::filesystem::path temp = boost::filesystem::unique_path();

    Acquisition acq = Acquisition(256, 32, 2);

    std::vector<Acquisition> acqs(10240, acq);
    for (size_t i = 0; i < acqs.size(); i++) {
        std::generate((float *)acqs[i].data_begin(), (float *)acqs[i].data_end(), create_random_float);
        std::generate((float *)acqs[i].traj_begin(), (float *)acqs[i].traj_end(), create_random_float);
    }

    {

        auto start = std::chrono::high_resolution_clock::now();
        Dataset dataset = Dataset(temp.string().c_str(), "/test", true);

        std::for_each(acqs.begin(),acqs.end(),[&dataset](Acquisition& acq){ dataset.appendAcquisition(acq);});

        auto duration = std::chrono::duration<double>(std::chrono::high_resolution_clock::now() - start);
        std::cout << "Write duration: " << duration.count() << "s" << std::endl;
    }

    {
        auto start = std::chrono::high_resolution_clock::now();
        Dataset dataset = Dataset(temp.string().c_str(), "/test", false);
        for (size_t i = 0; i < acqs.size(); i++) {
            Acquisition acq;
            dataset.readAcquisition(uint32_t(i), acq);
        }
        auto duration = std::chrono::duration<double>(std::chrono::high_resolution_clock::now() - start);
        std::cout << "Read duration: " << duration.count() << "s" << std::endl;
    }

    boost::filesystem::remove(temp);
}