mathplotlibNode.py

# -*- coding: utf-8 -*-
#-------------------------------------------------
#-- animation workbench
#--
#-- microelly 2016 v 0.1
#--
#-- GNU Lesser General Public License (LGPL)
#-------------------------------------------------
from __future__ import unicode_literals

__vers__="08.04.2016  0.4"

import sys
import os
import random
import numpy as np
import time

__dir__ = os.path.dirname(__file__)	


#import matplotlib

#matplotlib.use('Qt4Agg')
#matplotlib.rcParams['backend.qt4']='PySide'

#from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
#from matplotlib.figure import Figure


import say
reload(say)
from say import *

import reconstruction
reload (reconstruction.projectiontools)
from reconstruction.projectiontools import *

import reconstruction.miki as miki
reload(miki)

import Animation

class _MPL(Animation._Actor):

	def __init__(self,obj):
		obj.Proxy = self
		self.Type = self.__class__.__name__
		self.obj2 = obj
		self.vals={}
		_ViewProviderMPL(obj.ViewObject) 


	def onChanged(self,obj,prop):
#		say(["onChanged " + str(self),obj,prop,obj.getPropertyByName(prop)])
		if prop == 'countSources':
			for i in range(obj.countSources):
				try:
					obj.getPropertyByName('source'+str(i+1)+'Object')
				except:
					obj.addProperty('App::PropertyLink','source'+str(i+1)+'Object',"Source " + str(i+1))
					obj.addProperty('App::PropertyString','source'+str(i+1)+'Data',"Source " + str(i+1))
					obj.addProperty('App::PropertyFloatList','source'+str(i+1)+'Values',"Source " + str(i+1))
					obj.addProperty('App::PropertyBool','source'+str(i+1)+'Off',"Source " + str(i+1))
					exec("self.vals"+str(i+1)+"={}")
			for i in range(10):
				if i<obj.countSources: mode=0
				else: mode=2
				try:
					obj.setEditorMode("source"+str(i+1)+"Object", mode)
					obj.setEditorMode("source"+str(i+1)+"Data", mode)
					obj.setEditorMode("source"+str(i+1)+"Values", mode)
					obj.setEditorMode("source"+str(i+1)+"Off", mode)
				except:
					break
		pass


	def execute(self,obj):

		if obj.mode=='histogram':
			# self.edit()
			say("plot ------------------")
			try:
				app=obj.Proxy.app
				app.plot()
			except:
				sayW("Error for call obj.Proxy.app.plot()")
			say("plot ----------------done   --")
			return

		if not obj.record:
			say(obj.Label+ " no recording")
			return
		

		try: t=self.vals
		except: self.vals={}
		src=obj.sourceObject
		vs='src.'+obj.sourceData
		v=eval(vs)
		self.vals[v]=v

		for i in range(obj.countSources):
			exec("src"+str(i+1)+"=obj.source"+str(i+1)+"Object")
			exec("ss=obj.source"+str(i+1)+"Object")
			if ss != None:
				vs2="obj.source"+str(i+1)+"Data"
				v2=eval(vs2)
				vs3="ss."+v2
				v3=eval(vs3)
				tt=eval("self.vals"+str(i+1))
				tt[v]=v3
		return


class _ViewProviderMPL(Animation._ViewProviderActor):


	def __init__(self,vobj):
		self.attach(vobj)
		self.Object = vobj.Object
		vobj.Proxy = self
		self.vers=__vers__


	def attach(self,vobj):
		self.emenu=[]
		self.cmenu=[]
		self.Object = vobj.Object
		vobj.Proxy = self
		self.vers=__vers__


	def getIcon(self):
		return  __dir__+ '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/icons/icon1.svg'


	def createDialog(self):
		app=MyApp()
		miki2=miki.Miki()
		miki2.app=app
		app.root=miki2
		app.obj=self.Object
		self.Object.Proxy.app=app
		self.edit= lambda:miki2.run(MyApp.s6,app.create2)

	def setupContextMenu(self, obj, menu):
		self.createDialog()

		cl=self.Object.Proxy.__class__.__name__
		action = menu.addAction("About " + cl)
		action.triggered.connect(self.showVersion)

		action = menu.addAction("Edit ...")
		action.triggered.connect(self.edit)


	def setEdit(self,vobj,mode=0):
		self.createDialog()
		self.edit()
		#FreeCAD.ActiveDocument.recompute()
		return True



