def test_basic_drawing(self):
self.start(gsys.func_name())
testmap = os.path.join(self.gx.temp_folder(), "test")
with gxmap.GXmap.new(testmap, overwrite=True) as gmap:
mapfile = gmap.filename
with gxv.GXview(gmap, "base", area=(0, 0, 25, 20),
scale=100.0) as view:
view.xy_rectangle(((0, 0), (25, 20)),
pen={
'line_thick': 0.1,
'line_color': 'R'
})
with gxv.GXview(gmap,
"data",
map_location=(4, 3),
area=(0, 0, 1800, 1500),
scale=10000) as view:
view.xy_rectangle(((0, 0), (1800, 1500)),
pen={
'line_thick': 5,
'line_color': 'G'
})
# TODO - the underlying grid function has a bug - it cannot do dotted lines
view.graticule(style=gxv.GRATICULE_DOT, pen={'line_thick': 50})
#gxvwr.map(mapfile)
self.assertEqual(
gxmap.crc_map(mapfile),
2258658382) # TODO replace with correct crc once fixed
def test_basic_grid(self):
self.start(gsys.func_name())
# test grid file
folder, files = gsys.unzip(os.path.join(os.path.dirname(__file__),
'testgrids.zip'),
folder=self.gx.temp_folder())
grid_file = os.path.join(folder, 'test_agg_utm.grd')
map_file = os.path.join(self.gx.temp_folder(), "test_agg_utm")
with gxmap.GXmap.new(map_file, overwrite=True) as gmap:
with gxgrd.GXgrd(grid_file) as grd:
mn, mx = grd.extent_2d()
cs = grd.cs
mapfile = gmap.filename
with gxv.GXview(gmap,
"data",
cs=cs,
area=(mn[0], mn[1], mx[0], mx[1]),
scale=(mx[0] - mn[0]) / 0.2) as view:
view.xy_rectangle((mn, mx),
pen={
'line_thick': 0.1,
'line_color': 'R'
})
with gxagg.GXagg(grid_file) as agg:
view.aggregate(agg)
self.assertEqual(gxmap.crc_map(mapfile), 3752814683)
#gxvwr.map(mapfile)
with gxmap.GXmap.new(map_file, overwrite=True) as gmap:
mapfile = gmap.filename
with gxv.GXview(gmap,
"data",
cs="AGD66 / AMG zone 53",
area=(mn[0], mn[1], mx[0], mx[1]),
scale=(mx[0] - mn[0]) / 0.2) as view:
with gxgrd.GXgrd(grid_file) as grd:
mn, mx = grd.extent_2d()
view.set_drawing_cs(grd.cs)
view.xy_rectangle((mn, mx),
pen={
'line_thick': 0.1,
'line_color': 'R'
})
with gxagg.GXagg(grid_file) as agg:
view.aggregate(agg)
#gxvwr.map(mapfile)
self.assertEqual(gxmap.crc_map(mapfile), 4001381093)
def test_cs(self):
self.start(gsys.func_name())
testmap = os.path.join(self.gx.temp_folder(), "test")
with gxmap.GXmap.new(testmap, overwrite=True) as gmap:
with gxv.GXview(gmap, "rectangle_test", cs="wgs 84") as view:
self.assertEqual("WGS 84", str(view.cs))
with gxv.GXview(gmap, "vcs",
cs="wgs 84 / UTM zone 15N [special]") as view:
self.assertTrue(
"WGS 84 / UTM zone 15N [special]" in str(view.cs))
def new_test_map(mapname='test', rescale=1.0, locate=None):
test_map = os.path.join(gx.GXpy().temp_folder(), mapname)
with gxmap.GXmap.new(test_map, overwrite=True) as gmap:
with gxv.GXview(gmap, "rectangle_test") as view:
view.start_group('rectangle')
view.xy_rectangle((gxgm.Point((0, 0)), gxgm.Point((250, 110))),
pen={'line_thick': 1})
p1 = gxgm.Point((5, 5)) * rescale
p2 = gxgm.Point((100, 100)) * rescale
poff = gxgm.Point((10, 5)) * rescale
view.pen = {'fill_color': gxapi.C_LT_GREEN}
view.xy_rectangle((p1, p2))
view.pen = {'line_style': (2, 2.0)}
view.xy_line((p1 + poff, p2 - poff))
with gxv.GXview(gmap, "poly") as view:
view.start_group('poly')
plinelist = [[110, 5], [120, 20], [130, 15], [150, 50], [160, 70],
[175, 35], [190, 65], [220, 50], [235, 18.5]]
pp = gxgm.PPoint.from_list(plinelist) * rescale
view.pen = {'line_style': (2, 2.0)}
view.xy_poly_line(pp)
view.pen = {
'line_style': (4, 2.0),
'line_smooth': gxv.SMOOTH_AKIMA
}
view.xy_poly_line(pp)
ppp = np.array(plinelist)
pp = gxgm.PPoint(ppp[3:, :]) * rescale
view.pen = {
'line_style': (5, 5.0),
'line_smooth': gxv.SMOOTH_CUBIC,
'line_color': gxapi.C_RED,
'line_thick': 0.25,
'fill_color': gxapi.C_LT_BLUE
}
view.xy_poly_line(pp, close=True)
view.pen = {'fill_color': gxapi.C_LT_GREEN}
pp = (pp - (100, 0, 0)) / 2 + (100, 0, 0)
view.xy_poly_line(pp, close=True)
pp += (0, 25, 0)
view.pen = {'fill_color': gxapi.C_LT_RED}
view.xy_poly_line(pp, close=True)
return gmap.filename
def test_zone(zone, crc, shade=False):
with gxmap.GXmap.new(map_file, overwrite=True) as gmap:
mapfile = gmap.filename
with gxv.GXview(gmap,
"data",
area=(mn[0], mn[1], mx[0], mx[1]),
scale=(mx[0] - mn[0]) / 0.2) as view:
with gxagg.GXagg(grid_file, zone=zone, shade=shade) as agg:
view.aggregate(agg)
#gxvwr.map(mapfile)
self.assertEqual(gxmap.crc_map(mapfile), crc)
def test_new_geosoft_map(self):
self.start(gsys.func_name())
# temp map
with gxmap.GXmap.new_standard_geosoft(data_area=(0, 0, 100,
80)) as gmap:
views = gmap.view_list(gxmap.LIST_ALL)
self.assertTrue('Base' in views)
self.assertTrue('Data' in views)
with gxmap.GXmap.new_standard_geosoft(
data_area=(0, 0, 100, 80),
cs=gxcs.GXcs("DHDN / Okarito 2000 [geodetic]")) as gmap:
with gxv.GXview(gmap, 'Data', mode=gxv.WRITE_OLD) as v:
self.assertEqual("DHDN / Okarito 2000 [geodetic]", str(v.cs))
def test_view(self):
self.start(gsys.func_name())
testmap = os.path.join(self.gx.temp_folder(), "test")
with gxmap.GXmap.new(testmap, overwrite=True) as gmap:
mapfile = gmap.filename
with gxv.GXview(gmap, "rectangle_test",
area=(0, 0, 250, 125)) as view:
view.start_group('test_group')
rect_line(view)
view.graticule(25, 20, style=gxv.GRATICULE_LINE)
view.pen = {'line_thick': 0.1}
view.xy_rectangle(((0, 0), (250, 125)),
pen={
'line_thick': 0.1,
'line_color': 'R'
})
with gxv.GXview(gmap, "poly") as view:
view.start_group('test_group')
draw_stuff(view)
#gxvwr.map(mapfile)
self.assertEqual(gxmap.crc_map(mapfile), 3148511381)
gxmap.delete_files(mapfile)
def test_3D(self):
self.start(gsys.func_name())
testmap = os.path.join(self.gx.temp_folder(), "test")
with gxmap.GXmap.new(testmap, overwrite=True) as gmap:
mapfile = gmap.filename
with gxv.GXview(gmap, "base", area=(0, 0, 250, 125),
scale=1000) as view:
view.start_group('test_group')
view.xy_rectangle(((0, 0), (100, 100)))
with gxv.GXview3d(gmap,
viewname='v3d_test',
area=(0, 0, 300, 300),
scale=1000,
cs="wgs 84 / UTM zone 15S") as view:
view.start_group('test_group')
rect_line(view)
draw_stuff(view)
view.box_3d(((0, 0, 10), (120, 100, 50)))
def test_create(self):
self.start(gsys.func_name())
with gxmap.GXmap.new() as gmap:
with gxv.GXview(gmap) as vw:
self.assertEqual(vw.viewname, "_unnamed_view")
with gxmap.GXmap.new() as gmap:
with gxv.GXview(gmap, "test") as vw:
self.assertEqual(vw.viewname, "test")
with gxmap.GXmap.new() as gmap:
area = (100, 500, 15100, 10500)
scale = 20000
location = (0, 0)
xcm = (area[2] - area[0]) * 100.0 / scale
ycm = (area[3] - area[1]) * 100.0 / scale
with gxv.GXview(gmap,
"test",
map_location=location,
area=area,
scale=scale,
cs="WGS 84 / UTM zone 34N") as vw:
self.assertEqual(vw.extent(), area)
self.assertEqual(vw.extent(gxv.EXTENT_MAP), (0, 0, xcm, ycm))
self.assertEqual(vw.scale(), (scale, scale))
self.assertTrue(
vw.cs.same_as(gxcs.GXcs("WGS 84 / UTM zone 34N")))
self.assertEqual(vw.units_per_m, 1.0)
self.assertEqual(vw.unit_name, 'm')
with gxmap.GXmap.new() as gmap:
area = (100, 500, 15100, 10500)
scale = 12000
loc = (7.5, 2.0)
mpu = 1.0 / float(
gxcs.parameters(gxcs.PARM_UNITS, 'ftUS')['FACTOR'])
xcm = 100.0 * ((area[2] - area[0]) / scale) / mpu
ycm = 100.0 * ((area[3] - area[1]) / scale) / mpu
with gxv.GXview(gmap,
"test",
map_location=loc,
area=area,
scale=scale,
cs=("WGS 84 / UTM zone 34N", '', '', 'ftUS',
'')) as vw:
self.assertEqual(vw.extent(), area)
mx = vw.extent(gxv.EXTENT_MAP)
self.assertAlmostEqual(mx[0], loc[0])
self.assertAlmostEqual(mx[1], loc[1])
self.assertAlmostEqual(mx[2], loc[0] + xcm)
self.assertAlmostEqual(mx[3], loc[1] + ycm)
self.assertAlmostEqual(vw.scale()[0], scale)
self.assertAlmostEqual(vw.scale()[1], scale)
self.assertFalse(
vw.cs.same_as(gxcs.GXcs("WGS 84 / UTM zone 34N")))
self.assertTrue(
vw.cs.same_as(
gxcs.GXcs(
("WGS 84 / UTM zone 34N", '', '', 'ftUS', ''))))
self.assertAlmostEqual(vw.units_per_m, 3.28083333333334)
self.assertEqual(vw.unit_name, 'ftUS')
with gxmap.GXmap.new() as gmap:
area = (100, 500, 15100, 10500)
scale = 12000
loc = (7.5, 2.0)
mpu = 1.0 / float(
gxcs.parameters(gxcs.PARM_UNITS, 'ftUS')['FACTOR'])
xcm = 100.0 * ((area[2] - area[0]) / scale) / mpu
ycm = 100.0 * ((area[3] - area[1]) / scale) / mpu
with gxv.GXview(gmap,
"test",
map_location=loc,
area=area,
scale=scale,
cs='ftUS') as vw:
self.assertEqual(vw.extent(), area)
mx = vw.extent(gxv.EXTENT_MAP)
self.assertAlmostEqual(mx[0], loc[0])
self.assertAlmostEqual(mx[1], loc[1])
self.assertAlmostEqual(mx[2], loc[0] + xcm)
self.assertAlmostEqual(mx[3], loc[1] + ycm)
self.assertAlmostEqual(vw.scale()[0], scale)
self.assertAlmostEqual(vw.scale()[1], scale)
self.assertTrue(
vw.cs.same_as(gxcs.GXcs(('', '', '', 'ftUS', ''))))
self.assertAlmostEqual(vw.units_per_m, 3.28083333333334)
self.assertEqual(vw.unit_name, 'ftUS')
with gxmap.GXmap.new() as gmap:
with gxv.GXview(gmap,
"test",
area=(100, 500, 15100, 10500),
scale=(50000, 10000),
map_location=(10, 25)) as vw:
self.assertEqual(vw.extent(), (100, 500, 15100, 10500))
self.assertEqual(vw.scale(), (50000, 10000))
self.assertEqual(vw.extent(extent=gxv.EXTENT_MAP),
(10., 25., 40., 125.))
self.assertTrue(vw.cs.same_as(gxcs.GXcs()))
test_map = os.path.join(self.gx.temp_folder(), "test_map")
with gxmap.GXmap.new(test_map) as gmap:
mapname = gmap.filename
with gxv.GXview(gmap, "test") as view:
self.assertEqual(view.viewname, "test")
self.assertEqual(view.mapfilename, mapname)
view.start_group('test_group')
def_pen = view.pen
pn = {"line_thick": 99, "pat_number": 88}
view.push_pen()
view.pen = pn
new_pen = view.pen
self.assertEqual(new_pen["line_thick"], 99)
self.assertEqual(new_pen["pat_number"], 88)
view.pop_pen()
pen = view.pen
self.assertEqual(pen, def_pen)
view.push_pen(pn)
view.pen = pn
new_pen = view.pen
self.assertEqual(new_pen["line_thick"], 99)
self.assertEqual(new_pen["pat_number"], 88)
view.pop_pen()
pen = view.pen
self.assertEqual(pen, def_pen)
def sample_map(gxp, mapname='test', rescale=1.0, locate=None):
with gxmap.GXmap.new(mapname, overwrite=True) as gmap:
"""
with gxv.GXview(gmap, 'base',area=(0, 0, 200, 150), scale=100.0)as vw:
vw.xy_rectangle(vw.extent(), pen={"line_color":'B'})
with gxv.GXview(gmap, 'data', map_location=(5,5), area=(0,0,300,250)) as vw:
vw.xy_rectangle(vw.extent())
"""
p2 = gxgm.Point((110, 110)) * rescale
with gxv.GXview(gmap,
"rectangle_test",
area=(0, 0, p2.x, p2.y),
scale=100.0) as view:
view.start_group('rect')
view.xy_rectangle(((0, 0), p2), pen={'line_thick': 1})
p1 = gxgm.Point((5, 5)) * rescale
p2 = gxgm.Point((100, 100)) * rescale
poff = gxgm.Point((10, 5)) * rescale
view.pen = {'fill_color': gxapi.C_LT_GREEN}
view.xy_rectangle((p1, p2))
view.pen = {'line_style': (2, 2.0)}
view.xy_line((p1 + poff, p2 - poff))
p2 = gxgm.Point((250, 110)) * rescale
with gxv.GXview3d(gmap,
viewname="poly",
area=(0, 0, p2.x, p2.y),
scale=100.0) as view:
view.start_group('stuff')
view.xy_rectangle(((0, 0), p2),
pen={
'line_thick': 3,
'line_color': 'B'
})
plinelist = [[110, 5], [120, 20], [130, 15], [150, 50], [160, 70],
[175, 35], [190, 65], [220, 50], [235, 18.5]]
pp = gxgm.PPoint.from_list(plinelist) * rescale
view.pen = {'line_style': (2, 2.0)}
view.xy_poly_line(pp)
view.pen = {
'line_style': (4, 2.0),
'line_smooth': gxv.SMOOTH_AKIMA
}
view.xy_poly_line(pp)
ppp = np.array(plinelist)
pp = gxgm.PPoint(ppp[3:, :]) * rescale
view.pen = {
'line_style': (5, 5.0),
'line_smooth': gxv.SMOOTH_CUBIC,
'line_color': gxapi.C_RED,
'line_thick': 0.25,
'fill_color': gxapi.C_LT_BLUE
}
view.xy_poly_line(pp, close=True)
view.pen = {'fill_color': gxapi.C_LT_GREEN}
pp = (pp - (100, 0, 0)) / 2 + (100, 0, 0)
view.xy_poly_line(pp, close=True)
pp += (0, 25, 0)
view.pen = {'fill_color': "B"}
view.xy_poly_line(pp, close=True)
#TODO - viewer does not properly disply items on the plane if the box_3d is called
view.box_3d(((20, 10, 30), (80, 50, 50)),
pen={'fill_color': 'R255G100B50'})
return gmap.filename