chiark / gitweb /
commitid.scad.pl: Break out argl_formal and argl_actual (nfc)
[reprap-play.git] / commitid.scad.pl
1 #!/usr/bin/perl -w
2
3 # commitid.scad.pl - a program for annotating solid models with commit info
4 # Copyright (C)2016 Ian Jackson.  See below.  There is NO WARRANTY.
5
6
7 # USAGE
8 # =====
9 #
10 #   .../commitid.scad.pl [OPTION...] [STRING...] >commitid.scad.new \
11 #     && mv -f commitid.scad.new commitid.scad
12 #
13 # Run without arguments, commitid.scad.pl will output an openscad file
14 # which contains 2D and 3D models of the current git commit count and
15 # commit object id (commit hash), useful for identifying printed
16 # parts.
17 #
18 # See below for details.  You probably want these two sections, as a
19 # quick starting point:
20 #    General form of provided openscad modules
21 #    Autoscaling modules
22 #
23 # We can also generate models of short mainly-numeric strings
24 # specified on the command line.
25 #
26 #
27 # Options:
28 #
29 #   --git    Generate git commit indications, as shown below
30 #            (this is the default if no strings are requested with -t).
31 #            Ie, produce the `Autoscaling modules' and `Specific layouts'.
32 #
33 #   --git=objid
34 #            Generate git commit indication based on commit object only
35 #            (ie avoid counting commits).  Ie, do not generate `Small'
36 #            and `Full' layouts (and never select them for `Best').
37 #
38 #   -i       Do not generate `+' dirty indication if git-untracked files
39 #            are present (ie, missing .gitignore entries).  The `*'
40 #            dirty tree indication (for modified files) cannot be disabled.
41 #
42 #   [-t[LAYOUT]] TEXT
43 #            Generate a layout LAYOUT containing TEXT.  TEXT can
44 #            contain newlines (a final newline usually undesirable, as
45 #            it will generate a blank line).  If LAYOUT is not specified,
46 #            generates Arg0, Arg1, Arg2, etc., for successive such
47 #            TEXTs.  The permissible character set in is TEXT is:
48 #                 space 0-9 a-f + *
49 #
50 #
51 # OPENSCAD INTERFACE
52 # ==================
53 #
54 # Dynamic variables for configuration
55 # -----------------------------------
56 #
57 # We honour the following variables to control various scaling factors:
58 #
59 #                             default value  notes
60 #    $Commitid_pixelsz         0.8             \ multiplied together
61 #    $Commitid_scale           1.0             /
62 #    $Commitid_depth           pixelsz/2       \ multiplied together
63 #    $Commitid_depth_scale     1.0             / 
64 #    $Commitid_max_best_scale  2.0             limits XY scaling in *Best*
65 #
66 # FYI the font is nominally 3x5 pixels, with 1-pixel inter-line and
67 # inter-character gaps.  (It's not strictly speaking a 3x5 bitmap
68 # font, size it contains partial pixels and diagonals.)
69 #
70 #
71 # Non-`module'-specific functions
72 # -------------------------------
73 #
74 # We provide the following functions (which depend on the config
75 # variables, but not on anything else) and compute useful values:
76 #
77 #   function Commitid_pixelsz()   Actual size of each nominal pixel
78 #   function Commitid_depth()     Depth to use (the amount characters
79 #                                  should be raised or sunken)
80 #
81 # General form of provided openscad modules
82 # -----------------------------------------
83 #
84 #   module Commitid_MODULE_2D(...)  A collection of polygons forming characters
85 #   module Commitid_MODULE(...)     The above, extruded up and down in Z
86 #   function Commitid_MODULE_sz()   A 2-vector giving the X,Y size
87 #
88 # Except for *Best* modules, the XY origin is in the bottom left
89 # corner without any margin.  Likewise Commitid_MODULE_sz does not
90 # include any margin.
91 #
92 # For 3D versions, the model is 2*depth deep and the XY plane bisects
93 # the model.  This means it's convenient to either add or subtract from
94 # a workpiece whose face is in the XY plane.
95 #
96 #
97 # Autoscaling modules
98 # -------------------
99 #
100 # These modules take a specification of the available XY space, and
101 # select and generate a suitable specific identification layout:
102
103 #   module Commitid_BestCount_2D(max_sz, margin=Commitid_pixelsz())
104 #   module Commitid_BestCount   (max_sz, margin=Commitid_pixelsz())
105 #   module Commitid_BestObjid_2D(max_sz, margin=Commitid_pixelsz())
106 #   module Commitid_BestObjid   (max_sz, margin=Commitid_pixelsz())
107 #
108 # max_sz should be [x,y].
109 #
110 # BestCount includes (as much as it can of) the git commit count,
111 # ie the result of
112 #     git rev-list --first-parent --count HEAD
113 # (and it may include some of the git revision ID too).
114 #
115 # BestObjid includes as much as it can of the git commit object hash,
116 # and never includes any of the count.
117 #
118 # All of these will autoscale and autorotate the selected model, and
119 # will include an internal margin of the specified size (by default,
120 # one pixel around each edge).  If no margin is needed, pass margin=0.
121 #
122 # There are no `function Commitid_Best*_sz'.  If they existed they
123 # would simply return max_sz.
124 #
125 #
126 # Output format
127 # -------------
128 #
129 # In general the output, although it may be over multiple lines,
130 # is always in this order
131 #     git commit object id (hash)
132 #     dirty indicator
133 #     git commit count
134 #
135 # Not all layouts have all these parts.  The commit object id may
136 # sometimes be split over multiple lines, but the count will not be.
137 # If both commit id and commit count appear they will be separated
138 # by (at least) a newline, or a dirty indicator, or a space.
139 #
140 # The commit id is truncated to fit, from the right.
141 #
142 # The commit count is truncated from the _left_, leaving the least
143 # significant decimal digits.
144 #
145 # The dirty indicator can be
146 #
147 #   *   meaning the working tree contains differences from HEAD
148 #
149 #   +   meaning the working tree contains untracked files
150 #       (ie files you have failed to `git add' and also failed
151 #       to add to gitignore).  (But see the -i option.)
152 #
153 #
154 # Specific layouts
155 # ----------------
156 #
157 # If you want to control the exact layout (and make space for it in
158 # your design), you can use these:
159 #
160 #    module Commitid_LAYOUT_2D()
161 #    module Commitid_LAYOUT()
162 #    function Commitid_LAYOUT_sz()
163 #
164 # Here LAYOUT is one of the following (giving for example, `module
165 # Commitid_Full8_2D').  In the examples, we will assume that the tree
166 # is dirty, the commit count is 123456, and the commit object id
167 # starts abcdeffedbcaabcdef...  In the examples `_' shows where a
168 # space would be printed.
169 #
170 #   Small2 Small3 ... Small10
171 #       A single line containing as much of the count will fit, eg:
172 #            Small5    3456*
173 #            Small8    _*123456
174 #       The objectid is included if more than one character of of it
175 #       will fit without makign the output ambiguous:
176 #            Small9    ab*123456
177 #
178 #   Small2S Small4S ... Small10S
179 #   Small3T Small9T
180 #       Same as Small but split into two lines (S)
181 #       or three lines (T).  Eg:
182 #            Small4S    *4       Small6T   _*
183 #                       56                 34
184 #                                          56
185 #   Git2 Git3 ... Git10
186 #   Git4S Git6S ... Git10S
187 #   Git6T Git9T
188 #       Just the commit object hash, in one, two (S) or three (T)
189 #       lines.  E.g.:
190 #            Git5    abcd*
191 #
192 #   Full4 Full6 ... Full20:
193 #       The commit object hash plus the commit count, on
194 #       separate lines, eg:
195 #            Full6    abcdef     Full8     abcdeffe
196 #                     *23456               _*123456
197 #
198 #   Full6T Full9T ... Full30T
199 #       As Full but the commit object id is split over two lines
200 #       producing a 3-line layout, eg:
201 #            Full9T    abc       Full21T   abcdeff
202 #                      de*                 edbcaa*
203 #                      456                 _123456
204 #
205 # Other LAYOUTs
206 # -------------
207 #
208 #   FontDemo
209 #
210 #       A demonstration of the built-in 18-character font
211 #
212 #   Arg0 Arg1, ...
213 #
214 #       Strings passed on command line (without -t, or bare -t,
215 #       rather than with -tLAYOUT).
216 #
217 #   LAYOUT
218 #
219 #       Generated by passing -tLAYOUT on the command line.
220 #
221
222
223 # COPYRIGHT, LICENCE AND LACK-OF-WARRANTY INFORMATION
224 # ===================================================
225 #
226 # This program is Free Software and a Free Cultural Work.
227 #
228 #   You can redistribute it and/or modify it under the terms of the
229 #   GNU General Public License as published by the Free Software
230 #   Foundation, either version 3 of the License, or (at your option)
231 #   any later version.
232 #
233 #   This program is distributed in the hope that it will be useful,
234 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
235 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
236 #   GNU General Public License for more details.
237 #
238 #   You should have received a copy of the GNU General Public License
239 #   along with this program.  If not, see <http://www.gnu.org/licenses/>.
240 #
241 # Alternatively, at your option:
242 #
243 #   This work is licensed under the Creative Commons
244 #   Attribution-ShareAlike 4.0 International License.
245 #
246 #   There is NO WARRANTY.
247
248
249 use strict;
250
251 $SIG{__WARN__} = sub { die @_; };
252
253 sub p { print @_ or die $!; }
254
255 p <<'END';
256 // *** AUTOGENERATED - DO NOT EDIT *** //
257 function Commitid_pixelsz() =
258   ($Commitid_pixelsz       ? $Commitid_pixelsz       : 0.8) *
259   ($Commitid_scale         ? $Commitid_scale         : 1.0);
260 function Commitid_depth() =
261   ($Commitid_depth         ? $Commitid_depth         : Commitid_pixelsz()/2) *
262   ($Commitid_depth_scale   ? $Commitid_depth_scale   : 1.0);
263 function Commitid__scale() =
264   Commitid_pixelsz() / 0.2;
265 END
266
267 sub chrmodname ($) {
268     my ($chr) = @_;
269     my $chrx = sprintf '%#x', ord $chr;
270     return "Commitid__chr_$chrx";
271 }
272
273 our $gtm_demo_i = -1;
274 our $gtm_demo_j;
275 our @gtm_demo_o;
276
277 sub gentextmodule_demo_start_batch () {
278     $gtm_demo_j = 0;
279     $gtm_demo_i++;
280 }
281
282 sub argl_formal (@) { join ', ', @_; }
283 sub argl_actual (@) { join ',', map { m/=/ ? $` : $_ } @_; }
284
285 sub gen3dmodule ($@) {
286     my ($modb,@argl) = (@_);
287     p "module ${modb}(".argl_formal(@argl)."){\n";
288     p " d=Commitid_depth();\n";
289     p " translate([0,0,-d]) linear_extrude(height=d*2)\n";
290     p "  ${modb}_2D(".argl_actual(@argl).");\n";
291     p "}\n";
292 }
293
294 sub gentextmodule ($@) {
295     my ($form, @lines) = @_;
296     my $modb = "Commitid_$form";
297     p "module ${modb}_2D(){\n";
298     p " // |$_|\n" foreach @lines;
299     p " scale(Commitid__scale()){\n";
300     my $y = @lines;
301     my $cols = 1;
302     foreach my $line (@lines) {
303         $y--;
304         my $x = 0;
305         foreach my $chr (split //, $line) {
306             p sprintf "  translate([%d * 0.8, %d * 1.2]) %s();\n",
307                 $x, $y, chrmodname $chr
308                 if $chr =~ m/\S/;
309             $x++;
310         }
311         $cols = $x if $x > $cols;
312     }
313     p " }\n";
314     p "}\n";
315     gen3dmodule($modb);
316
317     p sprintf "function %s_sz() = Commitid__scale() * 0.1 * [ %d, %d ];\n",
318         $modb, 2 * ($cols * 4 - 1), 2 * (@lines * 6 - 1);
319
320     push @gtm_demo_o, <<END;
321  translate([$gtm_demo_i * st[0], $gtm_demo_j * st[1]]) {
322   difference(){
323    color("blue") translate([-e,-e]) square(${modb}_sz() + 2*[e,e]);
324    square(${modb}_sz());
325   }
326   ${modb}_2D();
327 }
328 END
329     $gtm_demo_j++;
330 }
331
332 our @demo;
333
334 sub parsefont () {
335     my %cellmap;
336     for (;;) {
337         $_ = <DATA> // die;
338         last if %cellmap && !m/\S/;
339         next unless m/\S/;
340         chomp;
341         s{^(.) }{};
342         $cellmap{$1} = $_;
343     }
344     my %chrpolys;
345     while (<DATA>) {
346         next unless m/\S/;
347         chomp;
348         my @chrs = split / /, $_;
349         <DATA> !~ m/\S/ or die;
350         foreach my $row (reverse 0..4) {
351             $_ = <DATA>;
352             chomp;
353             s{^}{ };
354             $_ .= ' ' x 8;
355             m{\S/\S} and die;
356             s{/(?=\s)}{L}g;
357             s{/(?=\S)}{r}g;
358             s{\\(?=\s)}{l}g;
359             s{\\(?=\S)}{R}g;
360             p "// $_\n";
361             foreach my $chr (@chrs) {
362                 s{^ }{} or die "$chr $_ ?";
363                 foreach my $col (0..2) {
364                     my @verts;
365                     if (s{^ }{}) {
366                     } elsif (s{^\S}{}) {
367                         my $f = $cellmap{$&};
368                         die unless $f;
369                         $f =~ s/\b\d/ sprintf '%05d', $col*2000 + $&*1025 /ge;
370                         $f =~ s/\d\b/ sprintf '%05d', $row*2000 + $&*1025 /ge;
371                         push @{ $chrpolys{$chr} }, [ split / /, $f ];
372                     } else {
373                         die "$_ ?";
374                     }
375                 }
376             }
377             die "$_ ?" if m{\S};
378         }    
379     }
380
381     my $demo = '';
382     my $democols = 6;
383     foreach my $chr (sort keys %chrpolys) {
384         my $mod = chrmodname $chr;
385         p "module $mod () {\n";
386         foreach my $poly (@{ $chrpolys{$chr} }) {
387             p " polygon([";
388             my $delim = "";
389             foreach my $pt (@$poly) {
390                 p $delim;
391                 $pt =~ s{\d{5}}{$&,};
392                 $pt =~ s{\b\d}{$&.}g;
393                 p "[$pt]";
394                 $delim = ',';
395             }
396             p "]);\n";
397         }
398         p "}\n";
399         $demo .= $chr;
400     }
401     @demo = reverse $demo =~ m{.{1,$democols}}go;
402 }
403
404 parsefont();
405
406 our $do_git; # contains may chars 'c' (count) and/or 'o' (object)
407 our $do_git_untracked = 1;
408 our $argcounter;
409
410 our @forms;
411
412 sub rjustt ($$;$) { # right justify and truncate (ie, pad and truncate at left)
413                     # always includes prefix
414     my ($sz, $whole, $prefix) = @_;
415     $prefix //= '';
416     my $lw = length $whole;
417     my $spare = $sz - $lw - (length $prefix);
418     return
419         ($spare > 0 ? (' ' x $spare) : '').
420         $prefix.
421         substr($whole, ($spare < 0 ? -$spare : 0));
422 }
423
424 sub ljustt ($$;$) {
425     my ($sz, $whole, $suffix) = @_;
426     $suffix //= '';
427     $sz -= length $suffix;
428     return sprintf "%-${sz}.${sz}s%s", $whole, $suffix;
429 }
430
431 sub genform ($@) {
432     my ($form, @lines) = @_;
433     gentextmodule($form, @lines);
434     my $f = {
435         Form => $form,
436         Chars => (length join '', @lines),
437         Lines => (scalar @lines),
438         Ambiguous => ($form =~ m/Full/ && !grep { m/\W/ } @lines),
439     };
440     push @forms, $f;
441 }
442
443 sub genform_q ($$$) {
444     my ($form, $s, $lines) = @_;
445     $gtm_demo_j++;
446     my $l = length $s;
447     return if $l % $lines;
448     my $e = $l/$lines;
449     return if $e < 2;
450     $gtm_demo_j--;
451     genform($form, $s =~ m/.{$e}/g);
452 }
453
454 sub genform_plusq ($$) {
455     my ($form, $s) = @_;
456     genform($form, $s);
457     genform_q("${form}S", $s, 2);
458     genform_q("${form}T", $s, 3);
459 }
460
461 our @gcmd;
462
463 sub gitrun_start () {
464     open F, "-|", @gcmd or die "$gcmd[0]: start: $!";
465 }
466
467 sub gitrun_done (;$) {
468     my ($errok) = @_;
469     $?=0; $!=0;
470     return if close F;
471     return if $errok;
472     die $! if $!;
473     die "@gcmd failed ($?)\n";
474 }
475
476 sub gitoutput (@) {
477     (@gcmd) = (qw(git), @_);
478     gitrun_start;
479     $_ = <F>;
480     gitrun_done;
481     defined or die "@gcmd produced no output";
482     chomp or die "@gcmd produced no final newline";
483     $_;
484 }
485
486 sub do_git () {
487     return unless $do_git;
488
489     @gcmd = qw(git status --porcelain);
490     push @gcmd, qw(--untracked=no) unless $do_git_untracked;
491
492     my $git_dirty = '';
493     gitrun_start;
494     while (<F>) {
495         if (m/^\?\?/ && $do_git_untracked) {
496             $git_dirty = '+';
497             next;
498         }
499         $git_dirty = '*';
500         last;
501     }
502     gitrun_done($git_dirty eq '*');
503
504     my $git_count;
505     my $git_object;
506
507     if ($do_git =~ m/c/) {
508         $git_count = gitoutput qw(rev-list --first-parent --count HEAD);
509     }
510     if ($do_git =~ m/o/) {
511         $git_object = gitoutput qw(rev-parse HEAD);
512     }
513     print STDERR join ' ', map { $_ // '?' }
514         "-- commitid", $git_object, $git_dirty, $git_count, "--\n";
515
516     foreach my $sz (2..10) {
517         gentextmodule_demo_start_batch();
518
519         if (defined($git_count)) {
520             my $smallstr = rjustt($sz, $git_count, $git_dirty);
521             if (defined($git_object) && $sz >= length($git_count) + 3) {
522                 $smallstr = $git_object;
523                 $smallstr .= ($git_dirty || ' ');
524                 $smallstr .= $git_count;
525                 $smallstr = rjustt($sz, $smallstr);
526             }
527             genform_plusq("Small$sz", $smallstr);
528         }
529
530         genform_plusq("Git$sz", ljustt($sz, $git_object, $git_dirty))
531             if defined $git_object;
532
533         if (defined $git_count && defined $git_object) {
534             genform("Full".($sz*2),
535                     ljustt($sz, $git_object),
536                     rjustt($sz, $git_count, $git_dirty));
537
538             my $e = $sz;
539             genform("Full".($e*3)."T",
540                     ljustt($e*2, $git_object, $git_dirty)
541                     =~ m/.{$e}/g,
542                     rjustt($e, $git_count));
543         }
544     }
545 }
546
547 sub do_some_best ($$) {
548     my ($modname, $formre) = @_;
549     my $fullmodname = "Commitid_${modname}_2D";
550     my @argl = qw(max_sz margin=Commitid_pixelsz());
551     p "module $fullmodname(".argl_formal(@argl).") {\n";
552     my $mbs = '$Commitid_max_best_scale';
553     p " sc_max = $mbs ? $mbs : 2;\n";
554     p " sz = max_sz - 2*[margin,margin];\n";
555     my @do;
556     foreach my $f (
557         sort {
558             $b->{Chars} <=> $a->{Chars} or
559             $a->{Lines} <=> $b->{Chars}
560         }
561         grep {
562             $_->{Form} =~ m/$formre/ &&
563             !$_->{Ambiguous}
564         }
565         @forms
566     ) {
567         my $form = $f->{Form};
568         p " sz_$form = Commitid_${form}_sz();\n";
569         foreach my $rot (qw(0 1)) {
570             my $id = "${form}_r${rot}";
571             p " sc_$id = min(sc_max";
572             foreach my $xy (qw(0 1)) {
573                 p ",sz[$xy]/sz_$form","[",(($xy xor $rot)+0),"]";
574             }
575             p ");\n";
576             push @do, " if (sc_$id >= 1.0";
577             push @do, " && sc_$id >= sc_${form}_r1" if !$rot;
578             push @do, ") {\n";
579             push @do, "  translate([margin,margin]) scale(sc_$id)\n";
580             push @do, "   rotate(90) translate([0,-sz_$form"."[1]])\n" if $rot;
581             push @do, "   Commitid_${form}_2D();\n";
582             push @do, " } else";
583         }
584     }
585     push @do, <<END;
586  {
587   echo("$fullmodname could not fit anything in", max_sz);
588  }
589 END
590     p $_ foreach @do;
591     p "}\n";
592
593     gen3dmodule "Commitid_$modname", @argl;
594 }
595
596 sub do_git_best () {
597     return unless $do_git;
598
599     # Auto-computer for `best fit'
600     #
601     # We have two best fit approaches: with count, and git-object-id-only
602     #
603     # For `with count', we only ever include the git object id if the
604     # result would be unambigous.  That means that at least one space
605     # or punctuation was generated.
606     #
607     # We sort the options by firstly number of characters
608     # (decreasing), and then by number of lines (increasing) and
609     # try each one both ways round.
610
611     do_some_best('BestCount', 'Small|Full') if $do_git =~ m/c/;
612     do_some_best('BestObjid', 'Git|Full') if $do_git =~ m/o/;
613 }
614
615 while (@ARGV) {
616     $_ = shift;
617     if (m/^--(no)?-git$/) {
618         $do_git = $1 ? '' : 'co';
619     } elsif (m/^---git=objid$/i) {
620         $do_git = 'o';
621     } elsif (m/^-i$/) {
622         $do_git_untracked = 0;
623     } elsif (m/^-t(.*)$/) {
624         my $form = $1;
625         die "bad usage: -t needs string argument\n";
626         $_ = shift;
627         gentextmodule($form, split /\n/, $_);
628         $argcounter //= 0;
629     } elsif (m/^[^-]/) {
630         gentextmodule("Arg$argcounter", $_);
631         $argcounter++;
632     } else {
633         die "bad usage: unknown option \`$_'\n";
634     }
635 }
636
637 $do_git //= defined($argcounter) ? '' : 'co';
638
639 gentextmodule_demo_start_batch();
640 gentextmodule('FontDemo', @demo);
641
642 do_git();
643 do_git_best();
644
645 p "module Commitid_2DDemo(){\n";
646 p " st = Commitid__scale() * [ 10, 5 ];\n";
647 p " e  = Commitid_pixelsz();\n";
648 p $_ foreach @gtm_demo_o;
649 p "}\n";
650
651 flush STDOUT or die $!;
652 close STDOUT or die $!;
653
654 __DATA__
655
656 # 00 20 22 02
657 l 00 20 02
658 r 00 20 22
659 L 00 22 02
660 R 20 22 02
661 > 00 20 22 02 11
662 < 00 20 11 22 02
663
664 0 1 2 3 4 5 6 7 8 9
665
666 /#\  r  /#\ ##\ # # ### /#/ ### /#\ /#\
667 # # /#    #   # # # #   #     # # # # #
668 # #  #  /#/ ##< \## ##\ ##\  // >#< \##
669 # #  #  #     #   #   # # #  #  # #   #
670 \#/ /#\ ### ##/   # ##/ \#/  #  \#/ ##/
671
672 a b c d e f
673
674     #         #     /##
675     #   /##   # /#\ #
676 /## ##\ #   /## #r# ###
677 # # # # #   # # #/  #
678 \## ##/ \## \## \#/ #
679
680 + *
681
682     # #
683  #  \#/
684 ### ###
685  #  /#\
686     # #