summaryrefslogtreecommitdiff
path: root/gerber/__main__.py
diff options
context:
space:
mode:
authorChintalagiri Shashank <shashank.chintalagiri@gmail.com>2019-05-11 04:23:19 +0530
committerChintalagiri Shashank <shashank.chintalagiri@gmail.com>2019-05-11 04:24:09 +0530
commit9cc42d9b7752603ae409e7c95a8cb5bec4d7b5b2 (patch)
tree5d3979dced48ae6bb12958f07ab6342c8746829e /gerber/__main__.py
parent2601ae8eab8d7be807bdbed264cd943e441a8da0 (diff)
downloadgerbonara-9cc42d9b7752603ae409e7c95a8cb5bec4d7b5b2.tar.gz
gerbonara-9cc42d9b7752603ae409e7c95a8cb5bec4d7b5b2.tar.bz2
gerbonara-9cc42d9b7752603ae409e7c95a8cb5bec4d7b5b2.zip
Make __main__ functional again and install a script entry point.
Diffstat (limited to 'gerber/__main__.py')
-rw-r--r--gerber/__main__.py126
1 files changed, 103 insertions, 23 deletions
diff --git a/gerber/__main__.py b/gerber/__main__.py
index 6643b54..202c00b 100644
--- a/gerber/__main__.py
+++ b/gerber/__main__.py
@@ -15,27 +15,107 @@
# License for the specific language governing permissions and limitations under
# the License.
+import os
+import argparse
+from .render import available_renderers
+from .render import theme
+from .pcb import PCB
+from . import load_layer
+
+
+def main():
+ parser = argparse.ArgumentParser(
+ description='Render gerber files to image',
+ prog='gerber-render'
+ )
+ parser.add_argument(
+ 'filenames', metavar='FILENAME', type=str, nargs='+',
+ help='Gerber files to render. If a directory is provided, it should '
+ 'be provided alone and should contain the gerber files for a '
+ 'single PCB.'
+ )
+ parser.add_argument(
+ '--outfile', '-o', type=str, nargs='?', default='out',
+ help="Output Filename (extension will be added automatically)"
+ )
+ parser.add_argument(
+ '--backend', '-b', choices=available_renderers.keys(), default='cairo',
+ help='Choose the backend to use to generate the output.'
+ )
+ parser.add_argument(
+ '--theme', '-t', choices=theme.THEMES.keys(), default='default',
+ help='Select render theme.'
+ )
+ parser.add_argument(
+ '--width', type=int, default=1920, help='Maximum width.'
+ )
+ parser.add_argument(
+ '--height', type=int, default=1080, help='Maximum height.'
+ )
+ parser.add_argument(
+ '--verbose', '-v', action='store_true', default=False,
+ help='Increase verbosity of the output.'
+ )
+ # parser.add_argument(
+ # '--quick', '-q', action='store_true', default=False,
+ # help='Skip longer running rendering steps to produce lower quality'
+ # ' output faster. This only has an effect for the freecad backend.'
+ # )
+ # parser.add_argument(
+ # '--nox', action='store_true', default=False,
+ # help='Run without using any GUI elements. This may produce suboptimal'
+ # 'output. For the freecad backend, colors, transparancy, and '
+ # 'visibility cannot be set without a GUI instance.'
+ # )
+
+ args = parser.parse_args()
+
+ renderer = available_renderers[args.backend]()
+
+ if args.backend in ['cairo', ]:
+ outext = 'png'
+ else:
+ outext = None
+
+ if os.path.exists(args.filenames[0]) and os.path.isdir(args.filenames[0]):
+ directory = args.filenames[0]
+ pcb = PCB.from_directory(directory)
+
+ if args.backend in ['cairo', ]:
+ top = pcb.top_layers
+ bottom = pcb.bottom_layers
+ copper = pcb.copper_layers
+
+ outline = pcb.outline_layer
+ if outline:
+ top = [outline] + top
+ bottom = [outline] + bottom
+ copper = [outline] + copper + pcb.drill_layers
+
+ renderer.render_layers(
+ layers=top, theme=theme.THEMES[args.theme],
+ max_height=args.height, max_width=args.width,
+ filename='{0}.top.{1}'.format(args.outfile, outext)
+ )
+ renderer.render_layers(
+ layers=bottom, theme=theme.THEMES[args.theme],
+ max_height=args.height, max_width=args.width,
+ filename='{0}.bottom.{1}'.format(args.outfile, outext)
+ )
+ renderer.render_layers(
+ layers=copper, theme=theme.THEMES['Transparent Multilayer'],
+ max_height=args.height, max_width=args.width,
+ filename='{0}.copper.{1}'.format(args.outfile, outext))
+ else:
+ pass
+ else:
+ filenames = args.filenames
+ for filename in filenames:
+ layer = load_layer(filename)
+ settings = theme.THEMES[args.theme].get(layer.layer_class, None)
+ renderer.render_layer(layer, settings=settings)
+ renderer.dump(filename='{0}.{1}'.format(args.outfile, outext))
+
+
if __name__ == '__main__':
- from gerber.common import read
- from gerber.render import GerberCairoContext
- import sys
-
- if len(sys.argv) < 2:
- sys.stderr.write("Usage: python -m gerber <filename> <filename>...\n")
- sys.exit(1)
-
- ctx = GerberCairoContext()
- ctx.alpha = 0.95
- for filename in sys.argv[1:]:
- print("parsing %s" % filename)
- if 'GTO' in filename or 'GBO' in filename:
- ctx.color = (1, 1, 1)
- ctx.alpha = 0.8
- elif 'GTS' in filename or 'GBS' in filename:
- ctx.color = (0.2, 0.2, 0.75)
- ctx.alpha = 0.8
- gerberfile = read(filename)
- gerberfile.render(ctx)
-
- print('Saving image to test.svg')
- ctx.dump('test.svg')
+ main()