diff --git a/modules/prediction/prediction_component.cc b/modules/prediction/prediction_component.cc index 2531e79b0bda76e5064af7597bbb6c40e6e074c1..74aad661f88674312fbd8e16a44eafb7069ad441 100644 --- a/modules/prediction/prediction_component.cc +++ b/modules/prediction/prediction_component.cc @@ -23,8 +23,6 @@ #include "boost/filesystem.hpp" #include "boost/range/iterator_range.hpp" -// #include "rosbag/bag.h" -// #include "rosbag/view.h" #include "modules/common/adapters/adapter_gflags.h" #include "modules/common/math/vec2d.h" @@ -87,7 +85,8 @@ std::string PredictionComponent::Name() const { // } // bag.close(); // } -void PredictionComponent::ProcessRosbag(const std::string& filename) { + +void PredictionComponent::ProcessOfflineData(const std::string& filename) { // TODO(all) implement } @@ -154,14 +153,14 @@ bool PredictionComponent::Init() { apollo::common::util::split(FLAGS_prediction_offline_bags, ':', &inputs); for (const auto& input : inputs) { std::vector offline_bags; - GetDataFiles(boost::filesystem::path(input), &offline_bags); + GetDataFileNames(boost::filesystem::path(input), &offline_bags); std::sort(offline_bags.begin(), offline_bags.end()); AINFO << "For input " << input << ", found " << offline_bags.size() << " rosbags to process"; for (std::size_t i = 0; i < offline_bags.size(); ++i) { AINFO << "\tProcessing: [ " << i << " / " << offline_bags.size() << " ]: " << offline_bags[i]; - ProcessRosbag(offline_bags[i]); + ProcessOfflineData(offline_bags[i]); } } Stop(); diff --git a/modules/prediction/prediction_component.h b/modules/prediction/prediction_component.h index 1e83151770fdb247f14c0716b6d9012d7b73152e..e0b96569c871b79ab27ac2203ac295f888ba7f5e 100644 --- a/modules/prediction/prediction_component.h +++ b/modules/prediction/prediction_component.h @@ -93,11 +93,13 @@ class PredictionComponent : * @brief process rosbag in offline mode, mainly for extracting prediction * features. */ - void ProcessRosbag(const std::string &filename); + void ProcessOfflineData(const std::string &filename); private: double start_time_ = 0.0; + PredictionConf prediction_conf_; + common::adapter::AdapterManagerConfig adapter_conf_; }; diff --git a/modules/prediction/util/data_extraction.cc b/modules/prediction/util/data_extraction.cc index f9b6932cc4785974127e076a874ccbddc0615381..8459028ed4a21af21805d64623762fd068bf5b6c 100644 --- a/modules/prediction/util/data_extraction.cc +++ b/modules/prediction/util/data_extraction.cc @@ -24,7 +24,7 @@ namespace apollo { namespace prediction { -void GetDataFiles(const boost::filesystem::path& p, +void GetDataFileNames(const boost::filesystem::path& p, std::vector* bag_files) { CHECK(bag_files); if (!boost::filesystem::exists(p)) { @@ -41,7 +41,7 @@ void GetDataFiles(const boost::filesystem::path& p, if (boost::filesystem::is_directory(p)) { for (auto& entry : boost::make_iterator_range( boost::filesystem::directory_iterator(p), {})) { - GetDataFiles(entry.path(), bag_files); + GetDataFileNames(entry.path(), bag_files); } } } diff --git a/modules/prediction/util/data_extraction.h b/modules/prediction/util/data_extraction.h index 5c89285a26aa1d505b45fc44ef7d2974085e37e6..2472373139dffd75781b8fb90c5391da84037e1c 100644 --- a/modules/prediction/util/data_extraction.h +++ b/modules/prediction/util/data_extraction.h @@ -26,8 +26,8 @@ namespace apollo { namespace prediction { -void GetDataFiles(const boost::filesystem::path& p, - std::vector* bag_files); +void GetDataFileNames(const boost::filesystem::path& p, + std::vector* data_files); } // namespace prediction } // namespace apollo