""" Constructor.

Defines "--what" option of a script."""

def __init__(self):

# Call base class construtor.

inkex.Effect.__init__(self)

# Option parser:

# -f, --axis_from

# -t, --axis_to

# -N, --num_ticks

# -n, --sub_ticks

# -p, --decimals (precision)

# -T, --axis_title

# -d, --direction

# -o, --axis_offset

# -F, --text_font

# -s, --text_size_axis

# -S, --text_size_title

# -b, --tab

self.OptionParser.add_option("-f", "--axis_from",

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

dest="axis_from", default=0,

help="Axis starting value")

self.OptionParser.add_option("-t", "--axis_to",

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

dest="axis_to", default=10,

help="Axis ending value")

self.OptionParser.add_option("-n", "--num_ticks",

action="store", type="int",

dest="num_ticks", default=5,

help="Number of ticks")

self.OptionParser.add_option("-N", "--sub_ticks",

action="store", type="int",

dest="sub_ticks", default=0,

help="Number of sub-ticks")

self.OptionParser.add_option("-p", "--decimals",

action="store", type="int",

dest="decimals", default=1,

help="Number of ticks")

self.OptionParser.add_option("-o", "--axis_offset",

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

dest="axis_offset", default=5.0,

help="Offset of the axis from the bounding box of selected object")

self.OptionParser.add_option("-T", "--axis_title",

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

dest="text", default='',

help="Title of the axis")

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

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

dest="direction", default='',

help="Direction of the axis")

self.OptionParser.add_option("-F", "--text_font",

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

dest="text_font", default="Arial",

help="The font for the text")

self.OptionParser.add_option("-s", "--text_size_axis",

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

dest="text_size_axis", default=11.0,

help="The font for the text")

self.OptionParser.add_option("-S", "--text_size_title",

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

dest="text_size_title", default=13.0,

help="The font for the text")

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

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

dest="tab", default='',

help="The tab of the interface")

def draw_axis(self, x_pos, y_pos, width, height, axis_from, axis_to, num_ticks, sub_ticks, axis_offset,

decimals, text, text_font, text_size_axis, text_size_title, direction, grp):

# Length of ticks

ticklength = 5

subticklength = 3

font_axis_ticks = {

'font-size' : str(text_size_axis)+'px',

'fill' : '#000000',

'font-family' : text_font,

'text-align' : 'center',

'text-anchor' : 'middle'

}

font_axis_title = {

'font-size' : str(text_size_title)+'px',

'fill' : '#000000',

'font-family' : text_font,

'text-align' : 'center',

'text-anchor' : 'middle'

}

if direction == 'vertical':

font_axis_ticks['text-align'] = "end"

font_axis_ticks['text-anchor'] = "end"

# The relative amount per tick

# e.g. we're going from 0 to 8 and have 5 ticks

# each tick will be 2, so we have 0, 2, 4, 6, 8

tickamount = (axis_to - axis_from)/(num_ticks-1)

if direction=="horizontal":

# TODO: allow axis position on other sides of the bbox

# Align to bounding box

x1 = x_pos

x2 = x_pos + width

y1 = y2 = y_pos + axis_offset

draw_SVG_line(x1, y1, x2, y2, "myaxis", grp)

i = 0

tickspace = float(x2-x1)/float(num_ticks-1)

subtickspace = tickspace/float(sub_ticks+1)

# Draw the ticks

for i in range(num_ticks):

ticktxt = ('%0.'+str(decimals)+'f') % (i*tickamount + axis_from)

# Draw main ticks

draw_SVG_line(i*tickspace+x1, y1, i*tickspace+x1, y1+ticklength,

"myaxis_tick"+str(i), grp)

# Draw minor ticks

if i<(num_ticks-1): # Do not put minor ticks after the last major one!

for j in range(1, sub_ticks+1):

draw_SVG_line(i*tickspace+x1+j*subtickspace, y1, i*tickspace+x1+j*subtickspace,

y1+subticklength, "myaxis_subtick"+str(i), grp)

# Draw text associated to main ticks

draw_SVG_text(i*tickspace+x1, y1+ticklength+text_size_axis, ticktxt, grp, font_axis_ticks)

# Draw the title

draw_SVG_text((x1+x2)/2, y1+ticklength+text_size_axis+text_size_title, text, grp, font_axis_title)

else: # Vertical axis

# Align to bounding box

y1 = y_pos

y2 = y_pos - height

x1 = x2 = x_pos - axis_offset

draw_SVG_line(x1, y1, x2, y2, "myaxis", grp)

i = 0

tickspace = float(y2-y1)/float(num_ticks-1)

subtickspace = tickspace/float(sub_ticks+1)

# The maximum length (in charachters) of a tick text

maxlen = 0

for i in range(num_ticks):

ticktxt = ('%0.'+str(decimals)+'f') % (i*tickamount + axis_from)

maxlen = max(maxlen, len(ticktxt))

draw_SVG_line(x1, i*tickspace+y1, x1-ticklength, i*tickspace+y1,

"myaxis_tick"+str(i), grp)

# Draw minor ticks

if i<(num_ticks-1): # Do not put minor ticks after the last major one!

for j in range(1, sub_ticks+1):

draw_SVG_line(x1, i*tickspace+y1+j*subtickspace, x1-subticklength,

i*tickspace+y1+j*subtickspace, "myaxis_subtick"+str(i), grp)

# Draw text associated to main ticks

ticktxt = draw_SVG_text(x1-ticklength-0.3*text_size_axis, i*tickspace+y1+2.5, ticktxt, grp, font_axis_ticks)

# Calculate the width of the tick text and position the title on the left of them

# simpletransform.computeBBox does not support text (as of 2014/06/12)

# So another solution needs to be used...

# An approximate the position of (number of charachters * font size * 0.6) seems to be OK

fontsize = float(font_axis_ticks["font-size"].replace("px", ""))

draw_SVG_text(x1-ticklength-maxlen*fontsize*0.6,

(y1+y2)/2, text, grp, font_axis_title, -90)

def effect(self):

if len(self.options.ids) == 0:

inkex.errormsg("Please select an object.")

exit()

# Collect document ids

doc_ids = {}

docIdNodes = self.document.xpath('//@id')

for m in docIdNodes:

doc_ids[m] = 1

grpname = 'axis'

# Make sure that the id/name is inique

index = 0

while (doc_ids.has_key(grpname)):

grpname = 'axis' + str(index)

index = index + 1

grp_name = grpname

grp_attribs = {inkex.addNS('label','inkscape'):grp_name}

# The group to put everything in

grp = inkex.etree.SubElement(self.current_layer, 'g', grp_attribs)

# Get the bounding box

bbox = simpletransform.computeBBox(self.selected.values())

width = int(bbox[1] - bbox[0])

height = int(bbox[3] - bbox[2])

x_pos = bbox[0]

y_pos = bbox[3]

self.draw_axis(x_pos, y_pos, width, height,

self.options.axis_from, self.options.axis_to,

self.options.num_ticks, self.options.sub_ticks,

self.options.axis_offset, self.options.decimals,

self.options.text, self.options.text_font,

self.options.text_size_axis, self.options.text_size_title,