wordtree 0.3.1 → 0.4.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 7bcb5a59a130a24ca1bede26a9f3b8efa651230c
+  data.tar.gz: 03ef94ffd836a11f891065fd7eb3c73d4aa8bdd2
+SHA512:
+  metadata.gz: cbf2b847f90fdffd1a52a4f7ea2c2fd78ce339ded803cbf30720fba1dced77b2ea2e9b1b84f5fee147d472d0e673437ad0e58ccffcb1a5fd2794fa87ec739384
+  data.tar.gz: 0abb233dd846913d69e13087d84c174136e8a1511c417f5239e3395972f17af71f43091079e8fc605354843c09bb73b6e82ca358801303d77f692c3d8d140ea5

data/ext/Makefile

@@ -0,0 +1,239 @@
+
+SHELL = /bin/sh
+
+# V=0 quiet, V=1 verbose.  other values don't work.
+V = 0
+Q1 = $(V:1=)
+Q = $(Q1:0=@)
+ECHO1 = $(V:1=@:)
+ECHO = $(ECHO1:0=@echo)
+
+#### Start of system configuration section. ####
+
+srcdir = .
+topdir = /Users/duane/.rbenv/versions/2.1.3/include/ruby-2.1.0
+hdrdir = $(topdir)
+arch_hdrdir = /Users/duane/.rbenv/versions/2.1.3/include/ruby-2.1.0/x86_64-darwin13.0
+PATH_SEPARATOR = :
+VPATH = $(srcdir):$(arch_hdrdir)/ruby:$(hdrdir)/ruby
+prefix = $(DESTDIR)/Users/duane/.rbenv/versions/2.1.3
+rubysitearchprefix = $(rubylibprefix)/$(sitearch)
+rubyarchprefix = $(rubylibprefix)/$(arch)
+rubylibprefix = $(libdir)/$(RUBY_BASE_NAME)
+exec_prefix = $(prefix)
+vendorarchhdrdir = $(vendorhdrdir)/$(sitearch)
+sitearchhdrdir = $(sitehdrdir)/$(sitearch)
+rubyarchhdrdir = $(rubyhdrdir)/$(arch)
+vendorhdrdir = $(rubyhdrdir)/vendor_ruby
+sitehdrdir = $(rubyhdrdir)/site_ruby
+rubyhdrdir = $(includedir)/$(RUBY_VERSION_NAME)
+vendorarchdir = $(vendorlibdir)/$(sitearch)
+vendorlibdir = $(vendordir)/$(ruby_version)
+vendordir = $(rubylibprefix)/vendor_ruby
+sitearchdir = $(sitelibdir)/$(sitearch)
+sitelibdir = $(sitedir)/$(ruby_version)
+sitedir = $(rubylibprefix)/site_ruby
+rubyarchdir = $(rubylibdir)/$(arch)
+rubylibdir = $(rubylibprefix)/$(ruby_version)
+sitearchincludedir = $(includedir)/$(sitearch)
+archincludedir = $(includedir)/$(arch)
+sitearchlibdir = $(libdir)/$(sitearch)
+archlibdir = $(libdir)/$(arch)
+ridir = $(datarootdir)/$(RI_BASE_NAME)
+mandir = $(datarootdir)/man
+localedir = $(datarootdir)/locale
+libdir = $(exec_prefix)/lib
+psdir = $(docdir)
+pdfdir = $(docdir)
+dvidir = $(docdir)
+htmldir = $(docdir)
+infodir = $(datarootdir)/info
+docdir = $(datarootdir)/doc/$(PACKAGE)
+oldincludedir = $(DESTDIR)/usr/include
+includedir = $(prefix)/include
+localstatedir = $(prefix)/var
+sharedstatedir = $(prefix)/com
+sysconfdir = $(prefix)/etc
+datadir = $(datarootdir)
+datarootdir = $(prefix)/share
+libexecdir = $(exec_prefix)/libexec
+sbindir = $(exec_prefix)/sbin
+bindir = $(exec_prefix)/bin
+archdir = $(rubyarchdir)
+
+
+CC = clang
+CXX = g++
+LIBRUBY = $(LIBRUBY_A)
+LIBRUBY_A = lib$(RUBY_SO_NAME)-static.a
+LIBRUBYARG_SHARED = 
+LIBRUBYARG_STATIC = -l$(RUBY_SO_NAME)-static -framework CoreFoundation
+empty =
+OUTFLAG = -o $(empty)
+COUTFLAG = -o $(empty)
+
+RUBY_EXTCONF_H = 
+cflags   =  $(optflags) $(debugflags) $(warnflags)
+optflags = -O3 -fno-fast-math
+debugflags = -ggdb3
+warnflags = -Wall -Wextra -Wno-unused-parameter -Wno-parentheses -Wno-long-long -Wno-missing-field-initializers -Wunused-variable -Wpointer-arith -Wwrite-strings -Wdeclaration-after-statement -Wshorten-64-to-32 -Wimplicit-function-declaration -Wdivision-by-zero -Wextra-tokens
+CCDLFLAGS = -fno-common
+CFLAGS   = $(CCDLFLAGS)  -O3 -Wno-error=shorten-64-to-32  -pipe $(ARCH_FLAG)
+INCFLAGS = -I. -I$(arch_hdrdir) -I$(hdrdir)/ruby/backward -I$(hdrdir) -I$(srcdir)
+DEFS     = 
+CPPFLAGS =  -I/Users/duane/.rbenv/versions/2.1.3/include  -D_XOPEN_SOURCE -D_DARWIN_C_SOURCE -D_DARWIN_UNLIMITED_SELECT -D_REENTRANT $(DEFS) $(cppflags)
+CXXFLAGS = $(CCDLFLAGS) $(cxxflags) $(ARCH_FLAG)
+ldflags  = -L. -L/Users/duane/.rbenv/versions/2.1.3/lib  -fstack-protector -L/usr/local/lib
+dldflags = -Wl,-undefined,dynamic_lookup -Wl,-multiply_defined,suppress 
+ARCH_FLAG = 
+DLDFLAGS = $(ldflags) $(dldflags) $(ARCH_FLAG)
+LDSHARED = $(CC) -dynamic -bundle
+LDSHAREDXX = $(CXX) -dynamic -bundle
+AR = ar
+EXEEXT = 
+
+RUBY_INSTALL_NAME = ruby
+RUBY_SO_NAME = ruby
+RUBYW_INSTALL_NAME = 
+RUBY_VERSION_NAME = $(RUBY_BASE_NAME)-$(ruby_version)
+RUBYW_BASE_NAME = rubyw
+RUBY_BASE_NAME = ruby
+
+arch = x86_64-darwin13.0
+sitearch = $(arch)
+ruby_version = 2.1.0
+ruby = $(bindir)/ruby
+RUBY = $(ruby)
+ruby_headers = $(hdrdir)/ruby.h $(hdrdir)/ruby/ruby.h $(hdrdir)/ruby/defines.h $(hdrdir)/ruby/missing.h $(hdrdir)/ruby/intern.h $(hdrdir)/ruby/st.h $(hdrdir)/ruby/subst.h $(arch_hdrdir)/ruby/config.h
+
+RM = rm -f
+RM_RF = $(RUBY) -run -e rm -- -rf
+RMDIRS = rmdir -p
+MAKEDIRS = mkdir -p
+INSTALL = /usr/bin/install -c
+INSTALL_PROG = $(INSTALL) -m 0755
+INSTALL_DATA = $(INSTALL) -m 644
+COPY = cp
+TOUCH = exit >
+
+#### End of system configuration section. ####
+
+preload = 
+
+libpath = . $(libdir)
+LIBPATH =  -L. -L$(libdir)
+DEFFILE = 
+
+CLEANFILES = mkmf.log
+DISTCLEANFILES = 
+DISTCLEANDIRS = 
+
+extout = 
+extout_prefix = 
+target_prefix = 
+LOCAL_LIBS = 
+LIBS =   -lpthread -lgmp -ldl -lobjc 
+ORIG_SRCS = wordtree.cc
+SRCS = $(ORIG_SRCS) 
+OBJS = wordtree.o
+HDRS = 
+TARGET = wordtree
+TARGET_NAME = wordtree
+TARGET_ENTRY = Init_$(TARGET_NAME)
+DLLIB = $(TARGET).bundle
+EXTSTATIC = 
+STATIC_LIB = 
+
+TIMESTAMP_DIR = .
+BINDIR        = $(bindir)
+RUBYCOMMONDIR = $(sitedir)$(target_prefix)
+RUBYLIBDIR    = $(sitelibdir)$(target_prefix)
+RUBYARCHDIR   = $(sitearchdir)$(target_prefix)
+HDRDIR        = $(rubyhdrdir)/ruby$(target_prefix)
+ARCHHDRDIR    = $(rubyhdrdir)/$(arch)/ruby$(target_prefix)
+
+TARGET_SO     = $(DLLIB)
+CLEANLIBS     = $(TARGET).bundle 
+CLEANOBJS     = *.o  *.bak
+
+all:    $(DLLIB)
+static: $(STATIC_LIB)
+.PHONY: all install static install-so install-rb
+.PHONY: clean clean-so clean-static clean-rb
+
+clean-static::
+clean-rb-default::
+clean-rb::
+clean-so::
+clean: clean-so clean-static clean-rb-default clean-rb
+		-$(Q)$(RM) $(CLEANLIBS) $(CLEANOBJS) $(CLEANFILES) .*.time
+
+distclean-rb-default::
+distclean-rb::
+distclean-so::
+distclean-static::
+distclean: clean distclean-so distclean-static distclean-rb-default distclean-rb
+		-$(Q)$(RM) Makefile $(RUBY_EXTCONF_H) conftest.* mkmf.log
+		-$(Q)$(RM) core ruby$(EXEEXT) *~ $(DISTCLEANFILES)
+		-$(Q)$(RMDIRS) $(DISTCLEANDIRS) 2> /dev/null || true
+
+realclean: distclean
+install: install-so install-rb
+
+install-so: $(DLLIB) $(TIMESTAMP_DIR)/.RUBYARCHDIR.time
+	$(INSTALL_PROG) $(DLLIB) $(RUBYARCHDIR)
+clean-static::
+	-$(Q)$(RM) $(STATIC_LIB)
+install-rb: pre-install-rb install-rb-default
+install-rb-default: pre-install-rb-default
+pre-install-rb: Makefile
+pre-install-rb-default: Makefile
+pre-install-rb-default:
+	$(ECHO) installing default wordtree libraries
+$(TIMESTAMP_DIR)/.RUBYARCHDIR.time:
+	$(Q) $(MAKEDIRS) $(@D) $(RUBYARCHDIR)
+	$(Q) $(TOUCH) $@
+
+site-install: site-install-so site-install-rb
+site-install-so: install-so
+site-install-rb: install-rb
+
+.SUFFIXES: .c .m .cc .mm .cxx .cpp .C .o
+
+.cc.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.mm.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.cxx.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.cpp.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.C.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CXX) $(INCFLAGS) $(CPPFLAGS) $(CXXFLAGS) $(COUTFLAG)$@ -c $<
+
+.c.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -c $<
+
+.m.o:
+	$(ECHO) compiling $(<)
+	$(Q) $(CC) $(INCFLAGS) $(CPPFLAGS) $(CFLAGS) $(COUTFLAG)$@ -c $<
+
+$(DLLIB): $(OBJS) Makefile
+	$(ECHO) linking shared-object $(DLLIB)
+	-$(Q)$(RM) $(@)
+	$(Q) $(LDSHAREDXX) -o $@ $(OBJS) $(LIBPATH) $(DLDFLAGS) $(LOCAL_LIBS) $(LIBS)
+	$(Q) $(POSTLINK)
+
+
+
+$(OBJS): $(HDRS) $(ruby_headers)