class MatplotlibWidget(FigureCanvas):

	def __init__(self, parent=None, width=5, height=4, dpi=100):

		super(MatplotlibWidget, self).__init__(Figure())
		self.setParent(parent)
		self.figure = Figure(figsize=(width, height), dpi=dpi) 
		self.canvas = FigureCanvas(self.figure)

		FigureCanvas.setSizePolicy(self,
				QtGui.QSizePolicy.Expanding,
				QtGui.QSizePolicy.Expanding)
		FigureCanvas.updateGeometry(self)

		self.axes = self.figure.add_subplot(111)
		self.setMinimumSize(self.size())



class MyApp(object):


	s6='''
VerticalLayout:
		id:'main'
#		setFixedHeight: 500
#		setFixedWidth: 500
#		move:  PySide.QtCore.QPoint(3000,100)

#		QtGui.QLabel:
#			setText:"***    My    M A T P L O T L I B     ***"
#		

'''


	def plot(self):
		if self.obj.mode=='histogram':
			self.mpl.figure.clf()
			self.mpl.canvas = FigureCanvas(self.mpl.figure)
			FigureCanvas.updateGeometry(self.mpl)

			self.mpl.axes = self.mpl.figure.add_subplot(111)
			self.mpl.draw()
			FreeCAD.mpl=self.mpl


			k=self.plot_histogram()
			
#			FreeCAD.k=k
#			self.mpl.axes.set_xlabel('length')
#			self.mpl.axes.set_ylabel('count')
#			self.mpl.axes.title=self.obj.Label

			return
		
		self.mpl.figure.clf()
		self.mpl.canvas = FigureCanvas(self.mpl.figure)
		FigureCanvas.updateGeometry(self.mpl)

		self.mpl.axes = self.mpl.figure.add_subplot(111)
		self.mpl.draw()

		vals=self.obj.Proxy.vals
		x=[]
		y=[]

		for k in vals:
			x.append(k)
			y.append(vals[k])

		self.obj.sourceValues=y

		for i in range(self.obj.countSources):
			nr=str(i+1)
			ss=eval("self.obj.source"+nr+"Object")
			sf=eval("self.obj.source"+nr+"Off")
			if ss!=None and not sf:
				exec("vals=self.obj.Proxy.vals"+nr)
				x2=[k for k in vals]
				y1=[vals[k] for k in vals]
				exec("label=self.obj.source"+nr+"Object.Label + ': ' + self.obj.source"+nr+"Data")
				t=self.mpl.axes.plot(x,y1,label=label)
				exec("self.obj.source"+nr+"Values=y1")

			if ss== None and not sf and not self.obj.useNumpy:
				say("no sourcve .jijij")
				exec("vals=self.obj.source"+nr+"Values")
#				x2=[k for k in vals]
#				y1=[vals[k] for k in vals]
#				say(vals)
				y1=vals
				x=range(len(vals))
				exec("label=self.obj.source"+nr+"Data")
				# label="Label for " + str(nr) + ": "+ label 
				t=self.mpl.axes.plot(x,y1,label=label)
