1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# This file is part of Beremiz, a Integrated Development Environment for

     5

# programming IEC 61131-3 automates supporting plcopen standard and CanFestival.

     6

#

     7

# Copyright (C) 2016 - 2017: Andrey Skvortsov <andrej.skvortzov@gmail.com>

     8

#

     9

# See COPYING file for copyrights details.

    10

#

    11

# This program is free software; you can redistribute it and/or

    12

# modify it under the terms of the GNU General Public License

    13

# as published by the Free Software Foundation; either version 2

    14

# of the License, or (at your option) any later version.

    15

#

    16

# This program is distributed in the hope that it will be useful,

    17

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    18

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    19

# GNU General Public License for more details.

    20

#

    21

# You should have received a copy of the GNU General Public License

    22

# along with this program; if not, write to the Free Software

    23

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

    24

    25

    26

from __future__ import absolute_import

    27

from __future__ import print_function

    28

import os

    29

import sys

    30

import getopt

    31

from past.builtins import execfile

    32

    33

import wx

    34

from wx.lib.agw.advancedsplash import AdvancedSplash, AS_NOTIMEOUT, AS_CENTER_ON_SCREEN

    35

    36

import util.paths as paths

    37

    38

    39

class BeremizIDELauncher(object):

    40

    def __init__(self):

    41

        self.app = None

    42

        self.frame = None

    43

        self.updateinfo_url = None

    44

        self.extensions = []

    45

        self.app_dir = paths.AbsDir(__file__)

    46

        self.projectOpen = None

    47

        self.buildpath = None

    48

        self.splash = None

    49

        self.splashPath = self.Bpath("images", "splash.png")

    50

        self.modules = ["BeremizIDE"]

    51

        self.debug = os.path.exists("BEREMIZ_DEBUG")

    52

        self.handle_exception = None

    53

        self.logf = None

    54

    55

    def Bpath(self, *args):

    56

        return os.path.join(self.app_dir, *args)

    57

    58

    def Usage(self):

    59

        print("Usage:")

    60

        print("%s [Options] [Projectpath] [Buildpath]" % sys.argv[0])

    61

        print("")

    62

        print("Supported options:")

    63

        print("-h --help                    Print this help")

    64

        print("-u --updatecheck URL         Retrieve update information by checking URL")

    65

        print("-e --extend PathToExtension  Extend IDE functionality by loading at start additional extensions")

    66

        print("-l --log path                write content of console tab to given file")

    67

        print("")

    68

        print("")

    69

    70

    def SetCmdOptions(self):

    71

        self.shortCmdOpts = "hu:e:l:"

    72

        self.longCmdOpts = ["help", "updatecheck=", "extend=", "log="]

    73

    74

    def ProcessOption(self, o, a):

    75

        if o in ("-h", "--help"):

    76

            self.Usage()

    77

            sys.exit()

    78

        if o in ("-u", "--updatecheck"):

    79

            self.updateinfo_url = a

    80

        if o in ("-e", "--extend"):

    81

            self.extensions.append(a)

    82

        if o in ("-l", "--log"):

    83

            self.logf = open(a, 'a')

    84

    85

    def ProcessCommandLineArgs(self):

    86

        self.SetCmdOptions()

    87

        try:

    88

            opts, args = getopt.getopt(sys.argv[1:], self.shortCmdOpts, self.longCmdOpts)

    89

        except getopt.GetoptError:

    90

            # print help information and exit:

    91

            self.Usage()

    92

            sys.exit(2)

    93

    94

        for o, a in opts:

    95

            self.ProcessOption(o, a)

    96

    97

        if len(args) > 2:

    98

            self.Usage()

    99

            sys.exit()

   100

   101

        elif len(args) == 1:

   102

            self.projectOpen = args[0]

   103

            self.buildpath = None

   104

        elif len(args) == 2:

   105

            self.projectOpen = args[0]

   106

            self.buildpath = args[1]

   107

   108

    def CreateApplication(self):

   109

   110

        BeremizAppType = wx.App if wx.VERSION >= (3, 0, 0) else wx.PySimpleApp

   111

   112

        class BeremizApp(BeremizAppType):

   113

            def OnInit(_self):  # pylint: disable=no-self-argument

   114

                self.ShowSplashScreen()

   115

                return True

   116

   117

        self.app = BeremizApp(redirect=self.debug)

   118

        self.app.SetAppName('beremiz')

   119

        if wx.VERSION < (3, 0, 0):

   120

            wx.InitAllImageHandlers()

   121

   122

    def ShowSplashScreen(self):

   123

        class Splash(AdvancedSplash):

   124

            Painted = False

   125

   126

            def OnPaint(_self, event):  # pylint: disable=no-self-argument

   127

                AdvancedSplash.OnPaint(_self, event)

   128

                if not _self.Painted:  # trigger app start only once

   129

                    _self.Painted = True

   130

                    wx.CallAfter(self.AppStart)

   131

        bmp = wx.Image(self.splashPath).ConvertToBitmap()

   132

        self.splash = Splash(None,

   133

                             bitmap=bmp,

   134

                             agwStyle=AS_NOTIMEOUT | AS_CENTER_ON_SCREEN)

   135

   136

    def BackgroundInitialization(self):

   137

        self.InitI18n()

   138

        self.CheckUpdates()

   139

        self.LoadExtensions()

   140

        self.ImportModules()

   141

   142

    def InitI18n(self):

   143

        from util.misc import InstallLocalRessources

   144

        InstallLocalRessources(self.app_dir)

   145

   146

    def globals(self):

   147

        """

   148

        allows customizations to specify what globals

   149

        are passed to extensions

   150

        """

   151

        return globals()

   152

   153

    def LoadExtensions(self):

   154

        for extfilename in self.extensions:

   155

            from util.TranslationCatalogs import AddCatalog

   156

            from util.BitmapLibrary import AddBitmapFolder

   157

            extension_folder = os.path.split(os.path.realpath(extfilename))[0]

   158

            sys.path.append(extension_folder)

   159

            AddCatalog(os.path.join(extension_folder, "locale"))

   160

            AddBitmapFolder(os.path.join(extension_folder, "images"))

   161

            execfile(extfilename, self.globals())

   162

   163

    def CheckUpdates(self):

   164

        if self.updateinfo_url is not None:

   165

            self.updateinfo = _("Fetching %s") % self.updateinfo_url

   166

   167

            def updateinfoproc():

   168

                try:

   169

                    import urllib2

   170

                    self.updateinfo = urllib2.urlopen(self.updateinfo_url, None).read()

   171

                except Exception:

   172

                    self.updateinfo = _("update info unavailable.")

   173

   174

            from threading import Thread

   175

            self.splash.SetText(text=self.updateinfo)

   176

            updateinfoThread = Thread(target=updateinfoproc)

   177

            updateinfoThread.start()

   178

            updateinfoThread.join(2)

   179

            self.splash.SetText(text=self.updateinfo)

   180

   181

    def ImportModules(self):

   182

        for modname in self.modules:

   183

            mod = __import__(modname)

   184

            setattr(self, modname, mod)

   185

   186

    def InstallExceptionHandler(self):

   187

        import version

   188

        import util.ExceptionHandler

   189

        self.handle_exception = util.ExceptionHandler.AddExceptHook(version.app_version)

   190

   191

    def CreateUI(self):

   192

        self.frame = self.BeremizIDE.Beremiz(None, self.projectOpen, self.buildpath, logf=self.logf)

   193

   194

    def CloseSplash(self):

   195

        if self.splash:

   196

            self.splash.Close()

   197

   198

    def ShowUI(self):

   199

        self.frame.Show()

   200

   201

    def PreStart(self):

   202

        self.ProcessCommandLineArgs()

   203

        self.CreateApplication()

   204

   205

    def AppStart(self):

   206

        try:

   207

            self.BackgroundInitialization()

   208

            self.CreateUI()

   209

            self.CloseSplash()

   210

            self.ShowUI()

   211

        except (KeyboardInterrupt, SystemExit):

   212

            raise

   213

        except Exception:

   214

            if self.handle_exception is not None:

   215

                self.handle_exception(*sys.exc_info(), exit=True)

   216

            else:

   217

                raise

   218

   219

    def MainLoop(self):

   220

        self.app.MainLoop()

   221

   222

    def Start(self):

   223

        self.PreStart()

   224

        self.InstallExceptionHandler()

   225

        self.MainLoop()

   226

   227

   228

if __name__ == '__main__':

   229

    beremiz = BeremizIDELauncher()

   230

    beremiz.Start()