cdat/testing/vcs/test_vcs_meshfill_vertices_issue.py at 31d935edc81774c92b8d13b38cc259e21ad904a7 · CDAT/cdat · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import numpy
import vcs
import sys
import os
pth = os.path.join(os.path.dirname(__file__),"..")
sys.path.append(pth)
import checkimage

x=vcs.init()

x.setantialiasing(0)
x.drawlogooff()
x.setbgoutputdimensions(1200,1090,units="pixels")

data_values = [ 25, 45, 55.]

data_lon = [ 5., 10., 15.]
data_lat = [ 5., 10., 15.]

data_lon_vert = [
        # Triangle (last one missing because traingle has only 3 vertices
        [2.5,7.5,5.,1.e20],
        # Square
        [7.5,12.5,12.5,7.5],
        # Diamond
        [15.,17.5,15,12.5],
        ]

data_lat_vert = [
        # triangle
        [2.5,2.5,7.5,1.e20],
        # square
        [7.5,7.5,12.5,12.5],
        # diamond
        [12.5,15,17.5,15],
        ]

mesh = numpy.array([data_lat_vert,data_lon_vert])
print "MESH SHAPE:",mesh.shape,mesh.dtype
mesh = numpy.transpose(mesh,(1,0,2))
print "MESH SHAPE:",mesh.shape
mesh = numpy.ma.masked_greater(mesh,1.e19)

"""
print "Triangle lats:"
print mesh[0][0]
print "Triangle lons:"
print mesh[0][1]
"""
m = x.createmeshfill()
m.levels = [20,30,50,70,80]
m.mesh = True

x.plot(numpy.array(data_values,),mesh,m,bg=True)
x.png("test_vcs_meshfill_vertices_issue.png")
src = sys.argv[1]
ret = checkimage.check_result_image("test_vcs_meshfill_vertices_issue.png",src,checkimage.defaultThreshold)
sys.exit(ret)