This repository has been archived on 2023-03-25. You can view files and clone it, but cannot push or open issues or pull requests.
mightyscape-1.1-deprecated/extensions/fablabchemnitz_render_sudoku.py

125 lines
6.9 KiB
Python
Raw Normal View History

2020-07-30 01:16:18 +02:00
#!/usr/bin/env python3
'''
render_sudoku.py
A sudoku generator plugin for Inkscape, but also can be used as a standalone
command line application.
Copyright (C) 2011 Chris Savery <chrissavery(a)gmail.com>
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
'''
__version__ = "0.1"
import inkex
import subprocess
from lxml import etree
class SVGSudoku (inkex.Effect):
def __init__(self):
inkex.Effect.__init__(self)
self.arg_parser.add_argument("--tab")
self.arg_parser.add_argument("--difficulty",default="mixed", help='How difficult to make puzzles.')
self.arg_parser.add_argument("--rows", type=int, default=1, help='Number of puzzle rows.')
self.arg_parser.add_argument("--cols", type=int, default=1, help='Number of puzzle columns.')
self.arg_parser.add_argument("--puzzle_size", type=int, default=6, help='The width & height of each puzzle.')
self.arg_parser.add_argument("--puzzle_gap", type=int, default=1, help='The space between puzzles.')
self.arg_parser.add_argument("--color_text", type=int, default=0, help='Color for given numbers.')
self.arg_parser.add_argument("--color_bkgnd", type=int, default=-1, help='Color for the puzzle background.')
self.arg_parser.add_argument("--color_puzzle",default=0, help='Border color for the puzzles.')
self.arg_parser.add_argument("--color_boxes", default=0, help='Border color for puzzle boxes.')
self.arg_parser.add_argument("--color_cells", default=-1061109505, help='Border color for the puzzle cells.')
self.arg_parser.add_argument("--units", help="The unit of the dimensions")
def getColorString(self, pickerColor):
longcolor = int(pickerColor)
if longcolor < 0:
longcolor = longcolor & 0xFFFFFFFF
return '#' + format(longcolor >> 8, '06X')
def draw_grid(self, g_puz, x, y):
bkgnd_style = {'stroke':'none', 'stroke-width':'2', 'fill':self.options.color_bkgnd }
puzzle_style = {'stroke':self.options.color_puzzle, 'stroke-width':'2', 'fill':'none' }
boxes_style = {'stroke':self.options.color_boxes, 'stroke-width':'2', 'fill':'none' }
cells_style = {'stroke':self.options.color_cells, 'stroke-width':'1', 'fill':'none' }
g = etree.SubElement(g_puz, 'g')
self.draw_rect(g, bkgnd_style, self.left+x, self.top+y, self.size, self.size)
self.draw_rect(g, cells_style, self.left+x+self.size/9, self.top+y, self.size/9, self.size)
self.draw_rect(g, cells_style, self.left+x+self.size/3+self.size/9, self.top+y, self.size/9, self.size)
self.draw_rect(g, cells_style, self.left+x+2*self.size/3+self.size/9, self.top+y, self.size/9, self.size)
self.draw_rect(g, cells_style, self.left+x, self.top+y+self.size/9, self.size, self.size/9)
self.draw_rect(g, cells_style, self.left+x, self.top+y+self.size/3+self.size/9, self.size, self.size/9)
self.draw_rect(g, cells_style, self.left+x, self.top+y+2*self.size/3+self.size/9, self.size, self.size/9)
self.draw_rect(g, boxes_style, self.left+x+self.size/3, self.top+y, self.size/3, self.size)
self.draw_rect(g, boxes_style, self.left+x, self.top+y+self.size/3, self.size, self.size/3)
self.draw_rect(g, puzzle_style, self.left+x, self.top+y, self.size, self.size)
def draw_rect(self, g, style, x, y, w, h):
attribs = {'style':str(inkex.Style(style)), 'x':str(x), 'y':str(y), 'height':str(h), 'width':str(w) }
etree.SubElement(g, inkex.addNS('rect','svg'), attribs)
def fill_puzzle(self, g_puz, x, y, data):
cellsize = self.size / 9
txtsize = self.size / 12
offset = (cellsize + txtsize)/2.25
g = etree.SubElement(g_puz, 'g')
text_style = {'font-size':str(txtsize),
'fill':self.options.color_text,
'font-family':'arial',
'text-anchor':'middle', 'text-align':'center' }
#inkex.utils.debug(len(data))
for n in range(len(data)):
#inkex.utils.debug(str(n))
#inkex.utils.debug("data["+str(n)+"]="+str(data[n]))
if str(data[n]) in "123456789":
attribs = {'style': str(inkex.Style(text_style)),
'x': str(self.left + x + n%9 * cellsize + cellsize/2 ), 'y': str(self.top + y + n//9 * cellsize + offset ) }
etree.SubElement(g, 'text', attribs).text = str(data[n])
def effect(self):
args = ["./qqwing", "--one-line", "--generate", str(self.options.rows * self.options.cols)]
if self.options.difficulty != 'mixed':
args.extend(["--difficulty", self.options.difficulty])
data = subprocess.Popen(args, stdout=subprocess.PIPE).communicate()[0].splitlines()
#inkex.utils.debug(data)
#
# convert colors
#inkex.utils.debug("%s"%(self.options.color_cells))
self.options.color_text = self.getColorString(self.options.color_text)
self.options.color_bkgnd = self.getColorString(self.options.color_bkgnd)
self.options.color_puzzle = self.getColorString(self.options.color_puzzle)
self.options.color_boxes = self.getColorString(self.options.color_boxes)
self.options.color_cells = self.getColorString(self.options.color_cells)
#
parent = self.document.getroot()
self.doc_w = self.svg.unittouu(parent.get('width'))
self.doc_h = self.svg.unittouu(parent.get('height'))
self.size = self.svg.unittouu(str(self.options.puzzle_size) + self.options.units)
self.gap = self.svg.unittouu(str(self.options.puzzle_gap) + self.options.units)
self.shift = self.size + self.gap
self.left = (self.doc_w - (self.options.cols * self.shift - self.gap))/2
self.top = (self.doc_h - (self.options.rows * self.shift - self.gap))/2
self.sudoku_g = etree.SubElement(parent, 'g', {'id':'sudoku'})
for row in range(0, self.options.rows):
for col in range(0, self.options.cols):
g = etree.SubElement(self.sudoku_g, 'g', {'id':'puzzle_'+str(col)+str(row)})
self.draw_grid(g, col*self.shift, row*self.shift)
self.fill_puzzle(g, col*self.shift, row*self.shift, data[col+row*self.options.cols].decode('UTF-8'))
if __name__ == '__main__': #pragma: no cover
SVGSudoku().run()