renderstl command

src/c3nav/mapdata/management/commands/renderscad.py

@@ -1,9 +0,0 @@
-from django.core.management.base import BaseCommand
-
-
-class Command(BaseCommand):
-    help = 'render the map to openscad'
-
-    def handle(self, *args, **options):
-        from c3nav.mapdata.models import Level
-        Level.render_scad_all()

src/c3nav/mapdata/management/commands/renderstl.py

@@ -0,0 +1,68 @@
+import argparse
+import os
+
+from django.conf import settings
+from django.core.management.base import BaseCommand
+from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import ungettext_lazy
+
+from c3nav.mapdata.models import AccessRestriction, Level, Source
+from c3nav.mapdata.render import MapRenderer
+from c3nav.mapdata.render.engines import STLEngine
+
+
+class Command(BaseCommand):
+    help = 'render the map to stl files'
+
+    @staticmethod
+    def levels_value(value):
+        if value == '*':
+            return Level.objects.filter(on_top_of__isnull=True)
+
+        values = set(v for v in value.split(',') if v)
+        levels = Level.objects.filter(on_top_of__isnull=True, short_label__in=values)
+
+        not_found = values - set(level.short_label for level in levels)
+        if not_found:
+            raise argparse.ArgumentTypeError(
+                ungettext_lazy('Unknown level: %s', 'Unknown levels: %s', len(not_found)) % ', '.join(not_found)
+            )
+
+        return levels
+
+    @staticmethod
+    def permissions_value(value):
+        if value == '*':
+            return AccessRestriction.objects.all()
+        if value == '0':
+            return ()
+
+        values = set(v for v in value.split(',') if v)
+        permissions = AccessRestriction.objects.all().filter(pk__in=values)
+
+        not_found = values - set(str(permission.pk) for permission in permissions)
+        if not_found:
+            raise argparse.ArgumentTypeError(
+                ungettext_lazy('Unknown access restriction: %s',
+                               'Unknown access restrictions: %s', len(not_found)) % ', '.join(not_found)
+            )
+
+        return permissions
+
+    def add_arguments(self, parser):
+        parser.add_argument('--levels', default='*', type=self.levels_value,
+                            help=_('levels to render, e.g. 0,1,2 or * for all levels (default)'))
+        parser.add_argument('--permissions', default='0', type=self.permissions_value,
+                            help=_('permissions, e.g. 2,3 or * for all permissions or 0 for none (default)'))
+
+    def handle(self, *args, **options):
+        (minx, miny), (maxx, maxy) = Source.max_bounds()
+        for level in options['levels']:
+            renderer = MapRenderer(level.pk, minx, miny, maxx, maxy, access_permissions=options['permissions'])
+
+            stl = renderer.render(STLEngine)
+            data = stl.render()
+            filename = os.path.join(settings.RENDER_ROOT,
+                                    'level_%s_%s.stl' % (level.short_label,
+                                                         renderer.access_cache_key.replace('_', '-')))
+            open(filename, 'wb').write(data)