#				exec("self.obj.source"+nr+"Values=y1")
				say("DDone")

		if self.obj.useNumpy:
			self.obj.outTime=self.obj.sourceNumpy.outTime

		FreeCAD.activeDocument().recompute()

		for i in range(10):
				if eval("self.obj.useOut"+str(i)):
					try:
						y=self.obj.sourceNumpy.getPropertyByName('out'+str(i))
						label=self.obj.sourceNumpy.getPropertyByName('label'+str(i))
						if label=='':
							label="numpy " + str(i)

	#					if x == []: 
						x=range(len(y))
						if self.obj.outTime!=[]:
							x=self.obj.outTime
						say(("lens",len(x),len(y)))
						
						t=self.mpl.axes.plot(x,y,label=label)
						exec("self.obj.out"+str(i)+"="+str(y))
					except:
						sayexc("cannont calculate out"+str(i))

		legend = self.mpl.axes.legend(loc='upper right', shadow=True)
		self.mpl.draw()
		self.mpl.figure.canvas.draw()


	def plot_histogram(self): # for mode ==histogram
		
		self.mpl.figure.clf()
		self.mpl.canvas = FigureCanvas(self.mpl.figure)
		FigureCanvas.updateGeometry(self.mpl)

		self.mpl.axes = self.mpl.figure.add_subplot(111)
		self.mpl.draw()



		sob=self.obj.source1Object
		y="sob."+str(self.obj.source1Data)
		vals=eval(y)
		# Proxy.extras.linelengths2
		# say(vals)
		
		FreeCAD.mpl=self.mpl
		# self.mpl.axes.axis([-90, 90, 0, 100])
		n, bins, patches = self.mpl.axes.hist(vals, 180, normed=0, facecolor='green', alpha=0.75)
		self.mpl.axes.axis([0, 180, 0, np.max(n)])
		

		# legend = self.mpl.axes.legend(loc='upper right', shadow=True)
		
		self.mpl.draw()
		self.mpl.figure.canvas.draw()


	def reset(self):
		self.obj.Proxy.vals={}
		self.obj.sourceValues=[]
		for i in range(self.obj.countSources):
			nr=str(i+1)
			exec("self.obj.Proxy.vals"+nr+"={}")
			exec("self.obj.source"+nr+"Values=[]")

		self.mpl.figure.clf()
		self.mpl.canvas = FigureCanvas(self.mpl.figure)

		FigureCanvas.updateGeometry(self.mpl)
		self.mpl.axes = self.mpl.figure.add_subplot(111)
		self.mpl.draw()
		self.plot()


	def create2(self):
		par=self.root.ids['main']

		l=QtGui.QLabel(self.obj.Label)
		
		self.mpl=MatplotlibWidget()
		bt=QtGui.QPushButton("update diagram")
		bt.clicked.connect(self.plot)

		bt2=QtGui.QPushButton("reset data")
		bt2.clicked.connect(self.reset)
		self.root.ids['main'].layout.addWidget(l)
		self.root.ids['main'].layout.addWidget(self.mpl)
		self.root.ids['main'].layout.addWidget(bt)
		self.root.ids['main'].layout.addWidget(bt2)
		self.root.ids['main'].layout.setStretchFactor(self.mpl, 1)

		self.plot()



def createMPL(base=False):

	obj=FreeCAD.ActiveDocument.addObject('App::DocumentObjectGroupPython','Plot')
	obj.addProperty('App::PropertyString','mode',"Base")

	obj.addProperty('App::PropertyBool','record',"Base",'true record, false no record data')
	obj.addProperty('App::PropertyInteger','countSources',"Base")
	obj.countSources=0

	# base data time/step
	obj.addProperty('App::PropertyLink','sourceObject',"Time")
	obj.addProperty('App::PropertyString','sourceData',"Time")
	obj.addProperty('App::PropertyFloatList','sourceValues',"Time")

	obj.addProperty('App::PropertyBool','useNumpy',"numpy source" )
	obj.addProperty('App::PropertyLink','sourceNumpy',"numpy source" )

	for i in range(10):
		obj.addProperty('App::PropertyBool','useOut'+str(i),"numpy source" )
		obj.addProperty('App::PropertyFloatList','out'+str(i),"out values")

	obj.addProperty('App::PropertyFloatList','outTime',"out values")

	if not base:
		_MPL(obj)
		#_ViewProviderMPL(obj.ViewObject,__dir__+ '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/icons/icon1.svg') 
		
		obj.countSources=1

		app=MyApp()
		miki2=miki.Miki()
		miki2.app=app
		app.root=miki2
		app.obj=obj
		obj.Proxy.app=app

		obj.ViewObject.Proxy.edit= lambda:miki2.run(MyApp.s6,app.create2)
	return obj




if False:

	t=createMPL()
	t.sourceObject= App.ActiveDocument.My_Manager
	t.sourceData="step"

	t.source1Object= App.ActiveDocument.Box001
	t.source1Data="Placement.Rotation.Angle"


	t.source2Object= App.ActiveDocument.Box001
	t.source2Data="Placement.Base.x"

	t.source3Object= App.ActiveDocument.Box001
	t.source3Data="Placement.Base.y"

	# t.expression2="2* np.arctan2(in2,in3)"

	t.record=True

	t.source2Values=[1,2,3,4]
	t.source3Values=[1,3,2,5]
	App.ActiveDocument.Numpy.sourceObject=t


	t2=createMPL()
	t2.sourceObject= App.ActiveDocument.My_Manager
	t2.sourceData="step"

	t2.source1Object= App.ActiveDocument.Box001
	t2.source1Data="Placement.Rotation.Angle"


	t2.sourceObject= App.ActiveDocument.My_Manager
	t2.useNumpy=True
	t2.sourceNumpy=App.ActiveDocument.Numpy
	t2.useOut0=True
	t2.useOut1=True
	t2.useOut2=True
	t2.record=True