eaglediff

#!/usr/bin/python

import difflib
import os
from eagle_automation.config import config
from eagle_automation.export import get_extension, BadExtension, EaglePNGExport, EagleDirectoryExport
import sys
import tempfile
from PIL import Image, ImageOps, ImageChops 
from optparse import OptionParser

def to_png(in_path, page):

	workdir = tempfile.mkdtemp()

	extension = in_path.split('.')[-1].lower()
	if extension == 'brd':
		layers = config.LAYERS.values()
		out_paths = [	os.path.join(workdir, layer + '.png')
				for layer in config.LAYERS.iterkeys() ]
	elif extension == 'sch':
		layers = [{'layers': ['ALL']}]
		out_paths = [os.path.join(workdir, 'all.png')]
	else:
		os.rmdir(workdir)
		raise BadExtension

	export = EaglePNGExport(workdir=workdir)
	export.set_page(page)
	export.export(in_path, layers, out_paths)

	oim = None
	for i, out_path in enumerate(out_paths):
		im = Image.open(out_path).convert("L")
		if oim is None:
			oim = im
		else:
			oim = Image.blend(oim, im, 1.0/(1.0+i))

		os.unlink(out_path)

	os.rmdir(workdir)

	return oim

def to_txt(in_path):
	workdir = tempfile.mkdtemp()

	out_path = os.path.join(workdir, "out.txt")

	export = EagleDirectoryExport()
	export.export(in_path, None, [ out_path ])

	directory = open(out_path).read()

	os.unlink(out_path)
	os.rmdir(workdir)

	return directory

def diff_visual(from_file, to_file, page):

	a_im = to_png(from_file, page=page)
	b_im = to_png(to_file, page=page)

	# make the sizes equal
	# if a sheet contains the filename, it is updated with the temporary name
	# and may thus change the size of the image
	width  = max( (a_im.size[0], b_im.size[0]) )
	height = max( (a_im.size[1], b_im.size[1]) )
	a_im2 = Image.new( "L", (width,height) )
	a_im2.paste( a_im, (0,0) )
	a_im = a_im2
	a_im2 = None
	b_im2 = Image.new( "L", (width,height) )
	b_im2.paste( b_im, (0,0) )
	b_im = b_im2
	b_im2 = None

	added = ImageOps.autocontrast(ImageChops.subtract(b_im, a_im), 0)
	deled = ImageOps.autocontrast(ImageChops.subtract(a_im, b_im), 0)
	same = Image.blend(a_im, b_im, 0.5)

	deled = ImageOps.colorize(deled, "#000", "#f00")
	added = ImageOps.colorize(added, "#000", "#00f")
	same = ImageOps.colorize(same, "#000", "#777")

	im = ImageChops.add(ImageChops.add(added, deled), same)

	im.show()

def diff_text(from_file, to_file):
	a_txt = to_txt(from_file)
	b_txt = to_txt(to_file)

	a_lines = a_txt.split('\n')
	b_lines = b_txt.split('\n')

	diff = difflib.unified_diff(a_lines, b_lines, fromfile=from_file, tofile=to_file, lineterm='')
	print '\n'.join(list(diff))

def diff(from_file, to_file, page):
	extension = get_extension(from_file)

	if get_extension(to_file) != extension:
		print "%s: both files should have the same extension" % (to_file,)
		return

	if extension == 'brd':
		diff_visual(from_file, to_file, page)
	elif extension == 'sch':
		diff_visual(from_file, to_file, page)
	elif extension == 'lbr':
		diff_text(from_file, to_file)
	else:
		print "%s: skipping, not a board or schematic" % (from_file,)
		return

def main():
	usage = """eaglediff, compare CadSoft Eagle files
Copyright (C) 2014  Tomaz Solc <tomaz.solc@tablix.org>

USAGE: %prog [options] from-file to-file"""

	parser = OptionParser(usage=usage)
	parser.add_option("-p", "--page", dest="page", type=int, default=1,
			metavar="PAGE",
			help="page to compare on multi-page schematics")

	(options, args) = parser.parse_args()

	if len(args) != 2:
		parser.print_help()
		sys.exit(1)

	from_file, to_file = args

	diff(from_file, to_file, options.page)

main()