# -*- coding: utf-8 -*-

 import pylab

 import random

 class MiniPlotTool :

     '''

     A mini tool to draw lines using pylab

     '''

     basecolors = ['red','green','yellow','blue','black','cyan','magenta']

     def __init__(self, baseConfig) :

         self.figsize = baseConfig.get('figsize',None)

         self.axis = baseConfig.get('axis',None)

         self.title = baseConfig.get('title','NoName')

         self.ylabel = baseConfig.get('ylabel','NoName')

         self.grid = baseConfig.get('grid',False)

         self.xaxis_locator = baseConfig.get('xaxis_locator',None)

         self.yaxis_locator = baseConfig.get('yaxis_locator',None)

         self.legend_loc = baseConfig.get('legend_loc',0)

         if self.figsize != None :

             pylab.figure(figsize = self.figsize)

         if self.axis != None :

             pylab.axis(self.axis)

         pylab.title(self.title)

         pylab.ylabel(self.ylabel)

         ax = pylab.gca()

         pylab.grid(self.grid)

         if self.xaxis_locator != None :

             ax.xaxis.set_major_locator( pylab.MultipleLocator(self.xaxis_locator) )

         if self.yaxis_locator != None :

             ax.yaxis.set_major_locator( pylab.MultipleLocator(self.yaxis_locator) )

         self.lineList = []

         self.id = 1

     def addline(self, lineConf) :

         self.lineList.append((self.id, lineConf))

         self.id += 1

         return {'id' : self.id - 1}

     def removeline(self, lineId) :

         for i in range(len(self.lineList)) :

             id, conf = self.lineList[i]

             if id == lineId :

                 del self.lineList[i]

                 break

         else :

             return {'status' : -1}

         print len(self.lineList)

         return {'status' : 0}

     def __parselineConf(self, lineConf) :

         X = lineConf['X']

         Y = lineConf['Y']

         marker = lineConf.get('marker',None)

         color = lineConf.get('color', random.choice(MiniPlotTool.basecolors))

         markerfacecolor = lineConf.get('markerfacecolor',color)

         label = lineConf.get('label','NoName')

         linewidth = lineConf.get('linewidth',1)

         linestyle = lineConf.get('linestyle','-')

         return X, Y, marker, color, markerfacecolor, label, linewidth, linestyle

     def plotSingleLine(self, lineConf):

         X, Y, marker, color, markerfacecolor, label, linewidth, linestyle = self.__parselineConf(lineConf)

         pylab.plot(X, Y, marker = marker, color = color, markerfacecolor = markerfacecolor, label=label, linewidth = linewidth, linestyle = linestyle)

         pylab.legend(loc = self.legend_loc)

     def plot(self) :

         colors = [MiniPlotTool.basecolors[i % len(MiniPlotTool.basecolors)] for i in range(len(self.lineList))]

         for i in range(len(self.lineList)) :

             id, conf = self.lineList[i]

             if conf.get('color',None) :

                 conf['color'] = colors[i]

             X, Y, marker, color, markerfacecolor, label, linewidth, linestyle = self.__parselineConf(conf)

             pylab.plot(X, Y, marker = marker, color = color, markerfacecolor = markerfacecolor, label=label, linewidth = linewidth, linestyle = linestyle)

         pylab.legend(loc = self.legend_loc)

     def show(self) :

         pylab.show()

 if __name__ == '__main__' :

     #test

     baseConfig = {

         #'figsize' : (6,8),

         #'axis': [0,10,0,10],

         #'title' : 'hello title',

         #'ylabel' : 'hello ylabel',

         'grid' : True,

         #'xaxis_locator' : 0.5,

         #'yaxis_locator' : 1,

         #'legend_loc' : 'upper right'

     }

     tool = MiniPlotTool(baseConfig)

     X = [ i for i in range(10)]

     Y = [random.randint(1,10) for i in range(10)]

     Y2 = [random.randint(1,10) for i in range(10)]

     lineConf = {

         'X' : X,

         'Y' : Y

         #'marker' : 'x',

         #'color' : 'b',

         #'markerfacecolor' : 'r',

         #'label' : '222',

         #'linewidth' : 3,

         #'linestyle' : '--'

     }

     lineConf2 = {

         'X' : X,

         'Y' : Y2,

         'marker' : 'o',

         'color' : 'b',

         'markerfacecolor' : 'r',

         'label' : '',

         'linewidth' : 3,

         'linestyle' : '--'

     }

     #tool.plotSingleLine(lineConf)

     print tool.addline(lineConf)

     print tool.addline(lineConf2)

     #print tool.removeline(1)

     tool.plot()

     tool.show()