feat: first prototype of optimizers + alloca optimizer
This commit is contained in:
79
include/prelude/file.hpp
Normal file
79
include/prelude/file.hpp
Normal file
@@ -0,0 +1,79 @@
|
||||
#pragma once
|
||||
#include "prelude/string.hpp"
|
||||
#include <cstdio>
|
||||
|
||||
class File
|
||||
{
|
||||
public:
|
||||
enum class Mode
|
||||
{
|
||||
WRITE,
|
||||
READ,
|
||||
};
|
||||
|
||||
private:
|
||||
File(FILE* handle, Mode mode)
|
||||
: m_handle(handle), m_mode(mode) {}
|
||||
|
||||
public:
|
||||
static File Open(const StringView& filename, Mode mode)
|
||||
{
|
||||
StringBuilder sb;
|
||||
switch(mode)
|
||||
{
|
||||
case Mode::WRITE:
|
||||
sb.Append('w');
|
||||
break;
|
||||
case Mode::READ:
|
||||
sb.Append('r');
|
||||
break;
|
||||
}
|
||||
FILE* handle = fopen(filename.c_str(), sb.c_str());
|
||||
return File(handle, mode);
|
||||
}
|
||||
|
||||
private:
|
||||
inline bool Writable() const { return m_mode == File::Mode::WRITE; }
|
||||
inline bool Readable() const { return m_mode == File::Mode::READ; }
|
||||
|
||||
public:
|
||||
bool Write(const StringView& content)
|
||||
{
|
||||
assert(Writable() && "attempt to write to a file in read only mode");
|
||||
if (fwrite(content.c_str(), sizeof(char), content.size - 1, m_handle) == 0)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
public:
|
||||
StringView ReadAll(bool* status = nullptr)
|
||||
{
|
||||
auto fail = [&]() -> StringView {
|
||||
if (status) *status = false;
|
||||
return {};
|
||||
};
|
||||
|
||||
if (status) *status = true;
|
||||
|
||||
assert(Readable() && "attempt to read from a file in write only mode");
|
||||
if (fseek(m_handle, 0, SEEK_END) != 0) return fail();
|
||||
|
||||
auto size = ftell(m_handle);
|
||||
if (size == -1 && status) return fail();
|
||||
|
||||
StringBuilder sb;
|
||||
sb.ensure_extra(size);
|
||||
|
||||
size_t wrote = fread(sb.data, sizeof(char), size, m_handle);
|
||||
if (wrote == 0) return fail();
|
||||
|
||||
sb.size = wrote;
|
||||
return sb.view();
|
||||
}
|
||||
|
||||
private:
|
||||
FILE* m_handle;
|
||||
Mode m_mode;
|
||||
};
|
||||
Reference in New Issue
Block a user