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

Last change on this file since 460 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
RevLine 
[286]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.
[196]4
[180]5#include "stl-util.h"
6#include <stdarg.h>
7#include <stdlib.h>
8#include "nonstd_stdio.h"
[281]9#include "Convert.h"
[180]10#include "nonstd.h"
[375]11#include "log.h"
[220]12#include <assert.h>
[247]13#ifdef USE_VIRTFILE
[382]14#include <common/virtfile/virtfile.h>
[247]15#endif
16#ifdef __BORLANDC__
[257]17#define va_copy(to,from) to=from //borland does not have va_copy() at all; va_list is just a pointer in borland
[247]18#endif
[180]19
20string ssprintf_va(const char* format, va_list ap)
21{
[220]22        string s; //clang crashed when this declaration was in s=buf
[319]23        int size = 256;
[180]24        char* buf;
[247]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)
[220]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
[257]30        va_copy(ap_copy, ap);
[247]31        size = _vscprintf(format, ap_copy) + 1; //+1 for terminating null character
32        va_end(ap_copy);
[220]33#endif
34
[246]35        while (1)
36        {
37                buf = (char*)malloc(size);
38                assert(buf != NULL);
[257]39                va_copy(ap_copy, ap);
[247]40                int n = vsnprintf(buf, size, format, ap_copy);
41                va_end(ap_copy);
[246]42                if (n > -1 && n < size)
[220]43                {
[246]44                        s = buf;
[220]45                        free(buf);
46                        return s;
[246]47                }
[220]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);
[246]55        }
[180]56}
57
[257]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
[180]66string ssprintf(const char* format, ...)
67{
68        va_list ap;
69        va_start(ap, format);
[246]70        string ret = ssprintf_va(format, ap); //is it too wasteful? copying the string again... unless the compiler can handle it better
[180]71        va_end(ap);
72        return ret;
73}
74
75bool readCompleteFile(const char* filename, vector<char>& data, bool warn_on_missing_file)
76{
[257]77        bool ok = false;
[247]78#ifdef USE_VIRTFILE
79        if (!isAbsolutePath(filename))
[257]80        {
[247]81                VirtFILE *f=Vfopen(filename,FOPEN_READ_BINARY);
82                if (f)
[257]83                {
[247]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                }
[257]90        }
[247]91        else
92#endif
93        {
[257]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                }
[180]103        }
104        if (warn_on_missing_file && !ok)
[375]105                logPrintf("stl-util", "readCompleteFile", LOG_WARN, "Couldn't open file '%s'", filename);
[180]106        return ok;
107}
108
109bool readCompleteFile(const char* filename, string& out, bool warn_on_missing_file)
110{
[246]111        vector<char> data;
112        if (readCompleteFile(filename, data, warn_on_missing_file))
[180]113        {
[246]114                out = string(&data[0], data.size());
[180]115                return true;
116        }
117        return false;
118}
119
120bool writeCompleteFile(const char* filename, const string& text, bool warn_on_fail)
121{
[246]122        MFILE *f = mfopen(filename, FOPEN_WRITE_BINARY);
123        bool ok = f != NULL;
[180]124        if (f)
125        {
[246]126                int zapisane = mfwrite(text.c_str(), text.length(), 1, f);
[180]127                mfclose(f);
[246]128                ok &= zapisane == 1;
[180]129        }
130        if (warn_on_fail && !ok)
[375]131                logPrintf("stl-util", "writeCompleteFile", LOG_WARN, "couldn't write file '%s'", filename);
[180]132        return ok;
133}
134
135bool writeCompleteFile(const char* filename, vector<char>& data, bool warn_on_fail)
136{
[246]137        string s(&data[0], data.size());
[180]138        return writeCompleteFile(filename, s, warn_on_fail);
139}
140
141
142
[246]143string stripExt(const string& filename)
[180]144{
[319]145        size_t dot = filename.rfind('.');
[246]146        if (dot == string::npos) return filename;
[319]147        size_t sep = filename.rfind(PATH_SEPARATOR_CHAR);
[246]148        if ((sep == string::npos) || (sep < dot))
149                return filename.substr(0, dot);
[180]150        return filename;
151}
152
[460]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
[246]160string getFileExt(const string& filename)
[180]161{
[319]162        size_t dot = filename.rfind('.');
[246]163        if (dot == string::npos) return string("");
[319]164        size_t sep = filename.rfind(PATH_SEPARATOR_CHAR);
[246]165        if ((sep == string::npos) || (sep < dot))
[180]166                return filename.substr(dot);
167        return string("");
168}
[246]169
170string getFileDir(const string& filename)
171{
[319]172        size_t slash = filename.rfind(PATH_SEPARATOR_CHAR);
[246]173        if (slash == string::npos) return string("");
174        return filename.substr(0, slash);
175}
Note: See TracBrowser for help on using the repository browser.