Cheetah - SKA - PSS - Prototype Time Domain Search Pipeline
TimeFrequencyContainerTest.cpp
1 /*
2  * The MIT License (MIT)
3  *
4  * Copyright (c) 2016 The SKA organisation
5  *
6  * Permission is hereby granted, free of charge, to any person obtaining a copy
7  * of this software and associated documentation files (the "Software"), to deal
8  * in the Software without restriction, including without limitation the rights
9  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10  * copies of the Software, and to permit persons to whom the Software is
11  * furnished to do so, subject to the following conditions:
12  *
13  * The above copyright notice and this permission notice shall be included in all
14  * copies or substantial portions of the Software.
15  *
16  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22  * SOFTWARE.
23  */
24 #include "cheetah/data/test/TimeFrequencyContainerTest.h"
25 #include "cheetah/data/TimeFrequencyContainer.h"
26 
27 
28 namespace ska {
29 namespace cheetah {
30 namespace data {
31 namespace test {
32 
33 
34 TimeFrequencyContainerTest::TimeFrequencyContainerTest()
35  : ::testing::Test()
36 {
37 }
38 
39 TimeFrequencyContainerTest::~TimeFrequencyContainerTest()
40 {
41 }
42 
43 void TimeFrequencyContainerTest::SetUp()
44 {
45 }
46 
47 void TimeFrequencyContainerTest::TearDown()
48 {
49 }
50 
51 TEST_F(TimeFrequencyContainerTest, test_unequal_channels)
52 {
53  typedef TimeFrequency<Cpu, float> TimeFrequencyType;
54  data::DimensionSize<data::Frequency> number_of_channels(1000);
55  TimeFrequencyType::TimeType interval(10 * data::milliseconds);
56  data::DimensionSize<data::Time> n_samples(100);
57  std::shared_ptr<TimeFrequencyType> time_freq_1=std::make_shared<TimeFrequencyType>(n_samples, number_of_channels);
58  time_freq_1->sample_interval(interval);
59  time_freq_1->start_time(TimeFrequencyType::TimePointType(std::chrono::seconds(10)));
60  std::shared_ptr<TimeFrequencyType> time_freq_2=std::make_shared<TimeFrequencyType>(n_samples, number_of_channels - 1);
61  time_freq_2->sample_interval(interval);
62  time_freq_2->start_time(time_freq_1->end_time() + interval);
63 
64  TimeFrequencyContainer<Cpu, float> tfs;
65  tfs.add(time_freq_1);
66  ASSERT_THROW(tfs.add(time_freq_2), panda::Error);
67 }
68 
69 TEST_F(TimeFrequencyContainerTest, test_iterator)
70 {
71  typedef TimeFrequency<Cpu, float> TimeFrequencyType;
72  DimensionSize<Frequency> number_of_channels(1000);
73  TimeFrequencyType::TimeType interval(10 * data::milliseconds);
74  std::shared_ptr<TimeFrequencyType> time_freq_1=std::make_shared<TimeFrequencyType>(data::DimensionSize<data::Time>(100), number_of_channels);
75  time_freq_1->sample_interval(interval);
76  time_freq_1->start_time(TimeFrequencyType::TimePointType(std::chrono::seconds(10)));
77  ASSERT_EQ(time_freq_1->end_time(), time_freq_1->start_time() + 99 * interval);
78  std::shared_ptr<TimeFrequencyType> time_freq_2=std::make_shared<TimeFrequencyType>(data::DimensionSize<data::Time>(500), number_of_channels);
79  time_freq_2->sample_interval(interval);
80  time_freq_2->start_time(time_freq_1->end_time() + interval);
81  std::shared_ptr<TimeFrequencyType> time_freq_3=std::make_shared<TimeFrequencyType>(data::DimensionSize<data::Time>(1000), number_of_channels);
82  time_freq_3->sample_interval(interval);
83  time_freq_3->start_time(time_freq_2->end_time() + interval + interval);
84 
85  TimeFrequencyContainer<Cpu, float> tfs;
86  tfs.add(time_freq_1);
87  tfs.add(*time_freq_2);
88  tfs.add(*time_freq_3);
89 
90  std::size_t channel_count=0;
91  std::size_t td_count=0;
92  std::vector<TimeFrequencyType::TimePointType> time_points;
93  auto channel_end_handler = [&]() { ++channel_count; };
94  auto time_discontinuity_handler = [&](TimeFrequencyType::TimePointType const& tp, typename TimeFrequencyType::TimeType const&) {
95  ++td_count;
96  time_points.push_back(tp);
97  };
98  auto it = tfs.begin(channel_end_handler, time_discontinuity_handler);
99  auto end = tfs.end();
100  std::size_t count=0;
101  while(it!=end) {
102  ++count;
103  float val = *it;
104  ++val; // use so we don't get a warning msg
105  static_assert(std::is_same<decltype(*it), const float&>::value, "unexpected type from iterator dereference");
106  ++it;
107  }
108  ASSERT_EQ(time_freq_1->data_size() + time_freq_2->data_size() + time_freq_3->data_size(), count);
109  ASSERT_EQ(channel_count, time_freq_1->number_of_spectra() + time_freq_2->number_of_spectra() + time_freq_3->number_of_spectra());
110  ASSERT_EQ(2U, td_count);
111  ASSERT_EQ(time_points.size(), td_count);
112  ASSERT_EQ(time_freq_1->start_time(), time_points[0]);
113  ASSERT_EQ(time_freq_3->start_time(), time_points[1]);
114 }
115 
116 } // namespace test
117 } // namespace data
118 } // namespace cheetah
119 } // namespace ska
Some limits and constants for FLDO.
Definition: Brdz.h:35