def plotContinents(self,x1,x2,y1,y2,projection,wrap,tmpl): contData = vcs2vtk.prepContinents(self.canvas._continents) contMapper = vtk.vtkPolyDataMapper() contMapper.SetInputData(contData) contActor = vtk.vtkActor() contActor.SetMapper(contMapper) contActor.GetProperty().SetColor(0.,0.,0.) contActor = vcs2vtk.doWrap(contActor,[x1,x2,y1,y2],wrap) if projection.type!="linear": contData=contActor.GetMapper().GetInput() cpts = contData.GetPoints() geo, gcpts = vcs2vtk.project(cpts,projection,[x1,x2,y1,y2]) contData.SetPoints(gcpts) contMapper = vtk.vtkPolyDataMapper() contMapper.SetInputData(contData) contActor = vtk.vtkActor() contActor.SetMapper(contMapper) contActor.GetProperty().SetColor(0.,0.,0.) else: geo=None ren = vtk.vtkRenderer() self.renWin.AddRenderer(ren) self.setLayer(ren,tmpl.data.priority) vcs2vtk.fitToViewport(contActor,ren,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2],wc=[x1,x2,y1,y2],geo=geo) if tmpl.data.priority!=0: ren.AddActor(contActor)
def plotContinents(self,x1,x2,y1,y2,projection,wrap,ren,tmpl): contData = vcs2vtk.prepContinents(self.canvas._continents) contMapper = vtk.vtkPolyDataMapper() contMapper.SetInputData(contData) contActor = vtk.vtkActor() contActor.SetMapper(contMapper) contActor.GetProperty().SetColor(0.,0.,0.) cpts = contData.GetPoints() gcpts = vcs2vtk.project(cpts,projection) contData.SetPoints(gcpts) contActor = vcs2vtk.doWrap(contActor,[x1,x2,y1,y2],wrap) vcs2vtk.fitToViewport(contActor,ren,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2],[x1,x2,y1,y2]) if tmpl.data.priority!=0: ren.AddActor(contActor)
def plotContinents(self,x1,x2,y1,y2,projection,wrap,tmpl): contData = vcs2vtk.prepContinents(self.canvas._continents) contMapper = vtk.vtkPolyDataMapper() contMapper.SetInputData(contData) contActor = vtk.vtkActor() contActor.SetMapper(contMapper) contActor.GetProperty().SetColor(0.,0.,0.) contActor = vcs2vtk.doWrap(contActor,[x1,x2,y1,y2],wrap,fastClip=False) if projection.type!="linear": contData=contActor.GetMapper().GetInput() cpts = contData.GetPoints() geo, gcpts = vcs2vtk.project(cpts,projection,[x1,x2,y1,y2]) contData.SetPoints(gcpts) contMapper = vtk.vtkPolyDataMapper() contMapper.SetInputData(contData) contActor = vtk.vtkActor() contActor.SetMapper(contMapper) contActor.GetProperty().SetColor(0.,0.,0.) else: geo=None ren = self.fitToViewport(contActor,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2], wc=[x1,x2,y1,y2],geo=geo,priority=tmpl.data.priority) return {}
def update_input(self,vtkobjects,array1,array2=None,update=True): if vtkobjects.has_key("vtk_backend_grid"): ## Ok ths is where we update the input data vg=vtkobjects["vtk_backend_grid"] data = vcs2vtk.numpy_to_vtk_wrapper(array1.filled(0.).flat, deep=False) pData= vg.GetPointData().GetScalars() if pData is not None: vg.GetPointData().SetScalars(data) else: vg.GetCellData().SetScalars(data) if vtkobjects.has_key("vtk_backend_filter"): vtkobjects["vtk_backend_filter"].Update() if vtkobjects.has_key("vtk_backend_missing_mapper"): missingMapper,color,cellData = vtkobjects["vtk_backend_missing_mapper"] missingMapper2 = vcs2vtk.putMaskOnVTKGrid(array1,vg,color,cellData,deep=False) else: missingMapper = None if vtkobjects.has_key("vtk_backend_contours"): for c in vtkobjects["vtk_backend_contours"]: c.Update() ports=vtkobjects["vtk_backend_contours"] elif vtkobjects.has_key("vtk_backend_geofilters"): ports=vtkobjects["vtk_backend_geofilters"] else: # Vector plot ports=vtkobjects["vtk_backend_glyphfilters"] w = vcs2vtk.generateVectorArray(array1,array2,vg) vg.GetPointData().AddArray(w) ports[0].SetInputData(vg) if vtkobjects.has_key("vtk_backend_actors"): i=0 for a in vtkobjects["vtk_backend_actors"]: act = a[0] wrp = a[1] if a[1] is missingMapper: i-=1 mapper = missingMapper2 wrp = a[2] else: ## Labeled contours are a different kind if vtkobjects.has_key("vtk_backend_luts"): lut,rg = vtkobjects["vtk_backend_luts"][i] mapper = vtk.vtkPolyDataMapper() elif vtkobjects.has_key("vtk_backend_labeled_luts"): lut,rg = vtkobjects["vtk_backend_labeled_luts"][i] mapper = vtk.vtkLabeledContourMapper() if lut is None: mapper.SetInputConnection(ports[i].GetOutputPort()) else: if mapper.IsA("vtkPolyDataMapper"): mapper.SetInputConnection(ports[i].GetOutputPort()) mapper.SetLookupTable(lut) mapper.SetScalarModeToUsePointData() else: stripper = vtk.vtkStripper() stripper.SetInputConnection(ports[i].GetOutputPort()) mapper.SetInputConnection(stripper.GetOutputPort()) stripper.Update() tprops = vtkobjects["vtk_backend_contours_labels_text_properties"] mapper.GetPolyDataMapper().SetLookupTable(lut) mapper.GetPolyDataMapper().SetScalarModeToUsePointData() mapper.GetPolyDataMapper().SetScalarRange(rg[0],rg[1]) mapper.SetLabelVisibility(1) mapper.SetTextProperties(tprops) if rg[2]: mapper.SetScalarModeToUseCellData() mapper.SetScalarRange(rg[0],rg[1]) act.SetMapper(mapper) act = vcs2vtk.doWrap(a[0],wrp) a[0].SetMapper(act.GetMapper()) i+=1 taxis = array1.getTime() if taxis is not None: tstr = str(cdtime.reltime(taxis[0],taxis.units).tocomp(taxis.getCalendar())) else: tstr = None ## Min/Max/Mean for att in ["Min","Max","Mean","crtime","crdate","zvalue"]: if vtkobjects.has_key("vtk_backend_%s_text_actor" % att): t = vtkobjects["vtk_backend_%s_text_actor" % att] if att == "Min": t.SetInput("Min %g" % array1.min()) elif att == "Max": t.SetInput("Max %g" % array1.max()) elif att == "Mean": if not inspect.ismethod(getattr(array1,'mean')): meanstring = "Mean: %s" % getattr(array1,"mean") else: try: meanstring='Mean %.4g'% float(cdutil.averager(array1, axis = " ".join(["(%s)" % S for S in array1.getAxisIds()]))) except Exception,err: meanstring='Mean %.4g'%array1.mean() t.SetInput(meanstring) elif att=="crdate" and tstr is not None: t.SetInput(tstr.split()[0].replace("-","/")) elif att=="crtime" and tstr is not None: t.SetInput(tstr.split()[1]) elif att=="zvalue": if len(array1.shape)>2: l=array1.getAxis(-3) if l.isTime(): t.SetInput(str(l.asComponentTime()[0])) else: t.SetInput("%g" % l[0])
def plotVector(self,data1,data2,tmpl,gm,ren): self.setLayer(ren,tmpl.data.priority) ug,xm,xM,ym,yM,continents,wrap = vcs2vtk.genUnstructuredGrid(data1,data2,gm) if ug.IsA("vtkUnstructuredGrid"): c2p = vtk.vtkCellDataToPointData() c2p.SetInputData(ug) c2p.Update() #For contouring duplicate points seem to confuse it cln = vtk.vtkCleanUnstructuredGrid() cln.SetInputConnection(c2p.GetOutputPort()) missingMapper = vcs2vtk.putMaskOnVTKGrid(data1,ug,None) u=numpy.ma.ravel(data1) v=numpy.ma.ravel(data2) sh = list(u.shape) sh.append(1) u = numpy.reshape(u,sh) v = numpy.reshape(v,sh) z = numpy.zeros(u.shape) w = numpy.concatenate((u,v),axis=1) w = numpy.concatenate((w,z),axis=1) w = VN.numpy_to_vtk(w,deep=True) w.SetName("vectors") ug.GetPointData().AddArray(w) ## Vector attempt arrow = vtk.vtkArrowSource() l = gm.line if l is None: l = "default" try: l = vcs.getline(l) lwidth = l.width[0] lcolor = l.color[0] lstyle = l.type[0] except: lstyle = "solid" lwidth = 1. lcolor = 0 if gm.linewidth is not None: lwidth = gm.linewidth if gm.linecolor is not None: lcolor = gm.linecolor arrow.SetTipRadius(.1*lwidth) arrow.SetShaftRadius(.03*lwidth) arrow.Update() glyphFilter = vtk.vtkGlyph2D() glyphFilter.SetSourceConnection(arrow.GetOutputPort()) glyphFilter.OrientOn() glyphFilter.SetVectorModeToUseVector() glyphFilter.SetInputArrayToProcess(1,0,0,0,"vectors") glyphFilter.SetScaleFactor(2.*gm.scale) if ug.IsA("vtkUnstructuredGrid"): glyphFilter.SetInputConnection(cln.GetOutputPort()) else: glyphFilter.SetInputData(ug) mapper = vtk.vtkPolyDataMapper() mapper.SetInputConnection(glyphFilter.GetOutputPort()) act = vtk.vtkActor() act.SetMapper(mapper) try: cmap = vcs.elements["colormap"][cmap] except: cmap = vcs.elements["colormap"][self.canvas.getcolormapname()] r,g,b = cmap.index[lcolor] act.GetProperty().SetColor(r/100.,g/100.,b/100.) x1,x2,y1,y2 = vcs2vtk.getRange(gm,xm,xM,ym,yM) act = vcs2vtk.doWrap(act,[x1,x2,y1,y2],wrap) vcs2vtk.fitToViewport(act,ren,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2],[x1,x2,y1,y2]) if tmpl.data.priority!=0: ren.AddActor(act) self.renderTemplate(ren,tmpl,data1,gm) if self.canvas._continents is None: continents = False if continents: projection = vcs.elements["projection"][gm.projection] self.plotContinents(x1,x2,y1,y2,projection,wrap,ren,tmpl)
def plot2D(self,data1,data2,tmpl,gm): ug,xm,xM,ym,yM,continents,wrap,geo,cellData = vcs2vtk.genGrid(data1,data2,gm) #Now applies the actual data on each cell if isinstance(gm,boxfill.Gfb) and gm.boxfill_type=="log10": data1=numpy.ma.log10(data1) data = VN.numpy_to_vtk(data1.filled(0.).flat,deep=True) if cellData: ug.GetCellData().SetScalars(data) else: ug.GetPointData().SetScalars(data) try: cmap = vcs.elements["colormap"][cmap] except: cmap = vcs.elements["colormap"][self.canvas.getcolormapname()] color = getattr(gm,"missing",None) if color is not None: color = cmap.index[color] missingMapper = vcs2vtk.putMaskOnVTKGrid(data1,ug,color,cellData) lut = vtk.vtkLookupTable() mn,mx=vcs.minmax(data1) #Ok now we have grid and data let's use the mapper mapper = vtk.vtkPolyDataMapper() legend = None if isinstance(gm,(meshfill.Gfm,boxfill.Gfb)): geoFilter = vtk.vtkDataSetSurfaceFilter() if cellData: p2c = vtk.vtkPointDataToCellData() p2c.SetInputData(ug) geoFilter.SetInputConnection(p2c.GetOutputPort()) else: geoFilter.SetInputData(ug) geoFilter.Update() if isinstance(gm,(isofill.Gfi,isoline.Gi,meshfill.Gfm)) or \ (isinstance(gm,boxfill.Gfb) and gm.boxfill_type=="custom"): #Now this filter seems to create the good polydata sFilter = vtk.vtkDataSetSurfaceFilter() if cellData: # Sets data to point instead of just cells c2p = vtk.vtkCellDataToPointData() c2p.SetInputData(ug) c2p.Update() if self.debug: vcs2vtk.dump2VTK(c2p) #For contouring duplicate points seem to confuse it if ug.IsA("vtkUntructuredGrid"): cln = vtk.vtkCleanUnstructuredGrid() cln.SetInputConnection(c2p.GetOutputPort()) if self.debug: vcs2vtk.dump2VTK(cln) sFilter.SetInputConnection(cln.GetOutputPort()) else: sFilter.SetInputConnection(c2p.GetOutputPort()) else: sFilter.SetInputData(ug) sFilter.Update() if self.debug: vcs2vtk.dump2VTK(sFilter) if isinstance(gm,isoline.Gi): cot = vtk.vtkContourFilter() if cellData: cot.SetInputData(sFilter.GetOutput()) else: cot.SetInputData(ug) levs = gm.levels if (isinstance(gm,isoline.Gi) and numpy.allclose( levs[0],[0.,1.e20])) or numpy.allclose(levs,1.e20): levs = vcs.mkscale(mn,mx) Ncolors = len(levs) if isinstance(gm,(isofill.Gfi,meshfill.Gfm)): levs2 = vcs.mkscale(mn,mx) levs=[] for i in range(len(levs2)-1): levs.append([levs2[i],levs2[i+1]]) else: if isinstance(gm.levels[0],(list,tuple)): if isinstance(gm,isoline.Gi): levs = [x[0] for x in gm.levels] else: levs = gm.levels else: levs = [] levs2=gm.levels if numpy.allclose(levs2[0],1.e20): levs2[0]=-1.e20 for i in range(len(levs2)-1): levs.append([levs2[i],levs2[i+1]]) if isinstance(gm,isoline.Gi): levs = levs2 Nlevs=len(levs) Ncolors = Nlevs ## Figure out colors if isinstance(gm,boxfill.Gfb): cols = gm.fillareacolors if cols is None: cols = vcs.getcolors(levs2,split=0) elif isinstance(gm,(isofill.Gfi,meshfill.Gfm)): cols = gm.fillareacolors if cols==[1,]: cols = vcs.getcolors(levs2,split=0) elif isinstance(gm,isoline.Gi): cols = gm.linecolors if isinstance(gm,isoline.Gi): cot.SetNumberOfContours(Nlevs) if levs[0]==1.e20: levs[0]=-1.e20 for i in range(Nlevs): cot.SetValue(i,levs[i]) cot.SetValue(Nlevs,levs[-1]) cot.Update() mapper.SetInputConnection(cot.GetOutputPort()) mappers = [] else: mappers = [] LEVS = [] INDX = [] COLS = [] indices = gm.fillareaindices if indices is None: indices=[1,] while len(indices)<len(cols): indices.append(indices[-1]) for i,l in enumerate(levs): if i==0: C = [cols[i],] if numpy.allclose(levs[0][0],-1.e20): ## ok it's an extension arrow L=[mn-1.,levs[0][1]] else: L = levs[i] I = [indices[i],] else: if l[0] == L[-1] and I[-1]==indices[i]: # Ok same type lets keep going if numpy.allclose(l[1],1.e20): L.append(mx+1.) else: L.append(l[1]) C.append(cols[i]) else: # ok we need new contouring LEVS.append(L) COLS.append(C) INDX.append(I) C = [cols[i],] L = levs[i] I = [indices[i],] LEVS.append(L) COLS.append(C) INDX.append(I) for i,l in enumerate(LEVS): # Ok here we are trying to group together levels can be, a join will happen if: # next set of levels contnues where one left off AND pattern is identical if isinstance(gm,isofill.Gfi): mapper = vtk.vtkPolyDataMapper() lut = vtk.vtkLookupTable() cot = vtk.vtkBandedPolyDataContourFilter() cot.ClippingOn() cot.SetInputData(sFilter.GetOutput()) cot.SetNumberOfContours(len(l)) for j,v in enumerate(l): cot.SetValue(j,v) #cot.SetScalarModeToIndex() cot.Update() mapper.SetInputConnection(cot.GetOutputPort()) lut.SetNumberOfTableValues(len(COLS[i])) for j,color in enumerate(COLS[i]): r,g,b = cmap.index[color] lut.SetTableValue(j,r/100.,g/100.,b/100.) #print l[j],vcs.colors.rgb2str(r*2.55,g*2.55,b*2.55),l[j+1] mapper.SetLookupTable(lut) mapper.SetScalarRange(0,len(l)-1) mapper.SetScalarModeToUseCellData() else: for j,color in enumerate(COLS[i]): mapper = vtk.vtkPolyDataMapper() lut = vtk.vtkLookupTable() th = vtk.vtkThreshold() th.ThresholdBetween(l[j],l[j+1]) th.SetInputConnection(geoFilter.GetOutputPort()) geoFilter2 = vtk.vtkDataSetSurfaceFilter() geoFilter2.SetInputConnection(th.GetOutputPort()) mapper.SetInputConnection(geoFilter2.GetOutputPort()) lut.SetNumberOfTableValues(1) r,g,b = cmap.index[color] lut.SetTableValue(0,r/100.,g/100.,b/100.) mapper.SetLookupTable(lut) mapper.SetScalarRange(l[j],l[j+1]) mappers.append([mapper,]) #png = vtk.vtkPNGReader() #png.SetFileName("/git/uvcdat/Packages/vcs/Share/uvcdat_texture.png") #T=vtk.vtkTexture() #T.SetInputConnection(png.GetOutputPort()) if isinstance(gm,isofill.Gfi): mappers.append([mapper,]) else: #Boxfill (non custom)/Meshfill if isinstance(gm,boxfill.Gfb): if numpy.allclose(gm.level_1,1.e20) or numpy.allclose(gm.level_2,1.e20): levs = vcs.mkscale(mn,mx) legend = vcs.mklabels(levs) dx = (levs[-1]-levs[0])/(gm.color_2-gm.color_1+1) levs = numpy.arange(levs[0],levs[-1]+dx,dx) else: if gm.boxfill_type=="log10": levs = vcs.mkscale(numpy.ma.log10(gm.level_1),numpy.ma.log10(gm.level_2)) else: levs = vcs.mkscale(gm.level_1,gm.level_2) legend = vcs.mklabels(levs) if gm.boxfill_type=="log10": for k in legend.keys(): legend[float(numpy.ma.log10(legend[k]))] = legend[k] del(legend[k]) dx = (levs[-1]-levs[0])/(gm.color_2-gm.color_1+1) levs = numpy.arange(levs[0],levs[-1]+dx,dx) cols = range(gm.color_1,gm.color_2+1) else: if numpy.allclose(gm.levels,1.e20): levs = vcs.mkscale(mn,mx) else: levs = gm.levels if numpy.allclose(levs[0],1.e20): levs[0]=-1.e20 cols = gm.fillareacolors if cols==[1,]: cols = vcs.getcolors(levs) Nlevs = len(levs) Ncolors = Nlevs-1 #Prep mapper mappers=[] mapper = vtk.vtkPolyDataMapper() thr = vtk.vtkThreshold() thr.SetInputConnection(geoFilter.GetOutputPort()) if not gm.ext_1 in ["y",1,True] and not gm.ext_2 in ["y",1,True] : thr.ThresholdBetween(levs[0],levs[-1]) elif gm.ext_1 in ["y",1,True] and not gm.ext_2 in ["y",1,True] : thr.ThresholdByLower(levs[-1]) elif not gm.ext_1 in ["y",1,True] and gm.ext_2 in ["y",1,True] : thr.ThresholdByUpper(levs[0]) thr.Update() geoFilter2 = vtk.vtkDataSetSurfaceFilter() geoFilter2.SetInputConnection(thr.GetOutputPort()) if gm.ext_1 in ["y",1,True] and gm.ext_2 in ["y",1,True] : mapper.SetInputConnection(geoFilter.GetOutputPort()) else: mapper.SetInputConnection(geoFilter2.GetOutputPort()) if mappers == []: # ok didn't need to have special banded contours mappers=[mapper,] ## Colortable bit # make sure length match while len(cols)<Ncolors: cols.append(cols[-1]) lut.SetNumberOfTableValues(Ncolors) for i in range(Ncolors): r,g,b = cmap.index[cols[i]] lut.SetTableValue(i,r/100.,g/100.,b/100.) mapper.SetLookupTable(lut) if numpy.allclose(levs[0],-1.e20): lmn = mn-1. else: lmn= levs[0] if numpy.allclose(levs[-1],1.e20): lmx = mx+1. else: lmx= levs[-1] mapper.SetScalarRange(lmn,lmx) if missingMapper is not None: mappers.insert(0,missingMapper) x1,x2,y1,y2 = vcs2vtk.getRange(gm,xm,xM,ym,yM) if tmpl.data.priority != 0: # And now we need actors to actually render this thing for mapper in mappers: act = vtk.vtkActor() if isinstance(mapper,list): act.SetMapper(mapper[0]) else: mapper.Update() act.SetMapper(mapper) if geo is None: act = vcs2vtk.doWrap(act,[x1,x2,y1,y2],wrap) if isinstance(mapper,list): #act.GetMapper().ScalarVisibilityOff() #act.SetTexture(mapper[1]) pass # create a new renderer for this mapper (we need one for each mapper because of cmaera flips) ren = vtk.vtkRenderer() self.renWin.AddRenderer(ren) self.setLayer(ren,tmpl.data.priority) ren.AddActor(act) vcs2vtk.fitToViewport(act,ren,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2],wc=[x1,x2,y1,y2],geo=geo) if isinstance(gm,meshfill.Gfm): tmpl.plot(self.canvas,data1,gm,bg=self.bg,X=numpy.arange(xm,xM*1.1,(xM-xm)/10.),Y=numpy.arange(ym,yM*1.1,(yM-ym)/10.)) else: self.renderTemplate(tmpl,data1,gm) if isinstance(gm,(isofill.Gfi,meshfill.Gfm,boxfill.Gfb)): if getattr(gm,"legend",None) is not None: legend = gm.legend if gm.ext_1 in ["y",1,True] and not numpy.allclose(levs[0],1.e20): if isinstance(levs,numpy.ndarray): levs=levs.tolist() levs.insert(0,-1.e20) if gm.ext_2 in ["y",1,True] and not numpy.allclose(levs[0],1.e20): if isinstance(levs,numpy.ndarray): levs=levs.tolist() levs.append(1.e20) self.renderColorBar(tmpl,levs,cols,legend,cmap) if self.canvas._continents is None: continents = False if continents: projection = vcs.elements["projection"][gm.projection] self.plotContinents(x1,x2,y1,y2,projection,wrap,tmpl)
def plotVector(self,data1,data2,tmpl,gm): ug,xm,xM,ym,yM,continents,wrap,geo,cellData = vcs2vtk.genGrid(data1,data2,gm) if cellData: c2p = vtk.vtkCellDataToPointData() c2p.SetInputData(ug) c2p.Update() #For contouring duplicate points seem to confuse it if ug.IsA("vtkUnstructuredGrid"): cln = vtk.vtkCleanUnstructuredGrid() cln.SetInputConnection(c2p.GetOutputPort()) missingMapper = vcs2vtk.putMaskOnVTKGrid(data1,ug,None,cellData) u=numpy.ma.ravel(data1) v=numpy.ma.ravel(data2) sh = list(u.shape) sh.append(1) u = numpy.reshape(u,sh) v = numpy.reshape(v,sh) z = numpy.zeros(u.shape) w = numpy.concatenate((u,v),axis=1) w = numpy.concatenate((w,z),axis=1) w = VN.numpy_to_vtk(w,deep=True) w.SetName("vectors") ug.GetPointData().AddArray(w) ## Vector attempt arrow = vtk.vtkArrowSource() l = gm.line if l is None: l = "default" try: l = vcs.getline(l) lwidth = l.width[0] lcolor = l.color[0] lstyle = l.type[0] except: lstyle = "solid" lwidth = 1. lcolor = 0 if gm.linewidth is not None: lwidth = gm.linewidth if gm.linecolor is not None: lcolor = gm.linecolor arrow.SetTipRadius(.1*lwidth) arrow.SetShaftRadius(.03*lwidth) arrow.Update() glyphFilter = vtk.vtkGlyph2D() glyphFilter.SetSourceConnection(arrow.GetOutputPort()) glyphFilter.OrientOn() glyphFilter.SetVectorModeToUseVector() glyphFilter.SetInputArrayToProcess(1,0,0,0,"vectors") glyphFilter.SetScaleFactor(2.*gm.scale) if cellData: if ug.IsA("vtkUnstructuredGrid"): glyphFilter.SetInputConnection(cln.GetOutputPort()) else: glyphFilter.SetInputConnection(c2p.GetOutputPort()) else: glyphFilter.SetInputData(ug) mapper = vtk.vtkPolyDataMapper() mapper.SetInputConnection(glyphFilter.GetOutputPort()) act = vtk.vtkActor() act.SetMapper(mapper) try: cmap = vcs.elements["colormap"][cmap] except: cmap = vcs.elements["colormap"][self.canvas.getcolormapname()] r,g,b = cmap.index[lcolor] act.GetProperty().SetColor(r/100.,g/100.,b/100.) x1,x2,y1,y2 = vcs2vtk.getRange(gm,xm,xM,ym,yM) act = vcs2vtk.doWrap(act,[x1,x2,y1,y2],wrap) ren=vtk.vtkRenderer() self.renWin.AddRenderer(ren) self.setLayer(ren,tmpl.data.priority) vcs2vtk.fitToViewport(act,ren,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2],[x1,x2,y1,y2]) if tmpl.data.priority!=0: ren.AddActor(act) self.renderTemplate(tmpl,data1,gm) if self.canvas._continents is None: continents = False if continents: projection = vcs.elements["projection"][gm.projection] self.plotContinents(x1,x2,y1,y2,projection,wrap,tmpl)
def plotVector(self,data1,data2,tmpl,gm): #Preserve time and z axis for plotting these inof in rendertemplate taxis = data1.getTime() if data1.ndim>2: zaxis = data1.getAxis(-3) else: zaxis = None data1 = self.trimData2D(data1) # Ok get3 only the last 2 dims data2 = self.trimData2D(data2) ug,xm,xM,ym,yM,continents,wrap,geo = vcs2vtk.genGridOnPoints(data1,data2,gm,deep=False) missingMapper = vcs2vtk.putMaskOnVTKGrid(data1,ug,None,False,deep=False) u=numpy.ma.ravel(data1) v=numpy.ma.ravel(data2) sh = list(u.shape) sh.append(1) u = numpy.reshape(u,sh) v = numpy.reshape(v,sh) z = numpy.zeros(u.shape) w = numpy.concatenate((u,v),axis=1) w = numpy.concatenate((w,z),axis=1) # HACK The grid returned by vtk2vcs.genGrid is not the same size as the # data array. I'm not sure where the issue is...for now let's just zero-pad # data array so that we can at least test rendering until Charles gets # back from vacation: wLen = len(w) numPts = ug.GetNumberOfPoints() if wLen != numPts: warnings.warn("!!! Warning during vector plotting: Number of points does not "\ "match the number of vectors to be glyphed (%s points vs %s "\ "vectors). The vectors will be padded/truncated to match for "\ "rendering purposes, but the resulting image should not be "\ "trusted."%(numPts, wLen)) newShape = (numPts,) + w.shape[1:] w = numpy.ma.resize(w, newShape) w = vcs2vtk.numpy_to_vtk_wrapper(w,deep=False) w.SetName("vectors") ug.GetPointData().AddArray(w) ## Vector attempt l = gm.line if l is None: l = "default" try: l = vcs.getline(l) lwidth = l.width[0] lcolor = l.color[0] lstyle = l.type[0] except: lstyle = "solid" lwidth = 1. lcolor = 0 if gm.linewidth is not None: lwidth = gm.linewidth if gm.linecolor is not None: lcolor = gm.linecolor # Strip out masked points. if ug.IsA("vtkStructuredGrid"): if ug.GetCellBlanking(): visArray = ug.GetCellVisibilityArray() visArray.SetName("BlankingArray") ug.GetCellData().AddArray(visArray) thresh = vtk.vtkThreshold() thresh.SetInputData(ug) thresh.ThresholdByUpper(0.5) thresh.SetInputArrayToProcess(0, 0, 0, "vtkDataObject::FIELD_ASSOCIATION_CELLS", "BlankingArray") thresh.Update() ug = thresh.GetOutput() elif ug.GetPointBlanking(): visArray = ug.GetPointVisibilityArray() visArray.SetName("BlankingArray") ug.GetPointData().AddArray(visArray) thresh = vtk.vtkThreshold() thresh.SetInputData(ug) thresh.SetUpperThreshold(0.5) thresh.SetInputArrayToProcess(0, 0, 0, "vtkDataObject::FIELD_ASSOCIATION_POINTS", "BlankingArray") thresh.Update() ug = thresh.GetOutput() arrow = vtk.vtkGlyphSource2D() arrow.SetGlyphTypeToArrow() arrow.FilledOff() glyphFilter = vtk.vtkGlyph2D() glyphFilter.SetSourceConnection(arrow.GetOutputPort()) glyphFilter.SetVectorModeToUseVector() # Rotate arrows to match vector data: glyphFilter.OrientOn() # Scale to vector magnitude: glyphFilter.SetScaleModeToScaleByVector() # These are some unfortunately named methods. It does *not* clamp the scale # range to [min, max], but rather remaps the range [min, max]-->[0,1]. Bump # up min so that near-zero vectors will not be rendered, as these tend to # come out randomly oriented. glyphFilter.ClampingOn() glyphFilter.SetRange(0.01, 1.0) glyphFilter.SetInputArrayToProcess(1,0,0,0,"vectors") glyphFilter.SetScaleFactor(2.*gm.scale) #if cellData: # if ug.IsA("vtkUnstructuredGrid"): # glyphFilter.SetInputConnection(cln.GetOutputPort()) # else: # glyphFilter.SetInputConnection(c2p.GetOutputPort()) #else: # glyphFilter.SetInputData(ug) glyphFilter.SetInputData(ug) mapper = vtk.vtkPolyDataMapper() mapper.SetInputConnection(glyphFilter.GetOutputPort()) act = vtk.vtkActor() act.SetMapper(mapper) try: cmap = vcs.elements["colormap"][cmap] except: cmap = vcs.elements["colormap"][self.canvas.getcolormapname()] r,g,b = cmap.index[lcolor] act.GetProperty().SetColor(r/100.,g/100.,b/100.) x1,x2,y1,y2 = vcs2vtk.getRange(gm,xm,xM,ym,yM) act = vcs2vtk.doWrap(act,[x1,x2,y1,y2],wrap) ren = self.createRenderer() self.renWin.AddRenderer(ren) self.setLayer(ren,tmpl.data.priority) vcs2vtk.fitToViewport(act,ren,[tmpl.data.x1,tmpl.data.x2,tmpl.data.y1,tmpl.data.y2],[x1,x2,y1,y2]) if tmpl.data.priority!=0: ren.AddActor(act) self.renderTemplate(tmpl,data1,gm,taxis,zaxis) if self.canvas._continents is None: continents = False if continents: projection = vcs.elements["projection"][gm.projection] self.plotContinents(x1,x2,y1,y2,projection,wrap,tmpl)
if "all" in p.output_type: p.output_type = ["png","pdf","ps","svg"] f=cdms2.open(p.file) s=f(p.var) if p.continents: #Continents contData = vcs2vtk.continentsVCS2VTK(os.environ["HOME"]+"/.uvcdat/data_continent_political") contMapper = vtk.vtkPolyDataMapper() contMapper.SetInputData(contData) contActor = vtk.vtkActor() contActor.SetMapper(contMapper) contActor.GetProperty().SetColor(0.,0.,0.) if p.wrap: A0 = vcs2vtk.doWrap(contMapper,contActor) if 0: lat = s.getGrid().getLatitude() lon = s.getGrid().getLongitude() if lat.rank()==1: # rectilinear lat = lat[:,numpy.newaxis]*numpy.ones(lon.shape)[numpy.newaxis,:] lon = lon[numpy.newaxis,:]*numpy.ones(lat.shape)[:,numpy.newaxis] else: lat=numpy.ascontiguousarray(lat) lon=numpy.ascontiguousarray(lon) lat.resize(lat.shape[0]*lat.shape[1],1) lon.resize(lat.shape[0]*lat.shape[1],1) l = numpy.zeros(lon.shape) m3 = numpy.concatenate((lon,lat,l),axis=1)
def update_input(self,vtkobjects,array1,array2=None,update=True): if vtkobjects.has_key("vtk_backend_grid"): ## Ok ths is where we update the input data vg=vtkobjects["vtk_backend_grid"] data = vcs2vtk.numpy_to_vtk_wrapper(array1.filled(0.).flat, deep=False) pData= vg.GetPointData().GetScalars() if pData is not None: vg.GetPointData().SetScalars(data) else: vg.GetCellData().SetScalars(data) if vtkobjects.has_key("vtk_backend_filter"): vtkobjects["vtk_backend_filter"].Update() if vtkobjects.has_key("vtk_backend_missing_mapper"): missingMapper,color,cellData = vtkobjects["vtk_backend_missing_mapper"] missingMapper2 = vcs2vtk.putMaskOnVTKGrid(array1,vg,color,cellData,deep=False) else: missingMapper = None if vtkobjects.has_key("vtk_backend_contours"): for c in vtkobjects["vtk_backend_contours"]: c.Update() ports=vtkobjects["vtk_backend_contours"] elif vtkobjects.has_key("vtk_backend_geofilters"): ports=vtkobjects["vtk_backend_geofilters"] else: # Vector plot ports=vtkobjects["vtk_backend_glyphfilters"] w = vcs2vtk.generateVectorArray(array1,array2,vg) vg.GetPointData().AddArray(w) vg = vcs2vtk.stripGrid(vg) ports[0].SetInputData(vg) if vtkobjects.has_key("vtk_backend_actors"): i=0 for a in vtkobjects["vtk_backend_actors"]: act = a[0] wrp = a[1] if a[1] is missingMapper: i-=1 mapper = missingMapper2 wrp = a[2] else: ## Labeled contours are a different kind if vtkobjects.has_key("vtk_backend_luts"): lut,rg = vtkobjects["vtk_backend_luts"][i] mapper = vtk.vtkPolyDataMapper() elif vtkobjects.has_key("vtk_backend_labeled_luts"): lut,rg = vtkobjects["vtk_backend_labeled_luts"][i] mapper = vtk.vtkLabeledContourMapper() if lut is None: mapper.SetInputConnection(ports[i].GetOutputPort()) else: if mapper.IsA("vtkPolyDataMapper"): mapper.SetInputConnection(ports[i].GetOutputPort()) mapper.SetLookupTable(lut) mapper.SetScalarModeToUsePointData() else: stripper = vtk.vtkStripper() stripper.SetInputConnection(ports[i].GetOutputPort()) mapper.SetInputConnection(stripper.GetOutputPort()) stripper.Update() tprops = vtkobjects["vtk_backend_contours_labels_text_properties"] mapper.GetPolyDataMapper().SetLookupTable(lut) mapper.GetPolyDataMapper().SetScalarModeToUsePointData() mapper.GetPolyDataMapper().SetScalarRange(rg[0],rg[1]) mapper.SetLabelVisibility(1) mapper.SetTextProperties(tprops) if rg[2]: mapper.SetScalarModeToUseCellData() mapper.SetScalarRange(rg[0],rg[1]) act.SetMapper(mapper) act = vcs2vtk.doWrap(a[0],wrp) a[0].SetMapper(act.GetMapper()) i+=1 taxis = array1.getTime() if taxis is not None: tstr = str(cdtime.reltime(taxis[0],taxis.units).tocomp(taxis.getCalendar())) else: tstr = None ## Min/Max/Mean for att in ["Min","Max","Mean","crtime","crdate","zvalue"]: if vtkobjects.has_key("vtk_backend_%s_text_actor" % att): t = vtkobjects["vtk_backend_%s_text_actor" % att] if att == "Min": t.SetInput("Min %g" % array1.min()) elif att == "Max": t.SetInput("Max %g" % array1.max()) elif att == "Mean": if not inspect.ismethod(getattr(array1,'mean')): meanstring = "Mean: %s" % getattr(array1,"mean") else: try: meanstring='Mean %.4g'% float(cdutil.averager(array1, axis = " ".join(["(%s)" % S for S in array1.getAxisIds()]))) except Exception,err: meanstring='Mean %.4g'%array1.mean() t.SetInput(meanstring) elif att=="crdate" and tstr is not None: t.SetInput(tstr.split()[0].replace("-","/")) elif att=="crtime" and tstr is not None: t.SetInput(tstr.split()[1]) elif att=="zvalue": if len(array1.shape)>2: l=array1.getAxis(-3) if l.isTime(): t.SetInput(str(l.asComponentTime()[0])) else: t.SetInput("%g" % l[0])