import inkex, os, csv, math, random

from pathmodifier import zSort

inkex.localize()

try:

from subprocess import Popen, PIPE

bsubprocess = True

except:

bsubprocess = False

class Restack(inkex.Effect):

def __init__(self):

inkex.Effect.__init__(self)

self.OptionParser.add_option("-d", "--direction",

action="store", type="string",

dest="direction", default="tb",

help="direction to restack")

self.OptionParser.add_option("-a", "--angle",

action="store", type="float",

dest="angle", default=0.0,

help="arbitrary angle")

self.OptionParser.add_option("-x", "--xanchor",

action="store", type="string",

dest="xanchor", default="m",

help="horizontal point to compare")

self.OptionParser.add_option("-y", "--yanchor",

action="store", type="string",

dest="yanchor", default="m",

help="vertical point to compare")

self.OptionParser.add_option("--zsort",

action="store", type="string",

dest="zsort", default="rev",

help="Restack mode based on Z-Order")

self.OptionParser.add_option("--tab",

action="store", type="string",

dest="tab",

help="The selected UI-tab when OK was pressed")

self.OptionParser.add_option("--nb_direction",

action="store", type="string",

dest="nb_direction",

help="The selected UI-tab when OK was pressed")

def effect(self):

if self.options.tab == '"help"':

pass

elif len(self.selected) > 0:

if self.options.tab == '"positional"':

self.restack_positional()

elif self.options.tab == '"z_order"':

self.restack_z_order()

else:

inkex.errormsg(_("There is no selection to restack."))

def restack_positional(self):

objects = {}

objlist = []

file = self.args[ -1 ]

if self.options.nb_direction == '"custom"':

self.options.direction = "aa"

# process selection to get list of objects to be arranged

firstobject = self.selected[self.options.ids[0]]

if len(self.selected) == 1 and firstobject.tag == inkex.addNS('g', 'svg'):

parentnode = firstobject

for child in parentnode.iterchildren():

objects[child.get('id')] = child

else:

parentnode = self.current_layer

objects = self.selected

#get all bounding boxes in file by calling inkscape again with the --query-all command line option

#it returns a comma separated list structured id,x,y,w,h

if bsubprocess:

p = Popen('inkscape --query-all "%s"' % (file), shell=True, stdout=PIPE, stderr=PIPE)

err = p.stderr

f = p.communicate()[0]

try:

reader=csv.CSVParser().parse_string(f) #there was a module cvs.py in earlier inkscape that behaved differently

except:

reader=csv.reader(f.split( os.linesep ))

err.close()

else:

_,f,err = os.popen3('inkscape --query-all "%s"' % ( file ) )

reader=csv.reader( f )

err.close()

#build a dictionary with id as the key

dimen = dict()

for line in reader:

if len(line) > 0:

dimen[line[0]] = map( float, line[1:])

if not bsubprocess: #close file if opened using os.popen3

f.close

#find the center of all selected objects **Not the average!

x,y,w,h = dimen[objects.keys()[0]]

minx = x

miny = y

maxx = x + w

maxy = y + h

for id, node in objects.iteritems():

# get the bounding box

x,y,w,h = dimen[id]

if x < minx:

minx = x

if (x + w) > maxx:

maxx = x + w

if y < miny:

miny = y

if (y + h) > maxy:

maxy = y + h

midx = (minx + maxx) / 2

midy = (miny + maxy) / 2

#calculate distances for each selected object

for id, node in objects.iteritems():

# get the bounding box

x,y,w,h = dimen[id]

# calc the comparison coords

if self.options.xanchor == "l":

cx = x

elif self.options.xanchor == "r":

cx = x + w

else: # middle

cx = x + w / 2

if self.options.yanchor == "t":

cy = y

elif self.options.yanchor == "b":

cy = y + h

else: # middle

cy = y + h / 2

#direction chosen

if self.options.direction == "tb" or (self.options.direction == "aa" and self.options.angle == 270):

objlist.append([cy,id])

elif self.options.direction == "bt" or (self.options.direction == "aa" and self.options.angle == 90):

objlist.append([-cy,id])

elif self.options.direction == "lr" or (self.options.direction == "aa" and (self.options.angle == 0 or self.options.angle == 360)):

objlist.append([cx,id])

elif self.options.direction == "rl" or (self.options.direction == "aa" and self.options.angle == 180):

objlist.append([-cx,id])

elif self.options.direction == "aa":

distance = math.hypot(cx,cy)*(math.cos(math.radians(-self.options.angle)-math.atan2(cy, cx)))

objlist.append([distance,id])

elif self.options.direction == "ro":

distance = math.hypot(midx - cx, midy - cy)

objlist.append([distance,id])

elif self.options.direction == "ri":

distance = -math.hypot(midx - cx, midy - cy)

objlist.append([distance,id])

objlist.sort()

#move them to the top of the object stack in this order.

for item in objlist:

parentnode.append( objects[item[1]])

def restack_z_order(self):

parentnode = None

objects = []

if len(self.selected) == 1:

firstobject = self.selected[self.options.ids[0]]

if firstobject.tag == inkex.addNS('g', 'svg'):

parentnode = firstobject

for child in parentnode.iterchildren(reversed=False):

objects.append(child)

else:

parentnode = self.current_layer

for id_ in zSort(self.document.getroot(), self.selected.keys()):

objects.append(self.selected[id_])

if self.options.zsort == "rev":

objects.reverse()

elif self.options.zsort == "rand":

random.shuffle(objects)

if parentnode is not None:

for item in objects:

parentnode.append(item)

if __name__ == '__main__':

e = Restack()

e.affect()