data/ext/extconf.rb

@@ -0,0 +1,5 @@
+require "mkmf"
+
+dir_config("wordtree")
+
+create_makefile("wordtree")

data/ext/wordtree.cc

@@ -0,0 +1,125 @@
+#include <ruby.h>
+#include <ruby/encoding.h>
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+
+// for rubinius
+#ifndef rb_enc_fast_mbclen
+#   define rb_enc_fast_mbclen rb_enc_mbclen
+#endif
+
+static rb_encoding* u8_enc;
+static rb_encoding* bin_enc;
+
+/** Transforms text such as the following:
+ *
+ *   And behold, I said, "This is no good!"
+ *   What shall ye say unto these people, there-
+ *   fore?
+ *
+ * Into a cleaned up single line of text, like the following:
+ *
+ *   and behold i said this is no good.what shall ye say unto these people therefore.
+ *
+ * Spaces indicate word boundaries, while periods indicate sentence boundaries.
+ */
+size_t text_clean_cstr(char* text)
+{
+  if (*text == '\0') return 0;
+
+  char* read;
+  char* write = text;
+  uint8_t join_lines = false,
+          just_added_space = true,   // prevent prefix spaces
+          just_added_period = false;
+  for (read=text; *read; read++) {
+    char c = *read;
+    if (c >= 'A' && c <= 'Z') {
+      // Change upper case to lowercase
+      c += 32;
+    } else if (c == '\n') {
+      // Change newlines to spaces (i.e. both count as whitespace)
+      c = ' ';
+    } else if (c == '?' || c == '!') {
+      // Change exclamation, question marks to periods (i.e. sentence boundaries)
+      c = '.';
+    }
+
+    if (c == '-') {
+      join_lines = true;
+    } else if (join_lines && c == ' ') {
+      // ignore whitespace after a dash (i.e. including newlines, which is the
+      // most common case because words that are broken by syllables are dashed)
+    } else if (c == '.' && !just_added_period) {
+      // erase space before period
+      if (just_added_space) write--;
+      *write++ = '.';
+      just_added_period = true;
+      just_added_space = false;
+      join_lines = false;
+    } else if (c == ' ' && !just_added_space && !just_added_period) {
+      *write++ = ' ';
+      just_added_space = true;
+      just_added_period = false;
+    } else if (c >= 'a' && c <= 'z') {
+      *write++ = c;
+      just_added_space = false;
+      just_added_period = false;
+      join_lines = false;
+    }
+  }
+  // erase space at end of text
+  if (just_added_space) write--;
+
+  // Return the new length of the string
+  return (size_t)(write - text);
+}
+
+static VALUE text_common_trigrams(VALUE self, VALUE text) {
+  char* ptext = RSTRING_PTR(text);
+  long len = RSTRING_LEN(text);
+
+  if (len < 3) return INT2NUM(0);
+
+  /* 28 most common English trigrams, all squished together */
+  char common_trigrams[] = "allandedtentereforhashatherhisingionithmenncendeoftsthterthathethitiotisverwaswityou";
+
+  char* ptr = ptext;
+  char* tail = ptext + len;
+  int i = 0, common_matched = 0;
+  while (ptr < tail) {
+    for (i = 0; i < sizeof(common_trigrams); i += 3) {
+      if (memcmp(ptr, common_trigrams + i, 3) == 0) {
+        common_matched++;
+        break;
+      }
+    }
+    ptr++;
+  }
+
+  return INT2NUM(common_matched);
+}
+
+static VALUE text_clean(VALUE self, VALUE text) {
+    rb_str_modify(text);
+
+    char* ctext = StringValueCStr(text);
+    size_t new_length = text_clean_cstr(ctext);
+
+    rb_str_set_len(text, (long)new_length);
+
+    return text;
+}
+
+extern "C"
+void Init_wordtree() {
+    VALUE rb_mWordTree = rb_define_module("WordTree");
+    VALUE rb_mText = rb_define_module_under(rb_mWordTree, "Text");
+
+    u8_enc = rb_utf8_encoding();
+    bin_enc = rb_ascii8bit_encoding();
+
+    rb_define_module_function(rb_mText, "clean", RUBY_METHOD_FUNC(text_clean), 1);
+    rb_define_module_function(rb_mText, "common_trigrams", RUBY_METHOD_FUNC(text_common_trigrams), 1);
+}

