Serialize the state in a proto stream file. (#403)

This follows googlecartographer/cartographer#379.
master
Wolfgang Hess 2017-07-03 16:07:55 +02:00 committed by GitHub
parent 7c31232208
commit 4d218283de
2 changed files with 9 additions and 12 deletions

View File

@ -26,6 +26,7 @@
#include "cartographer/io/file_writer.h" #include "cartographer/io/file_writer.h"
#include "cartographer/io/points_processor.h" #include "cartographer/io/points_processor.h"
#include "cartographer/io/points_processor_pipeline_builder.h" #include "cartographer/io/points_processor_pipeline_builder.h"
#include "cartographer/io/proto_stream.h"
#include "cartographer/mapping/proto/sparse_pose_graph.pb.h" #include "cartographer/mapping/proto/sparse_pose_graph.pb.h"
#include "cartographer/sensor/point_cloud.h" #include "cartographer/sensor/point_cloud.h"
#include "cartographer/sensor/range_data.h" #include "cartographer/sensor/range_data.h"
@ -58,9 +59,8 @@ DEFINE_string(
DEFINE_string(bag_filenames, "", DEFINE_string(bag_filenames, "",
"Bags to process, must be in the same order as the trajectories " "Bags to process, must be in the same order as the trajectories "
"in 'pose_graph_filename'."); "in 'pose_graph_filename'.");
DEFINE_string( DEFINE_string(pose_graph_filename, "",
pose_graph_filename, "", "Proto stream file containing the pose graph.");
"Proto containing the pose graph written by /write_assets service.");
DEFINE_bool(use_bag_transforms, true, DEFINE_bool(use_bag_transforms, true,
"Whether to read and use the transforms from the bag."); "Whether to read and use the transforms from the bag.");
@ -148,9 +148,9 @@ void Run(const string& pose_graph_filename,
carto::common::LuaParameterDictionary lua_parameter_dictionary( carto::common::LuaParameterDictionary lua_parameter_dictionary(
code, std::move(file_resolver)); code, std::move(file_resolver));
std::ifstream stream(pose_graph_filename.c_str()); carto::io::ProtoStreamReader reader(pose_graph_filename);
carto::mapping::proto::SparsePoseGraph pose_graph_proto; carto::mapping::proto::SparsePoseGraph pose_graph_proto;
CHECK(pose_graph_proto.ParseFromIstream(&stream)); CHECK(reader.ReadProto(&pose_graph_proto));
CHECK_EQ(pose_graph_proto.trajectory_size(), bag_filenames.size()) CHECK_EQ(pose_graph_proto.trajectory_size(), bag_filenames.size())
<< "Pose graphs contains " << pose_graph_proto.trajectory_size() << "Pose graphs contains " << pose_graph_proto.trajectory_size()
<< " trajectories while " << bag_filenames.size() << " trajectories while " << bag_filenames.size()

View File

@ -16,6 +16,7 @@
#include "cartographer_ros/map_builder_bridge.h" #include "cartographer_ros/map_builder_bridge.h"
#include "cartographer/io/proto_stream.h"
#include "cartographer_ros/assets_writer.h" #include "cartographer_ros/assets_writer.h"
#include "cartographer_ros/color.h" #include "cartographer_ros/color.h"
#include "cartographer_ros/msg_conversion.h" #include "cartographer_ros/msg_conversion.h"
@ -62,13 +63,9 @@ void MapBuilderBridge::FinishTrajectory(const int trajectory_id) {
} }
void MapBuilderBridge::SerializeState(const std::string& stem) { void MapBuilderBridge::SerializeState(const std::string& stem) {
std::ofstream proto_file(stem + ".pb", cartographer::io::ProtoStreamWriter writer(stem + ".pbstream");
std::ios_base::out | std::ios_base::binary); map_builder_.SerializeState(&writer);
CHECK(map_builder_.sparse_pose_graph()->ToProto().SerializeToOstream( CHECK(writer.Close()) << "Could not write state.";
&proto_file))
<< "Could not serialize pose graph.";
proto_file.close();
CHECK(proto_file) << "Could not write pose graph.";
} }
void MapBuilderBridge::WriteAssets(const string& stem) { void MapBuilderBridge::WriteAssets(const string& stem) {