distort-stl: Break out project_cylinder_triangle_need_subdivide

[reprap-play.git] / distort-stl

diff --git a/distort-stl b/distort-stl
index 4820e03104d51485c3d7b00a9b3f945aa1f05961..cc03ec5e7aea0776a516e12f900159adbdb2fb6f 100755 (executable)
--- a/distort-stl
+++ b/distort-stl
@@ -18,62 +18,209 @@
 #       polar-rectangular coords:
 #          Z' = Z; R' = Y + RADIUS; theta' = X / RADIUS
 #       and then converts back into cartesian
+#       honours fa but not fs or fn
+#
+#   set-fa $FA
 
 use strict;
 use autodie;
 
-use File::Temp;
+use List::Util;
+use POSIX;
+use File::Temp ();
+use Data::Dumper;
+
+sub TAU () { M_PI * 2; }
+
+our $fa = 10;
+
+our $triangles;
+our $output;
 
 sub shift_arg () {
     die unless @ARGV;
     scalar shift @ARGV;
 }
 
-sub pointmap_project_cylinder {
-    my ($x,$y,$z) = @_;
-    my $radius = shift_arg;
-    my $r = $radius - $y;
-    my $theta = $x / $radius;
-    return ($r * sin($theta),
-           -$r * cos($theta),
-           $z);
+#no warnings qw(recursion);
+
+sub subdivide_triangle ($$) {
+    my ($t, $fn) = @_;
+
+    #print STDERR 'SUBDIV', Dumper($t, $fn);
+
+    my @mids;
+    foreach my $ix (0..2) {
+       my $jx = ($ix+1) % 3;
+       my @midp;
+       foreach my $ci (0..2) {
+           push @midp, 0.5 * ($t->[$ix][$ci] + $t->[$jx][$ci]);
+       }
+       push @mids, \@midp;
+    }
+    foreach my $ix (0..2) {
+       #print STDERR 'SUBDIV IX ', $ix, "\n";
+       my $kx = ($ix+2) % 3;
+       $fn->([ $t->[$ix], $mids[$ix], $mids[$kx] ]);
+    }
+    #print STDERR 'SUBDIV MID\n';
+    $fn->(\@mids);
+}
+
+sub append_triangle ($) {
+    my ($t) = @_;
+    push @$output, $t;
+}
+
+#---------- project-cylinder ----------
+
+our $project_cylinder_radius;
+our $project_cylinder_max_d_theta;
+
+sub project_cylinder_triangle_need_subdivide ($) {
+    my ($t) = @_;
+    my @thetas = map { $_->[0] / $project_cylinder_radius } @$t;
+
+    foreach my $ix (0..2) {
+       if (abs($thetas[$ix] - $thetas[($ix+1)%3])
+           > $project_cylinder_max_d_theta) {
+           return 1;
+       }
+    }
+    return 0;
+}
+
+sub project_cylinder_need_subdivide () {
+    foreach my $t (@$triangles) {
+       next unless project_cylinder_triangle_need_subdivide $t;
+       return 1;
+    }
+    return 0;
+}
+
+sub project_cylinder_tri {
+    my ($t) = @_;
+
+    #print STDERR 'PROJECT', Dumper($t);
+
+    my $radius = $project_cylinder_radius;
+
+    my @ot;
+    foreach my $p (@$t) {
+       my ($x,$y,$z) = @$p;
+       my $r = $radius - $y;
+       my $theta = $x / $radius;
+       push @ot, [ $r * sin($theta),
+                   -$r * cos($theta),
+                   $z ];
+    }
+    append_triangle \@ot;
 }
 
-if (@ARGV && $ARGV[0] =~ m/^-/) {
-    die "no options supported\n";
+sub op__project_cylinder () {
+    $project_cylinder_radius = shift_arg;
+    $project_cylinder_max_d_theta = $fa * TAU/360;
+
+    while (project_cylinder_need_subdivide()) {
+       $output = [];
+       foreach my $t (@$triangles) {
+           subdivide_triangle $t, \&append_triangle;
+       }
+       $triangles = $output;
+    }
+
+    $output = [];
+    foreach my $t (@$triangles) {
+       project_cylinder_tri $t;
+    }
+    $triangles = $output;
 }
 
-my $itmp = new File::Temp;
-my $otmp = new File::Temp;
+#---------- main program ----------
+
+our $raw;
+
+while (@ARGV && $ARGV[0] =~ m/^-/) {
+    $_ = shift @ARGV;
+    last if m/^--$/;
+    if (s/^--raw$//) {
+       $raw = 1;
+    } else {
+       die "$_ ?";
+    }
+}
 
-system "cat >$itmp";
+my $itmp;
+my $otmp;
 
 my $admesh_stdout = '--write-ascii-stl /dev/fd/3 3>&1 >/dev/null';
 
-open I, "admesh $admesh_stdout $itmp |";
+if ($raw) {
+    open I, "<& STDIN";
+    $otmp = *STDOUT;
+} else {
+    $itmp = new File::Temp;
+    $otmp = new File::Temp;
 
-our @saved_argv = @ARGV;
+    system "cat >$itmp";
+
+    open I, "admesh $admesh_stdout $itmp |";
+}
+
+my $triangle;
 
 while (<I>) {
-    @ARGV = @saved_argv;
-    if (s/^\s+vertex\s+//) {
+    s/^\s*//;
+    if (m/^outer\s+loop/) {
+       die if $triangle;
+       $triangle = [];
+    } elsif (s/^vertex\s+//) {
        my $lhs = $&;
        s/\s+$//;
        my @xyz = split /\s+/, $_;
-       while (@ARGV) {
-           my $op = shift_arg;
-           $op =~ y/-/_/;
-           @xyz = &{ ${*::}{"pointmap_$op"} }( @xyz );
-       }
-       @xyz = map { sprintf "%.18g", $_ } @xyz;
-       $_ = "$lhs@xyz\n";
+       die unless $triangle;
+       push @$triangle, \@xyz;
+    } elsif (m/^endloop/) {
+       die unless @$triangle == 3;
+       push @$triangles, $triangle;
+       undef $triangle;
+    } elsif (m/^(?:solid|facet\s+normal|endfacet|endsolid)\s/) {
+    } else {
+       die "$_ ?";
     }
-    print $otmp $_;
 }
 
 close I;
 <I> if 0; # suppresses Name "main::I" used only once
 
+while (@ARGV) {
+    my $op = shift_arg;
+    $op =~ y/-/_/;
+    &{ ${*::}{"op__$op"} };
+}
+
+select $otmp;
+
+print "solid distort-stl\n";
+
+foreach my $t (@$triangles) {
+    print "  facet normal 0 0 0\n";
+    print "    outer loop\n";
+    die unless @$t==3;
+    foreach my $p (@$t) {
+       die unless @$p==3;
+       print "      vertex";
+       printf " %.18g", $_ foreach @$p;
+       print "\n";
+    }
+    print "    endloop\n";
+    print "  endfacet\n";
+}
+
+print "endsolid distort-stl\n";
+
 flush $otmp;
 
-system "admesh --normal-values $admesh_stdout $otmp";
+if (!$raw) {
+    system "admesh --normal-values $admesh_stdout $otmp";
+}