forked from google/language-resources
-
Notifications
You must be signed in to change notification settings - Fork 0
/
proto-util.cc
65 lines (57 loc) · 1.88 KB
/
proto-util.cc
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
// festus/proto-util.cc
//
// 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:https://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.
//
// Copyright 2015 Google, Inc.
// Author: [email protected] (Martin Jansche)
#include "festus/proto-util.h"
#include <fcntl.h>
#include <unistd.h>
#include <fst/compat.h>
#include <google/protobuf/io/zero_copy_stream.h>
#include <google/protobuf/io/zero_copy_stream_impl.h>
#include <google/protobuf/message.h>
#include <google/protobuf/text_format.h>
using ::google::protobuf::Message;
using ::google::protobuf::TextFormat;
using ::google::protobuf::io::FileInputStream;
namespace festus {
bool GetTextProtoFromFile(const char *filename, Message *proto) {
DCHECK(filename != nullptr);
DCHECK(proto != nullptr);
int fd = open(filename, O_RDONLY);
if (-1 == fd) {
LOG(ERROR) << "Could not open file: " << filename;
return false;
}
FileInputStream input(fd);
bool status = TextFormat::Parse(&input, proto);
if (!status) {
LOG(ERROR) << "Could not parse file " << filename
<< " as " << proto->GetTypeName();
}
if (close(fd) != 0) {
LOG(ERROR) << "Error closing file: " << filename;
return false;
}
return status;
}
string ShortUtf8DebugString(const Message &message) {
TextFormat::Printer p;
p.SetSingleLineMode(true);
p.SetUseUtf8StringEscaping(true); // Surprising, but correct.
string s;
p.PrintToString(message, &s);
return s;
}
} // namespace festus