/[escript]/trunk/escriptcore/py_src/pdetools.py
ViewVC logotype

Diff of /trunk/escriptcore/py_src/pdetools.py

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

revision 6413 by jfenwick, Mon Jul 4 05:32:05 2016 UTC revision 6414 by gross, Wed Nov 2 02:44:31 2016 UTC
# Line 415  class Locator(object): Line 415  class Locator(object):
415          """          """
416          if isinstance(data,escore.Data):          if isinstance(data,escore.Data):
417             dat=util.interpolate(data,self.getFunctionSpace())             dat=util.interpolate(data,self.getFunctionSpace())
418             id=self.getId()             ii=self.getId()
419             r=data.getRank()             r=data.getRank()
420             if isinstance(id,list):             if isinstance(ii,list):
421                 out=[]                 out=[]
422                 for i in id:                 for i in ii:
423                    o=numpy.array(dat.getTupleForGlobalDataPoint(*i))                    # workaround for bug #391
424                      if dat.isComplex():
425                          o=numpy.array(dat.real().getTupleForGlobalDataPoint(*i))+1j*numpy.array(dat.imag().getTupleForGlobalDataPoint(*i))
426                      else:
427                         o=numpy.array(dat.getTupleForGlobalDataPoint(*i))
428                    if data.getRank()==0:                    if data.getRank()==0:
429                       out.append(o[0])                       out.append(o[0])
430                    else:                    else:
431                       out.append(o)                       out.append(o)
432                 return out                 return out
433             else:             else:
434               out=numpy.array(dat.getTupleForGlobalDataPoint(*id))               # workaround for bug #391
435                 if dat.isComplex():
436                     print "DDD"
437                     out=numpy.array(dat.real().getTupleForGlobalDataPoint(*ii))+1j*numpy.array(dat.imag().getTupleForGlobalDataPoint(*ii))
438                 else:
439                     out=numpy.array(dat.getTupleForGlobalDataPoint(*ii))
440               if data.getRank()==0:               if data.getRank()==0:
441                  return out[0]                  return out[0]
442               else:               else:

Legend:
Removed from v.6413  
changed lines
  Added in v.6414

  ViewVC Help
Powered by ViewVC 1.1.26