2 files changed, 77 insertions, 85 deletions

@@ -93,4 +93,24 @@ indicator in the mode-line."
              (> winum--window-count 1))
     (format winum-format (winum-get-number-string))))
 
+(defcustom acdw-modeline/word-count-modes
+  (mapcar (lambda (m) (cons m nil)) simple-modeline-word-count-modes)
+  "Alist of modes to functions that `acdw-modeline/word-count' should dispatch.
+If the cdr of the cons cell is nil, use the default function (`count-words').
+Otherwise, cdr should be a function that takes two points (see `count-words')."
+  :type '(alist :key-type (symbol :tag "Major-Mode")
+                :value-type function)
+  :group 'simple-modeline)
+
+(defun acdw-modeline/word-count ()
+  "Display a buffer word count, depending on the major mode.
+Uses `acdw-modeline/word-count-modes' to determine which function to use."
+  (when-let ((modefun
+              (assoc major-mode acdw-modeline/word-count-modes #'equal)))
+    (let ((fn (or (cdr modefun)
+                  #'count-words))
+          (min (if (region-active-p) (region-beginning) (point-min)))
+          (max (if (region-active-p) (region-end) (point-max))))
+      (format "%dW" (funcall fn min max)))))
+
 (provide 'acdw-modeline)
@@ -243,91 +243,63 @@ the deletion might narrow the column."
       (org-table-copy-down n)
     (acdw-org/return-dwim n)))
 
-(defun acdw-org/word-count (beg end
-                                &optional count-latex-macro-args?
-                                count-footnotes?)
-  "Report the number of words in the Org mode buffer or selected region.
-Ignores:
-- comments
-- tables
-- source code blocks (#+BEGIN_SRC ... #+END_SRC, and inline blocks)
-- hyperlinks (but does count words in hyperlink descriptions)
-- tags, priorities, and TODO keywords in headers
-- sections tagged as 'not for export'.
+(defun acdw-org/count-words (start end)
+  "Count words between START and END, respecting `org-mode' conventions."
+  (interactive (list nil nil))
+  (cond ((not (called-interactively-p 'any))
+         (let ((words 0))
+           (save-excursion
+             (save-restriction
+               (narrow-to-region start end)
+               (goto-char (point-min))
+               (while (< (point) (point-max))
+                 (cond
+                  ;; Ignore comments
+                  ((or (org-at-comment-p)
+                       (org-in-commented-heading-p)) nil)
+                  ;; Ignore tables
+                  ((org-at-table-p) nil)
+                  ;; Ignore hyperlinks, but count the descriptions
+                  ((looking-at org-bracket-link-analytic-regexp)
+                   (when-let ((desc (match-string-no-properties 5)))
+                     (save-match-data
+                       (setq words (+ words
+                                      (length (remove ""
+                                                      (org-split-string
+                                                       desc "\\W")))))))
+                   (goto-char (match-end 0)))
+                  ;; Ignore source code blocks
+                  ((org-in-src-block-p) nil)
+                  ;; Ignore footnotes
+                  ((or (org-footnote-at-definition-p)
+                       (org-footnote-at-reference-p))
+                   nil)
+                  ;; else... check the context
+                  (t (let ((contexts (org-context)))
+                       (cond
+                        ;; Ignore tags, TODO keywords, etc.
+                        ((or (assoc :todo-keyword contexts)
+                             (assoc :priority contexts)
+                             (assoc :keyword contexts)
+                             (assoc :checkbox contexts))
+                         nil)
+                        ;; Ignore sections tagged :no-export
+                        ((assoc :tags contexts)
+                         (if (intersection (org-get-tags-at)
+                                           org-export-exclude-tags
+                                           :test 'equal)
+                             (org-forward-same-level 1)
+                           nil))
+                        ;; else... count the word
+                        (t (setq words (1+ words)))))))
+                 (re-search-forward "\\w+\\W*")))
+             words)))
+        ((use-region-p)
+         (message "%d words in region"
+                  (acdw-org/count-words (region-beginning) (region-end))))
+        (t
+         (message "%d words in buffer"
+                  (acdw-org/count-words (point-min) (point-max))))))
 
-The text of footnote definitions is ignored, unless the optional argument
-COUNT-FOOTNOTES? is non-nil.
-
-If the optional argument COUNT-LATEX-MACRO-ARGS? is non-nil, the word count
-includes LaTeX macro arguments (the material between {curly braces}).
-Otherwise, and by default, every LaTeX macro counts as 1 word regardless
-of its arguments."
-  (interactive "r")
-  (unless mark-active
-    (setf beg (point-min)
-          end (point-max)))
-  (let ((wc 0)
-        (latex-macro-regexp "\\\\[A-Za-z]+\\(\\[[^]]*\\]\\|\\){\\([^}]*\\)}"))
-    (save-excursion
-      (goto-char beg)
-      (while (< (point) end)
-        (cond
-         ;; Ignore comments.
-         ((or (org-in-commented-line) (org-at-table-p))
-          nil)
-         ;; Ignore hyperlinks. But if link has a description, count
-         ;; the words within the description.
-         ((looking-at org-bracket-link-analytic-regexp)
-          (when (match-string-no-properties 5)
-            (let ((desc (match-string-no-properties 5)))
-              (save-match-data
-                (incf wc (length (remove "" (org-split-string
-                                             desc "\\W")))))))
-          (goto-char (match-end 0)))
-         ((looking-at org-any-link-re)
-          (goto-char (match-end 0)))
-         ;; Ignore source code blocks.
-         ((org-in-regexps-block-p "^#\\+BEGIN_SRC\\W" "^#\\+END_SRC\\W")
-          nil)
-         ;; Ignore inline source blocks, counting them as 1 word.
-         ((save-excursion
-            (backward-char)
-            (looking-at org-babel-inline-src-block-regexp))
-          (goto-char (match-end 0))
-          (setf wc (+ 2 wc)))
-         ;; Count latex macros as 1 word, ignoring their arguments.
-         ((save-excursion
-            (backward-char)
-            (looking-at latex-macro-regexp))
-          (goto-char (if count-latex-macro-args?
-                         (match-beginning 2)
-                       (match-end 0)))
-          (setf wc (+ 2 wc)))
-         ;; Ignore footnotes.
-         ((and (not count-footnotes?)
-               (or (org-footnote-at-definition-p)
-                   (org-footnote-at-reference-p)))
-          nil)
-         (t
-          (let ((contexts (org-context)))
-            (cond
-             ;; Ignore tags and TODO keywords, etc.
-             ((or (assoc :todo-keyword contexts)
-                  (assoc :priority contexts)
-                  (assoc :keyword contexts)
-                  (assoc :checkbox contexts))
-              nil)
-             ;; Ignore sections marked with tags that are
-             ;; excluded from export.
-             ((assoc :tags contexts)
-              (if (intersection (org-get-tags-at) org-export-exclude-tags
-                                :test 'equal)
-                  (org-forward-same-level 1)
-                nil))
-             (t
-              (incf wc))))))
-        (re-search-forward "\\w+\\W*")))
-    (message (format "%d words in %s." wc
-                     (if mark-active "region" "buffer")))))
 
 (provide 'acdw-org)