more refactoring

passphrase.nimble

@@ -16,7 +16,6 @@ requires "nimcrypto >= 0.4.8"
 
 # Tasks n scripts
 
-import strutils
 
 proc runCmd(command: string, input = "", cache = ""): string =
   let (output, exitCode) = gorgeEx(command, input, cache)
@@ -29,10 +28,10 @@ proc runCmd(command: string, input = "", cache = ""): string =
 
 task(dictionary, "Generate dictionary from BNC XML files"):
   echo "Building dictionary"
-  let output = runCmd("nim c --run --threads:on -d:release -d:lto src/process.nim BNC/2554/download/Texts src/")
+  let output = runCmd("nim c --run --threads:on -d:release -d:lto src/process.nim data/BNC/2554/download/Texts src/")
   echo output.strip().splitlines()[^1]
 
 
 before(build):
-  if not fileExists("src/dictionary.txt"):
+  if not fileExists("data/dictionary.txt"):
     dictionaryTask()

src/passphrase.nim

@@ -23,7 +23,7 @@ proc len(d: Dictionary): int =
 
 
 proc loadWords(): Dictionary =
-  for word in staticRead("dictionary.txt").strip().splitLines():
+  for word in staticRead("../data/dictionary.txt").strip().splitLines():
     let startIdx = result.words.len.uint32
     result.offsets.add(startIdx)
     result.words.add(word)

src/passphrase.nims

@@ -0,0 +1 @@
+--d: release

src/process.nim

@@ -27,6 +27,24 @@ proc save(wordCounts: CountTable; dictName, countName: string) =
       break
 
 
+type Config = object
+  srcPath: string,
+  dstPath: string,
+
+
+proc parseInput(): Config =
+  result.srcPath = r"../data/BNC/2554/download/Texts/"
+  result.dstPath = "."
+
+  if paramCount() > 0:
+    result.srcPath = paramStr(1)
+  if paramCount() > 1:
+    result.dstPath = paramStr(2)
+
+  if not dirExists(basePath):
+    quit("Could not locate datafiles: directory " & basePath & " does not exist.")
+
+
 var 
   threadResults: Channel[CountTable[string]]
   progress: Channel[int]
@@ -46,17 +64,10 @@ proc processFiles(filenames: seq[string]) =
 when isMainModule:
   let start = getMonoTime()
 
-  let basePath =
-    if paramCount() > 0:
-      paramStr(1)
-    else:
-      r"../BNC/2554/download/Texts/"
-
-  if not dirExists(basePath):
-    quit("Could not locate datafiles: directory " & basePath & " does not exist.")
+  let config = parseInput()
   
   var paths: seq[string]
-  for path in walkDirRec(basePath):
+  for path in walkDirRec(config.srcPath):
     if path.endsWith(".xml"):
       paths.add(path)
 
@@ -79,15 +90,10 @@ when isMainModule:
       counts.inc(word, count)
 
   counts.sort()
-  let outPath = 
-    if paramCount() > 1:
-      paramStr(2)
-    else:
-      "."
   
   let
-    dPath = joinPath(outPath, "dictionary.txt")
-    cPath = joinPath(outPath, "counts.txt")
+    dPath = joinPath(config.dstPath, "dictionary.txt")
+    cPath = joinPath(config.dstPath, "counts.txt")
   save(counts, dPath, cPath)
 
   echo "Done. Finished in ", (getMonoTime() - start).inMilliseconds.float / 1000, " seconds."