data/lib/wordtree/archdown.rb

@@ -23,10 +23,8 @@ module WordTree
 
     def content_for(archivist_book)
       [archivist_book.download, nil]
-    rescue Archivist::Model::Document::UnsupportedFormat => e
-      [nil, e.to_s]
-    rescue StandardError => e
-      [nil, e.to_s]
+    rescue StandardError, Archivist::Model::Document::UnsupportedFormat => e
+      [nil, e]
     end
 
     def download_all(search_terms, &each_book)

data/lib/wordtree/book.rb

@@ -1,7 +1,8 @@
 require 'virtus'
 require 'simhash'
+require 'set'
 
-require 'wordtree/text_utils'
+require 'wordtree/text'
 
 module WordTree
   class Book
@@ -18,12 +19,12 @@ module WordTree
     attribute :size_bytes, Integer, :default => :content_size
     # A simhash (locality-sensitive hash) of the content
     attribute :simhash, Integer
+    attribute :ngrams_counted, Set
 
     attribute :content, String
 
     def initialize(*args)
       super
-      @ngrams = {}
     end
 
     def self.create(id, metadata, content)
@@ -38,44 +39,17 @@ module WordTree
       attributes.select{ |k,v| !v.nil? && k != :content }
     end
 
-    def content_clean(wrap=120)
-      if @content_clean_wrap != wrap
-        # Memoize content_clean (using last wrap size)
-        @content_clean_wrap = wrap
-        @content_clean = TextUtils.clean_text(content, wrap)
-      end
-      @content_clean
+    # Modify and clean content in-place (slightly faster)
+    def content_clean!
+      WordTree::Text.clean(content)
     end
 
