d1d8b5fe8ab71fbee03dd07514b688c6b3f6d4a4
[blender.git] / extern / ceres / internal / ceres / stringprintf.cc
1 // Ceres Solver - A fast non-linear least squares minimizer
2 // Copyright 2015 Google Inc. All rights reserved.
3 // http://ceres-solver.org/
4 //
5 // Redistribution and use in source and binary forms, with or without
6 // modification, are permitted provided that the following conditions are met:
7 //
8 // * Redistributions of source code must retain the above copyright notice,
9 //   this list of conditions and the following disclaimer.
10 // * Redistributions in binary form must reproduce the above copyright notice,
11 //   this list of conditions and the following disclaimer in the documentation
12 //   and/or other materials provided with the distribution.
13 // * Neither the name of Google Inc. nor the names of its contributors may be
14 //   used to endorse or promote products derived from this software without
15 //   specific prior written permission.
16 //
17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
18 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
21 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
22 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
23 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
24 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
25 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
26 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27 // POSSIBILITY OF SUCH DAMAGE.
28 //
29 // Author: Sanjay Ghemawat
30
31 #include "ceres/stringprintf.h"
32
33 #include <cerrno>
34 #include <cstdarg>  // For va_list and related operations
35 #include <cstdio>   // MSVC requires this for _vsnprintf
36 #include <string>
37 #include <vector>
38
39 #include "ceres/internal/port.h"
40
41 namespace ceres {
42 namespace internal {
43
44 using std::string;
45
46 #ifdef _MSC_VER
47 enum { IS_COMPILER_MSVC = 1 };
48 #if _MSC_VER < 1800
49 #define va_copy(d, s) ((d) = (s))
50 #endif
51 #else
52 enum { IS_COMPILER_MSVC = 0 };
53 #endif
54
55 void StringAppendV(string* dst, const char* format, va_list ap) {
56   // First try with a small fixed size buffer
57   char space[1024];
58
59   // It's possible for methods that use a va_list to invalidate
60   // the data in it upon use.  The fix is to make a copy
61   // of the structure before using it and use that copy instead.
62   va_list backup_ap;
63   va_copy(backup_ap, ap);
64   int result = vsnprintf(space, sizeof(space), format, backup_ap);
65   va_end(backup_ap);
66
67   if (result < sizeof(space)) {
68     if (result >= 0) {
69       // Normal case -- everything fit.
70       dst->append(space, result);
71       return;
72     }
73
74     if (IS_COMPILER_MSVC) {
75       // Error or MSVC running out of space.  MSVC 8.0 and higher
76       // can be asked about space needed with the special idiom below:
77       va_copy(backup_ap, ap);
78       result = vsnprintf(NULL, 0, format, backup_ap);
79       va_end(backup_ap);
80     }
81
82     if (result < 0) {
83       // Just an error.
84       return;
85     }
86   }
87
88   // Increase the buffer size to the size requested by vsnprintf,
89   // plus one for the closing \0.
90   int length = result+1;
91   char* buf = new char[length];
92
93   // Restore the va_list before we use it again
94   va_copy(backup_ap, ap);
95   result = vsnprintf(buf, length, format, backup_ap);
96   va_end(backup_ap);
97
98   if (result >= 0 && result < length) {
99     // It fit
100     dst->append(buf, result);
101   }
102   delete[] buf;
103 }
104
105
106 string StringPrintf(const char* format, ...) {
107   va_list ap;
108   va_start(ap, format);
109   string result;
110   StringAppendV(&result, format, ap);
111   va_end(ap);
112   return result;
113 }
114
115 const string& SStringPrintf(string* dst, const char* format, ...) {
116   va_list ap;
117   va_start(ap, format);
118   dst->clear();
119   StringAppendV(dst, format, ap);
120   va_end(ap);
121   return *dst;
122 }
123
124 void StringAppendF(string* dst, const char* format, ...) {
125   va_list ap;
126   va_start(ap, format);
127   StringAppendV(dst, format, ap);
128   va_end(ap);
129 }
130
131 }  // namespace internal
132 }  // namespace ceres