Zero load time file formats
Sometimes you cannot afford to load data files from disk. Maybe you need
results immediately, or the data is simply too large to fit into memory. A
technique that I like to use is an on-disk data structure. Here is a toy
example for instantly accessing lists of related words.
In this article, I address the problem of the time needed to load data into memory from disk. However, I do not make any optimization for disk caches or blocks. I am not going to talk about B-Trees or cache-oblivious structures.
No waiting
Using an on-disk data structure, there is no need to load the whole file into memory or parse it. Instead of opening a file and reading its contents, we will use a memory mapped file. We tell the operating system the file name, and it will lazily load the parts of the file only when we access them. These parts remain in the disk cache even after our program exits. So if you later start the program again, it will execute similar queries more quickly. We let the operating system do the caching for us. In python, this is done using the mmap module. Mmap makes the file appear as a very long string.Toy example
Here is an on-disk structure for looking up related words that I prepared. (Download 11 MB of it). It has three sections: A header, an index, and a word section. The header contains the number of words. The index contains a list of pointers to word records. The word section contains the word records. It is constructed so that we can instantly query for words related to a given word by jumping around to different parts of the file.
--- header
4 bytes: number of words
--- index section. The words are listed in alphabetical order, so you can
--- look one up using binary search.
for each word:
4 byte ptr to word record
--- word section:
for each word:
null terminated text
4 bytes: number of related words
for each link,
ptr to linked word record
Here is a short python program for accessing the data file.
#!/usr/bin/python
# An on-disk data structure for finding related words
# By Steve Hanov. This code and data file are released to the public domain.
import sys, mmap, struct
class FrozenThesaurus:
def __init__(self, filename):
self.f = file(filename, "rb")
self.mmap = mmap.mmap( self.f.fileno(), 0, access=mmap.ACCESS_READ )
def getDword( self, ptr ):
# return the 32 bit number beginning at the given byte offset in the
# file.
return struct.unpack("<I", self.mmap[ptr:ptr+4])[0]
def getString( self, ptr ):
# return the null terminated string beginning at the given byte offset.
result = []
while self.mmap[ptr] != "\x00":
result.append(self.mmap[ptr])
ptr += 1
return "".join(result)
def getWordCount(self):
# Retrive the number of words in the file.
return self.getDword(0)
def getWord(self, index):
# Retrive a word, given its index. The index must be less then the word
# count.
return self.getString( self.getDword(4 + index * 4) )
def getIndexOf( self, word ):
# perform a binary search through the index for the given word.
high = self.getWordCount()
low = -1
while (high - low > 1):
probe = (high + low) / 2
candidate = self.getWord(probe)
if candidate == word:
return probe
elif candidate < word:
low = probe
else:
high = probe
return None
def getRelatedWords( self, word ):
# Returns the list of related words to the given word.
results = []
index = self.getIndexOf( word )
if index == None: return results
ptr = self.getDword( 4 + index * 4 )
# skip past the word text
while self.mmap[ptr] != '\x00': ptr += 1
ptr += 1
numRelated = self.getDword( ptr )
for i in range(numRelated):
ptr += 4
results.append( self.getString( self.getDword( ptr ) ) )
return results;
data = FrozenThesaurus("thesaurus.dat")
print data.getRelatedWords(sys.argv[1])
Comments