-    def content_size
-      content ? content.size : nil
-    end
-
-    def each_ngram(n=1, &block)
-      TextUtils.each_ngram(content_clean, n, &block)
-    end
-
-    def set_ngrams(n, lookup)
-      raise ArgumentError, "must be a Hash" unless lookup.is_a?(Hash)
-      @ngrams[n] = lookup
+    def content_clean
+      @content_clean ||= WordTree::Text.clean(content.dup)
     end
 
-    def ngrams(n=1)
-      # Memoize ngram counts
-      @ngrams[n] ||= count_ngrams(n)
-    end
-
-    def all_ngrams
-      @ngrams
-    end
-
-    def count_ngrams(n=1)
-      {}.tap do |tally|
-        each_ngram(n) do |ngram|
-          tally[ngram] ||= 0
-          tally[ngram] += 1
-        end
-      end
+    def content_size
+      content ? content.size : nil
     end
 
     def calculate_simhash

data/lib/wordtree/book_list.rb

@@ -0,0 +1,38 @@
+module WordTree
+  class BookList
+    include Enumerable
+
+    # can be initialized from the following sources:
+    # - a WordTree::Disk::Library object
+    # - an open File object (containing a list of files or paths to books)
+    # - a String directory (presumed to be the library on disk)
+    # - a String file (containing a list of files or paths to books)
+    def initialize(source)
+      @source = source
+      @iterable = iterable_from_source(source)
+    end
+
+    def iterable_from_source(source)
+      case source
+      when WordTree::Disk::Library then
+        source
+      when File then
+        source.read.split("\n").tap do |file|
+          file.close
+        end
+      when String then
+        if File.directory?(source)
+          WordTree::Disk::Library.new(source)
+        elsif File.exist?(source)
+          IO.read(source).split("\n")
+        else
+          raise Errno::ENOENT, "Unable to find source for BookList, #{source.inspect}"
+        end
+      end
+    end
+
+    def each(&block)
+      @iterable.each(&block)
+    end
+  end
+end

