diff options
Diffstat (limited to 'src/lib/util.cc')
-rw-r--r-- | src/lib/util.cc | 92 |
1 files changed, 92 insertions, 0 deletions
diff --git a/src/lib/util.cc b/src/lib/util.cc new file mode 100644 index 0000000..eeba92b --- /dev/null +++ b/src/lib/util.cc @@ -0,0 +1,92 @@ +// 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://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 2005-2010 Google, Inc. +// Author: riley@google.com (Michael Riley) +// +// \file +// FST utility definitions. + +#include <cctype> +#include <string> +#include <fst/util.h> + +// Utility flag definitions + +DEFINE_bool(fst_error_fatal, true, + "FST errors are fatal; o.w. return objects flagged as bad: " + " e.g., FSTs - kError prop. true, FST weights - not a member()"); + +namespace fst { + +int64 StrToInt64(const string &s, const string &src, size_t nline, + bool allow_negative = false, bool *error) { + int64 n; + const char *cs = s.c_str(); + char *p; + if (error) *error = false; + n = strtoll(cs, &p, 10); + if (p < cs + s.size() || (!allow_negative && n < 0)) { + FSTERROR() << "StrToInt64: Bad integer = " << s + << "\", source = " << src << ", line = " << nline; + if (error) *error = true; + return 0; + } + return n; +} + +void Int64ToStr(int64 n, string *s) { + ostringstream nstr; + nstr << n; + *s = nstr.str(); +} + +void ConvertToLegalCSymbol(string *s) { + for (string::iterator it = s->begin(); it != s->end(); ++it) + if (!isalnum(*it)) *it = '_'; +} + +// Skips over input characters to align to 'align' bytes. Returns +// false if can't align. +bool AlignInput(istream &strm, int align) { + char c; + for (int i = 0; i < align; ++i) { + int64 pos = strm.tellg(); + if (pos < 0) { + LOG(ERROR) << "AlignInput: can't determine stream position"; + return false; + } + if (pos % align == 0) break; + strm.read(&c, 1); + } + return true; +} + +// Write null output characters to align to 'align' bytes. Returns +// false if can't align. +bool AlignOutput(ostream &strm, int align) { + for (int i = 0; i < align; ++i) { + int64 pos = strm.tellp(); + if (pos < 0) { + LOG(ERROR) << "AlignOutput: can't determine stream position"; + return false; + } + if (pos % align == 0) break; + strm.write("", 1); + } + return true; +} + + +} // namespace fst |