source: cpp/common/stl-util.cpp @ 471

Last change on this file since 471 was 460, checked in by Maciej Komosinski, 8 years ago

Added a function to strip the path component from filename

  • Property svn:eol-style set to native
File size: 4.5 KB
Line 
1// This file is a part of Framsticks SDK.  http://www.framsticks.com/
2// Copyright (C) 1999-2015  Maciej Komosinski and Szymon Ulatowski.
3// See LICENSE.txt for details.
4
5#include "stl-util.h"
6#include <stdarg.h>
7#include <stdlib.h>
8#include "nonstd_stdio.h"
9#include "Convert.h"
10#include "nonstd.h"
11#include "log.h"
12#include <assert.h>
13#ifdef USE_VIRTFILE
14#include <common/virtfile/virtfile.h>
15#endif
16#ifdef __BORLANDC__
17#define va_copy(to,from) to=from //borland does not have va_copy() at all; va_list is just a pointer in borland
18#endif
19
20string ssprintf_va(const char* format, va_list ap)
21{
22        string s; //clang crashed when this declaration was in s=buf
23        int size = 256;
24        char* buf;
25        va_list ap_copy; // "va_list ap" can only by used once by printf-type functions as they advance the current argument pointer (crashed on linux x86_64)
26        // (does not apply to SString::sprintf, it does not have the va_list variant)
27
28        //almost like SString::sprintf, but there is no common code to share because SString can use its directWrite to avoid double allocating/copying
29#ifdef USE_VSCPRINTF
30        va_copy(ap_copy, ap);
31        size = _vscprintf(format, ap_copy) + 1; //+1 for terminating null character
32        va_end(ap_copy);
33#endif
34
35        while (1)
36        {
37                buf = (char*)malloc(size);
38                assert(buf != NULL);
39                va_copy(ap_copy, ap);
40                int n = vsnprintf(buf, size, format, ap_copy);
41                va_end(ap_copy);
42                if (n > -1 && n < size)
43                {
44                        s = buf;
45                        free(buf);
46                        return s;
47                }
48#ifdef VSNPRINTF_RETURNS_REQUIRED_SIZE
49                if (n > -1)    /* glibc 2.1 */
50                        size = n+1; /* precisely what is needed */
51                else           /* glibc 2.0 */
52#endif
53                        size *= 2;  /* twice the old size */
54                free(buf);
55        }
56}
57
58char* strmove(char *a, char *b) //strcpy that works well for overlapping strings ("Source and destination overlap")
59{
60        if (a == NULL || b == NULL)
61                return NULL;
62        memmove(a, b, strlen(b) + 1);
63        return a;
64}
65
66string ssprintf(const char* format, ...)
67{
68        va_list ap;
69        va_start(ap, format);
70        string ret = ssprintf_va(format, ap); //is it too wasteful? copying the string again... unless the compiler can handle it better
71        va_end(ap);
72        return ret;
73}
74
75bool readCompleteFile(const char* filename, vector<char>& data, bool warn_on_missing_file)
76{
77        bool ok = false;
78#ifdef USE_VIRTFILE
79        if (!isAbsolutePath(filename))
80        {
81                VirtFILE *f=Vfopen(filename,FOPEN_READ_BINARY);
82                if (f)
83                {
84                        int size=f->getSize();
85                        data.resize(size);
86                        int przeczytane = f->Vread(&data[0], size, 1);
87                        ok = przeczytane == 1;
88                        delete f;
89                }
90        }
91        else
92#endif
93        {
94                MFILE *f = mfopen(filename, FOPEN_READ_BINARY);
95                if (f)
96                {
97                        int size = getFileSize(f);
98                        data.resize(size);
99                        int przeczytane = mfread(&data[0], size, 1, f);
100                        mfclose(f);
101                        ok = przeczytane == 1;
102                }
103        }
104        if (warn_on_missing_file && !ok)
105                logPrintf("stl-util", "readCompleteFile", LOG_WARN, "Couldn't open file '%s'", filename);
106        return ok;
107}
108
109bool readCompleteFile(const char* filename, string& out, bool warn_on_missing_file)
110{
111        vector<char> data;
112        if (readCompleteFile(filename, data, warn_on_missing_file))
113        {
114                out = string(&data[0], data.size());
115                return true;
116        }
117        return false;
118}
119
120bool writeCompleteFile(const char* filename, const string& text, bool warn_on_fail)
121{
122        MFILE *f = mfopen(filename, FOPEN_WRITE_BINARY);
123        bool ok = f != NULL;
124        if (f)
125        {
126                int zapisane = mfwrite(text.c_str(), text.length(), 1, f);
127                mfclose(f);
128                ok &= zapisane == 1;
129        }
130        if (warn_on_fail && !ok)
131                logPrintf("stl-util", "writeCompleteFile", LOG_WARN, "couldn't write file '%s'", filename);
132        return ok;
133}
134
135bool writeCompleteFile(const char* filename, vector<char>& data, bool warn_on_fail)
136{
137        string s(&data[0], data.size());
138        return writeCompleteFile(filename, s, warn_on_fail);
139}
140
141
142
143string stripExt(const string& filename)
144{
145        size_t dot = filename.rfind('.');
146        if (dot == string::npos) return filename;
147        size_t sep = filename.rfind(PATH_SEPARATOR_CHAR);
148        if ((sep == string::npos) || (sep < dot))
149                return filename.substr(0, dot);
150        return filename;
151}
152
153string stripFileDir(const string& filename)
154{
155        size_t sep = filename.rfind(PATH_SEPARATOR_CHAR);
156        if (sep == string::npos) return filename;
157        return filename.substr(sep+1);
158}
159
160string getFileExt(const string& filename)
161{
162        size_t dot = filename.rfind('.');
163        if (dot == string::npos) return string("");
164        size_t sep = filename.rfind(PATH_SEPARATOR_CHAR);
165        if ((sep == string::npos) || (sep < dot))
166                return filename.substr(dot);
167        return string("");
168}
169
170string getFileDir(const string& filename)
171{
172        size_t slash = filename.rfind(PATH_SEPARATOR_CHAR);
173        if (slash == string::npos) return string("");
174        return filename.substr(0, slash);
175}
Note: See TracBrowser for help on using the repository browser.