mirror of
https://github.com/odin-lang/Odin.git
synced 2026-01-07 13:33:13 +00:00
Refactor init_tokenizer_with_data to file memory mapping (Windows only currently)
This commit is contained in:
121
src/common.cpp
121
src/common.cpp
@@ -850,6 +850,127 @@ ReadDirectoryError read_directory(String path, Array<FileInfo> *fi) {
|
||||
|
||||
|
||||
|
||||
struct MemoryMappedFile {
|
||||
void *handle;
|
||||
|
||||
void *data;
|
||||
i32 size;
|
||||
};
|
||||
enum MemoryMappedFileError {
|
||||
MemoryMappedFile_None,
|
||||
|
||||
MemoryMappedFile_Empty,
|
||||
MemoryMappedFile_FileTooLarge,
|
||||
MemoryMappedFile_Invalid,
|
||||
MemoryMappedFile_NotExists,
|
||||
MemoryMappedFile_Permission,
|
||||
|
||||
MemoryMappedFile_COUNT,
|
||||
};
|
||||
|
||||
MemoryMappedFileError memory_map_file_32(char const *fullpath, MemoryMappedFile *memory_mapped_file) {
|
||||
MemoryMappedFileError err = MemoryMappedFile_None;
|
||||
|
||||
#if defined(GB_SYSTEM_WINDOWS)
|
||||
isize w_len = 0;
|
||||
wchar_t *w_str = gb__alloc_utf8_to_ucs2(temporary_allocator(), fullpath, &w_len);
|
||||
if (w_str == nullptr) {
|
||||
return MemoryMappedFile_Invalid;
|
||||
}
|
||||
i64 file_size = 0;
|
||||
LARGE_INTEGER li_file_size = {};
|
||||
HANDLE handle = nullptr;
|
||||
HANDLE file_mapping = nullptr;
|
||||
void *file_data = nullptr;
|
||||
|
||||
handle = CreateFileW(w_str, GENERIC_READ, FILE_SHARE_READ, nullptr, OPEN_EXISTING, FILE_FLAG_SEQUENTIAL_SCAN, NULL);
|
||||
if (handle == INVALID_HANDLE_VALUE) {
|
||||
handle = nullptr;
|
||||
goto window_handle_file_error;
|
||||
}
|
||||
|
||||
li_file_size = {};
|
||||
if (!GetFileSizeEx(handle, &li_file_size)) {
|
||||
goto window_handle_file_error;
|
||||
}
|
||||
file_size = cast(i64)li_file_size.QuadPart;
|
||||
if (file_size > I32_MAX) {
|
||||
CloseHandle(handle);
|
||||
return MemoryMappedFile_FileTooLarge;
|
||||
}
|
||||
|
||||
if (file_size == 0) {
|
||||
CloseHandle(handle);
|
||||
err = MemoryMappedFile_Empty;
|
||||
memory_mapped_file->handle = nullptr;
|
||||
memory_mapped_file->data = nullptr;
|
||||
memory_mapped_file->size = 0;
|
||||
return err;
|
||||
}
|
||||
|
||||
file_mapping = CreateFileMappingW(handle, nullptr, PAGE_READONLY, 0, 0, nullptr);
|
||||
CloseHandle(handle);
|
||||
|
||||
file_data = MapViewOfFileEx(file_mapping, FILE_MAP_READ, 0, 0, 0/*file_size*/, nullptr/*base address*/);
|
||||
memory_mapped_file->handle = cast(void *)file_mapping;
|
||||
memory_mapped_file->data = file_data;
|
||||
memory_mapped_file->size = cast(i32)file_size;
|
||||
return err;
|
||||
|
||||
window_handle_file_error:;
|
||||
{
|
||||
DWORD handle_err = GetLastError();
|
||||
CloseHandle(handle);
|
||||
err = MemoryMappedFile_Invalid;
|
||||
switch (handle_err) {
|
||||
case ERROR_FILE_NOT_FOUND:
|
||||
case ERROR_PATH_NOT_FOUND:
|
||||
case ERROR_INVALID_DRIVE:
|
||||
err = MemoryMappedFile_NotExists;
|
||||
break;
|
||||
case ERROR_ACCESS_DENIED:
|
||||
case ERROR_INVALID_ACCESS:
|
||||
err = MemoryMappedFile_Permission;
|
||||
break;
|
||||
}
|
||||
return err;
|
||||
}
|
||||
|
||||
#else
|
||||
// TODO(bill): Memory map rather than copy contents
|
||||
gbFileContents fc = gb_file_read_contents(heap_allocator(), true, fullpath);
|
||||
|
||||
if (fc.size > I32_MAX) {
|
||||
err = MemoryMappedFile_FileTooLarge;
|
||||
gb_file_free_contents(&fc);
|
||||
} else if (fc.data != nullptr) {
|
||||
memory_mapped_file->handle = nullptr;
|
||||
memory_mapped_file->data = fc.data;
|
||||
memory_mapped_file->size = cast(i32)fc.size;
|
||||
} else {
|
||||
gbFile f = {};
|
||||
gbFileError file_err = gb_file_open(&f, fullpath);
|
||||
defer (gb_file_close(&f));
|
||||
|
||||
switch (file_err) {
|
||||
case gbFileError_Invalid: err = MemoryMappedFile_Invalid; break;
|
||||
case gbFileError_NotExists: err = MemoryMappedFile_NotExists; break;
|
||||
case gbFileError_Permission: err = MemoryMappedFile_Permission; break;
|
||||
}
|
||||
|
||||
if (err == MemoryMappedFile_None && gb_file_size(&f) == 0) {
|
||||
err = MemoryMappedFile_Empty;
|
||||
}
|
||||
}
|
||||
return err;
|
||||
#endif
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
#define USE_DAMERAU_LEVENSHTEIN 1
|
||||
|
||||
isize levenstein_distance_case_insensitive(String const &a, String const &b) {
|
||||
|
||||
Reference in New Issue
Block a user