autoloading eval-when forms
[bpt/emacs.git] / lisp / emacs-lisp / chart.el
CommitLineData
5db81e33 1;;; chart.el --- Draw charts (bar charts, etc) -*- lexical-binding: t -*-
6dd12ef2 2
ba318903
PE
3;; Copyright (C) 1996, 1998-1999, 2001, 2004-2005, 2007-2014 Free
4;; Software Foundation, Inc.
6dd12ef2
CY
5
6;; Author: Eric M. Ludlam <zappo@gnu.org>
7;; Version: 0.2
8;; Keywords: OO, chart, graph
9
10;; This file is part of GNU Emacs.
11
12;; GNU Emacs is free software: you can redistribute it and/or modify
13;; it under the terms of the GNU General Public License as published by
14;; the Free Software Foundation, either version 3 of the License, or
15;; (at your option) any later version.
16
17;; GNU Emacs is distributed in the hope that it will be useful,
18;; but WITHOUT ANY WARRANTY; without even the implied warranty of
19;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20;; GNU General Public License for more details.
21
22;; You should have received a copy of the GNU General Public License
23;; along with GNU Emacs. If not, see <http://www.gnu.org/licenses/>.
24
25;;; Commentary:
26;;
27;; This package is an experiment of mine aiding in the debugging of
28;; eieio, and proved to be neat enough that others may like to use
29;; it. To quickly see what you can do with chart, run the command
30;; `chart-test-it-all'.
31;;
32;; Chart current can display bar-charts in either of two
33;; directions. It also supports ranged (integer) axis, and axis
34;; defined by some set of strings or names. These name can be
35;; automatically derived from data sequences, which are just lists of
36;; anything encapsulated in a nice eieio object.
37;;
38;; Current example apps for chart can be accessed via these commands:
39;; `chart-file-count' - count files w/ matching extensions
40;; `chart-space-usage' - display space used by files/directories
41;; `chart-emacs-storage' - Emacs storage units used/free (garbage-collect)
42;; `chart-emacs-lists' - length of Emacs lists
43;; `chart-rmail-from' - who sends you the most mail (in -summary only)
44;;
45;; Customization:
46;;
47;; If you find the default colors and pixmaps unpleasant, or too
48;; short, you can change them. The variable `chart-face-color-list'
49;; contains a list of colors, and `chart-face-pixmap-list' contains
50;; all the pixmaps to use. The current pixmaps are those found on
51;; several systems I found. The two lists should be the same length,
52;; as the long list will just be truncated.
53;;
54;; If you would like to draw your own stipples, simply create some
55;; xbm's and put them in a directory, then you can add:
56;;
57;; (setq x-bitmap-file-path (cons "~/mybitmaps" x-bitmap-file-path))
58;;
59;; to your .emacs (or wherever) and load the `chart-face-pixmap-list'
60;; with all the bitmaps you want to use.
61
62(require 'eieio)
63
64;;; Code:
4d789d84 65(define-obsolete-variable-alias 'chart-map 'chart-mode-map "24.1")
e5bd0a28 66(defvar chart-mode-map (make-sparse-keymap) "Keymap used in chart mode.")
6dd12ef2
CY
67
68(defvar chart-local-object nil
69 "Local variable containing the locally displayed chart object.")
70(make-variable-buffer-local 'chart-local-object)
71
6dd12ef2
CY
72(defvar chart-face-color-list '("red" "green" "blue"
73 "cyan" "yellow" "purple")
74 "Colors to use when generating `chart-face-list'.
75Colors will be the background color.")
76
77(defvar chart-face-pixmap-list
78 (if (and (fboundp 'display-graphic-p)
79 (display-graphic-p))
80 '("dimple1" "scales" "dot" "cross_weave" "boxes" "dimple3"))
81 "If pixmaps are allowed, display these background pixmaps.
a8f316ca 82Useful if new Emacs is used on B&W display.")
6dd12ef2
CY
83
84(defcustom chart-face-use-pixmaps nil
fb7ada5f 85 "Non-nil to use fancy pixmaps in the background of chart face colors."
6dd12ef2
CY
86 :group 'eieio
87 :type 'boolean)
88
7bb3e768
GM
89(declare-function x-display-color-cells "xfns.c" (&optional terminal))
90
4d789d84 91(defvar chart-face-list
7bb3e768 92 (if (display-color-p)
4d789d84
SM
93 (let ((cl chart-face-color-list)
94 (pl chart-face-pixmap-list)
95 (faces ())
96 nf)
97 (while cl
98 (setq nf (make-face
99 (intern (concat "chart-" (car cl) "-" (car pl)))))
100 (set-face-background nf (if (condition-case nil
101 (> (x-display-color-cells) 4)
102 (error t))
103 (car cl)
104 "white"))
105 (set-face-foreground nf "black")
106 (if (and chart-face-use-pixmaps
107 pl
108 (fboundp 'set-face-background-pixmap))
109 (condition-case nil
110 (set-face-background-pixmap nf (car pl))
111 (error (message "Cannot set background pixmap %s" (car pl)))))
112 (push nf faces)
113 (setq cl (cdr cl)
114 pl (cdr pl)))
115 faces))
116 "Faces used to colorize charts.
117List is limited currently, which is ok since you really can't display
118too much in text characters anyways.")
119
120(define-derived-mode chart-mode fundamental-mode "CHART"
6dd12ef2 121 "Define a mode in Emacs for displaying a chart."
6dd12ef2
CY
122 (buffer-disable-undo)
123 (set (make-local-variable 'font-lock-global-modes) nil)
4d789d84 124 (font-lock-mode -1) ;Isn't it off already? --Stef
6dd12ef2
CY
125 )
126
127(defun chart-new-buffer (obj)
128 "Create a new buffer NAME in which the chart OBJ is displayed.
a8f316ca 129Returns the newly created buffer."
9a529312 130 (with-current-buffer (get-buffer-create (format "*%s*" (oref obj title)))
6dd12ef2
CY
131 (chart-mode)
132 (setq chart-local-object obj)
133 (current-buffer)))
134
135(defclass chart ()
136 ((title :initarg :title
137 :initform "Emacs Chart")
138 (title-face :initarg :title-face
139 :initform 'bold-italic)
140 (x-axis :initarg :x-axis
141 :initform nil )
142 (x-margin :initarg :x-margin
143 :initform 5)
144 (x-width :initarg :x-width
145 )
146 (y-axis :initarg :y-axis
147 :initform nil)
148 (y-margin :initarg :y-margin
149 :initform 5)
150 (y-width :initarg :y-width
151 )
152 (key-label :initarg :key-label
153 :initform "Key")
154 (sequences :initarg :sequences
155 :initform nil)
156 )
a8f316ca 157 "Superclass for all charts to be displayed in an Emacs buffer.")
6dd12ef2 158
5db81e33 159(defmethod initialize-instance :AFTER ((obj chart) &rest _fields)
6dd12ef2
CY
160 "Initialize the chart OBJ being created with FIELDS.
161Make sure the width/height is correct."
162 (oset obj x-width (- (window-width) 10))
163 (oset obj y-width (- (window-height) 12)))
164
165(defclass chart-axis ()
166 ((name :initarg :name
167 :initform "Generic Axis")
168 (loweredge :initarg :loweredge
169 :initform t)
170 (name-face :initarg :name-face
171 :initform 'bold)
53964682 172 (labels-face :initarg :labels-face
6dd12ef2
CY
173 :initform 'italic)
174 (chart :initarg :chart
175 :initform nil)
176 )
177 "Superclass used for display of an axis.")
178
179(defclass chart-axis-range (chart-axis)
180 ((bounds :initarg :bounds
181 :initform '(0.0 . 50.0))
182 )
a8f316ca 183 "Class used to display an axis defined by a range of values.")
6dd12ef2
CY
184
185(defclass chart-axis-names (chart-axis)
186 ((items :initarg :items
187 :initform nil)
188 )
a8f316ca 189 "Class used to display an axis which represents different named items.")
6dd12ef2
CY
190
191(defclass chart-sequece ()
192 ((data :initarg :data
193 :initform nil)
194 (name :initarg :name
195 :initform "Data")
196 )
a8f316ca 197 "Class used for all data in different charts.")
6dd12ef2
CY
198
199(defclass chart-bar (chart)
200 ((direction :initarg :direction
201 :initform vertical))
a8f316ca 202 "Subclass for bar charts (vertical or horizontal).")
6dd12ef2
CY
203
204(defmethod chart-draw ((c chart) &optional buff)
205 "Start drawing a chart object C in optional BUFF.
a8f316ca 206Erases current contents of buffer."
6dd12ef2
CY
207 (save-excursion
208 (if buff (set-buffer buff))
209 (erase-buffer)
210 (insert (make-string 100 ?\n))
211 ;; Start by displaying the axis
212 (chart-draw-axis c)
213 ;; Display title
214 (chart-draw-title c)
215 ;; Display data
216 (message "Rendering chart...")
217 (sit-for 0)
218 (chart-draw-data c)
219 ;; Display key
220 ; (chart-draw-key c)
221 (message "Rendering chart...done")
222 ))
223
224(defmethod chart-draw-title ((c chart))
225 "Draw a title upon the chart.
226Argument C is the chart object."
227 (chart-display-label (oref c title) 'horizontal 0 0 (window-width)
228 (oref c title-face)))
229
230(defmethod chart-size-in-dir ((c chart) dir)
231 "Return the physical size of chart C in direction DIR."
232 (if (eq dir 'vertical)
233 (oref c y-width)
234 (oref c x-width)))
235
236(defmethod chart-draw-axis ((c chart))
237 "Draw axis into the current buffer defined by chart C."
238 (let ((ymarg (oref c y-margin))
239 (xmarg (oref c x-margin))
240 (ylen (oref c y-width))
241 (xlen (oref c x-width)))
242 (chart-axis-draw (oref c y-axis) 'vertical ymarg
243 (if (oref (oref c y-axis) loweredge) nil xlen)
244 xmarg (+ xmarg ylen))
245 (chart-axis-draw (oref c x-axis) 'horizontal xmarg
246 (if (oref (oref c x-axis) loweredge) nil ylen)
247 ymarg (+ ymarg xlen)))
248 )
249
250(defmethod chart-axis-draw ((a chart-axis) &optional dir margin zone start end)
a8f316ca 251 "Draw some axis for A in direction DIR with MARGIN in boundary.
6dd12ef2
CY
252ZONE is a zone specification.
253START and END represent the boundary."
254 (chart-draw-line dir (+ margin (if zone zone 0)) start end)
255 (chart-display-label (oref a name) dir (if zone (+ zone margin 3)
256 (if (eq dir 'horizontal)
257 1 0))
258 start end (oref a name-face)))
259
260(defmethod chart-translate-xpos ((c chart) x)
261 "Translate in chart C the coordinate X into a screen column."
262 (let ((range (oref (oref c x-axis) bounds)))
263 (+ (oref c x-margin)
264 (round (* (float (- x (car range)))
265 (/ (float (oref c x-width))
266 (float (- (cdr range) (car range))))))))
267 )
268
269(defmethod chart-translate-ypos ((c chart) y)
270 "Translate in chart C the coordinate Y into a screen row."
271 (let ((range (oref (oref c y-axis) bounds)))
272 (+ (oref c x-margin)
273 (- (oref c y-width)
274 (round (* (float (- y (car range)))
275 (/ (float (oref c y-width))
276 (float (- (cdr range) (car range)))))))))
277 )
278
5db81e33 279(defmethod chart-axis-draw ((a chart-axis-range) &optional dir margin zone _start _end)
6dd12ef2
CY
280 "Draw axis information based upon a range to be spread along the edge.
281A is the chart to draw. DIR is the direction.
282MARGIN, ZONE, START, and END specify restrictions in chart space."
283 (call-next-method)
284 ;; We prefer about 5 spaces between each value
285 (let* ((i (car (oref a bounds)))
286 (e (cdr (oref a bounds)))
287 (z (if zone zone 0))
288 (s nil)
289 (rng (- e i))
290 ;; want to jump by units of 5 spaces or so
291 (j (/ rng (/ (chart-size-in-dir (oref a chart) dir) 4)))
292 p1)
293 (if (= j 0) (setq j 1))
294 (while (<= i e)
295 (setq s
296 (cond ((> i 999999)
297 (format "%dM" (/ i 1000000)))
298 ((> i 999)
299 (format "%dK" (/ i 1000)))
300 (t
301 (format "%d" i))))
302 (if (eq dir 'vertical)
303 (let ((x (+ (+ margin z) (if (oref a loweredge)
304 (- (length s)) 1))))
305 (if (< x 1) (setq x 1))
306 (chart-goto-xy x (chart-translate-ypos (oref a chart) i)))
307 (chart-goto-xy (chart-translate-xpos (oref a chart) i)
308 (+ margin z (if (oref a loweredge) -1 1))))
309 (setq p1 (point))
310 (insert s)
311 (chart-zap-chars (length s))
312 (put-text-property p1 (point) 'face (oref a labels-face))
313 (setq i (+ i j))))
314)
315
316(defmethod chart-translate-namezone ((c chart) n)
317 "Return a dot-pair representing a positional range for a name.
318The name in chart C of the Nth name resides.
045b9da7 319Automatically compensates for direction."
6dd12ef2
CY
320 (let* ((dir (oref c direction))
321 (w (if (eq dir 'vertical) (oref c x-width) (oref c y-width)))
322 (m (if (eq dir 'vertical) (oref c y-margin) (oref c x-margin)))
323 (ns (length
324 (oref (if (eq dir 'vertical) (oref c x-axis) (oref c y-axis))
325 items)))
326 (lpn (/ (+ 1.0 (float w)) (float ns)))
327 )
328 (cons (+ m (round (* lpn (float n))))
329 (+ m -1 (round (* lpn (+ 1.0 (float n))))))
330 ))
331
5db81e33 332(defmethod chart-axis-draw ((a chart-axis-names) &optional dir margin zone _start _end)
6dd12ef2 333 "Draw axis information based upon A range to be spread along the edge.
a8f316ca
JB
334Optional argument DIR is the direction of the chart.
335Optional arguments MARGIN, ZONE, START and END specify boundaries of the drawing."
6dd12ef2
CY
336 (call-next-method)
337 ;; We prefer about 5 spaces between each value
338 (let* ((i 0)
339 (s (oref a items))
340 (z (if zone zone 0))
341 (r nil)
342 (p nil)
343 (odd nil)
344 p1)
345 (while s
346 (setq odd (= (% (length s) 2) 1))
347 (setq r (chart-translate-namezone (oref a chart) i))
348 (if (eq dir 'vertical)
349 (setq p (/ (+ (car r) (cdr r)) 2))
350 (setq p (- (+ (car r) (/ (- (cdr r) (car r)) 2))
351 (/ (length (car s)) 2))))
352 (if (eq dir 'vertical)
353 (let ((x (+ (+ margin z) (if (oref a loweredge)
354 (- (length (car s)))
355 (length (car s))))))
356 (if (< x 1) (setq x 1))
357 (if (> (length (car s)) (1- margin))
358 (setq x (+ x margin)))
359 (chart-goto-xy x p))
360 (chart-goto-xy p (+ (+ margin z) (if (oref a loweredge)
361 (if odd -2 -1)
362 (if odd 2 1)))))
363 (setq p1 (point))
364 (insert (car s))
365 (chart-zap-chars (length (car s)))
366 (put-text-property p1 (point) 'face (oref a labels-face))
367 (setq i (+ i 1)
368 s (cdr s))))
369)
370
371(defmethod chart-draw-data ((c chart-bar))
372 "Display the data available in a bar chart C."
373 (let* ((data (oref c sequences))
374 (dir (oref c direction))
375 (odir (if (eq dir 'vertical) 'horizontal 'vertical))
376 )
377 (while data
378 (if (stringp (car (oref (car data) data)))
379 ;; skip string lists...
380 nil
381 ;; display number lists...
382 (let ((i 0)
383 (seq (oref (car data) data)))
384 (while seq
385 (let* ((rng (chart-translate-namezone c i))
386 (dp (if (eq dir 'vertical)
387 (chart-translate-ypos c (car seq))
388 (chart-translate-xpos c (car seq))))
389 (zp (if (eq dir 'vertical)
390 (chart-translate-ypos c 0)
391 (chart-translate-xpos c 0)))
392 (fc (if chart-face-list
393 (nth (% i (length chart-face-list)) chart-face-list)
394 'default))
395 )
396 (if (< dp zp)
397 (progn
398 (chart-draw-line dir (car rng) dp zp)
399 (chart-draw-line dir (cdr rng) dp zp))
400 (chart-draw-line dir (car rng) zp (1+ dp))
401 (chart-draw-line dir (cdr rng) zp (1+ dp)))
402 (if (= (car rng) (cdr rng)) nil
403 (chart-draw-line odir dp (1+ (car rng)) (cdr rng))
404 (chart-draw-line odir zp (car rng) (1+ (cdr rng))))
405 (if (< dp zp)
406 (chart-deface-rectangle dir rng (cons dp zp) fc)
407 (chart-deface-rectangle dir rng (cons zp dp) fc))
408 )
409 ;; find the bounds, and chart it!
410 ;; for now, only do one!
411 (setq i (1+ i)
412 seq (cdr seq)))))
413 (setq data (cdr data))))
414 )
415
416(defmethod chart-add-sequence ((c chart) &optional seq axis-label)
417 "Add to chart object C the sequence object SEQ.
418If AXIS-LABEL, then the axis stored in C is updated with the bounds of SEQ,
419or is created with the bounds of SEQ."
420 (if axis-label
421 (let ((axis (eieio-oref c axis-label)))
422 (if (stringp (car (oref seq data)))
423 (let ((labels (oref seq data)))
424 (if (not axis)
425 (setq axis (make-instance chart-axis-names
426 :name (oref seq name)
427 :items labels
428 :chart c))
429 (oset axis items labels)))
430 (let ((range (cons 0 1))
431 (l (oref seq data)))
432 (if (not axis)
433 (setq axis (make-instance chart-axis-range
434 :name (oref seq name)
435 :chart c)))
436 (while l
437 (if (< (car l) (car range)) (setcar range (car l)))
438 (if (> (car l) (cdr range)) (setcdr range (car l)))
439 (setq l (cdr l)))
440 (oset axis bounds range)))
441 (if (eq axis-label 'x-axis) (oset axis loweredge nil))
442 (eieio-oset c axis-label axis)
443 ))
444 (oset c sequences (append (oref c sequences) (list seq))))
445
446;;; Charting optimizers
447
448(defmethod chart-trim ((c chart) max)
449 "Trim all sequences in chart C to be at most MAX elements long."
450 (let ((s (oref c sequences)))
451 (while s
452 (let ((sl (oref (car s) data)))
453 (if (> (length sl) max)
454 (setcdr (nthcdr (1- max) sl) nil)))
455 (setq s (cdr s))))
456 )
457
458(defmethod chart-sort ((c chart) pred)
459 "Sort the data in chart C using predicate PRED.
a8f316ca 460See `chart-sort-matchlist' for more details."
6dd12ef2
CY
461 (let* ((sl (oref c sequences))
462 (s1 (car sl))
463 (s2 (car (cdr sl)))
464 (s nil))
465 (if (stringp (car (oref s1 data)))
466 (progn
467 (chart-sort-matchlist s1 s2 pred)
468 (setq s (oref s1 data)))
469 (if (stringp (car (oref s2 data)))
470 (progn
471 (chart-sort-matchlist s2 s1 pred)
472 (setq s (oref s2 data)))
e5d1916a 473 (error "Sorting of chart %s not supported" (eieio-object-name c))))
6dd12ef2
CY
474 (if (eq (oref c direction) 'horizontal)
475 (oset (oref c y-axis) items s)
476 (oset (oref c x-axis) items s)
477 ))
478 )
479
480(defun chart-sort-matchlist (namelst numlst pred)
a8f316ca 481 "Sort NAMELST and NUMLST (both sequence objects) based on predicate PRED.
6dd12ef2 482PRED should be the equivalent of '<, except it must expect two
a8f316ca 483cons cells of the form (NAME . NUM). See `sort' for more details."
6dd12ef2
CY
484 ;; 1 - create 1 list of cons cells
485 (let ((newlist nil)
486 (alst (oref namelst data))
487 (ulst (oref numlst data)))
488 (while alst
489 ;; this is reversed, but were are sorting anyway
490 (setq newlist (cons (cons (car alst) (car ulst)) newlist))
491 (setq alst (cdr alst)
492 ulst (cdr ulst)))
493 ;; 2 - Run sort routine on it
494 (setq newlist (sort newlist pred)
495 alst nil
496 ulst nil)
497 ;; 3 - Separate the lists
498 (while newlist
499 (setq alst (cons (car (car newlist)) alst)
500 ulst (cons (cdr (car newlist)) ulst))
501 (setq newlist (cdr newlist)))
502 ;; 4 - Store them back
503 (oset namelst data (reverse alst))
504 (oset numlst data (reverse ulst))))
505
506;;; Utilities
507
508(defun chart-goto-xy (x y)
509 "Move cursor to position X Y in buffer, and add spaces and CRs if needed."
510 (let ((indent-tabs-mode nil)
511 (num (progn (goto-char (point-min)) (forward-line y))))
512 (if (and (= 0 num) (/= 0 (current-column))) (newline 1))
513 (if (eobp) (newline num))
514 (if (< x 0) (setq x 0))
515 (if (< y 0) (setq y 0))
516 ;; Now, a quicky column moveto/forceto method.
517 (or (= (move-to-column x) x)
518 (let ((p (point)))
519 (indent-to x)
520 (remove-text-properties p (point) '(face))))))
521
522(defun chart-zap-chars (n)
9ffe3f52 523 "Zap up to N chars without deleting EOLs."
6dd12ef2 524 (if (not (eobp))
e180ab9f 525 (if (< n (- (point-at-eol) (point)))
6dd12ef2 526 (delete-char n)
e180ab9f 527 (delete-region (point) (point-at-eol)))))
6dd12ef2
CY
528
529(defun chart-display-label (label dir zone start end &optional face)
530 "Display LABEL in direction DIR in column/row ZONE between START and END.
531Optional argument FACE is the property we wish to place on this text."
532 (if (eq dir 'horizontal)
533 (let (p1)
534 (chart-goto-xy (+ start (- (/ (- end start) 2) (/ (length label) 2)))
535 zone)
536 (setq p1 (point))
537 (insert label)
538 (chart-zap-chars (length label))
539 (put-text-property p1 (point) 'face face)
540 )
541 (let ((i 0)
542 (stz (+ start (- (/ (- end start) 2) (/ (length label) 2)))))
543 (while (< i (length label))
544 (chart-goto-xy zone (+ stz i))
545 (insert (aref label i))
546 (chart-zap-chars 1)
547 (put-text-property (1- (point)) (point) 'face face)
548 (setq i (1+ i))))))
549
550(defun chart-draw-line (dir zone start end)
551 "Draw a line using line-drawing characters in direction DIR.
a8f316ca 552Use column or row ZONE between START and END."
6dd12ef2
CY
553 (chart-display-label
554 (make-string (- end start) (if (eq dir 'vertical) ?| ?\-))
555 dir zone start end))
556
557(defun chart-deface-rectangle (dir r1 r2 face)
558 "Colorize a rectangle in direction DIR across range R1 by range R2.
559R1 and R2 are dotted pairs. Colorize it with FACE."
560 (let* ((range1 (if (eq dir 'vertical) r1 r2))
561 (range2 (if (eq dir 'vertical) r2 r1))
562 (y (car range2)))
563 (while (<= y (cdr range2))
564 (chart-goto-xy (car range1) y)
565 (put-text-property (point) (+ (point) (1+ (- (cdr range1) (car range1))))
566 'face face)
567 (setq y (1+ y)))))
568
569;;; Helpful `I don't want to learn eieio just now' washover functions
570
571(defun chart-bar-quickie (dir title namelst nametitle numlst numtitle
572 &optional max sort-pred)
a8f316ca 573 "Wash over the complex EIEIO stuff and create a nice bar chart.
9ffe3f52 574Create it going in direction DIR ['horizontal 'vertical] with TITLE
6dd12ef2
CY
575using a name sequence NAMELST labeled NAMETITLE with values NUMLST
576labeled NUMTITLE.
577Optional arguments:
a8f316ca 578Set the chart's max element display to MAX, and sort lists with
6dd12ef2
CY
579SORT-PRED if desired."
580 (let ((nc (make-instance chart-bar
581 :title title
582 :key-label "8-m" ; This is a text key pic
583 :direction dir
584 ))
585 (iv (eq dir 'vertical)))
586 (chart-add-sequence nc
587 (make-instance chart-sequece
588 :data namelst
589 :name nametitle)
590 (if iv 'x-axis 'y-axis))
591 (chart-add-sequence nc
592 (make-instance chart-sequece
593 :data numlst
594 :name numtitle)
595 (if iv 'y-axis 'x-axis))
596 (if sort-pred (chart-sort nc sort-pred))
597 (if (integerp max) (chart-trim nc max))
598 (switch-to-buffer (chart-new-buffer nc))
599 (chart-draw nc)))
600
601;;; Test code
602
603(defun chart-test-it-all ()
604 "Test out various charting features."
605 (interactive)
606 (chart-bar-quickie 'vertical "Test Bar Chart"
607 '( "U1" "ME2" "C3" "B4" "QT" "EZ") "Items"
608 '( 5 -10 23 20 30 -3) "Values")
609 )
610
611;;; Sample utility function
612
613(defun chart-file-count (dir)
9ffe3f52 614 "Draw a chart displaying the number of different file extensions in DIR."
6dd12ef2
CY
615 (interactive "DDirectory: ")
616 (if (not (string-match "/$" dir))
617 (setq dir (concat dir "/")))
618 (message "Collecting statistics...")
619 (let ((flst (directory-files dir nil nil t))
620 (extlst (list "<dir>"))
621 (cntlst (list 0)))
622 (while flst
623 (let* ((j (string-match "[^\\.]\\(\\.[a-zA-Z]+\\|~\\|#\\)$" (car flst)))
624 (s (if (file-accessible-directory-p (concat dir (car flst)))
625 "<dir>"
626 (if j
627 (substring (car flst) (match-beginning 1) (match-end 1))
628 nil)))
629 (m (member s extlst)))
630 (if (not s) nil
631 (if m
632 (let ((cell (nthcdr (- (length extlst) (length m)) cntlst)))
633 (setcar cell (1+ (car cell))))
634 (setq extlst (cons s extlst)
635 cntlst (cons 1 cntlst)))))
636 (setq flst (cdr flst)))
c7015153 637 ;; Let's create the chart!
6dd12ef2
CY
638 (chart-bar-quickie 'vertical "Files Extension Distribution"
639 extlst "File Extensions"
db9e401b 640 cntlst "# of occurrences"
6dd12ef2 641 10
4f91a816 642 (lambda (a b) (> (cdr a) (cdr b))))
6dd12ef2
CY
643 ))
644
645(defun chart-space-usage (d)
646 "Display a top usage chart for directory D."
647 (interactive "DDirectory: ")
648 (message "Collecting statistics...")
649 (let ((nmlst nil)
650 (cntlst nil)
651 (b (get-buffer-create " *du-tmp*")))
652 (set-buffer b)
653 (erase-buffer)
654 (insert "cd " d ";du -sk * \n")
655 (message "Running `cd %s;du -sk *'..." d)
656 (call-process-region (point-min) (point-max) shell-file-name t
657 (current-buffer) nil)
658 (goto-char (point-min))
659 (message "Scanning output ...")
660 (while (re-search-forward "^\\([0-9]+\\)[ \t]+\\([^ \n]+\\)$" nil t)
661 (let* ((nam (buffer-substring (match-beginning 2) (match-end 2)))
662 (num (buffer-substring (match-beginning 1) (match-end 1))))
663 (setq nmlst (cons nam nmlst)
664 ;; * 1000 to put it into bytes
665 cntlst (cons (* (string-to-number num) 1000) cntlst))))
666 (if (not nmlst)
667 (error "No files found!"))
668 (chart-bar-quickie 'vertical (format "Largest files in %s" d)
669 nmlst "File Name"
670 cntlst "File Size"
671 10
4f91a816 672 (lambda (a b) (> (cdr a) (cdr b))))
6dd12ef2
CY
673 ))
674
675(defun chart-emacs-storage ()
676 "Chart the current storage requirements of Emacs."
677 (interactive)
5db81e33 678 (let* ((data (garbage-collect)))
c7015153 679 ;; Let's create the chart!
6dd12ef2 680 (chart-bar-quickie 'vertical "Emacs Runtime Storage Usage"
5db81e33
SM
681 (mapcar (lambda (x) (symbol-name (car x))) data)
682 "Storage Items"
683 (mapcar (lambda (x) (* (nth 1 x) (nth 2 x)))
684 data)
685 "Bytes")))
6dd12ef2
CY
686
687(defun chart-emacs-lists ()
688 "Chart out the size of various important lists."
689 (interactive)
690 (let* ((names '("buffers" "frames" "processes" "faces"))
691 (nums (list (length (buffer-list))
692 (length (frame-list))
693 (length (process-list))
694 (length (face-list))
695 )))
696 (if (fboundp 'x-display-list)
697 (setq names (append names '("x-displays"))
698 nums (append nums (list (length (x-display-list))))))
c7015153 699 ;; Let's create the chart!
6dd12ef2
CY
700 (chart-bar-quickie 'vertical "Emacs List Size Chart"
701 names "Various Lists"
702 nums "Objects")))
703
704(defun chart-rmail-from ()
705 "If we are in an rmail summary buffer, then chart out the froms."
706 (interactive)
707 (if (not (eq major-mode 'rmail-summary-mode))
708 (error "You must invoke chart-rmail-from in an rmail summary buffer"))
709 (let ((nmlst nil)
710 (cntlst nil))
711 (save-excursion
712 (goto-char (point-min))
713 (while (re-search-forward "\\-[A-Z][a-z][a-z] +\\(\\w+\\)@\\w+" nil t)
714 (let* ((nam (buffer-substring (match-beginning 1) (match-end 1)))
715 (m (member nam nmlst)))
716 (message "Scanned username %s" nam)
717 (if m
718 (let ((cell (nthcdr (- (length nmlst) (length m)) cntlst)))
719 (setcar cell (1+ (car cell))))
720 (setq nmlst (cons nam nmlst)
721 cntlst (cons 1 cntlst))))))
db9e401b 722 (chart-bar-quickie 'vertical "Username Occurrence in RMAIL box"
6dd12ef2 723 nmlst "User Names"
db9e401b 724 cntlst "# of occurrences"
6dd12ef2 725 10
4f91a816 726 (lambda (a b) (> (cdr a) (cdr b))))
6dd12ef2
CY
727 ))
728
729
730(provide 'chart)
731
732;;; chart.el ends here