[markers] generate cross and support for png and pdf

Author: simogasp

doc/sphinx/source/markers/markers.rst

@@ -44,24 +44,31 @@ Here is the usage and options:
 .. code::
 
     usage: generate.py [-h] [--rings N] [--outdir dir] [--margin N] [--radius N]
-                       [--addId] [--whiteBackground]
+                       [--addId] [--addCross] [--generatePng] [--generatePdf]
+                       [--whiteBackground]
 
     Generate the svg file for the markers.
 
     optional arguments:
       -h, --help         show this help message and exit
       --rings N          the number of rings (possible values {3, 4}, default: 3)
-      --outdir dir       the directory where to save the files
+      --outdir dir       the directory where to save the files (default: ./)
       --margin N         the margin to add around the external ring (default: 400)
       --radius N         the radius of the outer circle (default: 500)
       --addId            add the marker id on the top left corner
+      --addCross         add a small cross in the center of the marker
+      --generatePng      also generate a png file
+      --generatePdf      also generate a pdf file
       --whiteBackground  set the background (outside the marker) to white instead
                          of transparent
 
+
 For example, calling:
 
 .. code:: shell
 
    ./generate.py --outdir markers3 --margin 100 --addId
 
 it will create a directory :code:`markers3` where it saves an svg file for each marker with a margin around the marker of 100 and with the ID of the marker printed on the top left corner.
+
+To generate pdf and/or png file, use the flags :code:`--generatePdf` and :code:`--generatePng` .

markersToPrint/generators/generate.py

@@ -5,6 +5,9 @@
 
 import svgwrite
 
+from svglib.svglib import svg2rlg
+from reportlab.graphics import renderPDF, renderPM
+
 if __name__ == "__main__":
 
     parser = argparse.ArgumentParser(description='Generate the svg file for the markers.')
@@ -18,6 +21,12 @@
                         help='the radius of the outer circle (default: %(default)s)')
     parser.add_argument('--addId', action='store_true',
                         help='add the marker id on the top left corner')
+    parser.add_argument('--addCross', action='store_true',
+                        help='add a small cross in the center of the marker')
+    parser.add_argument('--generatePng', action='store_true',
+                        help='also generate a png file')
+    parser.add_argument('--generatePdf', action='store_true',
+                        help='also generate a pdf file')
     parser.add_argument('--whiteBackground', action='store_true',
                         help='set the background (outside the marker) to white instead of transparent')
 
@@ -28,17 +37,17 @@
     scale = args.radius / 100
 
     # size of the image, diameter + margin
-    width = size+args.margin
-    height = size+args.margin
+    width = size + args.margin
+    height = size + args.margin
 
     # id of the first marker
     markerId = 1
 
     # font size for the id
-    font_size = int(0.037*height)
+    font_size = int(0.037 * height)
 
     # center of the marker
-    center = (width/2, height/2)
+    center = (width / 2, height / 2)
 
     radius = args.radius
 
@@ -56,7 +65,8 @@
         for line in f:
 
             # name of the output file
-            out_filename = os.path.join(args.outdir, str(markerId).zfill(4)+'.svg')
+            base_filename = os.path.join(args.outdir, str(markerId).zfill(4))
+            out_filename = base_filename + '.svg'
 
             # create the svg
             dwg = svgwrite.Drawing(out_filename, profile='tiny', size=(width, height))
@@ -67,7 +77,7 @@
                 dwg.add(dwg.text(text=str(markerId), insert=(5, 50), font_size=font_size))
 
             # print the outer circle as black
-            dwg.add(dwg.circle(center=center, r=size/2, fill='black'))
+            dwg.add(dwg.circle(center=center, r=size / 2, fill='black'))
 
             fill_color = 'white'
             count = 0
@@ -76,7 +86,7 @@
             for r in line.split():
                 radius = int(r)
                 # print(r)
-                dwg.add(dwg.circle(center=center, r=scale*radius, fill=fill_color))
+                dwg.add(dwg.circle(center=center, r=scale * radius, fill=fill_color))
                 if fill_color == 'white':
                     fill_color = 'black'
                 else:
@@ -89,6 +99,18 @@
             else:
                 assert count == 7
 
+            if args.addCross:
+                # print a small cross in the center
+                dwg.add(dwg.line(start=(center[0] - 10, center[1]), end=(center[0] + 10, center[1]), stroke="gray"))
+                dwg.add(dwg.line(start=(center[0], center[1] - 10), end=(center[0], center[1] + 10), stroke="gray"))
+
             dwg.save(pretty=True)
 
-            markerId = markerId + 1
+            if args.generatePng or args.generatePdf:
+                drawing = svg2rlg(out_filename)
+                if args.generatePdf:
+                    renderPDF.drawToFile(drawing, base_filename + ".pdf")
+                if args.generatePng:
+                    renderPM.drawToFile(drawing, base_filename + ".png", fmt="PNG")
+
+            markerId = markerId + 1

markersToPrint/generators/requirements.txt

@@ -0,0 +1,2 @@
+svgwrite
+svglib