data/lib/wordtree/disk/librarian.rb

@@ -10,6 +10,8 @@ module WordTree
 
       attr_reader :library
 
+      MissingContent = Class.new(StandardError)
+
       # @library can be either a string (the path of the library) or a
       # WordTree::Disk::Library object
       def initialize(library)
@@ -20,7 +22,7 @@ module WordTree
         end
       end
 
-      def find_without_ngrams(book_id)
+      def find(book_id)
         begin
           retrieved = Preamble.load(library.path_to(book_id), :external_encoding => "utf-8")
           Book.create(book_id, retrieved.metadata, retrieved.content)
@@ -29,45 +31,17 @@ module WordTree
         end
       end
 
-      def find(book_id)
-        find_without_ngrams(book_id).tap do |book|
-          (1..9).each do |n|
-            path = library.path_to(book_id, :ngrams, :n => n)
-            if File.exist?(path)
-              File.open(path) do |f|
-                hash = JSON.load(f)
-                book.set_ngrams(n, hash)
-              end
-            end
-          end
-        end
-      end
-
       def each(file_suffix_re=/\.(md|txt)$/, &block)
-        library.each(file_suffix_re) do |path|
+        library.each_with_id(file_suffix_re) do |path, id|
           retrieved = Preamble.load(path, :external_encoding => "utf-8")
           yield Book.new(retrieved.metadata.merge("content" => retrieved.content))
         end
       end
 
