1 /* Copyright 2017 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 // This file is used to provide equivalents of internal util::format::FormatF 16 // and util::format::AppendF. Unfortunately, type safety is not as good as a 17 // a full C++ example. 18 // TODO(aselle): When absl adds support for StrFormat, use that instead. 19 #ifndef TENSORFLOW_CONTRIB_LITE_TOCO_FORMAT_PORT_H_ 20 #define TENSORFLOW_CONTRIB_LITE_TOCO_FORMAT_PORT_H_ 21 22 #include "tensorflow/contrib/lite/toco/toco_types.h" 23 #include "tensorflow/core/lib/strings/stringprintf.h" 24 25 namespace toco { 26 namespace port { 27 28 /// Identity (default case) 29 template <class T> 30 T IdentityOrConvertStringToRaw(T foo) { 31 return foo; 32 } 33 34 // Overloaded case where we return std::string. 35 inline const char* IdentityOrConvertStringToRaw(const std::string& foo) { 36 return foo.c_str(); 37 } 38 39 #if defined(PLATFORM_GOOGLE) && defined(HAS_GLOBAL_STRING) 40 // Overloaded case where we return string. 41 inline const char* IdentityOrConvertStringToRaw(const string& foo) { 42 return foo.c_str(); 43 } 44 #endif // PLATFORM_GOOGLE 45 // Delegate to TensorFlow Appendf function until absl has an equivalent. 46 template <typename... Args> 47 inline void AppendFHelper(string* destination, const char* fmt, 48 Args&&... args) { 49 tensorflow::strings::Appendf(destination, fmt, args...); 50 } 51 52 // Specialization for no argument format string (avoid security bug). 53 inline void AppendFHelper(string* destination, const char* fmt) { 54 tensorflow::strings::Appendf(destination, "%s", fmt); 55 } 56 57 // Append formatted string (with format fmt and args args) to the string 58 // pointed to by destination. fmt follows C printf semantics. 59 // One departure is that %s can be driven by a std::string or string. 60 template <typename... Args> 61 inline void AppendF(string* destination, const char* fmt, Args&&... args) { 62 AppendFHelper(destination, fmt, IdentityOrConvertStringToRaw(args)...); 63 } 64 65 // Return formatted string (with format fmt and args args). fmt follows C printf 66 // semantics. One departure is that %s can be driven by a std::string or string. 67 template <typename... Args> 68 inline string StringF(const char* fmt, Args&&... args) { 69 string result; 70 AppendFHelper(&result, fmt, IdentityOrConvertStringToRaw(args)...); 71 return result; 72 } 73 74 } // namespace port 75 } // namespace toco 76 77 #endif // TENSORFLOW_CONTRIB_LITE_TOCO_FORMAT_PORT_H_ 78
