summaryrefslogtreecommitdiff
path: root/gerber/tests/test_cairo_backend.py
blob: 38cffba430283f84a2ba67f94b61e9b4f85b77ae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
#! /usr/bin/env python
# -*- coding: utf-8 -*-

# Author: Garret Fick <garret@ficksworkshop.com>
import io
import os

from ..render.cairo_backend import GerberCairoContext
from ..rs274x import read, GerberFile
from .tests import *
from nose.tools import assert_tuple_equal

def test_render_two_boxes():
    """Umaco exapmle of two boxes"""
    _test_render('resources/example_two_square_boxes.gbr', 'golden/example_two_square_boxes.png')


def test_render_single_quadrant():
    """Umaco exapmle of a single quadrant arc"""
    _test_render('resources/example_single_quadrant.gbr', 'golden/example_single_quadrant.png')


def test_render_simple_contour():
    """Umaco exapmle of a simple arrow-shaped contour"""
    gerber = _test_render('resources/example_simple_contour.gbr', 'golden/example_simple_contour.png')
    
    # Check the resulting dimensions
    assert_tuple_equal(((2.0, 11.0), (1.0, 9.0)), gerber.bounding_box)

    
def test_render_single_contour_1():
    """Umaco example of a single contour
    
    The resulting image for this test is used by other tests because they must generate the same output."""
    _test_render('resources/example_single_contour_1.gbr', 'golden/example_single_contour.png')


def test_render_single_contour_2():
    """Umaco exapmle of a single contour, alternate contour end order
    
    The resulting image for this test is used by other tests because they must generate the same output."""
    _test_render('resources/example_single_contour_2.gbr', 'golden/example_single_contour.png')


def test_render_single_contour_3():
    """Umaco exapmle of a single contour with extra line"""
    _test_render('resources/example_single_contour_3.gbr', 'golden/example_single_contour_3.png')
    
    
def test_render_not_overlapping_contour():
    """Umaco example of D02 staring a second contour"""
    _test_render('resources/example_not_overlapping_contour.gbr', 'golden/example_not_overlapping_contour.png')
    

def test_render_not_overlapping_touching():
    """Umaco example of D02 staring a second contour"""
    _test_render('resources/example_not_overlapping_touching.gbr', 'golden/example_not_overlapping_touching.png')


def test_render_overlapping_touching():
    """Umaco example of D02 staring a second contour"""
    _test_render('resources/example_overlapping_touching.gbr', 'golden/example_overlapping_touching.png')


def test_render_overlapping_contour():
    """Umaco example of D02 staring a second contour"""
    _test_render('resources/example_overlapping_contour.gbr', 'golden/example_overlapping_contour.png')


def _DISABLED_test_render_level_holes():
    """Umaco example of using multiple levels to create multiple holes"""
    
    # TODO This is clearly rendering wrong. I'm temporarily checking this in because there are more
    # rendering fixes in the related repository that may resolve these.
    _test_render('resources/example_level_holes.gbr', 'golden/example_overlapping_contour.png')


def _DISABLED_test_render_cutin():
    """Umaco example of using a cutin"""

    # TODO This is clearly rendering wrong.
    _test_render('resources/example_cutin.gbr', 'golden/example_cutin.png')


def test_render_fully_coincident():
    """Umaco example of coincident lines rendering two contours"""

    _test_render('resources/example_fully_coincident.gbr', 'golden/example_fully_coincident.png')


def test_render_coincident_hole():
    """Umaco example of coincident lines rendering a hole in the contour"""

    _test_render('resources/example_coincident_hole.gbr', 'golden/example_coincident_hole.png')


def test_render_cutin_multiple():
    """Umaco example of a region with multiple cutins"""

    _test_render('resources/example_cutin_multiple.gbr', 'golden/example_cutin_multiple.png')
    

def test_flash_circle():
    """Umaco example a simple circular flash with and without a hole"""

    _test_render('resources/example_flash_circle.gbr', 'golden/example_flash_circle.png')


def test_flash_rectangle():
    """Umaco example a simple rectangular flash with and without a hole"""

    _test_render('resources/example_flash_rectangle.gbr', 'golden/example_flash_rectangle.png')


def test_flash_obround():
    """Umaco example a simple obround flash with and without a hole"""

    _test_render('resources/example_flash_obround.gbr', 'golden/example_flash_obround.png')


def test_flash_polygon():
    """Umaco example a simple polygon flash with and without a hole"""

    _test_render('resources/example_flash_polygon.gbr', 'golden/example_flash_polygon.png', 'golden/example_flash_polygon.png')

def _resolve_path(path):
    return os.path.join(os.path.dirname(__file__),
                                path)


def _test_render(gerber_path, png_expected_path, create_output_path = None):
    """Render the gerber file and compare to the expected PNG output.
    
    Parameters
    ----------
    gerber_path : string
        Path to Gerber file to open
    png_expected_path : string
        Path to the PNG file to compare to
    create_output : string|None
        If not None, write the generated PNG to the specified path.
        This is primarily to help with 
    """
    
    gerber_path = _resolve_path(gerber_path)
    png_expected_path = _resolve_path(png_expected_path)
    if create_output_path:
        create_output_path = _resolve_path(create_output_path)
    
    gerber = read(gerber_path)

    # Create PNG image to the memory stream
    ctx = GerberCairoContext()
    gerber.render(ctx)

    actual_bytes = ctx.dump(None)
    
    # If we want to write the file bytes, do it now. This happens
    if create_output_path:
        with open(create_output_path, 'wb') as out_file:
            out_file.write(actual_bytes)
        # Creating the output is dangerous - it could overwrite the expected result.
        # So if we are creating the output, we make the test fail on purpose so you
        # won't forget to disable this
        assert_false(True, 'Test created the output %s. This needs to be disabled to make sure the test behaves correctly' % (create_output_path,))
    
    # Read the expected PNG file
    
    with open(png_expected_path, 'rb') as expected_file:
        expected_bytes = expected_file.read()
    
    assert_equal(expected_bytes, actual_bytes)
    
    return gerber