105 lines
4.3 KiB
Python
105 lines
4.3 KiB
Python
import os
|
|
from pcbnew import (PLOT_FORMAT_HPGL, PLOT_FORMAT_POST, PLOT_FORMAT_GERBER, PLOT_FORMAT_DXF, PLOT_FORMAT_SVG,
|
|
PLOT_FORMAT_PDF, wxPoint)
|
|
from .optionable import (Optionable, BaseOptions)
|
|
from kiplot.macros import macros, document # noqa: F401
|
|
from . import log
|
|
|
|
logger = log.get_logger(__name__)
|
|
|
|
|
|
class DrillMap(Optionable):
|
|
_out_def = '%f-%i.%x'
|
|
|
|
def __init__(self):
|
|
super().__init__()
|
|
with document:
|
|
self.output = self._out_def
|
|
""" name for the map file, KiCad defaults if empty (%i='PTH_drill_map') """
|
|
self.type = 'pdf'
|
|
""" [hpgl,ps,gerber,dxf,svg,pdf] format for a graphical drill map """ # pragma: no cover
|
|
self._unkown_is_error = True
|
|
|
|
|
|
class DrillReport(Optionable):
|
|
def __init__(self):
|
|
super().__init__()
|
|
with document:
|
|
self.filename = ''
|
|
""" name of the drill report. Not generated unless a name is specified. (%i='drill_report' %x='txt') """ # pragma: no cover
|
|
self._unkown_is_error = True
|
|
|
|
|
|
class AnyDrill(BaseOptions):
|
|
def __init__(self):
|
|
super().__init__()
|
|
# Options
|
|
with document:
|
|
self.use_aux_axis_as_origin = False
|
|
""" use the auxiliar axis as origin for coordinates """
|
|
self.map = DrillMap
|
|
""" [dict|string] [hpgl,ps,gerber,dxf,svg,pdf] format for a graphical drill map.
|
|
Not generated unless a format is specified """
|
|
self.report = DrillReport
|
|
""" [dict|string] name of the drill report. Not generated unless a name is specified """ # pragma: no cover
|
|
# Mappings to KiCad values
|
|
self._map_map = {
|
|
'hpgl': PLOT_FORMAT_HPGL,
|
|
'ps': PLOT_FORMAT_POST,
|
|
'gerber': PLOT_FORMAT_GERBER,
|
|
'dxf': PLOT_FORMAT_DXF,
|
|
'svg': PLOT_FORMAT_SVG,
|
|
'pdf': PLOT_FORMAT_PDF
|
|
}
|
|
self._map_ext = { 'hpgl': 'plt', 'ps': 'ps', 'gerber': 'gbr', 'dxf': 'dxf', 'svg': 'svg', 'pdf': 'pdf' }
|
|
|
|
def config(self, tree):
|
|
super().config(tree)
|
|
# Solve the map for both cases
|
|
if isinstance(self.map, str):
|
|
self.map_ext = self._map_ext[self.map]
|
|
self.map_output = DrillMap._out_def
|
|
self.map = self._map_map[self.map]
|
|
elif isinstance(self.map, DrillMap):
|
|
self.map_ext = self._map_ext[self.map.type]
|
|
self.map_output = self.map.output
|
|
self.map = self._map_map[self.map.type]
|
|
else:
|
|
self.map = None
|
|
# Solve the report for both cases
|
|
if isinstance(self.report, DrillReport):
|
|
self.report = self.report.filename
|
|
elif not isinstance(self.report, str):
|
|
self.report = None
|
|
|
|
def run(self, output_dir, board):
|
|
# dialog_gendrill.cpp:357
|
|
if self.use_aux_axis_as_origin:
|
|
offset = board.GetAuxOrigin()
|
|
else:
|
|
offset = wxPoint(0, 0)
|
|
drill_writer = self._configure_writer(board, offset)
|
|
|
|
logger.debug("Generating drill files in "+output_dir)
|
|
gen_map = self.map is not None
|
|
if gen_map:
|
|
drill_writer.SetMapFileFormat(self.map)
|
|
logger.debug("Generating drill map type {} in {}".format(self.map, output_dir))
|
|
# We always generate the drill file
|
|
drill_writer.CreateDrillandMapFilesSet(output_dir, True, gen_map)
|
|
# Rename the files
|
|
if gen_map and self.map_output:
|
|
id = 'PTH_drill_map'
|
|
k_file = self.expand_filename(output_dir, '%f-PTH-drl_map.%x', '', self.map_ext)
|
|
file = self.expand_filename(output_dir, self.map_output, id, self.map_ext)
|
|
os.rename(k_file, file)
|
|
id = 'N'+id
|
|
k_file = self.expand_filename(output_dir, '%f-NPTH-drl_map.%x', '', self.map_ext)
|
|
file = self.expand_filename(output_dir, self.map_output, id, self.map_ext)
|
|
os.rename(k_file, file)
|
|
|
|
if self.report:
|
|
drill_report_file = self.expand_filename(output_dir, self.report, 'drill_report', 'txt')
|
|
logger.debug("Generating drill report: "+drill_report_file)
|
|
drill_writer.GenDrillReportFile(drill_report_file)
|