229 lines
7.2 KiB
Python
229 lines
7.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
# remove_duplicates.py
|
|
#
|
|
# Copyright (C) 2019 Mitja Nemec, Stephen Walker-Weinshenker
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
|
# MA 02110-1301, USA.
|
|
#
|
|
#
|
|
import pcbnew
|
|
import os
|
|
import sys
|
|
import logging
|
|
from collections import defaultdict
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def zones_equal(zone1, zone2):
|
|
z1_layer = zone1.GetLayer()
|
|
z1_corners = zone1.GetNumCorners()
|
|
|
|
z2_layer = zone2.GetLayer()
|
|
z2_corners = zone2.GetNumCorners()
|
|
|
|
if z1_layer == z2_layer:
|
|
if z1_corners == z2_corners:
|
|
z1_corners = [zone1.GetCornerPosition(index) for index in range(z1_corners)]
|
|
z2_corners = [zone2.GetCornerPosition(index) for index in range(z2_corners)]
|
|
z1z2_corners = zip(z1_corners, z2_corners)
|
|
if all(x[0] == x[1] for x in z1z2_corners):
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def remove_duplicate_zones(board):
|
|
# load all zones
|
|
zones = board.Zones()
|
|
|
|
# build a dictionary wih a list of tracks for each net
|
|
zone_dict = defaultdict(list)
|
|
for zone in zones:
|
|
zone_dict[zone.GetNetCode()].append(zone)
|
|
|
|
# go through all the keys
|
|
for key in zone_dict:
|
|
zones = zone_dict[key]
|
|
|
|
# for each key compare each track agianst each other and if there is no equal
|
|
# remove it from the list and add to the new list
|
|
|
|
for index in range(len(zones)):
|
|
z1 = zones[index]
|
|
bools = map(lambda x: zones_equal(x, z1), zones[index+1:])
|
|
if sum(bools) != 0:
|
|
board.RemoveNative(z1)
|
|
|
|
|
|
def tracks_equal(track1, track2):
|
|
t1_layer = track1.GetLayer()
|
|
t1_pos = track1.GetPosition()
|
|
t1_start = track1.GetStart()
|
|
t1_end = track1.GetEnd()
|
|
|
|
t2_layer = track2.GetLayer()
|
|
t2_pos = track2.GetPosition()
|
|
t2_start = track2.GetStart()
|
|
t2_end = track2.GetEnd()
|
|
|
|
if t1_layer == t2_layer:
|
|
if t1_pos == t2_pos:
|
|
if t1_start == t2_start:
|
|
if t1_end == t2_end:
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def remove_duplicate_tracks(board):
|
|
# load all tracks
|
|
tracks = board.GetTracks()
|
|
|
|
# build a dictionary wih a list of tracks for each net
|
|
track_dict = defaultdict(list)
|
|
for track in tracks:
|
|
track_dict[track.GetNetCode()].append(track)
|
|
|
|
# go through all the keys
|
|
for key in track_dict:
|
|
tracks = track_dict[key]
|
|
|
|
# for each key compare each track agianst each other and if there is no equal
|
|
# remove it from the list and add to the new list
|
|
|
|
for index in range(len(tracks)):
|
|
t1 = tracks[index]
|
|
bools = map(lambda x: tracks_equal(x, t1), tracks[index+1:])
|
|
if sum(bools) != 0:
|
|
board.RemoveNative(t1)
|
|
|
|
|
|
def text_equal(text1, text2):
|
|
t1_properties = []
|
|
t1_properties.append(text1.GetLayer())
|
|
t1_properties.append(text1.GetPosition())
|
|
t1_properties.append(text1.GetTextPos())
|
|
t1_properties.append(text1.GetText())
|
|
#t1_properties.append(text1.GetThickness())
|
|
t1_properties.append(text1.GetTextAngle())
|
|
t1_properties.append(text1.IsItalic())
|
|
t1_properties.append(text1.IsBold())
|
|
t1_properties.append(text1.IsVisible())
|
|
t1_properties.append(text1.IsMirrored())
|
|
t1_properties.append(text1.GetVertJustify())
|
|
t1_properties.append(text1.GetHorizJustify())
|
|
t1_properties.append(text1.GetTextSize())
|
|
t1_properties.append(text1.GetTextWidth())
|
|
t1_properties.append(text1.GetTextHeight())
|
|
|
|
t2_properties = []
|
|
t2_properties.append(text2.GetLayer())
|
|
t2_properties.append(text2.GetPosition())
|
|
t2_properties.append(text2.GetTextPos())
|
|
t2_properties.append(text2.GetText())
|
|
#t2_properties.append(text2.GetThickness())
|
|
t2_properties.append(text2.GetTextAngle())
|
|
t2_properties.append(text2.IsItalic())
|
|
t2_properties.append(text2.IsBold())
|
|
t2_properties.append(text2.IsVisible())
|
|
t2_properties.append(text2.IsMirrored())
|
|
t2_properties.append(text2.GetVertJustify())
|
|
t2_properties.append(text2.GetHorizJustify())
|
|
t2_properties.append(text2.GetTextSize())
|
|
t2_properties.append(text2.GetTextWidth())
|
|
t2_properties.append(text2.GetTextHeight())
|
|
|
|
t1t2_properties = zip(t1_properties, t2_properties)
|
|
if all(x[0] == x[1] for x in t1t2_properties):
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def remove_duplicate_text(board):
|
|
# load text items
|
|
drawings = board.GetDrawings()
|
|
|
|
text_items = []
|
|
for drawing in drawings:
|
|
if isinstance(drawing, pcbnew.PCB_TEXT):
|
|
text_items.append(drawing)
|
|
|
|
for index in range(len(text_items)):
|
|
t1 = text_items[index]
|
|
bools = map(lambda x: text_equal(x, t1), text_items[index+1:])
|
|
if sum(bools) != 0:
|
|
board.RemoveNative(t1)
|
|
|
|
|
|
def drawings_equal(drawing1, drawing2):
|
|
d1_properties = []
|
|
d1_properties.append(drawing1.GetLayer())
|
|
d1_properties.append(drawing1.GetPosition())
|
|
d1_properties.append(drawing1.GetStart())
|
|
d1_properties.append(drawing1.GetEnd())
|
|
d1_properties.append(drawing1.GetClass())
|
|
d1_properties.append(drawing1.GetLength())
|
|
d1_properties.append(drawing1.GetWidth())
|
|
d1_properties.append(drawing1.GetAngle())
|
|
d1_properties.append(drawing1.GetType())
|
|
d1_properties.append(drawing1.GetShape())
|
|
|
|
d2_properties = []
|
|
d2_properties.append(drawing2.GetLayer())
|
|
d2_properties.append(drawing2.GetPosition())
|
|
d2_properties.append(drawing2.GetStart())
|
|
d2_properties.append(drawing2.GetEnd())
|
|
d2_properties.append(drawing2.GetClass())
|
|
d2_properties.append(drawing2.GetLength())
|
|
d2_properties.append(drawing2.GetWidth())
|
|
d2_properties.append(drawing2.GetAngle())
|
|
d2_properties.append(drawing2.GetType())
|
|
d2_properties.append(drawing2.GetShape())
|
|
|
|
d1d2_properties = zip(d1_properties, d2_properties)
|
|
if all(x[0] == x[1] for x in d1d2_properties):
|
|
return 1
|
|
return 0
|
|
|
|
|
|
def remove_duplicate_drawings(board):
|
|
# load text items
|
|
drawings = board.GetDrawings()
|
|
|
|
drawing_items = []
|
|
for drawing in drawings:
|
|
if isinstance(drawing, pcbnew.DRAWINGS):
|
|
drawing_items.append(drawing)
|
|
|
|
for index in range(len(drawing_items)):
|
|
d1 = drawing_items[index]
|
|
bools = map(lambda x: drawings_equal(x, d1), drawing_items[index+1:])
|
|
if sum(bools) != 0:
|
|
board.RemoveNative(d1)
|
|
|
|
|
|
def remove_duplicates(board):
|
|
# remove duplicate tracks
|
|
remove_duplicate_tracks(board)
|
|
# remove duplicate zones
|
|
remove_duplicate_zones(board)
|
|
# remove duplicate text
|
|
remove_duplicate_text(board)
|
|
# remove duplicate drawings
|
|
remove_duplicate_drawings(board)
|
|
|
|
|
|
|