class MapFrame(wx.Frame):
def __init__(self, parent, id=wx.ID_ANY, title="", pos=wx.DefaultPosition,
size=(900, 500), style=wx.SYSTEM_MENU|wx.CAPTION|wx.CLOSE_BOX, name="Frame"):
super(MapFrame, self).__init__(parent, id, title, pos, size, style, name)
# Attributes
self.panel = wx.Panel(self, id=wx.ID_ANY, pos=wx.DefaultPosition,
size=(900, 500), style=wx.EXPAND, name="")
vsizer = wx.BoxSizer(wx.HORIZONTAL)
mapsizer = wx.BoxSizer(wx.VERTICAL)
textsizer = wx.BoxSizer(wx.VERTICAL)
img_path = os.path.abspath("./UnsuiGUI/apartmentdesign.jpg")
bitmap = wx.Bitmap(img_path, type=wx.BITMAP_TYPE_JPEG)
img = wx.StaticBitmap(self.panel, wx.ID_ANY, bitmap=bitmap)
self.start_button = wx.Button(self.panel, label="Start")
self.load_button = wx.Button(self.panel, label="Load")
self.load_button.Disable()
self.end_button = wx.Button(self.panel, label="Quit")
self.start_button.Bind(wx.EVT_BUTTON, self.OnStart)
self.load_button.Bind(wx.EVT_BUTTON, self.OnLoad)
self.end_button.Bind(wx.EVT_BUTTON, self.QuitButton)
self.shell = PyShell(self.panel, -1, size=(400, -1))
self.shell.clear()
self.shell.SetLexer(wx.stc.STC_LEX_NULL)
self.shell.redirectStdin(True)
self.shell.redirectStdout(True)
self.shell.SetInsertionPointEnd()
self.shell.Bind(wx.EVT_COMMAND_ENTER, self.OnEnter)
textsizer.Add(self.shell, 5, wx.EXPAND | wx.TE_MULTILINE, 5)
mapsizer.Add(img, 5, wx.EXPAND | wx.ALIGN_CENTER, 5)
mapsizerbuttonsizer = wx.BoxSizer(wx.HORIZONTAL)
mapsizerbuttonsizer.Add(self.start_button, 5, wx.ALIGN_LEFT, 5)
mapsizerbuttonsizer.Add(self.load_button, 5, wx.ALIGN_LEFT, 5)
mapsizerbuttonsizer.Add(self.end_button, 5, wx.ALIGN_LEFT, 5)
mapsizer.Add(mapsizerbuttonsizer, 0, wx.EXPAND|wx.ALIGN_CENTER, 5)
vsizer.Add(mapsizer, 0, wx.EXPAND | wx.ALIGN_LEFT, 5)
vsizer.Add(textsizer, 0, wx.EXPAND | wx.ALIGN_RIGHT, 5)
self.panel.SetSizer(vsizer)
def OnStart(self, evt):
self.load_button.Enable()
self.shell.execStartupScript("Unsui.py")
def OnLoad(self, evt):
self.shell.Execute("load")
def QuitButton(self, evt):
sys.exit()
def OnEnter(self, evt):
self.shell.ScrollToEnd()
class ShellPanel(wx.Panel):
def __init__(self, parent):
wx.Panel.__init__(self, parent)
self.shell = Shell(self)
self.shell.clear()
self.sizer = wx.BoxSizer(wx.VERTICAL)
self.sizer.Add(self.shell, 1, wx.GROW)
# self.sizer.Add(wx.StaticText(self, -1, 'Linestyle: '), 0)
self.SetSizer(self.sizer)
self.Layout()
self.Fit()
class PyShellWindow(wx.Panel):
"""Python Shell Window"""
def __init__(self, parent, giface, id=wx.ID_ANY, simpleEditorHandler=None, **kwargs):
self.parent = parent
self.giface = giface
wx.Panel.__init__(self, parent=parent, id=id, **kwargs)
self.intro = _("Welcome to wxGUI Interactive Python Shell %s") % VERSION + "\n\n" + \
_("Type %s for more GRASS scripting related information.") % "\"help(gs)\"" + "\n" + \
_("Type %s to add raster or vector to the layer tree.") % "\"AddLayer()\"" + "\n\n"
self.shell = PyShell(parent=self, id=wx.ID_ANY,
introText=self.intro,
locals={'gs': grass,
'AddLayer': self.AddLayer})
sys.displayhook = self._displayhook
self.btnClear = Button(self, wx.ID_CLEAR)
self.btnClear.Bind(wx.EVT_BUTTON, self.OnClear)
self.btnClear.SetToolTip(_("Delete all text from the shell"))
self.simpleEditorHandler = simpleEditorHandler
if simpleEditorHandler:
self.btnSimpleEditor = Button(
self, id=wx.ID_ANY, label=_("Simple &editor"))
self.btnSimpleEditor.Bind(wx.EVT_BUTTON, simpleEditorHandler)
self.btnSimpleEditor.SetToolTip(
_("Open a simple Python code editor"))
self._layout()
def _displayhook(self, value):
print(value) # do not modify __builtin__._
def _layout(self):
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(self.shell, proportion=1,
flag=wx.EXPAND)
btnSizer = wx.BoxSizer(wx.HORIZONTAL)
if self.simpleEditorHandler:
btnSizer.Add(self.btnSimpleEditor, proportion=0,
flag=wx.EXPAND | wx.LEFT | wx.RIGHT, border=5)
btnSizer.AddStretchSpacer()
btnSizer.Add(self.btnClear, proportion=0,
flag=wx.EXPAND, border=5)
sizer.Add(btnSizer, proportion=0,
flag=wx.ALL | wx.EXPAND, border=5)
sizer.Fit(self)
sizer.SetSizeHints(self)
self.SetSizer(sizer)
self.Fit()
self.SetAutoLayout(True)
self.Layout()
def AddLayer(self, name, ltype='auto'):
"""Add selected map to the layer tree
:param name: name of raster/vector map to be added
:param type: map type ('raster', 'vector', 'auto' for autodetection)
"""
fname = None
if ltype == 'raster' or ltype != 'vector':
# check for raster
fname = grass.find_file(name, element='cell')['fullname']
if fname:
ltype = 'raster'
lcmd = 'd.rast'
if not fname and (ltype == 'vector' or ltype != 'raster'):
# if not found check for vector
fname = grass.find_file(name, element='vector')['fullname']
if fname:
ltype = 'vector'
lcmd = 'd.vect'
if not fname:
return _("Raster or vector map <%s> not found") % (name)
self.giface.GetLayerTree().AddLayer(ltype=ltype,
lname=fname,
lchecked=True,
lcmd=[lcmd, 'map=%s' % fname])
if ltype == 'raster':
return _('Raster map <%s> added') % fname
return _('Vector map <%s> added') % fname
def OnClear(self, event):
"""Delete all text from the shell
"""
self.shell.clear()
self.shell.showIntro(self.intro)
self.shell.prompt()
class PyShellWindow(wx.Panel):
"""Python Shell Window"""
def __init__(self, parent, id = wx.ID_ANY, **kwargs):
self.parent = parent # GMFrame
wx.Panel.__init__(self, parent = parent, id = id, **kwargs)
self.intro = _("Welcome to wxGUI Interactive Python Shell %s") % VERSION + "\n\n" + \
_("Type %s for more GRASS scripting related information.") % "\"help(grass)\"" + "\n" + \
_("Type %s to add raster or vector to the layer tree.") % "\"AddLayer()\"" + "\n\n"
self.shell = PyShell(parent = self, id = wx.ID_ANY,
introText = self.intro,
locals={'grass': grass,
'AddLayer': self.AddLayer})
sys.displayhook = self._displayhook
self.btnClear = wx.Button(self, wx.ID_CLEAR)
self.btnClear.Bind(wx.EVT_BUTTON, self.OnClear)
self.btnClear.SetToolTipString(_("Delete all text from the shell"))
self._layout()
def _displayhook(self, value):
print value # do not modify __builtin__._
def _layout(self):
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(item = self.shell, proportion = 1,
flag = wx.EXPAND)
btnSizer = wx.BoxSizer(wx.HORIZONTAL)
btnSizer.Add(item = self.btnClear, proportion = 0,
flag = wx.EXPAND | wx.RIGHT, border = 5)
sizer.Add(item = btnSizer, proportion = 0,
flag = wx.ALIGN_RIGHT | wx.ALL, border = 5)
sizer.Fit(self)
sizer.SetSizeHints(self)
self.SetSizer(sizer)
self.Fit()
self.SetAutoLayout(True)
self.Layout()
def AddLayer(self, name, ltype = 'auto'):
"""Add selected map to the layer tree
:param name: name of raster/vector map to be added
:param type: map type ('raster', 'vector', 'auto' for autodetection)
"""
fname = None
if ltype == 'raster' or ltype != 'vector':
# check for raster
fname = grass.find_file(name, element = 'cell')['fullname']
if fname:
ltype = 'raster'
lcmd = 'd.rast'
if not fname and (ltype == 'vector' or ltype != 'raster'):
# if not found check for vector
fname = grass.find_file(name, element = 'vector')['fullname']
if fname:
ltype = 'vector'
lcmd = 'd.vect'
if not fname:
return _("Raster or vector map <%s> not found") % (name)
self.parent.GetLayerTree().AddLayer(ltype = ltype,
lname = fname,
lchecked = True,
lcmd = [lcmd, 'map=%s' % fname])
if ltype == 'raster':
return _('Raster map <%s> added') % fname
return _('Vector map <%s> added') % fname
def OnClear(self, event):
"""Delete all text from the shell
"""
self.shell.clear()
self.shell.showIntro(self.intro)
self.shell.prompt()
class PyShellWindow(wx.Panel):
"""Python Shell Window"""
def __init__(self,
parent,
giface,
id=wx.ID_ANY,
simpleEditorHandler=None,
**kwargs):
self.parent = parent
self.giface = giface
wx.Panel.__init__(self, parent=parent, id=id, **kwargs)
self.intro = (
_("Welcome to wxGUI Interactive Python Shell %s") % VERSION +
"\n\n" +
_("Type %s for more GRASS scripting related information.") %
'"help(gs)"' + "\n" +
_("Type %s to add raster or vector to the layer tree.") %
"\"AddLayer('map_name')\"" + "\n\n")
shellargs = dict(
parent=self,
id=wx.ID_ANY,
introText=self.intro,
locals={
"gs": grass,
"AddLayer": self.AddLayer,
"help": self.Help
},
)
# useStockId (available since wxPython 4.0.2) should be False on macOS
if sys.platform == "darwin" and CheckWxVersion([4, 0, 2]):
shellargs["useStockId"] = False
self.shell = PyShell(**shellargs)
if IsDark():
SetDarkMode(self.shell)
sys.displayhook = self._displayhook
self.btnClear = ClearButton(self)
self.btnClear.Bind(wx.EVT_BUTTON, self.OnClear)
self.btnClear.SetToolTip(_("Delete all text from the shell"))
self.simpleEditorHandler = simpleEditorHandler
if simpleEditorHandler:
self.btnSimpleEditor = Button(self,
id=wx.ID_ANY,
label=_("Simple &editor"))
self.btnSimpleEditor.Bind(wx.EVT_BUTTON, simpleEditorHandler)
self.btnSimpleEditor.SetToolTip(
_("Open a simple Python code editor"))
self._layout()
def _displayhook(self, value):
print(value) # do not modify __builtin__._
def _layout(self):
sizer = wx.BoxSizer(wx.VERTICAL)
sizer.Add(self.shell, proportion=1, flag=wx.EXPAND)
btnSizer = wx.BoxSizer(wx.HORIZONTAL)
if self.simpleEditorHandler:
btnSizer.Add(
self.btnSimpleEditor,
proportion=0,
flag=wx.EXPAND | wx.LEFT | wx.RIGHT,
border=5,
)
btnSizer.AddStretchSpacer()
btnSizer.Add(self.btnClear, proportion=0, flag=wx.EXPAND, border=5)
sizer.Add(btnSizer, proportion=0, flag=wx.ALL | wx.EXPAND, border=5)
sizer.Fit(self)
sizer.SetSizeHints(self)
self.SetSizer(sizer)
self.Fit()
self.SetAutoLayout(True)
self.Layout()
def AddLayer(self, name, ltype="auto"):
"""Add selected map to the layer tree
:param name: name of raster/vector map to be added
:param type: map type ('raster', 'vector', 'auto' for autodetection)
"""
fname = None
if ltype == "raster" or ltype != "vector":
# check for raster
fname = grass.find_file(name, element="cell")["fullname"]
if fname:
ltype = "raster"
lcmd = "d.rast"
if not fname and (ltype == "vector" or ltype != "raster"):
# if not found check for vector
fname = grass.find_file(name, element="vector")["fullname"]
if fname:
ltype = "vector"
lcmd = "d.vect"
if not fname:
return _("Raster or vector map <%s> not found") % (name)
self.giface.GetLayerTree().AddLayer(ltype=ltype,
lname=fname,
lchecked=True,
lcmd=[lcmd, "map=%s" % fname])
if ltype == "raster":
return _("Raster map <%s> added") % fname
return _("Vector map <%s> added") % fname
def Help(self, obj):
"""Override help() function
:param obj object/str: generate the help of the given object
return str: help str of the given object
"""
with redirect_stdout(io.StringIO()) as f:
help(obj)
return f.getvalue()
def OnClear(self, event):
"""Delete all text from the shell"""
self.shell.clear()
self.shell.showIntro(self.intro)
self.shell.prompt()
