mightyscape-1.1-deprecated/extensions/fablabchemnitz/sudoku/render_sudoku.py

#!/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
from inkex import Color

class SVGSudoku (inkex.EffectExtension):

    def add_arguments(self, pars):
        pars.add_argument("--tab")
        pars.add_argument("--difficulty",default="mixed", help='How difficult to make puzzles.')
        pars.add_argument("--rows", type=int, default=1, help='Number of puzzle rows.')
        pars.add_argument("--cols", type=int, default=1, help='Number of puzzle columns.') 
        pars.add_argument("--puzzle_size", type=int, default=6, help='The width & height of each puzzle.')
        pars.add_argument("--puzzle_gap", type=int, default=1, help='The space between puzzles.')
        pars.add_argument("--color_text", type=Color, default=255, help='Color for given numbers.') 
        pars.add_argument("--color_bkgnd", type=Color, default=4243148799, help='Color for the puzzle background.')
        pars.add_argument("--color_puzzle",type=Color, default=2290779647, help='Border color for the puzzles.')
        pars.add_argument("--color_boxes", type=Color, default=3298820351, help='Border color for puzzle boxes.')
        pars.add_argument("--color_cells", type=Color, default=1923076095, help='Border color for the puzzle cells.')
        pars.add_argument("--units", help="The unit of the dimensions")

    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)

        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__':
    SVGSudoku().run()