Add the ability to select type of output.

flags/parse.go

@@ -6,6 +6,7 @@ import (
 	"log"
 	"os"
 	"runtime"
+	"strings"
 )
 
 // Parse command-line parameters, test input, store them in the Controller.
@@ -24,6 +25,7 @@ func (flags *Flags) ParseFlags() {
 	flag.IntVar(&flags.Controller.NumCPUs, "numcpu", runtime.NumCPU(), "Number of CPU cores to assign to this task.")
 	flag.IntVar(&flags.Controller.Split, "split", 1, "Split the tasks in n parts. This depends on the availability of the first row.")
 	flag.IntVar(&flags.Controller.Part, "part", 1, "Process part x in n parts. Cannot be lower than 1, or higher than specified in split.")
+	flag.StringVar(&flags.Controller.Output, "output", "human", "Type of output. 'human' for human readable. 'flat' for flat as stored in memory output.")
 
 	// Parse the flags
 	flag.Parse()
@@ -72,6 +74,14 @@ func (flags *Flags) ParseFlags() {
 		os.Exit(1)
 	}
 
+	// Process output selection
+	flags.Controller.Output = strings.ToLower(flags.Controller.Output)
+	if flags.Controller.Output != "human" && flags.Controller.Output != "flat" {
+		log.Printf("ERROR: Invalid output, can only be 'human' or 'flat'.\n")
+		flags.printUsage()
+		os.Exit(1)
+	}
+
 }
 
 // Validate if a row is properly set.