/[escript]/trunk/escript/py_src/modelframe.py
ViewVC logotype

Diff of /trunk/escript/py_src/modelframe.py

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 838 by gross, Tue Sep 5 22:45:55 2006 UTC revision 874 by elspeth, Tue Oct 17 12:06:11 2006 UTC
# Line 24  __date__="$Date$" Line 24  __date__="$Date$"
24    
25  from types import StringType,IntType,FloatType,BooleanType,ListType,DictType  from types import StringType,IntType,FloatType,BooleanType,ListType,DictType
26  from sys import stdout  from sys import stdout
27    import numarray
28    import operator
29  import itertools  import itertools
30  # import modellib  temporarily removed!!!  # import modellib  temporarily removed!!!
31    
# Line 364  class ParameterSet(LinkableObject): Line 366  class ParameterSet(LinkableObject):
366       - a ParameterSet object       - a ParameterSet object
367       - a Simulation object       - a Simulation object
368       - a Model object       - a Model object
369       - any other object (not considered by writeESySXML and writeXML)       - a numarray object
370             - a list of booleans
371            - any other object (not considered by writeESySXML and writeXML)
372            
373      Example how to create an ESySParameters object::      Example how to create an ESySParameters object::
374            
# Line 476  class ParameterSet(LinkableObject): Line 480  class ParameterSet(LinkableObject):
480    
481              val = document.createElement('Value')              val = document.createElement('Value')
482    
483              if isinstance(value,ParameterSet):              if isinstance(value,(ParameterSet,Link)):
484                  value.toDom(document, val)                  value.toDom(document, val)
485                  param.appendChild(val)                  param.appendChild(val)
486              elif isinstance(value, Link):              elif isinstance(value, numarray.NumArray):
487                  value.toDom(document, val)                  shape = value.getshape()
488                    if isinstance(shape, tuple):
489                        size = reduce(operator.mul, shape)
490                        shape = ' '.join(map(str, shape))
491                    else:
492                        size = shape
493                        shape = str(shape)
494    
495                    arraytype = value.type()
496                    numarrayElement = document.createElement('NumArray')
497                    numarrayElement.appendChild(dataNode(document, 'ArrayType', str(arraytype)))
498                    numarrayElement.appendChild(dataNode(document, 'Shape', shape))
499                    numarrayElement.appendChild(dataNode(document, 'Data', ' '.join(
500                        [str(x) for x in numarray.reshape(value, size)])))
501                    val.appendChild(numarrayElement)
502                  param.appendChild(val)                  param.appendChild(val)
503              elif isinstance(value,StringType):              elif isinstance (value, list):
504                  param.appendChild(dataNode(document, 'Value', value))                  param.appendChild(dataNode(document, 'Value', ' '.join(
505                        [str(x) for x in value])
506                    ))
507              else:              else:
508                  param.appendChild(dataNode(document, 'Value', str(value)))                  param.appendChild(dataNode(document, 'Value', str(value)))
509    
# Line 496  class ParameterSet(LinkableObject): Line 516  class ParameterSet(LinkableObject):
516              """              """
517              Remove the empty nodes from the children of this node.              Remove the empty nodes from the children of this node.
518              """              """
519              return [x for x in node.childNodes              ret = []
520                      if not isinstance(x, minidom.Text) or x.nodeValue.strip()]              for x in node.childNodes:
521                    if isinstance(x, minidom.Text):
522                        if x.nodeValue.strip():
523                            ret.append(x)
524                    else:
525                        ret.append(x)
526                return ret
527    
528          def _floatfromValue(doc):          def _floatfromValue(doc):
529              return float(doc.nodeValue.strip())              return float(doc.nodeValue.strip())
# Line 509  class ParameterSet(LinkableObject): Line 535  class ParameterSet(LinkableObject):
535              return int(doc.nodeValue.strip())              return int(doc.nodeValue.strip())
536    
537          def _boolfromValue(doc):          def _boolfromValue(doc):
538              return bool(doc.nodeValue.strip())              return _boolfromstring(doc.nodeValue.strip())
539    
540          def _nonefromValue(doc):          def _nonefromValue(doc):
541              return None              return None
542          
543            def _numarrayfromValue(doc):
544                for node in _children(doc):
545                    if node.tagName == 'ArrayType':
546                        arraytype = node.firstChild.nodeValue.strip()
547                    if node.tagName == 'Shape':
548                        shape = node.firstChild.nodeValue.strip()
549                        shape = [int(x) for x in shape.split()]
550                    if node.tagName == 'Data':
551                        data = node.firstChild.nodeValue.strip()
552                        data = [float(x) for x in data.split()]
553                return numarray.reshape(numarray.array(data, type=getattr(numarray, arraytype)),
554                                        shape)
555          
556            def _listfromValue(doc):
557                return [_boolfromstring(x) for x in doc.nodeValue.split()]
558    
559    
560            def _boolfromstring(s):
561                if s == 'True':
562                    return True
563                else:
564                    return False
565          # Mapping from text types in the xml to methods used to process trees of that type          # Mapping from text types in the xml to methods used to process trees of that type
566          ptypemap = {"Simulation": Simulation.fromDom,          ptypemap = {"Simulation": Simulation.fromDom,
567                      "Model":Model.fromDom,                      "Model":Model.fromDom,
# Line 523  class ParameterSet(LinkableObject): Line 571  class ParameterSet(LinkableObject):
571                      "int":_intfromValue,                      "int":_intfromValue,
572                      "str":_stringfromValue,                      "str":_stringfromValue,
573                      "bool":_boolfromValue,                      "bool":_boolfromValue,
574                      "NoneType":_nonefromValue                      "list":_listfromValue,
575                        "NumArray":_numarrayfromValue,
576                        "NoneType":_nonefromValue,
577                      }                      }
578    
579  #        print doc.toxml()  #        print doc.toxml()
# Line 540  class ParameterSet(LinkableObject): Line 590  class ParameterSet(LinkableObject):
590    
591                  if childnode.tagName == "Value":                  if childnode.tagName == "Value":
592                      nodes = _children(childnode)                      nodes = _children(childnode)
593                    #    if ptype == 'NumArray':
594                     #       pvalue = _numarrayfromValue(nodes)
595                     #   else:
596                      pvalue = ptypemap[ptype](nodes[0])                      pvalue = ptypemap[ptype](nodes[0])
597    
598              parameters[pname] = pvalue              parameters[pname] = pvalue

Legend:
Removed from v.838  
changed lines
  Added in v.874

  ViewVC Help
Powered by ViewVC 1.1.26