FileStream.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstUtil/StringView.hpp"
#include "AstUtil/Encode.hpp"   // for aUtf8ToWide
#include <ios>                  // for std::ios_base, std::ios::in, ...
#include <fstream>              // for std::filebuf, std::ifstream, std::ofstream, ...
#include <string>               // for std::string
 
AST_NAMESPACE_BEGIN
 
namespace cxx
{
 
#if defined(_WIN32) || defined(AST_PARSED_BY_DOXYGEN)
namespace detail
{
 
inline std::filebuf* openfilebuf(std::filebuf& buf, const char* filepath, std::ios_base::openmode mode)
{
#if defined(_WIN32) 
    std::wstring wpath;
    errc_t rc = aUtf8ToWide(filepath, wpath);
    if(rc == eNoError)
        return buf.open(wpath.c_str(), mode);
    else
        return nullptr;
#else
    return buf.open(filepath, mode);
#endif
}
 
}   // namespace detail
 
 
class ifstream : public std::ifstream
{
public:
    ifstream() = default;
 
    explicit ifstream(const char* filepath, std::ios_base::openmode mode = std::ios::in)
    {
        open(filepath, mode);
    }
 
    explicit ifstream(const std::string& filepath, std::ios_base::openmode mode = std::ios::in)
    {
        open(filepath.c_str(), mode);
    }
 
    void open(const char* filepath, std::ios_base::openmode mode = std::ios::in)
    {
        // 与 std::ifstream::open 一致：始终补上 in
        if (detail::openfilebuf(*rdbuf(), filepath, mode | std::ios::in))
            clear();
        else
            setstate(std::ios::failbit);
    }
};
 
 
class ofstream : public std::ofstream
{
public:
    ofstream() = default;
 
    explicit ofstream(const char* filepath, std::ios_base::openmode mode = std::ios::out)
    {
        open(filepath, mode);
    }
 
    explicit ofstream(const std::string& filepath, std::ios_base::openmode mode = std::ios::out)
    {
        open(filepath.c_str(), mode);
    }
 
    void open(const char* filepath, std::ios_base::openmode mode = std::ios::out)
    {
        // 与 std::ofstream::open 一致：始终补上 out
        if (detail::openfilebuf(*rdbuf(), filepath, mode | std::ios::out))
            clear();
        else
            setstate(std::ios::failbit);
    }
};
 
 
class fstream : public std::fstream
{
public:
    fstream() = default;
 
    explicit fstream(const char* filepath, std::ios_base::openmode mode = std::ios::in | std::ios::out)
    {
        open(filepath, mode);
    }
 
    explicit fstream(const std::string& filepath, std::ios_base::openmode mode = std::ios::in | std::ios::out)
    {
        open(filepath.c_str(), mode);
    }
 
    void open(const char* filepath, std::ios_base::openmode mode = std::ios::in | std::ios::out)
    {
        if (detail::openfilebuf(*rdbuf(), filepath, mode))
            clear();
        else
            setstate(std::ios::failbit);
    }
};
 
#else
 
using std::ifstream;
using std::ofstream;
using std::fstream;
 
#endif
 
}   // namespace cxx
 
 
AST_NAMESPACE_END