def creategraphicsmethod(self, typeName, name, nameSource):
"""Creates a graphicsmethod 'name' with type 'typeName' as a copy of 'nameSource'"""
typeName = str(typeName)
name = str(name)
if (nameSource is None):
nameSource = 'default'
else:
nameSource = str(nameSource)
vcs.creategraphicsmethod(typeName, nameSource, name)
return self.getgraphicsmethod(typeName, name)
def creategraphicsmethod(self, typeName, name, nameSource):
"""Creates a graphicsmethod 'name' with type 'typeName' as a copy of 'nameSource'"""
typeName = str(typeName)
name = str(name)
if (nameSource is None):
nameSource = 'default'
else:
nameSource = str(nameSource)
vcs.creategraphicsmethod(typeName, nameSource, name)
return self.getgraphicsmethod(typeName, name)
def projected(self, gm_type, projtype="default"):
s = None
if gm_type == "meshfill":
f = cdms2.open(os.path.join(vcs.sample_data,
"sampleCurveGrid4.nc"))
s2 = f("sample")
s = MV2.resize(s2, (4, 32, 48))
s.id = 'sample'
t = cdms2.createAxis(range(4))
t.units = "months since 2015"
t.id = "time"
t.designateTime()
s.setAxis(0, t)
s.setAxis(1, s2.getAxis(0))
s.setAxis(2, s2.getAxis(1))
s.setGrid(s2.getGrid())
for i in range(4):
s[i] = s[i] * (1 + float(i) / 10.)
else:
s = self.clt("clt", slice(
0, 12)) # read only 12 times steps to speed up things
gm = vcs.creategraphicsmethod(gm_type, "default")
if projtype != "default":
p = vcs.createprojection()
try:
ptype = int(projtype)
except:
ptype = projtype
p.type = ptype
gm.projection = p
self.x.clear()
self.x.plot(s, gm, bg=self.bg)
self.x.animate.create()
prefix = "test_vcs_animate_%s_%s" % (gm_type.lower(), projtype.lower())
self.x.animate.save("%s.mp4" % prefix)
pngs = self.x.animate.close(
preserve_pngs=True) # so we can look at them again
pdir = os.path.split(pngs[0])[0]
p = os.path.join(pdir, "anim_0.png")
src = os.path.join(
self.basedir, "test_vcs_animate_projected_%s_%s.png" %
(gm_type.lower(), projtype.lower()))
self.checkImage(p, pngReady=True, src=src)
for f in pngs:
if os.path.isfile(f):
os.remove(f)
os.removedirs(pdir)
os.remove("%s.mp4" % prefix)
def testVCSreset1only(self):
for gtype in vcs.listelements():
b0 = vcs.listelements(gtype)
if gtype == 'colormap':
b = vcs.createcolormap()
xtra = vcs.createisofill()
untouched = vcs.listelements("isofill")
elif gtype == "template":
b = vcs.createtemplate()
elif gtype == "textcombined":
b = vcs.createtextcombined()
elif gtype == "textorientation":
b = vcs.createtextorientation()
elif gtype == "texttable":
b = vcs.createtexttable()
elif gtype == "fillarea":
b = vcs.createfillarea()
elif gtype == "projection":
b = vcs.createprojection()
elif gtype == "marker":
b = vcs.createmarker()
elif gtype == "line":
b = vcs.createline()
elif gtype in ["display", "font", "fontNumber", "list", "format"]:
vcs.manageElements.reset()
continue
else:
b = vcs.creategraphicsmethod(gtype)
if gtype != "colormap":
xtra = vcs.createcolormap()
untouched = vcs.listelements("colormap")
b1 = vcs.listelements(gtype)
self.assertNotEqual(b0, b1)
vcs.manageElements.reset(gtype)
b2 = vcs.listelements(gtype)
self.assertEqual(b0, b2)
if gtype == "colormap":
self.assertEqual(untouched, vcs.listelements("isofill"))
else:
self.assertEqual(untouched, vcs.listelements("colormap"))
vcs.manageElements.reset()
if gtype == "colormap":
self.assertNotEqual(untouched, vcs.listelements("isofill"))
else:
self.assertNotEqual(untouched, vcs.listelements("colormap"))
# case for 1d weirdness
sc = vcs.createscatter()
vcs.manageElements.reset()
def setGraphicsMethod(self, gm):
if (isinstance(gm, list)):
self._plot.graphics_method = vcs.getgraphicsmethod(gm[0], gm[1])
return
for t in vcs.listelements():
if len(vcs.listelements(t)):
o = vcs.elements[t].values()[0]
if hasattr(o, "g_name"):
if o.g_name == gm["g_name"]:
break
else:
return False
my_gm = vcs.creategraphicsmethod(t)
for k in gm:
if k == "name":
continue
if gm[k] == 100000000000000000000:
gm[k] = 1e20
if gm[k] == -100000000000000000000:
gm[k] = -1e20
if isinstance(gm[k], list):
conv = []
for v in gm[k]:
if v == 100000000000000000000:
conv.append(1e20)
elif v == -100000000000000000000:
conv.append(-1e20)
else:
conv.append(v)
gm[k] = conv
if hasattr(my_gm, k):
try:
setattr(my_gm, k, gm[k])
except:
print "Could not set attribute %s on graphics method of type %s" % (
k, t)
if "ext_1" in gm:
my_gm.ext_1 = gm["ext_1"]
if "ext_2" in gm:
my_gm.ext_2 = gm["ext_2"]
self._plot.graphics_method = my_gm
def setGraphicsMethod(self, gm):
if (isinstance(gm, list)):
self._plot.graphics_method = vcs.getgraphicsmethod(gm[0], gm[1])
return
for t in vcs.listelements():
if len(vcs.listelements(t)):
o = list(vcs.elements[t].values())[0]
if hasattr(o, "g_name"):
if o.g_name == gm["g_name"]:
break
else:
return False
my_gm = vcs.creategraphicsmethod(t)
updateGraphicsMethodProps(gm, my_gm)
if "ext_1" in gm:
my_gm.ext_1 = gm["ext_1"]
if "ext_2" in gm:
my_gm.ext_2 = gm["ext_2"]
self._plot.graphics_method = my_gm
def isSlowingDown(self, gmtype):
n = 100
print("Testing slow down for:", gmtype)
maxpct = 100.
fastest = 100000.
avg = 0.
times = []
for i in range(n):
start = time.time()
gm = vcs.creategraphicsmethod(gmtype)
gm.datawc_x1 = 0.
gm.datawc_x2 = 2.
gm.datawc_y1 = 0.
gm.datawc_y2 = 2.
self.x.plot(gm, *self.gmtypes[gmtype])
end = time.time()
elapsed = end - start
if elapsed < fastest:
fastest = elapsed
pct = (elapsed / fastest * 100.)
times.append(elapsed)
if i > 5 and pct > maxpct: # skip the first 5 times to make sure system gets in groovy mode
maxpct = pct
avg += elapsed
self.x.clear()
self.x.removeobject(gm)
print(elapsed)
a, b = numpy.polyfit(numpy.arange(n - 5),
numpy.array(times[5:]) / fastest, 1)
print("\tMax percentage", maxpct)
print("\tFastest time:", fastest)
print("\tAvg time:", avg / n)
print("\tFit coeff:", a, b)
if a > 25.E-2:
return True
return False
def gmPatternHatch(self,
gm_type="isofill",
fill_style="pattern",
projtype="default",
lat1=-90,
lat2=90,
lon1=-180,
lon2=180,
contig=True):
self.x.setcolormap("classic")
gm = vcs.creategraphicsmethod(gm_type, "default")
if projtype != "default":
p = vcs.createprojection()
try:
ptype = int(projtype)
except:
ptype = projtype
p.type = ptype
gm.projection = p
if contig:
gm.levels = [220, 230, 240, 250, 260, 270, 280, 290, 300, 310, 320]
else:
gm.levels = [[230, 235], [240, 245], [250, 255], [260, 265],
[270, 275], [280, 285], [290, 295], [300, 305],
[310, 315], [320, 325]]
gm.fillareastyle = fill_style
gm.fillareacolors = [242, 244, 237, 248, 250, 252, 44, 243, 139, 247]
if fill_style == "hatch":
gm.fillareaindices = [2, 4, 6, 8, 10, 12, 14, 16, 18, 20]
gm.fillareaopacity = [50, 75, 20, 100, 25, 30, 40, 80, 60, 100]
else:
gm.fillareaindices = [1, 3, 5, 7, 9, 11, 18, 15, 17, 19]
gm.fillareaopacity = [50, 75, 20, 0, 25, 30, 100, 0, 60, 0]
if gm_type == "boxfill":
gm.boxfill_type = "custom"
if gm_type == "meshfill":
gm.mesh = True
nm_xtra = ""
xtra = {}
if lat1 != lat2:
gm.datawc_y1 = lat1
gm.datawc_y2 = lat2
xtra["latitude"] = (lat1, lat2)
if lat1 < 0:
nm_xtra += "_SH"
else:
nm_xtra += "_NH"
if lon1 != lon2:
xtra["longitude"] = (lon1, lon2)
nm_xtra += "_%i_%i" % (lon1, lon2)
if not contig:
nm_xtra += "_non-contig"
xtra["time"] = slice(0, 1)
xtra["squeeze"] = 1
f = cdms2.open(
os.path.join(vcs.sample_data, 'tas_ccsr-95a_1979.01-1979.12.nc'))
s = f("tas", **xtra)
f.close()
self.x.clear()
self.x.plot(s, gm, bg=self.bg)
fnm = "test_vcs_%s_%s" % (gm_type.lower(), fill_style.lower())
if projtype != "default":
fnm += "_%s_proj" % projtype
fnm += nm_xtra
self.checkImage(fnm + ".png")
p.add_argument("--non-contiguous", dest="contig", default=True, action="store_false", help="use non contiguous levels")
args = p.parse_args(sys.argv[1:])
gm_type = args.gm
src = args.src
bg = not args.show
x = vcs.init()
x.setantialiasing(0)
x.setcolormap("classic")
if bg:
x.setbgoutputdimensions(1200, 1091, units="pixels")
gm = vcs.creategraphicsmethod(gm_type, "default")
if args.projtype != "default":
p = vcs.createprojection()
try:
ptype = int(args.projtype)
except:
ptype = args.projtype
p.type = ptype
gm.projection = p
if args.contig:
gm.levels = [220, 230, 240, 250, 260, 270, 280, 290, 300, 310, 320]
else:
gm.levels = [[230, 235], [240, 245], [250, 255], [260, 265], [270, 275],
[280, 285], [290, 295], [300, 305], [310, 315], [320, 325]]
gm.fillareastyle = args.fill_style
help="use non contiguous levels")
args = p.parse_args(sys.argv[1:])
gm_type = args.gm
src = args.src
bg = not args.show
x = vcs.init()
x.setantialiasing(0)
x.setcolormap("classic")
if bg:
x.setbgoutputdimensions(1200, 1091, units="pixels")
gm = vcs.creategraphicsmethod(gm_type, "default")
if args.projtype != "default":
p = vcs.createprojection()
try:
ptype = int(args.projtype)
except:
ptype = args.projtype
p.type = ptype
gm.projection = p
if args.contig:
gm.levels = [220, 230, 240, 250, 260, 270, 280, 290, 300, 310, 320]
else:
gm.levels = [[230, 235], [240, 245], [250, 255], [260, 265], [270, 275],
[280, 285], [290, 295], [300, 305], [310, 315], [320, 325]]
gm.fillareastyle = args.fill_style