-      def save_without_ngrams(book)
-        library.mkdir(book.id)
-        Preamble.new(book.metadata, book.content || "").save(library.path_to(book.id))
-      end
-
-      def save_ngrams(book)
-        book.all_ngrams.each_pair do |n, hash|
-          path = library.path_to(book.id, :ngrams, :n => n)
-          File.open(path, "w") do |file|
-            file.write hash.to_json
-          end
-        end
-      end
-
       def save(book)
-        save_without_ngrams(book).tap do
-          save_ngrams(book)
-        end
+        library.mkdir(book.id)
+        raise MissingContent, "book #{book.id} is missing content" unless book.content
+        Preamble.new(book.metadata, book.content).save(library.path_to(book.id))
       end
 
       def archive_org_get(*book_ids, &block)
@@ -92,6 +66,7 @@ module WordTree
             if failure
               #TODO: logging
               $stderr.puts "Unable to download from archive.org: #{failure}"
+              raise failure
             else
               book = Book.create(metadata["archive_org_id"], metadata, content)
               save(book)

data/lib/wordtree/disk/library.rb

@@ -45,7 +45,7 @@ module WordTree
       end
 
       # Breadth-first search of the directory structure, operating on each book
-      def each(file_suffix_re=/\.(md|txt)$/, &block)
+      def each_with_id(file_suffix_re=/\.(md|txt)$/, &block)
         Find.find(@root) do |path|
           if FileTest.directory?(path)
             if File.basename(path)[0] == ?.
@@ -60,6 +60,10 @@ module WordTree
         end
       end
 
+      def each(&block)
+        each_with_id{ |path, id| yield path }
+      end
+
     end
   end
 end

data/lib/wordtree/ngrams.rb

@@ -0,0 +1,12 @@
+module WordTree
+  class Ngrams
+    def initialize
+      @trie = Trie.new
+    end
+
+    def inc(ngram)
+      value = @trie.get(ngram) || 0
+      @trie.set(ngram, value + 1)
+    end
+  end
+end

data/lib/wordtree/text.rb

@@ -0,0 +1,37 @@
+require 'strscan'
+require_relative "../../ext/wordtree"
+
+module WordTree
+  module Text
+    def self.split_near(text, split_index)
+      if split_index >= text.size
+        return [text, ""]
+      else
+        index = split_index
+        while index >= 0
+          if text[index] == ' '
+            return [text[0...index], text[(index+1)..-1]]
+          end
+          index -= 1
+        end
+        return [text[0...split_index], text[split_index..-1]]
+      end
+    end
+
+    # Remove punctuation an non-alphabetical characters from a text, and return
+    # a cleaned-up version wrapped at +wrap+ characters per line.
+    def self.word_wrap(input, wrap=120)
+      output_line = String.new
+      wrapped_output = String.new
+      begin
+        output_line, remainder = split_near(input, wrap)
+        wrapped_output << output_line + "\n"
+        output = remainder
+      end while remainder.size > wrap
+      wrapped_output << remainder + "\n" unless remainder.empty?
+
+      return wrapped_output
+    end
+
+  end
+end

data/lib/wordtree/version.rb

@@ -1,3 +1,3 @@
-module Wordtree
-  VERSION = "0.3.1"
+module WordTree
+  VERSION = "0.4.0"
 end