Project 1 Inverted Index

For this project, you will write a Java program that processes all text files in a directory and its subdirectories, cleans and parses the text into word stems, and builds an in-memory inverted index to store the mapping from word stems to the documents and position within those documents where those word stems were found.

For example, suppose we have the following mapping stored in our inverted index:

{
	"capybara": {
		"input/mammals.txt": [
			11
		]
	},
	"platypus": {
		"input/mammals.txt": [
			3,
			8
		],
		"input/dangerous/venomous.txt": [
			2
		]
	}
}

This indicates that after processing the word stems from files, the word capybara is found in the file input/mammals.html in position 11. The word platypus is found in two files, input/mammals.html and input/dangerous/venomous.html. In the file input/mammals.html, the word platypus appears twice in positions 3 and 8. In file input/dangerous/venomous.html, the word platypus is in position 2 in the file.

The process of stemming reduces a word to a base form (or “stem”), so that words like interesting, interested, and interests all map to the stem interest. Stemming is a common preprocessing step in many web search engines.

Functionality

The core functionality of your project must satisfy the following requirements:

• Process command-line arguments to determine the input to process and output to produce. See the Input and Output sections below for specifics.

• Create a custom inverted index data structure that stores a mapping from a word stem to the file(s) the word was found, and the position(s) in that file the word is located. The positions should start at 1. This will require nesting multiple built-in data structures.

• If provided a directory as input, find all files within that directory and all subdirectories and parse each text file found. If provided a single text file as input, only parse that individual file. Any files that end in the .text or .txt extension (case insensitive) should be considered a text file.

  • Use the UTF-8 character encoding for all file processing, including reading and writing.

• Efficiently process text files into word stems by removing any non-letter symbols (including digits, punctuation, accents, special characters), convert the remaining alphabetic characters to lowercase, splitting the text into words by whitespace, and then stemming the word using the Apache OpenNLP toolkit.

  • Use the regular expression (?U)[^\\p{Alpha}\\p{Space}]+ to remove special characters from text.

  • Use the regular expression (?U)\\p{Space}+ to split text into words by whitespace.

  • Use the SnowballStemmer English stemming algorithm in OpenNLP to stem words.

• If the appropriate command-line arguments are provided, output the inverted index in pretty JSON format. See the Output section below for specifics.

• Output user-friendly error messages in the case of exceptions or invalid input. Under no circumstance should your main() method output a stack trace to the user!

The functionality of your project will be evaluated with the Project1Test.java group of JUnit tests.

Input

Your main method must be placed in a class named Driver. The Driver class should accept the following command-line arguments:

• -path path where the flag -path indicates the next argument is a path to either a single text file or a directory of text files that must be processed and added to the inverted index

• -index path where the flag -index is an optional flag that indicates the next argument is the path to use for the inverted index output file. If the path argument is not provided, use index.json as the default output path. If the -index flag is not provided, do not produce an output file.

The command-line flag/value pairs may be provided in any order.

Output

The output of your program should be a file that contains the contents of your inverted index in alphabetically sorted order as a nested JSON object using a “pretty” format using tab characters for indentation. For example:

{
	"capybara": {
		"input/mammals.html": [
			11
		]
	},
	"platypus": {
		"input/mammals.html": [
			3,
			8
		],
		"input/dangerous/venomous.html": [
			2
		]
	}
}

According to the JSON standard, numbers like integers should never be quoted. Any string or object key, however, should always be surrounded by " quotes. Objects (similar to maps) should use curly braces { and } and arrays should use square brackets [ and ].

Each path should be a normalized relative path. Make sure there are no trailing commas after the last element. See the JSON files in the project tests expected directory for more examples.

Examples

The following are a few examples (non-comprehensive) to illustrate the usage of the command-line arguments. Consider the following example:

java Driver -path ../project-tests/text/simple/hello.txt
            -index index-path-hello.json

The above arguments indicate that Driver should build an inverted index from the single hello.txt file in the text/simple subdirectory of the current working directory’s parent directory project-tests, and output the inverted index as JSON to the index-path-hello.json file in the current working directory.

java Driver -path ../project-tests/text/simple -index

The above arguments indicate that Driver should build an inverted index from all of the text files found in the text/simple subdirectory of the current working directory’s parent directory, and output the inverted index as JSON to the default path index.json in the current working directory.

java Driver -path ../project-tests/text/simple

The above arguments indicate that Driver should build an inverted index from all of the HTML files found in the text/simple subdirectory of the current working directory’s parent directory, but it should NOT produce an output file. It must still build the inverted index however! (This will be useful in the future when we add the ability to search.)

Hints

It is important to develop the project iteratively. In fact, you may already have certain components complete thanks to the homework assignments. One possible breakdown of tasks are:

• Create code that handles parsing command-line arguments into flag/value pairs, and supports default values if a flag is missing a value.

• Create code that is able to traverse a directory and return a list of all the text files found within that directory. The File I/O (Featuring NIO.2) tutorials might be useful for this.

• Create code that is able to parse text into words, including converting that text to lowercase, replacing special characters and digits, splitting that text into words by whitespaces, and finally stemming the words.

• Create code that handles storing a word, file path, and location into an inverted index data structure.

• Create code that is capable of writing a nested data structure (matching your inverted index data structure) to a file in JSON format.

Test each part of your code. Keep in mind the tests provided only test the final output of your Java program. You are responsible for testing the individual components of your code.

Once you have the above components working, then you can work on integrating your code together. This part must still be done iteratively! Do not try to combine all of the code at once. First, work on being able to traverse a directory based on command-line parameters. Then, clean the text files found and output the cleaned words to the console. Finally, add those words to the index and write that index as JSON.