135 lines
4.3 KiB
C++
135 lines
4.3 KiB
C++
/*
|
|
* Copyright (C) 2012 Open Source Robotics Foundation
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*/
|
|
|
|
#include <google/protobuf/descriptor.h>
|
|
#include <google/protobuf/compiler/code_generator.h>
|
|
#include <google/protobuf/io/printer.h>
|
|
#include <google/protobuf/io/zero_copy_stream.h>
|
|
#include <google/protobuf/descriptor.pb.h>
|
|
#include <boost/algorithm/string/replace.hpp>
|
|
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <utility>
|
|
#include <iostream>
|
|
|
|
#include "gazebo/msgs/generator/GazeboGenerator.hh"
|
|
|
|
namespace google {
|
|
namespace protobuf {
|
|
namespace compiler {
|
|
namespace cpp {
|
|
|
|
GazeboGenerator::GazeboGenerator(const std::string &/*_name*/) {}
|
|
GazeboGenerator::~GazeboGenerator() {}
|
|
bool GazeboGenerator::Generate(const FileDescriptor *_file,
|
|
const string &/*parameter*/,
|
|
OutputDirectory *_generator_context,
|
|
std::string * /*_error*/) const
|
|
{
|
|
std::string headerFilename = _file->name();
|
|
boost::replace_last(headerFilename, ".proto", ".pb.h");
|
|
|
|
std::string sourceFilename = _file->name();
|
|
boost::replace_last(sourceFilename, ".proto", ".pb.cc");
|
|
|
|
// Add boost shared point include
|
|
{
|
|
std::unique_ptr<io::ZeroCopyOutputStream> output(
|
|
_generator_context->OpenForInsert(headerFilename, "includes"));
|
|
io::Printer printer(output.get(), '$');
|
|
|
|
printer.Print("#pragma GCC system_header", "name", "includes");
|
|
}
|
|
|
|
// Add boost shared point include
|
|
{
|
|
std::unique_ptr<io::ZeroCopyOutputStream> output(
|
|
_generator_context->OpenForInsert(sourceFilename, "includes"));
|
|
io::Printer printer(output.get(), '$');
|
|
|
|
printer.Print("#pragma GCC diagnostic ignored \"-Wshadow\"", "name",
|
|
"includes");
|
|
}
|
|
|
|
|
|
{
|
|
std::unique_ptr<io::ZeroCopyOutputStream> output(
|
|
_generator_context->OpenForInsert(headerFilename, "includes"));
|
|
io::Printer printer(output.get(), '$');
|
|
|
|
printer.Print("#include <boost/shared_ptr.hpp>\n", "name", "includes");
|
|
printer.Print("#include <gazebo/util/system.hh>\n", "name", "includes");
|
|
printer.Print("#include \"gazebo/msgs/MsgFactory.hh\"\n",
|
|
"name", "includes");
|
|
}
|
|
|
|
// Add boost shared typedef
|
|
{
|
|
std::unique_ptr<io::ZeroCopyOutputStream> output(
|
|
_generator_context->OpenForInsert(headerFilename, "namespace_scope"));
|
|
io::Printer printer(output.get(), '$');
|
|
|
|
std::string package = _file->package();
|
|
boost::replace_all(package, ".", "::");
|
|
|
|
std::string ptrType = "typedef boost::shared_ptr<" + package
|
|
+ "::" + _file->message_type(0)->name() + "> "
|
|
+ _file->message_type(0)->name() + "Ptr;\n";
|
|
|
|
printer.Print(ptrType.c_str(), "name", "namespace_scope");
|
|
}
|
|
|
|
// Add const boost shared typedef
|
|
{
|
|
std::unique_ptr<io::ZeroCopyOutputStream> output(
|
|
_generator_context->OpenForInsert(headerFilename, "global_scope"));
|
|
io::Printer printer(output.get(), '$');
|
|
|
|
std::string package = _file->package();
|
|
boost::replace_all(package, ".", "::");
|
|
|
|
std::string constType = "typedef const boost::shared_ptr<" + package
|
|
+ "::" + _file->message_type(0)->name() + " const> Const"
|
|
+ _file->message_type(0)->name() + "Ptr;";
|
|
|
|
printer.Print(constType.c_str(), "name", "global_scope");
|
|
}
|
|
|
|
// Add Message Factory register
|
|
{
|
|
std::unique_ptr<io::ZeroCopyOutputStream> output(
|
|
_generator_context->OpenForInsert(sourceFilename, "global_scope"));
|
|
io::Printer printer(output.get(), '$');
|
|
|
|
std::string package = _file->package();
|
|
boost::replace_all(package, ".", "::");
|
|
|
|
std::string name = _file->message_type(0)->name();
|
|
|
|
std::string constType = "GZ_REGISTER_STATIC_MSG(\"" + _file->package() +
|
|
"." + name + "\", " + name + ")\n";
|
|
printer.Print(constType.c_str(), "name", "namespace_scope");
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|