blakeridgway/supercell-wx
1
0
Fork 0
mirror of https://github.com/ciphervance/supercell-wx.git synced 2025-10-31 02:20:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Generic product record retrieval for level 2 or level 3 data

Browse source
This commit is contained in:
Dan Paulat 2022-04-03 23:01:16 -05:00
parent 1b3dc1ca67
commit c7b5a659ab
1 changed files with 53 additions and 28 deletions
Download patch file Download diff file Expand all files Collapse all files

81
scwx-qt/source/scwx/qt/manager/radar_product_manager.cpp
View file

@ -26,6 +26,9 @@ static const std::string logPrefix_ =
typedef std::function<std::shared_ptr<wsr88d::NexradFile>()>
CreateNexradFileFunction;
typedef std::map<std::chrono::system_clock::time_point,
std::shared_ptr<types::RadarProductRecord>>
RadarProductRecordMap;
static constexpr uint32_t NUM_RADIAL_GATES_0_5_DEGREE =
common::MAX_0_5_DEGREE_RADIALS * common::MAX_DATA_MOMENT_GATES;
@ -63,9 +66,15 @@ public:
}
~RadarProductManagerImpl() = default;
static std::shared_ptr<types::RadarProductRecord>
GetRadarProductRecord(RadarProductRecordMap& map,
std::chrono::system_clock::time_point time);
std::shared_ptr<types::RadarProductRecord>
GetLevel2ProductRecord(std::chrono::system_clock::time_point time);
std::shared_ptr<types::RadarProductRecord>
GetLevel3ProductRecord(const std::string& product,
std::chrono::system_clock::time_point time);
std::shared_ptr<types::RadarProductRecord>
StoreRadarProductRecord(std::shared_ptr<types::RadarProductRecord> record);
static void
@ -80,13 +89,8 @@ public:
std::vector<float> coordinates0_5Degree_;
std::vector<float> coordinates1Degree_;
std::map<std::chrono::system_clock::time_point,
std::shared_ptr<types::RadarProductRecord>>
level2ProductRecords_;
std::unordered_map<std::string,
std::map<std::chrono::system_clock::time_point,
std::shared_ptr<types::RadarProductRecord>>>
level3ProductRecords_;
RadarProductRecordMap level2ProductRecords_;
std::unordered_map<std::string, RadarProductRecordMap> level3ProductRecords_;
};
RadarProductManager::RadarProductManager(const std::string& radarId) :
@ -302,46 +306,67 @@ void RadarProductManagerImpl::LoadNexradFile(
}
std::shared_ptr<types::RadarProductRecord>
RadarProductManagerImpl::GetLevel2ProductRecord(
std::chrono::system_clock::time_point time)
RadarProductManagerImpl::GetRadarProductRecord(
RadarProductRecordMap& map, std::chrono::system_clock::time_point time)
{
std::shared_ptr<types::RadarProductRecord> record = nullptr;
// TODO: Round to minutes
// Find the first product record greater than the time requested
auto it = level2ProductRecords_.upper_bound(time);
auto it = map.upper_bound(time);
// A product record with a time greater was found
if (it != level2ProductRecords_.cend())
if (it != map.cend())
{
// Are there product records prior to this record?
if (it != level2ProductRecords_.cbegin())
if (it != map.cbegin())
{
// Get the product record immediately preceding, this the record we are
// looking for
--it;
// Does the record contain the time we are looking for?
if (it->second->level2_file()->start_time() <= time &&
time <= it->second->level2_file()->end_time())
{
record = it->second;
}
record = (--it)->second;
}
}
else if (level2ProductRecords_.size() > 0)
else if (map.size() > 0)
{
// A product record with a time greater was not found. If it exists, it
// must be the last record.
auto rit = level2ProductRecords_.rbegin();
record = map.rbegin()->second;
}
// Does the record contain the time we are looking for?
if (rit->second->level2_file()->start_time() <= time &&
time <= rit->second->level2_file()->end_time())
{
record = rit->second;
}
return record;
}
std::shared_ptr<types::RadarProductRecord>
RadarProductManagerImpl::GetLevel2ProductRecord(
std::chrono::system_clock::time_point time)
{
std::shared_ptr<types::RadarProductRecord> record =
GetRadarProductRecord(level2ProductRecords_, time);
// TODO: Round to minutes
// Does the record contain the time we are looking for?
if (record != nullptr && (time < record->level2_file()->start_time() ||
record->level2_file()->end_time() < time))
{
record = nullptr;
}
return record;
}
std::shared_ptr<types::RadarProductRecord>
RadarProductManagerImpl::GetLevel3ProductRecord(
const std::string& product, std::chrono::system_clock::time_point time)
{
std::shared_ptr<types::RadarProductRecord> record = nullptr;
auto it = level3ProductRecords_.find(product);
if (it != level3ProductRecords_.cend())
{
record = GetRadarProductRecord(it->second, time);
}
return record;