summaryrefslogtreecommitdiff
path: root/gerber
diff options
context:
space:
mode:
authorPaulo Henrique Silva <ph.silva@gmail.com>2019-11-27 11:43:19 -0300
committerGitHub <noreply@github.com>2019-11-27 11:43:19 -0300
commit77a156cea2411411922d0a755e4b4efab2a1b3d0 (patch)
tree4233452d40a23ab99d0aab5f977e63df632d3c70 /gerber
parent0024c0b045a29621845f4f1b5ee1283e32e38887 (diff)
parent5288a8314b535d455e8635ee9a3a537cd795845b (diff)
downloadgerbonara-77a156cea2411411922d0a755e4b4efab2a1b3d0.tar.gz
gerbonara-77a156cea2411411922d0a755e4b4efab2a1b3d0.tar.bz2
gerbonara-77a156cea2411411922d0a755e4b4efab2a1b3d0.zip
Merge pull request #102 from chintal/script
Make __main__ functional again and install a script entry point.
Diffstat (limited to 'gerber')
-rw-r--r--gerber/__main__.py127
-rw-r--r--gerber/render/__init__.py5
2 files changed, 109 insertions, 23 deletions
diff --git a/gerber/__main__.py b/gerber/__main__.py
index 50c5a42..988adff 100644
--- a/gerber/__main__.py
+++ b/gerber/__main__.py
@@ -15,27 +15,108 @@
# 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.cairo_backend 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()
+
diff --git a/gerber/render/__init__.py b/gerber/render/__init__.py
index fe08d50..c7dbdd5 100644
--- a/gerber/render/__init__.py
+++ b/gerber/render/__init__.py
@@ -24,3 +24,8 @@ SVG is the only supported format.
"""
from .render import RenderSettings
+from .cairo_backend import GerberCairoContext
+
+available_renderers = {
+ 'cairo': GerberCairoContext,
+}