modellib/py_src/visualization.py

# $Id$

from escript.modelframe import Model

class Visualization(Model):
       """

             generic visualization of scalar, vector and tensorial data (not implemeted yet)

             t:      current time
             scalar: scalar data set
             vector: vector data set
             tensor: tensor data set
             stride: visulaization is done every strides time step
             filename: name of the movie file

       """
       def __init__(self,debug=False):
            Model.__init__(self,debug=debug)
            self.declareParameter(t=0.,scalar=None,vector=None,tensor=None,stride=1,movie="movie.mpg",counter=0)

       def doInitialization(self):
           self.__n=0
           self.__scene=None

       def doStepPostprocessing(self,dt):
           self.__n+=1
           if self.__n%self.stride:
               data=self.scalar
               if data!=None:
                   pass
               data=self.vector
               if data!=None:
                   pass
               data=self.tensor
               if data!=None:
                   pass

       def doFinalization(self):
            # make the movie into self.filename
            pass

class WriteVTK(Visualization):
       """ writes data into a VTK file for further processing. Currently data are written in several files for
           each data type. This may change in the future.

             scalar: scalar data set
             vector: vector data set
             tensor: tensor data set
             stride: file is written every stride-th time step
             filename: name of the data files. use %s for indication of data type (s,v,t) and time step id.

       """
       def __init__(self,debug=False):
            Visualization.__init__(self,debug=debug)
            self.declareParameter(filename="data.%s.xml")

       def doStepPostprocessing(self,dt):
           self.counter+=1
           if self.counter%self.stride==0:
               n=self.counter/self.stride
               data=self.scalar
               if hasattr(data,"saveVTK"): data.saveVTK(self.filename%("s.%d"%n))
               data=self.vector
               if hasattr(data,"saveVTK"): data.saveVTK(self.filename%("v.%d"%n))
               data=self.tensor
               if hasattr(data,"saveVTK"): data.saveVTK(self.filename%("t.%d"%n))
1	# $Id$
2
3	from escript.modelframe import Model
4
5	class Visualization(Model):
6	"""
7
8	generic visualization of scalar, vector and tensorial data (not implemeted yet)
9
10	t: current time
11	scalar: scalar data set
12	vector: vector data set
13	tensor: tensor data set
14	stride: visulaization is done every strides time step
15	filename: name of the movie file
16
17	"""
18	def __init__(self,debug=False):
19	Model.__init__(self,debug=debug)
20	self.declareParameter(t=0.,scalar=None,vector=None,tensor=None,stride=1,movie="movie.mpg",counter=0)
21
22	def doInitialization(self):
23	self.__n=0
24	self.__scene=None
25
26	def doStepPostprocessing(self,dt):
27	self.__n+=1
28	if self.__n%self.stride:
29	data=self.scalar
30	if data!=None:
31	pass
32	data=self.vector
33	if data!=None:
34	pass
35	data=self.tensor
36	if data!=None:
37	pass
38
39	def doFinalization(self):
40	# make the movie into self.filename
41	pass
42
43	class WriteVTK(Visualization):
44	""" writes data into a VTK file for further processing. Currently data are written in several files for
45	each data type. This may change in the future.
46
47	scalar: scalar data set
48	vector: vector data set
49	tensor: tensor data set
50	stride: file is written every stride-th time step
51	filename: name of the data files. use %s for indication of data type (s,v,t) and time step id.
52
53	"""
54	def __init__(self,debug=False):
55	Visualization.__init__(self,debug=debug)
56	self.declareParameter(filename="data.%s.xml")
57
58	def doStepPostprocessing(self,dt):
59	self.counter+=1
60	if self.counter%self.stride==0:
61	n=self.counter/self.stride
62	data=self.scalar
63	if hasattr(data,"saveVTK"): data.saveVTK(self.filename%("s.%d"%n))
64	data=self.vector
65	if hasattr(data,"saveVTK"): data.saveVTK(self.filename%("v.%d"%n))
66	data=self.tensor
67	if hasattr(data,"saveVTK"): data.saveVTK(self.filename%("t.%d"%n))
Name	Value
svn:eol-style	native
svn:keywords	Author Date Id Revision