printer.py

from __future__ import division
import numpy as np
import matplotlib.pyplot as plt
import sdxf
import os.path

class PDFPrinter:
    def __init__(self):
        plt.figure(figsize=[6, 6])
        plt.hold(True)

    def save(self, filename):
        plt.axis('equal')
        plt.tick_params(colors='w')
        plt.savefig(os.path.abspath(filename+'.pdf'), bbox_inches = 'tight')

    def draw_arc(self, center, r, angles = [np.pi/6, 5*np.pi/6], **kwargs):
        if r > 0:
            color = 'b'
        else:
            color = 'r'
        angles = np.linspace(angles[0], angles[1])
        plt.plot(center[0] + r * np.cos(angles), center[1] + r * np.sin(angles),
                linewidth=.5,
                 color = color,
                **kwargs)
        plt.plot([center[0],
                  center[0] + r * np.cos(angles[0])],
                 [center[1],
                  center[1] + r * np.sin(angles[0])],
                 linewidth = .5,
                 color = color,
                 marker = '+',
                 markersize = 2,
                 **kwargs)

    def draw_line(self, center, length, angle, style='k-', **kwargs):
        plt.plot([center[0] - length/2 * np.cos(angle),
                  center[0] + length/2 * np.cos(angle)],
                 [center[1] - length/2 * np.sin(angle),
                  center[1] + length/2 * np.sin(angle)], style, **kwargs)

    def draw_circle(self, center, r):
        angles = np.linspace(0, 2*np.pi, 100)
        plt.plot(center[0] + r * np.cos(angles),
                 center[1] + r * np.sin(angles), 'k-', linewidth=.5)

    def draw_point(self, center, **kwargs):
        plt.plot(center[0], center[1], **kwargs)

class DXFPrinter:
    def __init__(self):
        self.dxf = sdxf.Drawing()

    def save(self, filename):
        self.dxf.saveas(filename+'.dxf')

    def draw_arc(self, center, r, angles = [np.pi/6, 5*np.pi/6], **kwargs):
        if r < 0:
            r = -r
            angles = np.array(angles) + np.pi
        startAngle = min(angles) * 180/np.pi
        endAngle = max(angles) * 180/np.pi
        self.dxf.append(sdxf.Arc(center = center + [0], radius = r,
                                startAngle = startAngle,
                                endAngle = endAngle,
                                layer = "drawinglayer"))

    def draw_line(self, center, length, angle, style='k-', **kwargs):
        self.dxf.append(sdxf.Line(points=
            [[center[0] - length/2 * np.cos(angle),
              center[1] - length/2 * np.sin(angle)],
             [center[0] + length/2 * np.cos(angle),
              center[1] + length/2 * np.sin(angle)]]))