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

    54

    def Bpath(self, *args):

    55

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

    56

    57

    def Usage(self):

    58

        print("Usage:")

    59

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

    60

        print("")

    61

        print("Supported options:")

    62

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

    63

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

    64

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

    65

        print("")

    66

        print("")

    67

    68

    def SetCmdOptions(self):

    69

        self.shortCmdOpts = "hu:e:"

    70

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

    71

    72

    def ProcessOption(self, o, a):

    73

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

    74

            self.Usage()

    75

            sys.exit()

    76

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

    77

            self.updateinfo_url = a

    78

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

    79

            self.extensions.append(a)

    80

    81

    def ProcessCommandLineArgs(self):

    82

        self.SetCmdOptions()

    83

        try:

    84

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

    85

        except getopt.GetoptError:

    86

            # print help information and exit:

    87

            self.Usage()

    88

            sys.exit(2)

    89

    90

        for o, a in opts:

    91

            self.ProcessOption(o, a)

    92

    93

        if len(args) > 2:

    94

            self.Usage()

    95

            sys.exit()

    96

    97

        elif len(args) == 1:

    98

            self.projectOpen = args[0]

    99

            self.buildpath = None

   100

        elif len(args) == 2:

   101

            self.projectOpen = args[0]

   102

            self.buildpath = args[1]

   103

   104

    def CreateApplication(self):

   105

   106

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

   107

   108

        class BeremizApp(BeremizAppType):

   109

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

   110

                self.ShowSplashScreen()

   111

                return True

   112

   113

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

   114

        self.app.SetAppName('beremiz')

   115

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

   116

            wx.InitAllImageHandlers()

   117

   118

    def ShowSplashScreen(self):

   119

        class Splash(AdvancedSplash):

   120

            Painted = False

   121

   122

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

   123

                AdvancedSplash.OnPaint(_self, event)

   124

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

   125

                    _self.Painted = True

   126

                    wx.CallAfter(self.AppStart)

   127

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

   128

        self.splash = Splash(None,

   129

                             bitmap=bmp,

   130

                             agwStyle=AS_NOTIMEOUT | AS_CENTER_ON_SCREEN)

   131

   132

    def BackgroundInitialization(self):

   133

        self.InitI18n()

   134

        self.CheckUpdates()

   135

        self.LoadExtensions()

   136

        self.ImportModules()

   137

   138

    def InitI18n(self):

   139

        from util.misc import InstallLocalRessources

   140

        InstallLocalRessources(self.app_dir)

   141

   142

    def globals(self):

   143

        """

   144

        allows customizations to specify what globals

   145

        are passed to extensions

   146

        """

   147

        return globals()

   148

   149

    def LoadExtensions(self):

   150

        for extfilename in self.extensions:

   151

            from util.TranslationCatalogs import AddCatalog

   152

            from util.BitmapLibrary import AddBitmapFolder

   153

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

   154

            sys.path.append(extension_folder)

   155

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

   156

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

   157

            execfile(extfilename, self.globals())

   158

   159

    def CheckUpdates(self):

   160

        if self.updateinfo_url is not None:

   161

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

   162

   163

            def updateinfoproc():

   164

                try:

   165

                    import urllib2

   166

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

   167

                except Exception:

   168

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

   169

   170

            from threading import Thread

   171

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

   172

            updateinfoThread = Thread(target=updateinfoproc)

   173

            updateinfoThread.start()

   174

            updateinfoThread.join(2)

   175

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

   176

   177

    def ImportModules(self):

   178

        for modname in self.modules:

   179

            mod = __import__(modname)

   180

            setattr(self, modname, mod)

   181

   182

    def InstallExceptionHandler(self):

   183

        import version

   184

        import util.ExceptionHandler

   185

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

   186

   187

    def CreateUI(self):

   188

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

   189

   190

    def CloseSplash(self):

   191

        if self.splash:

   192

            self.splash.Close()

   193

   194

    def ShowUI(self):

   195

        self.frame.Show()

   196

   197

    def PreStart(self):

   198

        self.ProcessCommandLineArgs()

   199

        self.CreateApplication()

   200

   201

    def AppStart(self):

   202

        try:

   203

            self.BackgroundInitialization()

   204

            self.CreateUI()

   205

            self.CloseSplash()

   206

            self.ShowUI()

   207

        except (KeyboardInterrupt, SystemExit):

   208

            raise

   209

        except Exception:

   210

            if self.handle_exception is not None:

   211

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

   212

            else:

   213

                raise

   214

   215

    def MainLoop(self):

   216

        self.app.MainLoop()

   217

   218

    def Start(self):

   219

        self.PreStart()

   220

        self.InstallExceptionHandler()

   221

        self.MainLoop()

   222

   223

   224

if __name__ == '__main__':

   225

    beremiz = BeremizIDELauncher()

   226

    beremiz.Start()