Data.DAWG.Packed

Fully minimized and bit-packed directed acyclic word graphs.

This implementation mainly focuses on compactness (<500 Kb space for ~150000 word dictionaries) rather than genericity or dynamic usage. There are no insertion or deletion operations.

A DAWG node is stored in four bytes, using 22 bits for indexing and 8 bits for data storage. This implies that

• The number of nodes shouldn't exceed 2^22, or 4194304.
• Input characters should be mapped to the 0-255 range.

Types

data Node

This data type points to a prefix in the DAWG. When a node is the root node it represents the whole DAWG. When it is non-root, it can be used to access the suffixes of the prefix pointed to by the node.

Instances

Eq Node

Show Node

Binary Node

NFData Node

Construction

fromList :: [String] -> Node

fromAscList :: [String] -> Node

Allows for faster DAWG generation than fromList. The ordering assumption is unchecked.

fromFile :: FilePath -> IO Node

Read a DAWG previously serialized with toFile from a file.

Accessors

char :: Node -> Char

Get the character char of a node. The root nodes have the null character as char.

endOfWord :: Node -> Bool

Indicates whether a prefix pointed to by the node is a valid word.

root :: Node -> Node

Get the root node from a node.

children :: Node -> [Node]

Generate a list of the direct children of a node.

lookupPrefixBy :: (Char -> Char -> Ordering) -> String -> Node -> Maybe Node

Lookup a prefix by memberwise applying a comparison function. It is useful for setting case sensitivity, e.g. insensitiveLookup = lookupPrefixBy (comparing toLower)

lookupPrefix :: String -> Node -> Maybe Node

lookupPrefix = lookupPrefixBy (==)

memberBy :: (Char -> Char -> Ordering) -> String -> Node -> Bool

Test for membership with a memberwise comparison function.

member :: String -> Node -> Bool

member = memberBy (==)

Conversions

toList :: Node -> [String]

Get the list of all suffixes that end on a valid word ending. When used on the root node this function enlists the original words. The resulting list is unsorted.

toFile :: FilePath -> Node -> IO ()

Serialize a DAWG.

Internal

pack :: Char -> Bool -> Bool -> Int -> Word32

Create a bit-packed Word32.

unpack :: Word32 -> NodeVector -> Node

Create a node from a Word32 and a NodeVector.

type NodeVector = Vector Word32

The underlying container of the DAWG data. Modifying it will most likely result in an invalid DAWG.

Each Word32 represents a node. The format of a node is the following:

• 22 bits: the index of the first child.
• 8 bits: character data.
• 1 bit: end-of-word flag.
• 1 bit: end-of-childlist flag.

The children of a node are laid out next to each other, so they can be iterated over by starting from the first child and incrementing the index until a node with the end-of-childlist flag is found.

nodeVector :: Node -> NodeVector

Get the underlying vector from a node.

endOfList :: Node -> Bool

Indicates whether a node is the last in a list of children nodes.

childIndex :: Node -> Word32

Get the index of a node's first child node.

getNodeAt :: NodeVector -> Word32 -> Node

Create a node from some member of a NodeVector.