PLCOpenEditor.py
author lbessard
Tue, 18 Sep 2007 18:05:52 +0200
changeset 95 ee66a9a1748b
parent 92 76d5001393df
child 97 28337cd092fd
permissions -rw-r--r--
Corrections for wx version 2.8.4 and Windows
#!/usr/bin/env python
# -*- coding: utf-8 -*-

#This file is part of PLCOpenEditor, a library implementing an IEC 61131-3 editor
#based on the plcopen standard. 
#
#Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
#
#See COPYING file for copyrights details.
#
#This library is free software; you can redistribute it and/or
#modify it under the terms of the GNU General Public
#License as published by the Free Software Foundation; either
#version 2.1 of the License, or (at your option) any later version.
#
#This library is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
#General Public License for more details.
#
#You should have received a copy of the GNU General Public
#License along with this library; if not, write to the Free Software
#Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

from time import localtime
from datetime import datetime
import wx
import wx.grid

from SFCViewer import *
from LDViewer import *
from Viewer import *
from TextViewer import *
from RessourceEditor import *
from PLCControler import *
from plcopen import OpenPDFDoc

import os, re, platform, sys, time, traceback, getopt

__version__ = "$Revision$"

CWD = os.path.split(__file__)[0]

[ID_PLCOPENEDITOR, ID_PLCOPENEDITORPROJECTTREE, 
 ID_PLCOPENEDITORMAINSPLITTER, ID_PLCOPENEDITOREDITORPANEL,
 ID_PLCOPENEDITORTABSOPENED, ID_PLCOPENEDITORTOOLBAR,
 ID_PLCOPENEDITORDEFAULTTOOLBAR, ID_PLCOPENEDITORSFCTOOLBAR, 
 ID_PLCOPENEDITORFBDTOOLBAR, ID_PLCOPENEDITORLDTOOLBAR,
] = [wx.NewId() for _init_ctrls in range(10)]

[ID_PLCOPENEDITORFILEMENUITEMS0, ID_PLCOPENEDITORFILEMENUITEMS1, 
 ID_PLCOPENEDITORFILEMENUITEMS2, ID_PLCOPENEDITORFILEMENUITEMS3, 
 ID_PLCOPENEDITORFILEMENUITEMS5, ID_PLCOPENEDITORFILEMENUITEMS6, 
 ID_PLCOPENEDITORFILEMENUITEMS7, ID_PLCOPENEDITORFILEMENUITEMS9, 
 ID_PLCOPENEDITORFILEMENUITEMS11,
] = [wx.NewId() for _init_coll_FileMenu_Items in range(9)]

[ID_PLCOPENEDITORHELPMENUITEMS0, ID_PLCOPENEDITORHELPMENUITEMS1, 
 ID_PLCOPENEDITORHELPMENUITEMS2, ID_PLCOPENEDITORHELPMENUITEMS3, 
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(4)]

[ID_PLCOPENEDITORSFCMENUITEMS0, ID_PLCOPENEDITORSFCMENUITEMS1, 
 ID_PLCOPENEDITORSFCMENUITEMS2, ID_PLCOPENEDITORSFCMENUITEMS3,
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(4)]

[ID_PLCOPENEDITORCONFIGMENUITEMS0, ID_PLCOPENEDITORCONFIGMENUITEMS1, 
] = [wx.NewId() for _init_coll_HelpMenu_Items in range(2)]

[ID_PLCOPENEDITOREDITMENUITEMS0, ID_PLCOPENEDITOREDITMENUITEMS1, 
 ID_PLCOPENEDITOREDITMENUITEMS11, ID_PLCOPENEDITOREDITMENUITEMS12, 
 ID_PLCOPENEDITOREDITMENUITEMS2, ID_PLCOPENEDITOREDITMENUITEMS4, 
 ID_PLCOPENEDITOREDITMENUITEMS5, ID_PLCOPENEDITOREDITMENUITEMS6, 
 ID_PLCOPENEDITOREDITMENUITEMS8, ID_PLCOPENEDITOREDITMENUITEMS9, 
] = [wx.NewId() for _init_coll_EditMenu_Items in range(10)]

[ID_PLCOPENEDITORSFCMENUITEMS0, ID_PLCOPENEDITORSFCMENUITEMS1, 
 ID_PLCOPENEDITORSFCMENUITEMS2, ID_PLCOPENEDITORSFCMENUITEMS3, 
] = [wx.NewId() for _init_coll_SFCMenu_Items in range(4)]

[ID_PLCOPENEDITORCONFIGMENUITEMS0, ID_PLCOPENEDITORCONFIGMENUITEMS1, 
] = [wx.NewId() for _init_coll_ConfigMenu_Items in range(2)]


#-------------------------------------------------------------------------------
#                            ToolBars definitions
#-------------------------------------------------------------------------------


[ID_PLCOPENEDITORTOOLBARSELECTION, ID_PLCOPENEDITORTOOLBARCOMMENT,
 ID_PLCOPENEDITORTOOLBARVARIABLE, ID_PLCOPENEDITORTOOLBARBLOCK,
 ID_PLCOPENEDITORTOOLBARCONNECTION, ID_PLCOPENEDITORTOOLBARWIRE,
 ID_PLCOPENEDITORTOOLBARPOWERRAIL, ID_PLCOPENEDITORTOOLBARRUNG,
 ID_PLCOPENEDITORTOOLBARCOIL, ID_PLCOPENEDITORTOOLBARCONTACT,
 ID_PLCOPENEDITORTOOLBARBRANCH, ID_PLCOPENEDITORTOOLBARINITIALSTEP,
 ID_PLCOPENEDITORTOOLBARSTEP, ID_PLCOPENEDITORTOOLBARTRANSITION,
 ID_PLCOPENEDITORTOOLBARACTIONBLOCK, ID_PLCOPENEDITORTOOLBARDIVERGENCE,
 ID_PLCOPENEDITORTOOLBARJUMP,
] = [wx.NewId() for _init_coll_DefaultToolBar_Items in range(17)]

ToolBarItems = {
    "FBD" : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOMMENT, "OnCommentTool", "comment.png", "Create a new comment"),
             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARVARIABLE, "OnVariableTool", "variable.png", "Create a new variable"),
             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBLOCK, "OnBlockTool", "block.png", "Create a new block"),
             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONNECTION, "OnConnectionTool", "connection.png", "Create a new connection"),
             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARWIRE, "OnWireTool", "wire.png", "Create a new wire")],
    "LD"  : [(True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOMMENT, "OnCommentTool", "comment.png", "Create a new comment"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARPOWERRAIL, "OnPowerRailTool", "powerrail.png", "Create a new power rail"),
             (False, DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARRUNG, "OnRungTool", "rung.png", "Create a new rung"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOIL, "OnCoilTool", "coil.png", "Create a new coil"),
             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONTACT, "OnContactTool", "contact.png", "Create a new contact"),
             (False, DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBRANCH, "OnBranchTool", "branch.png", "Create a new branch"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARVARIABLE, "OnVariableTool", "variable.png", "Create a new variable"),
             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBLOCK, "OnBlockTool", "block.png", "Create a new block"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONNECTION, "OnConnectionTool", "connection.png", "Create a new connection"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARWIRE, "OnWireTool", "wire.png", "Create a new wire")],
    "SFC" : [(True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCOMMENT, "OnCommentTool", "comment.png", "Create a new comment"),
             (True, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARINITIALSTEP, "OnInitialStepTool", "initial_step.png", "Create a new initial step"),
             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARSTEP, "OnStepTool", "step.png", "Create a new step"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARTRANSITION, "OnTransitionTool", "transition.png", "Create a new transition"),
             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARACTIONBLOCK, "OnActionBlockTool", "action.png", "Create a new action block"),
             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARDIVERGENCE, "OnDivergenceTool", "divergence.png", "Create a new divergence"),
             (False, FREEDRAWING_MODE|DRIVENDRAWING_MODE, ID_PLCOPENEDITORTOOLBARJUMP, "OnJumpTool", "jump.png", "Create a new jump"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARVARIABLE, "OnVariableTool", "variable.png", "Create a new variable"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARBLOCK, "OnBlockTool", "block.png", "Create a new block"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONNECTION, "OnConnectionTool", "connection.png", "Create a new connection"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARPOWERRAIL, "OnPowerRailTool", "powerrail.png", "Create a new power rail"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARCONTACT, "OnContactTool", "contact.png", "Create a new contact"),
             (True, FREEDRAWING_MODE, ID_PLCOPENEDITORTOOLBARWIRE, "OnWireTool", "wire.png", "Create a new wire")],
    "ST"  : [],
    "IL"  : []
}

class PLCOpenEditor(wx.Frame):
    _custom_classes = {'wx.SashWindow' : ['Viewer']}
    
    def _init_coll_EditMenu_Items(self, parent):
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS0,
              kind=wx.ITEM_NORMAL, text=u'Refresh\tCTRL+R')
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS1,
              kind=wx.ITEM_NORMAL, text=u'Undo\tCTRL+Z')
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS2,
              kind=wx.ITEM_NORMAL, text=u'Redo\tCTRL+Y')
        parent.AppendSeparator()
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS4,
              kind=wx.ITEM_NORMAL, text=u'Cut\tCTRL+X')
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS5,
              kind=wx.ITEM_NORMAL, text=u'Copy\tCTRL+C')
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS6,
              kind=wx.ITEM_NORMAL, text=u'Paste\tCTRL+V')
        parent.AppendSeparator()
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS8,
              kind=wx.ITEM_NORMAL, text=u'Add POU')
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS9,
              kind=wx.ITEM_NORMAL, text=u'Remove POU')
        parent.AppendSeparator()
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS11,
              kind=wx.ITEM_NORMAL, text=u'Add Configuration')
        parent.Append(help='', id=ID_PLCOPENEDITOREDITMENUITEMS12,
              kind=wx.ITEM_NORMAL, text=u'Remove Configuration')
        self.Bind(wx.EVT_MENU, self.OnRefreshMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS0)
        self.Bind(wx.EVT_MENU, self.OnUndoMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS1)
        self.Bind(wx.EVT_MENU, self.OnRedoMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS2)
        self.Bind(wx.EVT_MENU, self.OnCutMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS4)
        self.Bind(wx.EVT_MENU, self.OnCopyMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS5)
        self.Bind(wx.EVT_MENU, self.OnPasteMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS6)
        self.Bind(wx.EVT_MENU, self.OnAddPouMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS8)
        self.Bind(wx.EVT_MENU, self.OnRemovePouMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS9)
        self.Bind(wx.EVT_MENU, self.OnAddConfigurationMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS11)
        self.Bind(wx.EVT_MENU, self.OnRemoveConfigurationMenu,
              id=ID_PLCOPENEDITOREDITMENUITEMS12)

    def _init_coll_menuBar1_Menus(self, parent):
        if self.ModeSolo:
            parent.Append(menu=self.FileMenu, title=u'File')
        parent.Append(menu=self.EditMenu, title=u'Edit')
        parent.Append(menu=self.HelpMenu, title=u'Help')

    def _init_coll_ConfigMenu_Items(self, parent):
        parent.Append(help='', id=ID_PLCOPENEDITORCONFIGMENUITEMS0,
              kind=wx.ITEM_NORMAL, text=u'Add Resource')
        parent.Append(help='', id=ID_PLCOPENEDITORCONFIGMENUITEMS1,
              kind=wx.ITEM_NORMAL, text=u'Remove Resource')
        self.Bind(wx.EVT_MENU, self.OnAddResourceMenu,
              id=ID_PLCOPENEDITORCONFIGMENUITEMS0)
        self.Bind(wx.EVT_MENU, self.OnRemoveResourceMenu,
              id=ID_PLCOPENEDITORCONFIGMENUITEMS1)

    def _init_coll_HelpMenu_Items(self, parent):
        parent.Append(help='', id=ID_PLCOPENEDITORHELPMENUITEMS0,
              kind=wx.ITEM_NORMAL, text=u'PLCOpenEditor\tF1')
        parent.Append(help='', id=ID_PLCOPENEDITORHELPMENUITEMS1,
              kind=wx.ITEM_NORMAL, text=u'PLCOpen\tF2')
        parent.Append(help='', id=ID_PLCOPENEDITORHELPMENUITEMS2,
              kind=wx.ITEM_NORMAL, text=u'IEC 61131-3\tF3')
        if self.ModeSolo:
            parent.Append(help='', id=ID_PLCOPENEDITORHELPMENUITEMS3,
                  kind=wx.ITEM_NORMAL, text=u'About')
        self.Bind(wx.EVT_MENU, self.OnPLCOpenMenu,
              id=ID_PLCOPENEDITORHELPMENUITEMS1)

    def _init_coll_FileMenu_Items(self, parent):
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS0,
              kind=wx.ITEM_NORMAL, text=u'New\tCTRL+N')
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS1,
              kind=wx.ITEM_NORMAL, text=u'Open\tCTRL+O')
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS2,
              kind=wx.ITEM_NORMAL, text=u'Close Tab\tCTRL+W')
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS3,
              kind=wx.ITEM_NORMAL, text=u'Close Project')
        parent.AppendSeparator()
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS5,
              kind=wx.ITEM_NORMAL, text=u'Save\tCTRL+S')
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS6,
              kind=wx.ITEM_NORMAL, text=u'Save As...\tCTRL+SHIFT+S')
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS7,
              kind=wx.ITEM_NORMAL, text=u'Generate Program\tCTRL+G')
        parent.AppendSeparator()
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS9,
              kind=wx.ITEM_NORMAL, text=u'Properties')
        parent.AppendSeparator()
        parent.Append(help='', id=ID_PLCOPENEDITORFILEMENUITEMS11,
              kind=wx.ITEM_NORMAL, text=u'Quit\tCTRL+Q')
        self.Bind(wx.EVT_MENU, self.OnNewProjectMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS0)
        self.Bind(wx.EVT_MENU, self.OnOpenProjectMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS1)
        self.Bind(wx.EVT_MENU, self.OnCloseTabMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS2)
        self.Bind(wx.EVT_MENU, self.OnCloseProjectMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS3)
        self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS5)
        self.Bind(wx.EVT_MENU, self.OnSaveProjectAsMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS6)
        self.Bind(wx.EVT_MENU, self.OnGenerateProgramMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS7)
        self.Bind(wx.EVT_MENU, self.OnPropertiesMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS9)
        self.Bind(wx.EVT_MENU, self.OnQuitMenu,
              id=ID_PLCOPENEDITORFILEMENUITEMS11)

    def _init_coll_SFCMenu_Items(self, parent):
        parent.Append(help='', id=ID_PLCOPENEDITORSFCMENUITEMS0,
              kind=wx.ITEM_NORMAL, text=u'Add Transition')
        parent.Append(help='', id=ID_PLCOPENEDITORSFCMENUITEMS1,
              kind=wx.ITEM_NORMAL, text=u'Add Action')
        parent.Append(help='', id=ID_PLCOPENEDITORSFCMENUITEMS2,
              kind=wx.ITEM_NORMAL, text=u'Remove Transition')
        parent.Append(help='', id=ID_PLCOPENEDITORSFCMENUITEMS3,
              kind=wx.ITEM_NORMAL, text=u'Remove Action')
        self.Bind(wx.EVT_MENU, self.OnAddPouTransitionMenu,
              id=ID_PLCOPENEDITORSFCMENUITEMS0)
        self.Bind(wx.EVT_MENU, self.OnAddPouActionMenu,
              id=ID_PLCOPENEDITORSFCMENUITEMS1)
        self.Bind(wx.EVT_MENU, self.OnRemovePouTransitionMenu,
              id=ID_PLCOPENEDITORSFCMENUITEMS2)
        self.Bind(wx.EVT_MENU, self.OnRemovePouActionMenu,
              id=ID_PLCOPENEDITORSFCMENUITEMS3)

    def _init_utils(self):
        self.menuBar1 = wx.MenuBar()

        if self.ModeSolo:
            self.FileMenu = wx.Menu(title=u'')
        else:
            self.FileMenu = None
        self.EditMenu = wx.Menu(title=u'')
        self.HelpMenu = wx.Menu(title='')
        self.SFCMenu = wx.Menu(title='')
        self.ConfigMenu = wx.Menu(title='')

        self._init_coll_menuBar1_Menus(self.menuBar1)
        if self.ModeSolo:
            self._init_coll_FileMenu_Items(self.FileMenu)
        self._init_coll_EditMenu_Items(self.EditMenu)
        self._init_coll_HelpMenu_Items(self.HelpMenu)
        self._init_coll_SFCMenu_Items(self.SFCMenu)
        self._init_coll_ConfigMenu_Items(self.ConfigMenu)

    def _init_coll_EditorGridSizer_Items(self, parent):
        parent.AddWindow(self.ToolBar, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.TabsOpened, 0, border=0, flag=wx.GROW)

    def _init_coll_EditorGridSizer_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(1)

    def _init_sizers(self):
        if wx.Platform != '__WXMSW__':
            self.EditorGridSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=0)
        
            self._init_coll_EditorGridSizer_Growables(self.EditorGridSizer)
            self._init_coll_EditorGridSizer_Items(self.EditorGridSizer)
        
            self.EditorPanel.SetSizer(self.EditorGridSizer)

    def _init_ctrls(self, prnt):
        wx.Frame.__init__(self, id=ID_PLCOPENEDITOR, name=u'PLCOpenEditor',
              parent=prnt, pos=wx.Point(235, 287), size=wx.Size(1000, 600),
              style=wx.DEFAULT_FRAME_STYLE, title=u'PLCOpenEditor')
        self._init_utils()
        self.SetClientSize(wx.Size(1000, 600))
        self.SetMenuBar(self.menuBar1)
        self.Bind(wx.EVT_CLOSE, self.OnCloseFrame)
        if not self.ModeSolo:
            self.Bind(wx.EVT_MENU, self.OnSaveProjectMenu, id=ID_PLCOPENEDITORFILEMENUITEMS5)
            accel = wx.AcceleratorTable([wx.AcceleratorEntry(wx.ACCEL_CTRL, 83, ID_PLCOPENEDITORFILEMENUITEMS5)])
            self.SetAcceleratorTable(accel)
            self.Bind(wx.EVT_ACTIVATE, self.OnFrameActivated)
        
        self.MainSplitter = wx.SplitterWindow(id=ID_PLCOPENEDITORMAINSPLITTER,
              name='MainSplitter', parent=self, point=wx.Point(0, 0),
              size=wx.Size(0, 0), style=wx.SP_3D)
        self.MainSplitter.SetNeedUpdating(True)
        self.MainSplitter.SetMinimumPaneSize(1)
        
        self.ProjectTree = wx.TreeCtrl(id=ID_PLCOPENEDITORPROJECTTREE,
              name='treeCtrl1', parent=self.MainSplitter, pos=wx.Point(0, 0),
              size=wx.Size(-1, -1),
              style=wx.TR_HAS_BUTTONS|wx.TR_EDIT_LABELS|wx.TR_SINGLE|wx.SUNKEN_BORDER)
        self.ProjectTree.Bind(wx.EVT_RIGHT_UP, self.OnProjectTreeRightUp)
        self.Bind(wx.EVT_TREE_BEGIN_DRAG, self.OnProjectTreeBeginDrag,
              id=ID_PLCOPENEDITORPROJECTTREE)
        self.Bind(wx.EVT_TREE_BEGIN_LABEL_EDIT, self.OnProjectTreeItemBeginEdit,
              id=ID_PLCOPENEDITORPROJECTTREE)
        self.Bind(wx.EVT_TREE_END_LABEL_EDIT, self.OnProjectTreeItemEndEdit,
              id=ID_PLCOPENEDITORPROJECTTREE)
        self.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.OnProjectTreeItemActivated,
              id=ID_PLCOPENEDITORPROJECTTREE)
        self.Bind(wx.EVT_TREE_SEL_CHANGED, self.OnProjectTreeItemSelected,
              id=ID_PLCOPENEDITORPROJECTTREE)
        
        if wx.Platform == '__WXMSW__':
            self.ToolBar = self.CreateToolBar(wx.TB_HORIZONTAL|wx.TB_FLAT|wx.NO_BORDER, 
                  ID_PLCOPENEDITORTOOLBAR, 'ToolBar')
            self.ToolBar.SetToolBitmapSize(wx.Size(25, 25))
            self.ToolBar.AddRadioTool(ID_PLCOPENEDITORTOOLBARSELECTION, 
                  wx.Bitmap(os.path.join(CWD, 'Images', 'select.png')), wx.NullBitmap, "Select an object")
            self.ToolBar.Realize()
            self.Bind(wx.EVT_TOOL, self.OnSelectionTool, 
                  id=ID_PLCOPENEDITORTOOLBARSELECTION)
        
            self.TabsOpened = wx.Notebook(id=ID_PLCOPENEDITORTABSOPENED,
                  name='TabsOpened', parent=self.MainSplitter, pos=wx.Point(0,
                  0), size=wx.Size(0, 0), style=0)
            self.TabsOpened.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED,
                  self.OnPouSelectedChanged, id=ID_PLCOPENEDITORTABSOPENED)
        
            self.MainSplitter.SplitVertically(self.ProjectTree, self.TabsOpened, 200)
        
        else:
            self.EditorPanel = wx.Panel(id=ID_PLCOPENEDITOREDITORPANEL, 
                  name='TabPanel', parent=self.MainSplitter, pos=wx.Point(0, 0),
                  size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)
        
            self.ToolBar = wx.ToolBar(id=ID_PLCOPENEDITORTOOLBAR, name='ToolBar',
                  parent=self.EditorPanel, pos=wx.Point(0, 0), size=wx.Size(0, 40),
                  style=wx.TB_HORIZONTAL|wx.NO_BORDER)
            self.ToolBar.AddRadioTool(ID_PLCOPENEDITORTOOLBARSELECTION, 
                  wx.Bitmap(os.path.join(CWD, 'Images', 'select.png')), wx.NullBitmap, "Select an object")
            self.ToolBar.Realize()
            self.Bind(wx.EVT_TOOL, self.OnSelectionTool, 
                  id=ID_PLCOPENEDITORTOOLBARSELECTION)
        
            self.TabsOpened = wx.Notebook(id=ID_PLCOPENEDITORTABSOPENED,
                  name='TabsOpened', parent=self.EditorPanel, pos=wx.Point(0,
                  0), size=wx.Size(0, 0), style=0)
            self.TabsOpened.Bind(wx.EVT_NOTEBOOK_PAGE_CHANGED,
                  self.OnPouSelectedChanged, id=ID_PLCOPENEDITORTABSOPENED)
        
            self.MainSplitter.SplitVertically(self.ProjectTree, self.EditorPanel, 200)
        
        self._init_sizers()

    def __init__(self, parent, controler = None, fileOpen = None):
        self.ModeSolo = controler == None
        self._init_ctrls(parent)
        
        if wx.Platform != '__WXMSW__':
            self.TreeImageList = wx.ImageList(16, 16)
            for language in LANGUAGES:
                self.TreeImageList.Add(wx.Bitmap(os.path.join(CWD, 'Images','%s.png'%language)))
            self.ProjectTree.AssignImageList(self.TreeImageList)
        
        if self.ModeSolo:
            self.Controler = PLCControler()
            if fileOpen:
                self.Controler.OpenXMLFile(fileOpen)
                self.RefreshProjectTree()
        else:
            self.Controler = controler
        
        self.CurrentToolBar = []
        self.CurrentLanguage = ""
        self.DrawingMode = FREEDRAWING_MODE
        #self.DrawingMode = DRIVENDRAWING_MODE
        
        self.RefreshFileMenu()
        self.RefreshEditMenu()
        self.RefreshTitle()
        self.RefreshToolBar()

    def GetDrawingMode(self):
        return self.DrawingMode

    def RefreshTitle(self):
        if self.Controler.HasOpenedProject() > 0:
            self.SetTitle("PLCOpenEditor - %s"%self.Controler.GetFilename())
        else:
            self.SetTitle("PLCOpenEditor")

    def RefreshFileMenu(self):
        if self.FileMenu:
            if self.Controler.HasOpenedProject():
                if self.TabsOpened.GetPageCount() > 0:
                    self.FileMenu.FindItemByPosition(2).Enable(True)
                else:
                    self.FileMenu.FindItemByPosition(2).Enable(False)
                self.FileMenu.FindItemByPosition(3).Enable(True)
                self.FileMenu.FindItemByPosition(5).Enable(True)
                self.FileMenu.FindItemByPosition(6).Enable(True)
                self.FileMenu.FindItemByPosition(7).Enable(True)
                self.FileMenu.FindItemByPosition(9).Enable(True)
            else:
                self.FileMenu.FindItemByPosition(2).Enable(False)
                self.FileMenu.FindItemByPosition(3).Enable(False)
                self.FileMenu.FindItemByPosition(5).Enable(False)
                self.FileMenu.FindItemByPosition(6).Enable(False)
                self.FileMenu.FindItemByPosition(7).Enable(False)
                self.FileMenu.FindItemByPosition(9).Enable(False)

    def RefreshEditMenu(self):
        if self.EditMenu:
            if self.Controler.HasOpenedProject():
                undo, redo = self.Controler.GetBufferState()
                self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS1, undo)
                self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS2, redo)
            else:
                self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS1, False)
                self.EditMenu.Enable(ID_PLCOPENEDITOREDITMENUITEMS2, False)
            if self.Controler.HasOpenedProject():
                if self.TabsOpened.GetPageCount() > 0:
                    self.EditMenu.FindItemByPosition(0).Enable(True)
                else:
                    self.EditMenu.FindItemByPosition(0).Enable(False)
                self.EditMenu.FindItemByPosition(8).Enable(True)
                self.EditMenu.FindItemByPosition(9).Enable(True)
            else:
                self.EditMenu.FindItemByPosition(0).Enable(False)
                self.EditMenu.FindItemByPosition(8).Enable(False)
                self.EditMenu.FindItemByPosition(9).Enable(False)
            bodytype = self.Controler.GetCurrentElementEditingBodyType()
            if bodytype in ["IL","ST"]:
                self.EditMenu.FindItemByPosition(4).Enable(True)
                self.EditMenu.FindItemByPosition(5).Enable(True)
                self.EditMenu.FindItemByPosition(6).Enable(True)
            else:
                self.EditMenu.FindItemByPosition(4).Enable(False)
                self.EditMenu.FindItemByPosition(5).Enable(False)
            self.EditMenu.FindItemByPosition(6).Enable(False)

    def RefreshEditor(self):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).RefreshView()

    def ShowProperties(self):
        old_values = self.Controler.GetProjectProperties()
        dialog = ProjectDialog(self)
        dialog.SetValues(old_values)
        if dialog.ShowModal() == wx.ID_OK:
            new_values = dialog.GetValues()
            projectname = new_values.pop("projectName")
            new_values["creationDateTime"] = old_values["creationDateTime"]
            self.Controler.SetProjectProperties(projectname, new_values)
            self.RefreshProjectTree()
        dialog.Destroy()

    def OnFrameActivated(self, event):
        if not self.ModeSolo and event.GetActive():
            self.Controler.RefreshPluginsBlockLists()
            selected = self.TabsOpened.GetSelection()
            if selected >= 0:
                self.TabsOpened.GetPage(selected).RefreshView()
        event.Skip()

    def OnCloseFrame(self, event):
        if not self.ModeSolo and getattr(self, "_onclose", None) != None:
            self._onclose()
            event.Skip()
        elif not self.Controler.ProjectIsSaved():
            dialog = wx.MessageDialog(self, "There are changes, do you want to save?", "Close Application", wx.YES_NO|wx.CANCEL|wx.ICON_QUESTION)
            answer = dialog.ShowModal()
            dialog.Destroy()
            if answer == wx.ID_YES:
                self.SaveProject()
                event.Skip()
            elif answer == wx.ID_NO:
                self.Controler.Reset()
                wx.CallAfter(self.Close)
                event.Skip()
            else:
                event.Veto()
        else:
            event.Skip()

    def OnNewProjectMenu(self, event):
        dialog = ProjectDialog(self)
        if dialog.ShowModal() == wx.ID_OK:
            values = dialog.GetValues()
            projectname = values.pop("projectName")
            values["creationDateTime"] = datetime(*localtime()[:6])
            self.Controler.CreateNewProject(projectname)
            self.Controler.SetProjectProperties(projectname, values)
            self.RefreshTitle()
            self.RefreshFileMenu()
            self.RefreshEditMenu()
            self.RefreshProjectTree()
        event.Skip()

    def OnOpenProjectMenu(self, event):
        filepath = self.Controler.GetFilePath()
        if filepath != "":
            directory = os.path.dirname(filepath)
        else:
            directory = os.getcwd()
        dialog = wx.FileDialog(self, "Choose a file", directory, "",  "PLCOpen files (*.xml)|*.xml|All files|*.*", wx.OPEN)
        if dialog.ShowModal() == wx.ID_OK:
            filepath = dialog.GetPath()
            if os.path.isfile(filepath):
                self.Controler.OpenXMLFile(filepath)
                self.TabsOpened.DeleteAllPages()
                self.RefreshProjectTree()
            self.RefreshTitle()
            self.RefreshFileMenu()
            self.RefreshEditMenu()
            self.RefreshToolBar()
        dialog.Destroy()
        event.Skip()

    def OnCloseTabMenu(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected >= 0:
            self.Controler.CloseElementEditing()
            self.TabsOpened.DeletePage(selected)
            if self.TabsOpened.GetPageCount() > 0:
                self.TabsOpened.SetSelection(min(selected, self.TabsOpened.GetPageCount() - 1))
            self.RefreshTitle()
            self.RefreshFileMenu()
            self.RefreshEditMenu()
            self.RefreshToolBar()
        event.Skip()
    
    def OnCloseProjectMenu(self, event):
        self.Controler.Reset()
        self.TabsOpened.DeleteAllPages()
        self.ProjectTree.DeleteAllItems()
        self.RefreshTitle()
        self.RefreshFileMenu()
        self.RefreshEditMenu()
        self.RefreshToolBar()
        event.Skip()

    def OnSaveProjectMenu(self, event):
        if not self.ModeSolo and getattr(self, "_onsave", None) != None:
            self._onsave()
        else:
            self.SaveProject()
        event.Skip()

    def OnSaveProjectAsMenu(self, event):
        self.SaveProjectAs()
        event.Skip()

    def OnGenerateProgramMenu(self, event):
        dialog = wx.FileDialog(self, "Choose a file", os.getcwd(), self.Controler.GetProgramFilePath(),  "ST files (*.st)|*.st|All files|*.*", wx.SAVE|wx.CHANGE_DIR)
        if dialog.ShowModal() == wx.ID_OK:
            filepath = dialog.GetPath()
            if os.path.isdir(os.path.dirname(filepath)):
                result = self.Controler.GenerateProgram(filepath)
                if not result:
                    message = wx.MessageDialog(self, "Can't generate program to file %s!"%filepath, "Error", wx.OK|wx.ICON_ERROR)
                    message.ShowModal()
                    message.Destroy()
                else:
                    message = wx.MessageDialog(self, "Program was successfully generated!", "Done", wx.OK|wx.ICON_INFORMATION)
                    message.ShowModal()
                    message.Destroy()
            else:
                message = wx.MessageDialog(self, "%s is not a valid folder!"%os.path.dirname(filepath), "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
        dialog.Destroy()
        event.Skip()

    def SaveProject(self):
        result = self.Controler.SaveXMLFile()
        if not result:
            self.SaveProjectAs()
        else:
            self.RefreshTitle()
    
    def SaveProjectAs(self):
        filepath = self.Controler.GetFilePath()
        if filepath != "":
            directory, filename = os.path.split(filepath)
        else:
            directory, filename = os.getcwd(), "%(projectName)s.xml"%self.Controler.GetProjectProperties()
        dialog = wx.FileDialog(self, "Choose a file", directory, filename,  "PLCOpen files (*.xml)|*.xml|All files|*.*", wx.SAVE|wx.OVERWRITE_PROMPT)
        if dialog.ShowModal() == wx.ID_OK:
            filepath = dialog.GetPath()
            if os.path.isdir(os.path.dirname(filepath)):
                result = self.Controler.SaveXMLFile(filepath)
                if not result:
                    message = wx.MessageDialog(self, "Can't save project to file %s!"%filepath, "Error", wx.OK|wx.ICON_ERROR)
                    message.ShowModal()
                    message.Destroy()
            else:
                message = wx.MessageDialog(self, "%s is not a valid folder!"%os.path.dirname(filepath), "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
            self.RefreshTitle()
        dialog.Destroy()

    def OnPropertiesMenu(self, event):
        self.ShowProperties()
        event.Skip()

    def OnQuitMenu(self, event):
        self.Close()
        event.Skip()

    def ResetCurrentMode(self):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            window = self.TabsOpened.GetPage(selected)
            window.SetMode(MODE_SELECTION)
        self.ToolBar.ToggleTool(ID_PLCOPENEDITORTOOLBARSELECTION, True)

    def ResetToolToggle(self, id):
        tool = self.ToolBar.FindById(id)
        tool.SetToggle(False)

    def OnSelectionTool(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_SELECTION)
        event.Skip()
    
    def OnCommentTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCOMMENT)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_COMMENT)
        event.Skip()
    
    def OnVariableTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARVARIABLE)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_VARIABLE)
        event.Skip()
    
    def OnBlockTool(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_BLOCK)
        event.Skip()
        
    def OnConnectionTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCONNECTION)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_CONNECTION)
        event.Skip()

    def OnWireTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARWIRE)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_WIRE)
        event.Skip()

    def OnPowerRailTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARPOWERRAIL)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_POWERRAIL)
        event.Skip()

    def OnRungTool(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).Viewer.AddLadderRung()
        event.Skip()
    
    def OnCoilTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCOIL)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_COIL)
        event.Skip()
    
    def OnContactTool(self, event):
        if self.DrawingMode == FREEDRAWING_MODE:
            self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARCONTACT)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            if self.DrawingMode == FREEDRAWING_MODE:
                self.TabsOpened.GetPage(selected).SetMode(MODE_CONTACT)
            else:
                self.TabsOpened.GetPage(selected).Viewer.AddLadderContact()
        event.Skip()
    
    def OnBranchTool(self, event): 
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).Viewer.AddLadderBranch()
        event.Skip()    
    
    def OnInitialStepTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARINITIALSTEP)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_INITIALSTEP)
        event.Skip()
    
    def OnStepTool(self, event):
        if self.GetDrawingMode() == FREEDRAWING_MODE:
            self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARSTEP)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            if self.GetDrawingMode() == FREEDRAWING_MODE:
                self.TabsOpened.GetPage(selected).SetMode(MODE_STEP)
            else:
                self.TabsOpened.GetPage(selected).Viewer.AddStep()
        event.Skip()

    def OnActionBlockTool(self, event):
        if self.GetDrawingMode() == FREEDRAWING_MODE:
            self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARACTIONBLOCK)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            if self.GetDrawingMode() == FREEDRAWING_MODE:
                self.TabsOpened.GetPage(selected).SetMode(MODE_ACTION)
            else:
                self.TabsOpened.GetPage(selected).Viewer.AddStepAction()
        event.Skip()

    def OnTransitionTool(self, event):
        self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARTRANSITION)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).SetMode(MODE_TRANSITION)
        event.Skip()

    def OnDivergenceTool(self, event):
        if self.GetDrawingMode() == FREEDRAWING_MODE:
            self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARDIVERGENCE)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            if self.GetDrawingMode() == FREEDRAWING_MODE:
                self.TabsOpened.GetPage(selected).SetMode(MODE_DIVERGENCE)
            else:
                self.TabsOpened.GetPage(selected).Viewer.AddDivergence()
        event.Skip()
    
    def OnJumpTool(self, event):
        if self.GetDrawingMode() == FREEDRAWING_MODE:
            self.ResetToolToggle(ID_PLCOPENEDITORTOOLBARJUMP)
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            if self.GetDrawingMode() == FREEDRAWING_MODE:
                self.TabsOpened.GetPage(selected).SetMode(MODE_JUMP)
            else:
                self.TabsOpened.GetPage(selected).Viewer.AddJump()
        event.Skip()
        
    def OnPouSelectedChanged(self, event):
        old_selected = self.TabsOpened.GetSelection()
        if old_selected >= 0:
            self.TabsOpened.GetPage(old_selected).ResetBuffer()
        selected = event.GetSelection()
        if selected >= 0:
            self.Controler.RefreshCurrentElementEditing(selected)
            found = False
            name = self.TabsOpened.GetPageText(selected)
            root = self.ProjectTree.GetRootItem()
            item, root_cookie = self.ProjectTree.GetFirstChild(root)
            while item.IsOk() and not found:
                if self.ProjectTree.GetItemText(item) == name:
                    self.ProjectTree.SelectItem(item)
                item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
            self.TabsOpened.GetPage(selected).RefreshView()
            self.RefreshFileMenu()
            self.RefreshEditMenu()
            self.RefreshToolBar()
        event.Skip()

    def OnProjectTreeBeginDrag(self, event):
        item = event.GetItem()
        if self.ProjectTree.GetPyData(item) == ITEM_POU:
            block_name = self.ProjectTree.GetItemText(item)
            block_type = self.Controler.GetPouType(block_name)
            if block_type != "program":
                data = wx.TextDataObject(str((block_name, block_type, "")))
                dragSource = wx.DropSource(self.ProjectTree)
                dragSource.SetData(data)
                dragSource.DoDragDrop()

    def OnProjectTreeItemEndEdit(self, event):
        message = None
        abort = False
        new_name = event.GetLabel()
        if new_name != "":
            if not TestIdentifier(new_name):
                message = "\"%s\" is not a valid identifier!"%new_name
            elif new_name.upper() in IEC_KEYWORDS:
                message = "\"%s\" is a keyword. It can't be used!"%new_name
            else:
                item = event.GetItem()
                itemtype = self.ProjectTree.GetPyData(item)
                if itemtype == ITEM_PROJECT:
                    self.Controler.SetProjectProperties(name = new_name)
                elif itemtype == ITEM_POU:
                    if new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
                        message = "\"%s\" pou already exists!"%new_name
                        abort = True
                    elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables()]:
                        messageDialog = wx.MessageDialog(self, "A variable is defined with \"%s\" as name. It can generate a conflict. Do you wish to continue?"%new_name, "Error", wx.YES_NO|wx.ICON_QUESTION)
                        if messageDialog.ShowModal() == wx.ID_NO:
                            abort = True
                        messageDialog.Destroy()
                    if not abort:
                        old_name = self.ProjectTree.GetItemText(item)
                        self.Controler.ChangePouName(old_name, new_name)
                        self.RefreshTabsOpenedTitles()
                elif itemtype == ITEM_TRANSITION:
                    category = self.ProjectTree.GetItemParent(item)
                    pou = self.ProjectTree.GetItemParent(category)
                    pou_name = self.ProjectTree.GetItemText(pou)
                    if new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
                        message = "A pou with \"%s\" as name exists!"%new_name
                    elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables(pou_name)]:
                        message = "A variable with \"%s\" as name already exists in this pou!"%new_name
                    else:
                        old_name = self.ProjectTree.GetItemText(item)
                        self.Controler.ChangePouTransitionName(pou_name, old_name, new_name)
                        self.RefreshTabsOpenedTitles()
                elif itemtype == ITEM_ACTION:
                    category = self.ProjectTree.GetItemParent(item)
                    pou = self.ProjectTree.GetItemParent(category)
                    pou_name = self.ProjectTree.GetItemText(pou)
                    if new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouNames()]:
                        message = "A pou with \"%s\" as name exists!"%new_name
                    elif new_name.upper() in [name.upper() for name in self.Controler.GetProjectPouVariables(pou_name)]:
                        message = "A variable with \"%s\" as name already exists in this pou!"%new_name
                    else:
                        old_name = self.ProjectTree.GetItemText(item)
                        self.Controler.ChangePouActionName(pou_name, old_name, new_name)
                        self.RefreshTabsOpenedTitles()
            if message or abort:
                if message:
                    messageDialog = wx.MessageDialog(self, message, "Error", wx.OK|wx.ICON_ERROR)
                    messageDialog.ShowModal()
                    messageDialog.Destroy()
                item = event.GetItem()
                wx.CallAfter(self.ProjectTree.EditLabel, item)
                event.Veto()
            else:
                wx.CallAfter(self.RefreshProjectTree)
                event.Skip()

    def OnProjectTreeItemBeginEdit(self, event):
        selected = event.GetItem()
        if self.ProjectTree.GetPyData(selected) == ITEM_UNEDITABLE:
            event.Veto()
        else:
            event.Skip()
    
    def OnProjectTreeItemActivated(self, event):
        selected = event.GetItem()
        if self.ProjectTree.IsExpanded(selected):
            self.ProjectTree.Collapse(selected)
        else:
            self.ProjectTree.Expand(selected)
        name = self.ProjectTree.GetItemText(selected)
        data = self.ProjectTree.GetPyData(selected)
        if name == "Properties":
            self.ShowProperties()
        elif data in [ITEM_POU, ITEM_TRANSITION, ITEM_ACTION, ITEM_RESOURCE, ITEM_CONFIGURATION]:
            idx = None
            if data == ITEM_CONFIGURATION:
                idx = self.Controler.OpenConfigurationEditing(name)
                if idx != None:
                    new_window = PouEditorPanel(self.TabsOpened, self, self.Controler, "config", config_name = name)
                    self.TabsOpened.AddPage(new_window, "")
                else:
                    idx = self.Controler.ChangeConfigurationEditing(name)
            elif data == ITEM_RESOURCE:
                item = self.ProjectTree.GetItemParent(selected)
                item_type = self.ProjectTree.GetPyData(item)
                while item_type != ITEM_CONFIGURATION:
                    item = self.ProjectTree.GetItemParent(item)
                    item_type = self.ProjectTree.GetPyData(item)
                config_name = self.ProjectTree.GetItemText(item)
                idx = self.Controler.OpenConfigurationResourceEditing(config_name, name)
                if idx != None:
                    new_window = PouEditorPanel(self.TabsOpened, self, self.Controler, "resource", config_name = config_name, resource_name = name)
                    self.TabsOpened.AddPage(new_window, "")
                else:
                    idx = self.Controler.ChangeConfigurationResourceEditing(config_name, name)
            elif data == ITEM_POU:
                idx = self.Controler.OpenPouEditing(name)
                if idx != None:
                    new_window = PouEditorPanel(self.TabsOpened, self, self.Controler, self.Controler.GetPouType(name), pou_name = name)
                    self.TabsOpened.AddPage(new_window, "")
                else:
                    idx = self.Controler.ChangePouEditing(name)
            else:
                item = self.ProjectTree.GetItemParent(selected)
                item_type = self.ProjectTree.GetPyData(item)
                while item_type != ITEM_POU:
                    item = self.ProjectTree.GetItemParent(item)
                    item_type = self.ProjectTree.GetPyData(item)
                pou_name = self.ProjectTree.GetItemText(item)
                if data == ITEM_TRANSITION:
                    idx = self.Controler.OpenPouTransitionEditing(pou_name, name)
                    if idx != None:
                        new_window = PouEditorPanel(self.TabsOpened, self, self.Controler, "transition", pou_name = pou_name, transition_name = name)
                        self.TabsOpened.AddPage(new_window, "")
                    else:
                        idx = self.Controler.ChangePouTransitionEditing(pou_name, name)
                elif data == ITEM_ACTION:
                    idx = self.Controler.OpenPouActionEditing(pou_name, name)
                    if idx != None:
                        new_window = PouEditorPanel(self.TabsOpened, self, self.Controler, "action", pou_name = pou_name, action_name = name)
                        self.TabsOpened.AddPage(new_window, "")
                    else:
                        idx = self.Controler.ChangePouActionEditing(pou_name, name)
            if idx != None:
                old_selected = self.TabsOpened.GetSelection()
                if old_selected >= 0:
                    self.TabsOpened.GetPage(old_selected).ResetBuffer()
                self.TabsOpened.SetSelection(idx)
                window = self.TabsOpened.GetPage(idx)
                window.RefreshView()
                self.RefreshTabsOpenedTitles()
                self.RefreshFileMenu()
                self.RefreshEditMenu()
                self.RefreshToolBar()
    
    def OnProjectTreeRightUp(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_POU:
            name = self.ProjectTree.GetItemText(selected)
            if self.Controler.GetPouBodyType(name) == "SFC":
                self.PopupMenu(self.SFCMenu)
        elif self.ProjectTree.GetPyData(selected) == ITEM_CONFIGURATION:
            self.PopupMenu(self.ConfigMenu)
        event.Skip()
    
    def OnProjectTreeItemSelected(self, event):
        selected = event.GetItem()
        name = self.ProjectTree.GetItemText(selected)
        data = self.ProjectTree.GetPyData(selected)
        if data in [ITEM_POU, ITEM_TRANSITION, ITEM_ACTION, ITEM_RESOURCE, ITEM_CONFIGURATION]:
            idx = None
            if data == ITEM_CONFIGURATION:
                if self.Controler.IsConfigurationEditing(name):
                    idx = self.Controler.ChangeConfigurationEditing(name)
            elif data == ITEM_RESOURCE:
                item = self.ProjectTree.GetItemParent(selected)
                item_type = self.ProjectTree.GetPyData(item)
                while item_type != ITEM_CONFIGURATION:
                    item = self.ProjectTree.GetItemParent(item)
                    item_type = self.ProjectTree.GetPyData(item)
                config_name = self.ProjectTree.GetItemText(item)
                if self.Controler.IsConfigurationResourceEditing(config_name, name):
                    idx = self.Controler.ChangeConfigurationResourceEditing(config_name, name)
            elif data == ITEM_POU:
                if self.Controler.IsPouEditing(name):
                    idx = self.Controler.ChangePouEditing(name)
            else:
                item = self.ProjectTree.GetItemParent(selected)
                item_type = self.ProjectTree.GetPyData(item)
                while item_type != ITEM_POU:
                    item = self.ProjectTree.GetItemParent(item)
                    item_type = self.ProjectTree.GetPyData(item)
                pou_name = self.ProjectTree.GetItemText(item)
                if data == ITEM_TRANSITION:
                    if self.Controler.IsPouTransitionEditing(pou_name, name):
                        idx = self.Controler.ChangePouTransitionEditing(pou_name, name)
                elif data == ITEM_ACTION:
                    if self.Controler.IsPouActionEditing(pou_name, name):
                        idx = self.Controler.ChangePouActionEditing(pou_name, name)
            if idx != None:
                old_selected = self.TabsOpened.GetSelection()
                if old_selected >= 0:
                    self.TabsOpened.GetPage(old_selected).ResetBuffer()
                self.TabsOpened.SetSelection(idx)
                window = self.TabsOpened.GetPage(idx)
                window.RefreshView()
                self.RefreshTabsOpenedTitles()
                self.RefreshFileMenu()
                self.RefreshEditMenu()
                self.RefreshToolBar()
        event.Skip()

    def RefreshProjectTree(self):
        infos = self.Controler.GetProjectInfos()
        root = self.ProjectTree.GetRootItem()
        if not root.IsOk():
            root = self.ProjectTree.AddRoot(infos["name"])
        self.GenerateTreeBranch(root, infos)
        self.ProjectTree.Expand(self.ProjectTree.GetRootItem())

    def GenerateTreeBranch(self, root, infos):
        to_delete = []
        self.ProjectTree.SetItemText(root, infos["name"])
        self.ProjectTree.SetPyData(root, infos["type"])
        if infos["type"] == ITEM_POU and wx.Platform != '__WXMSW__':
            self.ProjectTree.SetItemImage(root, LANGUAGES.index(self.Controler.GetPouBodyType(infos["name"])))
        item, root_cookie = self.ProjectTree.GetFirstChild(root)
        for values in infos["values"]:
            if not item.IsOk():
                item = self.ProjectTree.AppendItem(root, "")
                if wx.VERSION < (2, 7, 0):
                    item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
            self.GenerateTreeBranch(item, values)
            item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
        while item.IsOk():
            to_delete.append(item)
            item, root_cookie = self.ProjectTree.GetNextChild(root, root_cookie)
        for item in to_delete:
            self.ProjectTree.Delete(item)
    
    def ResetToolBar(self):
        for item in self.CurrentToolBar:
            self.Unbind(wx.EVT_MENU, id=item)
            self.ToolBar.DeleteTool(item)

    def RefreshToolBar(self):
        language = self.Controler.GetCurrentElementEditingBodyType()
        if language and language != self.CurrentLanguage:
            self.ResetToolBar()
            self.CurrentLanguage = language
            self.CurrentToolBar = []
            for radio, modes, id, method, picture, help in ToolBarItems[language]:
                if modes & self.DrawingMode:
                    if radio or self.DrawingMode == FREEDRAWING_MODE:
                        self.ToolBar.AddRadioTool(id, wx.Bitmap(os.path.join(CWD, "Images", picture)), wx.NullBitmap, help)
                    else:
                        self.ToolBar.AddSimpleTool(id, wx.Bitmap(os.path.join(CWD, "Images", picture)), help)
                    self.Bind(wx.EVT_TOOL, getattr(self, method), id=id)
                    self.CurrentToolBar.append(id)
            self.ToolBar.Realize()
        elif not language:
            self.ResetToolBar()
            self.CurrentLanguage = language
        self.ResetCurrentMode()

    def RefreshTabsOpenedTitles(self):
        pous = self.Controler.GetElementsOpenedNames()
        for i, pou in enumerate(pous):
            self.TabsOpened.SetPageText(i, pou)
    
    def OnRefreshMenu(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).RefreshView()
            self.TabsOpened.GetPage(selected).Refresh()
        event.Skip()

    def OnUndoMenu(self, event):
        self.Controler.LoadPrevious()
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).RefreshView()
            self.TabsOpened.GetPage(selected).Refresh()
        self.RefreshTitle()
        self.RefreshEditMenu()
        self.RefreshProjectTree()
        event.Skip()
    
    def OnRedoMenu(self, event):
        self.Controler.LoadNext()
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            self.TabsOpened.GetPage(selected).RefreshView()
            self.TabsOpened.GetPage(selected).Refresh()
        self.RefreshTitle()
        self.RefreshEditMenu()
        self.RefreshProjectTree()
        event.Skip()

    def OnCutMenu(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            try:
                self.TabsOpened.GetPage(selected).Viewer.Cut()
            except:
                pass
        event.Skip()
    
    def OnCopyMenu(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            try:
                self.TabsOpened.GetPage(selected).Viewer.Copy()
            except:
                pass
        event.Skip()
    
    def OnPasteMenu(self, event):
        selected = self.TabsOpened.GetSelection()
        if selected != -1:
            try:
                self.TabsOpened.GetPage(selected).Viewer.Paste()
            except:
                pass
        event.Skip()
    
    def OnAddPouMenu(self, event):
        dialog = PouDialog(self)
        dialog.SetPouNames(self.Controler.GetProjectPouNames())
        dialog.SetPouElementNames(self.Controler.GetProjectPouVariables())
        if dialog.ShowModal() == wx.ID_OK:
            values = dialog.GetValues()
            self.Controler.ProjectAddPou(values["pouName"], values["pouType"], values["language"])
            self.RefreshTitle()
            self.RefreshEditMenu()
            self.RefreshProjectTree()
        dialog.Destroy()
        event.Skip()

    def OnRemovePouMenu(self, event):
        pous = self.Controler.GetProjectPouNames()
        dialog = wx.SingleChoiceDialog(self, "Select POU to remove:", "POU Remove", pous, wx.OK|wx.CANCEL)
        if dialog.ShowModal() == wx.ID_OK:
            selected = dialog.GetStringSelection()
            if not self.Controler.PouIsUsed(selected):
                self.Controler.ProjectRemovePou(selected)
                deleted = None
                for i in xrange(self.TabsOpened.GetPageCount()):
                    if self.TabsOpened.GetPageText(i) == selected:
                        deleted = i
                if deleted != None:
                    self.TabsOpened.DeletePage(i)
                self.RefreshTitle()
                self.RefreshEditMenu()
                self.RefreshProjectTree()
                self.RefreshToolBar()
            else:
                message = wx.MessageDialog(self, "%s is used by one or more POUs. It can't be removed!"%selected, "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
        event.Skip()

    def OnAddConfigurationMenu(self, event):
        dialog = ConfigurationNameDialog(self, "Please enter configuration name", "Add new configuration")
        dialog.SetPouNames(self.Controler.GetProjectPouNames())
        dialog.SetPouElementNames(self.Controler.GetProjectPouVariables())
        if dialog.ShowModal() == wx.ID_OK:
            value = dialog.GetValue()
            self.Controler.ProjectAddConfiguration(value)
            self.RefreshTitle()
            self.RefreshEditMenu()
            self.RefreshProjectTree()
        dialog.Destroy()
        event.Skip()

    def OnRemoveConfigurationMenu(self, event):
        configs = self.Controler.GetProjectConfigNames()
        dialog = wx.SingleChoiceDialog(self, "Please select a configuration", "Remove configuration", configs, wx.OK|wx.CANCEL)
        if dialog.ShowModal() == wx.ID_OK:
            selected = dialog.GetStringSelection()
            self.Controler.ProjectRemoveConfiguration(selected)
            self.RefreshTitle()
            self.RefreshEditMenu()
            self.RefreshProjectTree()
        event.Skip()

    def OnAddPouTransitionMenu(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_POU:
            pouname = self.ProjectTree.GetItemText(selected)
            if self.Controler.GetPouBodyType(pouname) == "SFC":
                dialog = PouTransitionDialog(self)
                dialog.SetPouNames(self.Controler.GetProjectPouNames())
                dialog.SetPouElementNames(self.Controler.GetProjectPouVariables(pouname))
                if dialog.ShowModal() == wx.ID_OK: 
                    values = dialog.GetValues()
                    self.Controler.ProjectAddPouTransition(pouname, values["transitionName"], values["language"])
                    self.RefreshTitle()
                    self.RefreshEditMenu()
                    self.RefreshProjectTree()
                dialog.Destroy()
        event.Skip()

    def OnRemovePouTransitionMenu(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_POU:
            pouname = self.ProjectTree.GetItemText(selected)
            if self.Controler.GetPouBodyType(pouname) == "SFC":
                transitions = self.Controler.GetPouTransitions(pouname)
                dialog = wx.SingleChoiceDialog(self, "Select Transition to remove:", "Remove transition", transitions, wx.OK|wx.CANCEL)
                if dialog.ShowModal() == wx.ID_OK: 
                    selected = dialog.GetStringSelection()
                    self.Controler.ProjectRemovePouTransition(pouname, selected)
                    self.RefreshTitle()
                    self.RefreshEditMenu()
                    self.RefreshProjectTree()
                dialog.Destroy()
        event.Skip()

    def OnAddPouActionMenu(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_POU:
            pouname = self.ProjectTree.GetItemText(selected)
            if self.Controler.GetPouBodyType(pouname) == "SFC":
                dialog = PouActionDialog(self)
                dialog.SetPouNames(self.Controler.GetProjectPouNames())
                dialog.SetPouElementNames(self.Controler.GetProjectPouVariables(pouname))
                if dialog.ShowModal() == wx.ID_OK:
                    values = dialog.GetValues()
                    self.Controler.ProjectAddPouAction(pouname, values["actionName"], values["language"])
                    self.RefreshTitle()
                    self.RefreshEditMenu()
                    self.RefreshProjectTree()
                dialog.Destroy()
        event.Skip()

    def OnRemovePouActionMenu(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_POU:
            pouname = self.ProjectTree.GetItemText(selected)
            if self.Controler.GetPouBodyType(pouname) == "SFC":
                actions = self.Controler.GetPouActions(pouname)
                dialog = wx.SingleChoiceDialog(self, "Select Action to remove:", "Remove action", actions, wx.OK|wx.CANCEL)
                if dialog.ShowModal() == wx.ID_OK: 
                    selected = dialog.GetStringSelection()
                    self.Controler.ProjectRemovePouAction(pouname, selected)
                    self.RefreshTitle()
                    self.RefreshEditMenu()
                    self.RefreshProjectTree()
                dialog.Destroy()
        event.Skip()

    def OnAddResourceMenu(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_CONFIGURATION:
            config_name = self.ProjectTree.GetItemText(selected)
            dialog = ResourceNameDialog(self, "Please enter resource name", "Add new resource")
            dialog.SetPouNames(self.Controler.GetProjectPouNames())
            dialog.SetPouElementNames(self.Controler.GetProjectPouVariables())
            if dialog.ShowModal() == wx.ID_OK:
                value = dialog.GetValue()
                self.Controler.ProjectAddConfigurationResource(config_name, value)
                self.RefreshTitle()
                self.RefreshEditMenu()
                self.RefreshProjectTree()
            dialog.Destroy()
        event.Skip()

    def OnRemoveResourceMenu(self, event):
        selected = self.ProjectTree.GetSelection()
        if self.ProjectTree.GetPyData(selected) == ITEM_CONFIGURATION:
            config_name = self.ProjectTree.GetItemText(selected)
            infos = self.Controler.GetProjectInfos()
            resources = []
            for config in infos["configs"]:
                if config["name"] == config_name:
                    resources = config["resources"]
            dialog = wx.SingleChoiceDialog(self, "Select Resource to remove:", "Remove resource", resources, wx.OK|wx.CANCEL)
            if dialog.ShowModal() == wx.ID_OK:
                resource = dialog.GetStringSelection()
                self.Controler.ProjectRemoveConfigurationResource(config_name, resource)
                self.RefreshTitle()
                self.RefreshEditMenu()
                self.RefreshProjectTree()
            dialog.Destroy()
        event.Skip()

    def OnPLCOpenMenu(self, event):
        result = OpenPDFDoc()
        if type(result) == StringType:
            message = wx.MessageDialog(self, result, "ERROR", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        event.Skip()

current_num = 0
def GetNewNum():
    global current_num
    current_num += 1
    return current_num

#-------------------------------------------------------------------------------
#                            Create Project Dialog
#-------------------------------------------------------------------------------

[ID_PROJECTDIALOG, ID_PROJECTDIALOGPROJECTNAME, 
 ID_PROJECTDIALOGCOMPANYNAME, ID_PROJECTDIALOGCOMPANYURL, 
 ID_PROJECTDIALOGPRODUCTNAME, ID_PROJECTDIALOGPRODUCTVERSION, 
 ID_PROJECTDIALOGPRODUCTRELEASE, ID_PROJECTDIALOGCONTENTDESCRIPTION, 
 ID_PROJECTDIALOGSTATICTEXT1, ID_PROJECTDIALOGSTATICTEXT2, 
 ID_PROJECTDIALOGSTATICTEXT3, ID_PROJECTDIALOGSTATICTEXT4, 
 ID_PROJECTDIALOGSTATICTEXT5, ID_PROJECTDIALOGSTATICTEXT6, 
 ID_PROJECTDIALOGSTATICTEXT7, 
] = [wx.NewId() for _init_ctrls in range(15)]
        
class ProjectDialog(wx.Dialog):
    def _init_coll_flexGridSizer1_Items(self, parent):
        parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
        
    def _init_coll_flexGridSizer1_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(0)
    
    def _init_coll_MainSizer_Items(self, parent):
        parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.ProjectName, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.CompanyName, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.CompanyURL, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText4, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.ProductName, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText5, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.ProductVersion, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText6, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.ProductRelease, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText7, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.ContentDescription, 0, border=0, flag=wx.GROW)
        
    def _init_coll_MainSizer_Growables(self, parent):
        parent.AddGrowableCol(1)
        parent.AddGrowableRow(6)
        
    def _init_sizers(self):
        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
        self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=7, vgap=15)

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
        self._init_coll_MainSizer_Items(self.MainSizer)
        self._init_coll_MainSizer_Growables(self.MainSizer)

        self.SetSizer(self.flexGridSizer1)

    def _init_ctrls(self, prnt):
        wx.Dialog.__init__(self, id=ID_PROJECTDIALOG,
              name='ProjectDialog', parent=prnt, pos=wx.Point(376, 223),
              size=wx.Size(550, 450), style=wx.DEFAULT_DIALOG_STYLE,
              title='Create a new project')
        self.SetClientSize(wx.Size(550, 450))

        self.staticText1 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT1,
              label='Project Name (required):', name='staticText1', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.ProjectName = wx.TextCtrl(id=ID_PROJECTDIALOGPROJECTNAME,
              name='ProjectName', parent=self, pos=wx.Point(0, 0), 
              size=wx.Size(0, 24), style=0)

        self.staticText2 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT2,
              label='Company Name (required):', name='staticText2', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.CompanyName = wx.TextCtrl(id=ID_PROJECTDIALOGCOMPANYNAME,
              name='CompanyName', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText3 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT3,
              label='Company URL (optional):', name='staticText3', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.CompanyURL = wx.TextCtrl(id=ID_PROJECTDIALOGCOMPANYURL,
              name='CompanyURL', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText4 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT4,
              label='Product Name (required):', name='staticText4', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.ProductName = wx.TextCtrl(id=ID_PROJECTDIALOGPRODUCTNAME,
              name='ProductName', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText5 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT5,
              label='Product Version (required):', name='staticText5', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.ProductVersion = wx.TextCtrl(id=ID_PROJECTDIALOGPRODUCTVERSION,
              name='ProductVersion', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText6 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT6,
              label='Product Release (optional):', name='staticText6', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.ProductRelease = wx.TextCtrl(id=ID_PROJECTDIALOGPRODUCTRELEASE,
              name='ProductRelease', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText7 = wx.StaticText(id=ID_PROJECTDIALOGSTATICTEXT7,
              label='Content Description (optional):', name='staticText7', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(200, 17), style=0)

        self.ContentDescription = wx.TextCtrl(id=ID_PROJECTDIALOGCONTENTDESCRIPTION,
              name='ProductRelease', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 120), style=wx.TE_MULTILINE)

        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
        
        self._init_sizers()

    def __init__(self, parent):
        self._init_ctrls(parent)
        
    def OnOK(self, event):
        error = []
        if self.ProjectName.GetValue() == "":
            error.append("Project Name")
        if self.CompanyName.GetValue() == "":
            error.append("Company Name")
        if self.ProductName.GetValue() == "":
            error.append("Product Name")
        if self.ProductVersion.GetValue() == "":
            error.append("Product Version")
        if len(error) > 0:
            text = ""
            for i, item in enumerate(error):
                if i == 0:
                    text += item
                elif i == len(error) - 1:
                    text += " and %s"%item
                else:
                    text += ", %s"%item 
            message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        else:
            self.EndModal(wx.ID_OK)

    def SetValues(self, values):
        for item, value in values.items():
            if item == "projectName":
                self.ProjectName.SetValue(value)
            elif item == "companyName":
                self.CompanyName.SetValue(value)
            elif item == "companyURL":
                self.CompanyURL.SetValue(value)
            elif item == "productName":
                self.ProductName.SetValue(value)
            elif item == "productVersion":
                self.ProductVersion.SetValue(value)
            elif item == "productRelease":
                self.ProductRelease.SetValue(value)
            elif item == "contentDescription":
                self.ContentDescription.SetValue(value)

    def GetValues(self):
        values = {}
        values["projectName"] = self.ProjectName.GetValue()
        values["companyName"] = self.CompanyName.GetValue()
        if self.CompanyURL.GetValue() != None:
            values["companyURL"] = self.CompanyURL.GetValue()
        values["productName"] = self.ProductName.GetValue()
        values["productVersion"] = self.ProductVersion.GetValue()
        if self.ProductRelease.GetValue() != None:
            values["productRelease"] = self.ProductRelease.GetValue()
        if self.ProductRelease.GetValue() != None:
            values["contentDescription"] = self.ContentDescription.GetValue()
        return values

#-------------------------------------------------------------------------------
#                            Create Pou Dialog
#-------------------------------------------------------------------------------

[ID_POUDIALOG, ID_POUDIALOGPOUNAME, 
 ID_POUDIALOGPOUTYPE, ID_POUDIALOGLANGUAGE, ID_POUDIALOGSTATICTEXT1,
 ID_POUDIALOGSTATICTEXT2, ID_POUDIALOGSTATICTEXT3, 
] = [wx.NewId() for _init_ctrls in range(7)]

class PouDialog(wx.Dialog):
    def _init_coll_flexGridSizer1_Items(self, parent):
        parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
        
    def _init_coll_flexGridSizer1_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(0)
    
    def _init_coll_MainSizer_Items(self, parent):
        parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.PouName, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.PouType, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText3, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.Language, 0, border=0, flag=wx.GROW)
        
    def _init_coll_MainSizer_Growables(self, parent):
        parent.AddGrowableCol(1)
        
    def _init_sizers(self):
        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
        self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=3, vgap=15)

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
        self._init_coll_MainSizer_Items(self.MainSizer)
        self._init_coll_MainSizer_Growables(self.MainSizer)

        self.SetSizer(self.flexGridSizer1)
        
    def _init_ctrls(self, prnt):
        wx.Dialog.__init__(self, id=ID_POUDIALOG,
              name='PouDialog', parent=prnt, pos=wx.Point(376, 223),
              size=wx.Size(300, 200), style=wx.DEFAULT_DIALOG_STYLE,
              title='Create a new POU')
        self.SetClientSize(wx.Size(300, 200))

        self.staticText1 = wx.StaticText(id=ID_POUDIALOGSTATICTEXT1,
              label='POU Name:', name='staticText1', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)

        self.PouName = wx.TextCtrl(id=ID_POUDIALOGPOUNAME,
              name='POUName', parent=self, pos=wx.Point(0, 0), 
              size=wx.Size(0, 24), style=0)

        self.staticText2 = wx.StaticText(id=ID_POUDIALOGSTATICTEXT2,
              label='POU Type:', name='staticText2', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)

        self.PouType = wx.Choice(id=ID_POUDIALOGPOUTYPE,
              name='POUType', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)
        self.Bind(wx.EVT_CHOICE, self.OnTypeChanged, id=ID_POUDIALOGPOUTYPE)

        self.staticText3 = wx.StaticText(id=ID_POUDIALOGSTATICTEXT3,
              label='Language:', name='staticText3', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)

        self.Language = wx.Choice(id=ID_POUDIALOGLANGUAGE,
              name='Language', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)
        
        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
            
        self._init_sizers()

    def __init__(self, parent):
        self._init_ctrls(parent)
        
        for option in ["function","functionBlock","program"]:
            self.PouType.Append(option)
        self.RefreshLanguage()

        self.PouNames = []
        self.PouElementNames = []

    def OnOK(self, event):
        error = []
        pou_name = self.PouName.GetValue()
        if pou_name == "":
            error.append("POU Name")
        if self.PouType.GetStringSelection() == "":
            error.append("POU Type")
        if self.Language.GetStringSelection() == "":
            error.append("Language")
        if len(error) > 0:
            text = ""
            for i, item in enumerate(error):
                if i == 0:
                    text += item
                elif i == len(error) - 1:
                    text += " and %s"%item
                else:
                    text += ", %s"%item 
            message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif not TestIdentifier(pou_name):
            message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%pou_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif pou_name.upper() in IEC_KEYWORDS:
            message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%pou_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif pou_name.upper() in self.PouNames:
            message = wx.MessageDialog(self, "\"%s\" pou already exists!"%pou_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif pou_name.upper() in self.PouElementNames:
            message = wx.MessageDialog(self, "A pou has an element with \"%s\" for name. It can generate a conflict. Do you wish to continue?"%pou_name, "Warning", wx.YES_NO|wx.ICON_EXCLAMATION)
            result = message.ShowModal()
            message.Destroy()
            if result == wx.ID_YES:
                self.EndModal(wx.ID_OK)
        else:
            self.EndModal(wx.ID_OK)

    def RefreshLanguage(self):
        selection = self.Language.GetStringSelection()
        self.Language.Clear()
        for option in ["IL","ST","LD","FBD","SFC"]:
            if option != "SFC" or self.PouType.GetStringSelection() != "function":
                self.Language.Append(option)
        if self.Language.FindString(selection) != wx.NOT_FOUND:
            self.Language.SetStringSelection(selection)

    def OnTypeChanged(self, event):
        self.RefreshLanguage()
        event.Skip()

    def SetPouNames(self, pou_names):
        self.PouNames = [pou_name.upper() for pou_name in pou_names]

    def SetPouElementNames(self, element_names):
        self.PouElementNames = [element_name.upper() for element_name in element_names]

    def SetValues(self, values):
        for item, value in values.items():
            if item == "pouName":
                self.PouName.SetValue(value)
            elif item == "pouType":
                self.PouType.SetStringSelection(value)
            elif item == "language":
                self.Language.SetStringSelection(value)
                
    def GetValues(self):
        values = {}
        values["pouName"] = self.PouName.GetValue()
        values["pouType"] = self.PouType.GetStringSelection()
        values["language"] = self.Language.GetStringSelection()
        return values


#-------------------------------------------------------------------------------
#                          Create Pou Transition Dialog
#-------------------------------------------------------------------------------

[ID_POUTRANSITIONDIALOG, ID_POUTRANSITIONDIALOGTRANSITIONNAME, 
 ID_POUTRANSITIONDIALOGLANGUAGE, ID_POUTRANSITIONDIALOGSTATICTEXT1, 
 ID_POUTRANSITIONDIALOGSTATICTEXT2,
] = [wx.NewId() for _init_ctrls in range(5)]

class PouTransitionDialog(wx.Dialog):
    def _init_coll_flexGridSizer1_Items(self, parent):
        parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
        
    def _init_coll_flexGridSizer1_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(0)
    
    def _init_coll_MainSizer_Items(self, parent):
        parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.TransitionName, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.Language, 0, border=0, flag=wx.GROW)
        
    def _init_coll_MainSizer_Growables(self, parent):
        parent.AddGrowableCol(1)
        
    def _init_sizers(self):
        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
        self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=2, vgap=15)

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
        self._init_coll_MainSizer_Items(self.MainSizer)
        self._init_coll_MainSizer_Growables(self.MainSizer)

        self.SetSizer(self.flexGridSizer1)

    def _init_ctrls(self, prnt):
        wx.Dialog.__init__(self, id=ID_POUTRANSITIONDIALOG,
              name='PouTransitionDialog', parent=prnt, pos=wx.Point(376, 223),
              size=wx.Size(350, 200), style=wx.DEFAULT_DIALOG_STYLE,
              title='Create a new transition')
        self.SetClientSize(wx.Size(350, 160))

        self.staticText1 = wx.StaticText(id=ID_POUTRANSITIONDIALOGSTATICTEXT1,
              label='Transition Name:', name='staticText1', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(120, 17), style=0)

        self.TransitionName = wx.TextCtrl(id=ID_POUTRANSITIONDIALOGTRANSITIONNAME,
              name='TransitionName', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText2 = wx.StaticText(id=ID_POUTRANSITIONDIALOGSTATICTEXT2,
              label='Language:', name='staticText2', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(120, 17), style=0)

        self.Language = wx.Choice(id=ID_POUTRANSITIONDIALOGLANGUAGE,
              name='Language', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)
        
        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
        
        self._init_sizers()

    def __init__(self, parent):
        self._init_ctrls(parent)
        
        for option in ["IL","ST","LD","FBD"]:
            self.Language.Append(option)
            
        self.PouNames = []
        self.PouElementNames = []
        
    def OnOK(self, event):
        error = []
        transition_name = self.TransitionName.GetValue()
        if self.TransitionName.GetValue() == "":
            error.append("Transition Name")
        if self.Language.GetStringSelection() == "":
            error.append("Language")
        if len(error) > 0:
            text = ""
            for i, item in enumerate(error):
                if i == 0:
                    text += item
                elif i == len(error) - 1:
                    text += " and %s"%item
                else:
                    text += ", %s"%item 
            message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif not TestIdentifier(transition_name):
            message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif transition_name.upper() in IEC_KEYWORDS:
            message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif transition_name.upper() in self.PouNames:
            message = wx.MessageDialog(self, "A pou with \"%s\" for name exists!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif transition_name.upper() in self.PouElementNames:
            message = wx.MessageDialog(self, "\"%s\" element for this pou already exists!"%transition_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        else:
            self.EndModal(wx.ID_OK)
    
    def SetPouNames(self, pou_names):
        self.PouNames = [pou_name.upper() for pou_name in pou_names]
    
    def SetPouElementNames(self, pou_names):
        self.PouElementNames = [pou_name.upper() for pou_name in pou_names]
    
    def SetValues(self, values):
        for item, value in values.items():
            if item == "transitionName":
                self.TransitionName.SetValue(value)
            elif item == "language":
                self.Language.SetStringSelection(value)
                
    def GetValues(self):
        values = {}
        values["transitionName"] = self.TransitionName.GetValue()
        values["language"] = self.Language.GetStringSelection()
        return values

#-------------------------------------------------------------------------------
#                          Create Pou Action Dialog
#-------------------------------------------------------------------------------

[ID_POUACTIONDIALOG, ID_POUACTIONDIALOGACTIONNAME, 
 ID_POUACTIONDIALOGLANGUAGE, ID_POUACTIONDIALOGSTATICTEXT1, 
 ID_POUACTIONDIALOGSTATICTEXT2, 
] = [wx.NewId() for _init_ctrls in range(5)]

class PouActionDialog(wx.Dialog):
    def _init_coll_flexGridSizer1_Items(self, parent):
        parent.AddSizer(self.MainSizer, 0, border=20, flag=wx.GROW|wx.TOP|wx.LEFT|wx.RIGHT)
        parent.AddSizer(self.ButtonSizer, 0, border=20, flag=wx.ALIGN_RIGHT|wx.BOTTOM|wx.LEFT|wx.RIGHT)
        
    def _init_coll_flexGridSizer1_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(0)
    
    def _init_coll_MainSizer_Items(self, parent):
        parent.AddWindow(self.staticText1, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.ActionName, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.GROW)
        parent.AddWindow(self.Language, 0, border=0, flag=wx.GROW)
        
    def _init_coll_MainSizer_Growables(self, parent):
        parent.AddGrowableCol(1)
        
    def _init_sizers(self):
        self.flexGridSizer1 = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
        self.MainSizer = wx.FlexGridSizer(cols=2, hgap=0, rows=2, vgap=15)

        self._init_coll_flexGridSizer1_Items(self.flexGridSizer1)
        self._init_coll_flexGridSizer1_Growables(self.flexGridSizer1)
        self._init_coll_MainSizer_Items(self.MainSizer)
        self._init_coll_MainSizer_Growables(self.MainSizer)

        self.SetSizer(self.flexGridSizer1)

    def _init_ctrls(self, prnt):
        wx.Dialog.__init__(self, id=ID_POUACTIONDIALOG,
              name='PouActionDialog', parent=prnt, pos=wx.Point(376, 223),
              size=wx.Size(320, 200), style=wx.DEFAULT_DIALOG_STYLE|wx.RESIZE_BORDER,
              title='Create a new action')
        self.SetClientSize(wx.Size(320, 160))

        self.staticText1 = wx.StaticText(id=ID_POUACTIONDIALOGSTATICTEXT1,
              label='Action Name:', name='staticText1', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)

        self.ActionName = wx.TextCtrl(id=ID_POUACTIONDIALOGACTIONNAME,
              name='ActionName', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)

        self.staticText2 = wx.StaticText(id=ID_POUACTIONDIALOGSTATICTEXT2,
              label='Language:', name='staticText2', parent=self,
              pos=wx.Point(0, 0), size=wx.Size(100, 17), style=0)

        self.Language = wx.Choice(id=ID_POUACTIONDIALOGLANGUAGE,
              name='Language', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 24), style=0)
        
        self.ButtonSizer = self.CreateButtonSizer(wx.OK|wx.CANCEL|wx.CENTRE)
        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.ButtonSizer.GetAffirmativeButton().GetId())
        
        self._init_sizers()

    def __init__(self, parent):
        self._init_ctrls(parent)
        
        for option in ["IL","ST","LD","FBD"]:
            self.Language.Append(option)
        
        self.PouNames = []
        self.PouElementNames = []
    
    def OnOK(self, event):
        error = []
        action_name = self.ActionName.GetValue()
        if action_name == "":
            error.append("Action Name")
        if self.Language.GetStringSelection() == "":
            error.append("Language")
        if len(error) > 0:
            text = ""
            for i, item in enumerate(error):
                if i == 0:
                    text += item
                elif i == len(error) - 1:
                    text += " and %s"%item
                else:
                    text += ", %s"%item 
            message = wx.MessageDialog(self, "Form isn't complete. %s must be filled!"%text, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif not TestIdentifier(action_name):
            message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif action_name.upper() in IEC_KEYWORDS:
            message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif action_name.upper() in self.PouNames:
            message = wx.MessageDialog(self, "A pou with \"%s\" for name exists!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif action_name.upper() in self.PouElementNames:
            message = wx.MessageDialog(self, "\"%s\" element for this pou already exists!"%action_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        else:
            self.EndModal(wx.ID_OK)
    
    def SetPouNames(self, pou_names):
        self.PouNames = [pou_name.upper() for pou_name in pou_names]
        
    def SetPouElementNames(self, element_names):
        self.PouElementNames = [element_name.upper() for element_name in element_names]
        
    def SetValues(self, values):
        for item, value in values.items():
            if item == "actionName":
                self.ActionName.SetValue(value)
            elif item == "language":
                self.Language.SetStringSelection(value)
                
    def GetValues(self):
        values = {}
        values["actionName"] = self.ActionName.GetValue()
        values["language"] = self.Language.GetStringSelection()
        return values

#-------------------------------------------------------------------------------
#                          Configuration Name Dialog
#-------------------------------------------------------------------------------

class ConfigurationNameDialog(wx.TextEntryDialog):

    def __init__(self, parent, message, caption = "Please enter configuration name", defaultValue = "", 
                       style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition):
        wx.TextEntryDialog.__init__(self, parent, message, caption, defaultValue, style, pos)
        
        self.PouNames = []
        self.PouElementNames = []
        
        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())
        
    def OnOK(self, event):
        config_name = self.GetSizer().GetItem(1).GetWindow().GetValue()
        if config_name == "":
            message = wx.MessageDialog(self, "You must type a name!", "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif not TestIdentifier(config_name):
            message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%config_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif config_name.upper() in IEC_KEYWORDS:
            message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%config_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif config_name.upper() in self.PouNames:
            message = wx.MessageDialog(self, "A pou with \"%s\" as name exists!"%config_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif config_name.upper() in self.PouElementNames:
            message = wx.MessageDialog(self, "A pou has an element with \"%s\" for name. It can generate a conflict. Do you wish to continue?"%config_name, "Warning", wx.YES_NO|wx.ICON_EXCLAMATION)
            result = message.ShowModal()
            message.Destroy()
            if result == wx.ID_YES:
                self.EndModal(wx.ID_OK)
        else:
            self.EndModal(wx.ID_OK)

    def SetPouNames(self, pou_names):
        self.PouNames = [pou_name.upper() for pou_name in pou_names]
    
    def SetPouElementNames(self, pou_names):
        self.PouElementNames = [pou_name.upper() for pou_name in pou_names]
    
    def GetValue(self):
        return self.GetSizer().GetItem(1).GetWindow().GetValue()

#-------------------------------------------------------------------------------
#                          Resource Name Dialog
#-------------------------------------------------------------------------------

class ResourceNameDialog(wx.TextEntryDialog):

    def __init__(self, parent, message, caption = "Please enter resource name", defaultValue = "", 
                       style = wx.OK|wx.CANCEL|wx.CENTRE, pos = wx.DefaultPosition):
        wx.TextEntryDialog.__init__(self, parent, message, caption, defaultValue, style, pos)
        
        self.PouNames = []
        self.PouElementNames = []
        
        self.Bind(wx.EVT_BUTTON, self.OnOK, id=self.GetSizer().GetItem(3).GetSizer().GetAffirmativeButton().GetId())
        
    def OnOK(self, event):
        resource_name = self.GetSizer().GetItem(1).GetWindow().GetValue()
        if resource_name == "":
            message = wx.MessageDialog(self, "You must type a name!", "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif not TestIdentifier(resource_name):
            message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%resource_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif resource_name.upper() in IEC_KEYWORDS:
            message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%resource_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif resource_name.upper() in self.PouNames:
            message = wx.MessageDialog(self, "A pou with \"%s\" as name exists!"%resource_name, "Error", wx.OK|wx.ICON_ERROR)
            message.ShowModal()
            message.Destroy()
        elif resource_name.upper() in self.PouElementNames:
            message = wx.MessageDialog(self, "A pou has an element with \"%s\" for name. It can generate a conflict. Do you wish to continue?"%resource_name, "Warning", wx.YES_NO|wx.ICON_EXCLAMATION)
            result = message.ShowModal()
            message.Destroy()
            if result == wx.ID_YES:
                self.EndModal(wx.ID_OK)
        else:
            self.EndModal(wx.ID_OK)

    def SetPouNames(self, pou_names):
        self.PouNames = [pou_name.upper() for pou_name in pou_names]
    
    def SetPouElementNames(self, pou_names):
        self.PouElementNames = [pou_name.upper() for pou_name in pou_names]
    
    def GetValue(self):
        return self.GetSizer().GetItem(1).GetWindow().GetValue()

#-------------------------------------------------------------------------------
#                            Pou Editor Panel
#-------------------------------------------------------------------------------

class VariableTable(wx.grid.PyGridTableBase):
    
    """
    A custom wx.grid.Grid Table using user supplied data
    """
    def __init__(self, parent, data, colnames):
        # The base class must be initialized *first*
        wx.grid.PyGridTableBase.__init__(self)
        self.data = data
        self.old_value = None
        self.colnames = colnames
        self.Parent = parent
        # XXX
        # we need to store the row length and collength to
        # see if the table has changed size
        self._rows = self.GetNumberRows()
        self._cols = self.GetNumberCols()
    
    def GetNumberCols(self):
        return len(self.colnames)
        
    def GetNumberRows(self):
        return len(self.data)

    def GetColLabelValue(self, col):
        if col < len(self.colnames):
            return self.colnames[col]

    def GetRowLabelValues(self, row):
        return row

    def GetValue(self, row, col):
        if row < self.GetNumberRows():
            if col == 0:
                return self.data[row]["Number"]
            name = str(self.data[row].get(self.GetColLabelValue(col), ""))
            return name
    
    def SetValue(self, row, col, value):
        if col < len(self.colnames):
            colname = self.GetColLabelValue(col)
            if colname == "Name":
                self.old_value = self.data[row][colname]
            self.data[row][colname] = value
    
    def GetValueByName(self, row, colname):
        if row < self.GetNumberRows():
            return self.data[row].get(colname)

    def SetValueByName(self, row, colname, value):
        if row < self.GetNumberRows():
            self.data[row][colname] = value

    def GetOldValue(self):
        return self.old_value
    
    def ResetView(self, grid):
        """
        (wx.grid.Grid) -> Reset the grid view.   Call this to
        update the grid if rows and columns have been added or deleted
        """
        grid.BeginBatch()
        for current, new, delmsg, addmsg in [
            (self._rows, self.GetNumberRows(), wx.grid.GRIDTABLE_NOTIFY_ROWS_DELETED, wx.grid.GRIDTABLE_NOTIFY_ROWS_APPENDED),
            (self._cols, self.GetNumberCols(), wx.grid.GRIDTABLE_NOTIFY_COLS_DELETED, wx.grid.GRIDTABLE_NOTIFY_COLS_APPENDED),
        ]:
            if new < current:
                msg = wx.grid.GridTableMessage(self,delmsg,new,current-new)
                grid.ProcessTableMessage(msg)
            elif new > current:
                msg = wx.grid.GridTableMessage(self,addmsg,new-current)
                grid.ProcessTableMessage(msg)
                self.UpdateValues(grid)
        grid.EndBatch()

        self._rows = self.GetNumberRows()
        self._cols = self.GetNumberCols()
        # update the column rendering scheme
        self._updateColAttrs(grid)

        # update the scrollbars and the displayed part of the grid
        grid.AdjustScrollbars()
        grid.ForceRefresh()

    def UpdateValues(self, grid):
        """Update all displayed values"""
        # This sends an event to the grid table to update all of the values
        msg = wx.grid.GridTableMessage(self, wx.grid.GRIDTABLE_REQUEST_VIEW_GET_VALUES)
        grid.ProcessTableMessage(msg)

    def _updateColAttrs(self, grid):
        """
        wx.grid.Grid -> update the column attributes to add the
        appropriate renderer given the column name.

        Otherwise default to the default renderer.
        """
        
        for col in range(self.GetNumberCols()):
            attr = wx.grid.GridCellAttr()
            attr.SetAlignment(self.Parent.ColAlignements[col], wx.ALIGN_CENTRE)
            grid.SetColAttr(col, attr)
            grid.SetColSize(col, self.Parent.ColSizes[col])
        
        typelist = None
        accesslist = None
        for row in range(self.GetNumberRows()):
            for col in range(self.GetNumberCols()):
                editor = None
                renderer = None
                colname = self.GetColLabelValue(col)
                if col != 0 and self.GetValueByName(row, "Edit"):
                    grid.SetReadOnly(row, col, False)
                    if colname == "Name":
                        if self.Parent.PouIsUsed and self.GetValueByName(row, "Class") in ["Input", "Output", "InOut"]:
                            grid.SetReadOnly(row, col, True)
                        else:
                            editor = wx.grid.GridCellTextEditor()
                            renderer = wx.grid.GridCellStringRenderer()
                    elif colname in ["Initial Value","Location"]:
                        editor = wx.grid.GridCellTextEditor()
                        renderer = wx.grid.GridCellStringRenderer()
                    elif colname == "Class":
                        if len(self.Parent.ClassList) == 1 or self.Parent.PouIsUsed and self.GetValueByName(row, "Class") in ["Input", "Output", "InOut"]:
                            grid.SetReadOnly(row, col, True)
                        else:
                            editor = wx.grid.GridCellChoiceEditor()
                            excluded = []
                            if self.Parent.PouIsUsed:
                                excluded.extend(["Input","Output","InOut"])    
                            editor.SetParameters(",".join([choice for choice in self.Parent.ClassList if choice not in excluded]))
                    elif colname in ["Retain", "Constant"]:
                        editor = wx.grid.GridCellChoiceEditor()
                        editor.SetParameters(self.Parent.OptionList)
                    elif colname == "Type":
                        editor = wx.grid.GridCellTextEditor()
                else:
                    grid.SetReadOnly(row, col, True)
                
                grid.SetCellEditor(row, col, editor)
                grid.SetCellRenderer(row, col, renderer)
                
                grid.SetCellBackgroundColour(row, col, wx.WHITE)
    
    def SetData(self, data):
        self.data = data
    
    def GetData(self):
        return self.data
    
    def GetCurrentIndex(self):
        return self.CurrentIndex
    
    def SetCurrentIndex(self, index):
        self.CurrentIndex = index
    
    def AppendRow(self, row_content):
        self.data.append(row_content)

    def RemoveRow(self, row_index):
        self.data.pop(row_index)

    def GetRow(self, row_index):
        return self.data[row_index]

    def Empty(self):
        self.data = []
        self.editors = []

class VariableDropTarget(wx.TextDropTarget):
    
    def __init__(self, parent):
        wx.TextDropTarget.__init__(self)
        self.ParentWindow = parent
    
    def OnDropText(self, x, y, data):
        col = self.ParentWindow.VariablesGrid.XToCol(x)
        row = self.ParentWindow.VariablesGrid.YToRow(y)
        if col != wx.NOT_FOUND and row != wx.NOT_FOUND:
            if self.ParentWindow.Table.GetColLabelValue(col) != "Location":
                return
            try:
                values = eval(data)    
            except:
                values = None
            if values and values[1] == "location":
                dialog = wx.SingleChoiceDialog(self.ParentWindow, "Select a variable class:", "Variable class", ["Input", "Output", "Memory"], wx.OK|wx.CANCEL)
                if dialog.ShowModal() == wx.ID_OK:
                    selected = dialog.GetSelection()
                    if selected == 0:
                        location = "%I" + values[0]
                    elif selected == 1:
                        location = "%Q" + values[0]
                    else:
                        location = "%M" + values[0]
                    self.ParentWindow.Table.SetValue(row - 1, col, location)
                    self.ParentWindow.Table.ResetView(self.ParentWindow.VariablesGrid)
                    self.ParentWindow.SaveValues()
                dialog.Destroy()    

[ID_POUEDITORPANEL, ID_POUEDITORPANELVIEWER, 
 ID_POUEDITORPANELVARIABLESGRID, ID_POUEDITORPANELRETURNTYPE, 
 ID_POUEDITORPANELCLASSFILTER, ID_POUEDITORPANELADDBUTTON,
 ID_POUEDITORPANELDELETEBUTTON, ID_POUEDITORPANELUPBUTTON, 
 ID_POUEDITORPANELDOWNBUTTON, ID_POUEDITORPANELSTATICTEXT1, 
 ID_POUEDITORPANELSTATICTEXT2, ID_POUEDITORPANELSTATICTEXT3,
] = [wx.NewId() for _init_ctrls in range(12)]

class PouEditorPanel(wx.SplitterWindow):
    def _init_coll_VariablePanelSizer_Items(self, parent):
        parent.AddWindow(self.VariablesGrid, 0, border=0, flag=wx.GROW)
        parent.AddSizer(self.ControlPanelSizer, 0, border=0, flag=wx.GROW)
    
    def _init_coll_VariablePanelSizer_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(0)
    
    def _init_coll_ControlPanelSizer_Items(self, parent):
        parent.AddSizer(self.ChoicePanelSizer, 0, border=0, flag=wx.GROW)
        parent.AddSizer(self.ButtonPanelSizer, 0, border=0, flag=wx.ALIGN_CENTER)

    def _init_coll_ControlPanelSizer_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableRow(0)
        parent.AddGrowableRow(1)

    def _init_coll_ChoicePanelSizer_Items(self, parent):
        parent.AddWindow(self.staticText1, 0, border=0, flag=wx.ALIGN_BOTTOM)
        parent.AddWindow(self.ReturnType, 0, border=0, flag=0)
        parent.AddWindow(self.staticText2, 0, border=0, flag=wx.ALIGN_BOTTOM)
        parent.AddWindow(self.ClassFilter, 0, border=0, flag=0)

    def _init_coll_ButtonPanelSizer_Items(self, parent):
        parent.AddWindow(self.UpButton, 0, border=0, flag=0)
        parent.AddWindow(self.AddButton, 0, border=0, flag=0)
        parent.AddWindow(self.DownButton, 0, border=0, flag=0)
        parent.AddWindow(self.DeleteButton, 0, border=0, flag=0)
        
    def _init_coll_ButtonPanelSizer_Growables(self, parent):
        parent.AddGrowableCol(0)
        parent.AddGrowableCol(1)
        parent.AddGrowableCol(2)
        parent.AddGrowableCol(3)
        parent.AddGrowableRow(0)

    def _init_sizers(self):
        self.VariablePanelSizer = wx.FlexGridSizer(cols=2, hgap=10, rows=1, vgap=0)
        self.ControlPanelSizer = wx.FlexGridSizer(cols=1, hgap=0, rows=2, vgap=10)
        self.ChoicePanelSizer = wx.GridSizer(cols=1, hgap=5, rows=4, vgap=5)
        self.ButtonPanelSizer = wx.FlexGridSizer(cols=2, hgap=5, rows=2, vgap=0)
        
        self._init_coll_VariablePanelSizer_Items(self.VariablePanelSizer)
        self._init_coll_VariablePanelSizer_Growables(self.VariablePanelSizer)
        self._init_coll_ControlPanelSizer_Items(self.ControlPanelSizer)
        self._init_coll_ControlPanelSizer_Growables(self.ControlPanelSizer)
        self._init_coll_ChoicePanelSizer_Items(self.ChoicePanelSizer)
        self._init_coll_ButtonPanelSizer_Items(self.ButtonPanelSizer)
        self._init_coll_ButtonPanelSizer_Growables(self.ButtonPanelSizer)
        
        self.VariablePanel.SetSizer(self.VariablePanelSizer)

    def _init_ctrls(self, prnt, element_type):
        wx.SplitterWindow.__init__(self, id=ID_POUEDITORPANEL,
              name='EditVariablePanel', parent=prnt, pos=wx.Point(0, 0),
              size=wx.Size(0, 0), style=wx.SP_3D)
        self.SetNeedUpdating(True)
        self.SetMinimumPaneSize(1)
        
        if element_type == "config":
            self.Viewer = wx.Panel(id=ID_POUEDITORPANELVIEWER,
              name='ConfigPanel', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 0), style=0)
            self.Viewer.SetSizer(wx.BoxSizer(wx.VERTICAL))
            self.Viewer.ResetBuffer = lambda: None
            self.Viewer.RefreshView = lambda: None
        elif element_type == "resource":
            self.Viewer = ResourceEditor(self, self.ParentWindow, self.Controler)
        elif element_type == "FBD":
            self.Viewer = Viewer(self, self.ParentWindow, self.Controler)
        elif element_type == "LD":
            self.Viewer = LD_Viewer(self, self.ParentWindow, self.Controler)
        elif element_type == "SFC":
            self.Viewer = SFC_Viewer(self, self.ParentWindow, self.Controler)
        else:
            self.Viewer = TextViewer(self, self.ParentWindow, self.Controler)
            self.Viewer.SetTextSyntax(element_type)
            if element_type == "IL":
                self.Viewer.SetKeywords(IL_KEYWORDS)
            else:
                self.Viewer.SetKeywords(ST_KEYWORDS)
        
        self.VariablePanel = wx.Panel(id=ID_POUEDITORPANELVIEWER,
              name='VariablePanel', parent=self, pos=wx.Point(0, 0),
              size=wx.Size(0, 0), style=wx.TAB_TRAVERSAL)
        
        self.staticText1 = wx.StaticText(id=ID_POUEDITORPANELSTATICTEXT1,
              label='Return Type:', name='staticText1', parent=self.VariablePanel,
              pos=wx.Point(0, 0), size=wx.Size(95, 17), style=0)

        self.ReturnType = wx.Choice(id=ID_POUEDITORPANELRETURNTYPE,
              name='ReturnType', parent=self.VariablePanel, pos=wx.Point(0, 0),
              size=wx.Size(145, 24), style=0)
        self.Bind(wx.EVT_CHOICE, self.OnReturnTypeChanged, id=ID_POUEDITORPANELRETURNTYPE)

        self.staticText2 = wx.StaticText(id=ID_POUEDITORPANELSTATICTEXT2,
              label='Class Filter:', name='staticText2', parent=self.VariablePanel,
              pos=wx.Point(0, 0), size=wx.Size(95, 17), style=0)

        self.ClassFilter = wx.Choice(id=ID_POUEDITORPANELCLASSFILTER,
              name='ClassFilter', parent=self.VariablePanel, pos=wx.Point(0, 0),
              size=wx.Size(145, 24), style=0)
        self.Bind(wx.EVT_CHOICE, self.OnClassFilter, id=ID_POUEDITORPANELCLASSFILTER)

        self.VariablesGrid = wx.grid.Grid(id=ID_POUEDITORPANELVARIABLESGRID,
              name='VariablesGrid', parent=self.VariablePanel, pos=wx.Point(0, 0), 
              size=wx.Size(0, 150), style=wx.VSCROLL)
        self.VariablesGrid.SetFont(wx.Font(12, 77, wx.NORMAL, wx.NORMAL, False,
              'Sans'))
        self.VariablesGrid.SetLabelFont(wx.Font(10, 77, wx.NORMAL, wx.NORMAL,
              False, 'Sans'))
        self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_CHANGE, self.OnVariablesGridCellChange)
        self.VariablesGrid.Bind(wx.grid.EVT_GRID_SELECT_CELL, self.OnVariablesGridSelectCell)
        self.VariablesGrid.Bind(wx.grid.EVT_GRID_CELL_LEFT_CLICK, self.OnVariablesGridCellLeftClick)
        self.VariablesGrid.Bind(wx.grid.EVT_GRID_EDITOR_SHOWN, self.OnVariablesGridEditorShown)
        self.VariablesGrid.SetDropTarget(VariableDropTarget(self))
        
        self.AddButton = wx.Button(id=ID_POUEDITORPANELADDBUTTON, label='Add',
              name='AddButton', parent=self.VariablePanel, pos=wx.Point(345, 340),
              size=wx.Size(72, 32), style=0)
        self.Bind(wx.EVT_BUTTON, self.OnAddButton, id=ID_POUEDITORPANELADDBUTTON)

        self.DeleteButton = wx.Button(id=ID_POUEDITORPANELDELETEBUTTON, label='Delete',
              name='DeleteButton', parent=self.VariablePanel, pos=wx.Point(425, 340),
              size=wx.Size(72, 32), style=0)
        self.Bind(wx.EVT_BUTTON, self.OnDeleteButton, id=ID_POUEDITORPANELDELETEBUTTON)

        self.UpButton = wx.Button(id=ID_POUEDITORPANELUPBUTTON, label='^',
              name='UpButton', parent=self.VariablePanel, pos=wx.Point(505, 340),
              size=wx.Size(32, 32), style=0)
        self.Bind(wx.EVT_BUTTON, self.OnUpButton, id=ID_POUEDITORPANELUPBUTTON)

        self.DownButton = wx.Button(id=ID_POUEDITORPANELDOWNBUTTON, label='v',
              name='DownButton', parent=self.VariablePanel, pos=wx.Point(545, 340),
              size=wx.Size(32, 32), style=0)
        self.Bind(wx.EVT_BUTTON, self.OnDownButton, id=ID_POUEDITORPANELDOWNBUTTON)

        self.SplitHorizontally(self.Viewer, self.VariablePanel, -200)

        self._init_sizers()

    def __init__(self, parent, window, controler, element_type, pou_name = None, transition_name = None, action_name = None, config_name = None, resource_name = None):
        self.ParentWindow = window
        self.Controler = controler
        self.ElementType = element_type
        self.PouName = pou_name
        self.TransitionName = transition_name
        self.ActionName = action_name
        self.ConfigName = config_name
        self.ResourceName = resource_name
        
        if element_type == "config":
            self._init_ctrls(parent, "config")
            pou_type = "config"
        elif element_type == "resource":
            self._init_ctrls(parent, "resource")
            pou_type = "resource"
        else:
            if element_type == "transition":
                self._init_ctrls(parent, self.Controler.GetTransitionBodyType(pou_name, transition_name))
            elif element_type == "action":
                self._init_ctrls(parent, self.Controler.GetActionBodyType(pou_name, action_name))
            else:
                self._init_ctrls(parent, self.Controler.GetPouBodyType(pou_name))
            pou_type = self.Controler.GetPouType(pou_name)
        
        self.Filter = "All"
        self.FilterChoices = []
        self.FilterChoiceTransfer = {"All" : "All", "Interface" : "Interface", 
            "   Input" : "Input", "   Output" : "Output", "   InOut" : "InOut", 
            "   External" : "External", "Variables" : "Variables", "   Local" : "Local",
            "   Temp" : "Temp", "Global" : "Global", "Access" : "Access"}
        
        if pou_type in ["config", "resource"]:
            self.DefaultTypes = {"All" : "Global"}
            self.DefaultValue = {"Name" : "", "Class" : "", "Type" : "INT", "Location" : "", "Initial Value" : "", "Retain" : "No", "Constant" : "No", "Edit" : True}
        else:
            self.DefaultTypes = {"All" : "Local", "Interface" : "Input", "Variables" : "Local"}
            self.DefaultValue = {"Name" : "", "Class" : "", "Type" : "INT", "Location" : "", "Initial Value" : "", "Retain" : "No", "Constant" : "No", "Edit" : True}
        if pou_type in ["config", "resource"] or pou_type == "program":
            self.Table = VariableTable(self, [], ["#", "Name", "Class", "Type", "Location", "Initial Value", "Retain", "Constant"])
            if pou_type not in ["config", "resource"]:
                self.FilterChoices = ["All","Interface","   Input","   Output","   InOut","   External","Variables","   Local","   Temp","Global","Access"]
            else:
                self.FilterChoices = ["All","Global","Access"]
            self.ColSizes = [40, 80, 70, 80, 80, 80, 60, 70]
            self.ColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_CENTER, wx.ALIGN_CENTER]
        else:
            self.Table = VariableTable(self, [], ["#", "Name", "Class", "Type", "Initial Value", "Retain", "Constant"])
            self.FilterChoices = ["All","Interface","   Input","   Output","   InOut","   External","Variables","   Local","   Temp"]
            self.ColSizes = [40, 120, 70, 80, 120, 60, 70]
            self.ColAlignements = [wx.ALIGN_CENTER, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_LEFT, wx.ALIGN_CENTER, wx.ALIGN_CENTER]
        for choice in self.FilterChoices:
            self.ClassFilter.Append(choice)
        reverse_transfer = {}
        for filter, choice in self.FilterChoiceTransfer.items():
            reverse_transfer[choice] = filter
        self.ClassFilter.SetStringSelection(reverse_transfer[self.Filter])
        self.RefreshTypeList()

        self.OptionList = "Yes,No"
        self.TypeList = [value for value, parent in TypeHierarchy_list if not value.startswith("ANY")]
        
        if pou_type == "function":
            for value, parent in TypeHierarchy_list:
                if not value.startswith("ANY"):
                    self.ReturnType.Append(value)
            self.ReturnType.Enable(True)
        else:
            self.ReturnType.Enable(False)
            self.staticText1.Hide()
            self.ReturnType.Hide()
        
        self.VariablesGrid.SetTable(self.Table)
        self.VariablesGrid.SetRowLabelSize(0)
    
    def SetMode(self, mode):
        if self.ElementType not in ["resource", "config"]:
            if self.ElementType == "transition":
                language = self.Controler.GetTransitionBodyType(self.PouName, self.TransitionName)
            elif self.ElementType == "action":
                language = self.Controler.GetActionBodyType(self.PouName, self.ActionName)
            else:
                language = self.Controler.GetPouBodyType(self.PouName)
            if language not in ["IL", "ST"]:
                self.Viewer.SetMode(mode)

    def ResetBuffer(self):
        self.Viewer.ResetBuffer()

    def RefreshView(self):
        self.PouNames = self.Controler.GetProjectPouNames()
        
        if self.ElementType == "config":
            self.PouIsUsed = False
            returnType = None
            self.Values = self.Controler.GetConfigurationGlobalVars(self.ConfigName)
        elif self.ElementType == "resource":
            self.PouIsUsed = False
            returnType = None
            self.Values = self.Controler.GetConfigurationResourceGlobalVars(self.ConfigName, self.ResourceName)
        else:
            self.PouIsUsed = self.Controler.PouIsUsed(self.PouName)
            returnType = self.Controler.GetCurrentElementEditingInterfaceReturnType()
            self.Values = self.Controler.GetCurrentElementEditingInterfaceVars()
        
        if returnType and self.ReturnType.IsEnabled():
            self.ReturnType.SetStringSelection(returnType)
        
        self.RefreshValues()
        self.RefreshViewerVarList()
        self.RefreshButtons()
        self.Viewer.RefreshView()
    
    def RefreshViewerVarList(self):
        if self.ElementType not in ["config", "resource"]:
            varlist = [var["Name"] for var in self.Values]
            if self.ElementType == "transition":
                language = self.Controler.GetTransitionBodyType(self.PouName, self.TransitionName)
                varlist.append(self.TransitionName)
            elif self.ElementType == "action":
                language = self.Controler.GetActionBodyType(self.PouName, self.ActionName)
                varlist.append(self.ActionName)
            else:
                language = self.Controler.GetPouBodyType(self.PouName)
                varlist.append(self.PouName)
            if language in ["IL", "ST"]:
                self.Viewer.SetVariables(varlist)
                self.Viewer.SetFunctions(self.Controler.GetBlockTypes())
    
    def OnReturnTypeChanged(self, event):
        self.Controler.SetPouInterfaceReturnType(self.PouName, self.ReturnType.GetStringSelection())
        self.Controler.BufferProject()
        self.ParentWindow.RefreshTitle()
        self.ParentWindow.RefreshEditMenu()
        event.Skip()
    
    def OnClassFilter(self, event):
        self.Filter = self.FilterChoiceTransfer[self.ClassFilter.GetStringSelection()]
        self.RefreshTypeList()
        self.RefreshValues()
        self.RefreshButtons()
        event.Skip()

    def RefreshTypeList(self):
        if self.Filter == "All":
            self.ClassList = [self.FilterChoiceTransfer[choice] for choice in self.FilterChoices if self.FilterChoiceTransfer[choice] not in ["All","Interface","Variables"]]
        elif self.Filter == "Interface":
            self.ClassList = ["Input","Output","InOut","External"]
        elif self.Filter == "Variables":
            self.ClassList = ["Local","Temp"]
        else:
            self.ClassList = [self.Filter]

    def RefreshButtons(self):
        if self.Table:
            table_length = len(self.Table.data)
            row_class = None
            if table_length and self.PouIsUsed:
                row = self.VariablesGrid.GetGridCursorRow()
                row_class = self.Table.GetValueByName(row, "Class")
            self.AddButton.Enable(not self.PouIsUsed or self.Filter not in ["Interface", "Input", "Output", "InOut"])
            self.DeleteButton.Enable(table_length > 0 and row_class not in ["Input", "Output", "InOut"])
            self.UpButton.Enable(table_length > 0 and self.Filter == "All" and row_class not in ["Input", "Output", "InOut"])
            self.DownButton.Enable(table_length > 0 and self.Filter == "All" and row_class not in ["Input", "Output", "InOut"])

    def OnAddButton(self, event):
        new_row = self.DefaultValue.copy()
        if self.Filter in self.DefaultTypes:
            new_row["Class"] = self.DefaultTypes[self.Filter]
        else:
            new_row["Class"] = self.Filter
        self.Values.append(new_row)
        self.SaveValues()
        self.RefreshValues()
        self.RefreshButtons()
        event.Skip()

    def OnDeleteButton(self, event):
        row = self.Table.GetRow(self.VariablesGrid.GetGridCursorRow())
        self.Values.remove(row)
        self.SaveValues()
        self.RefreshValues()
        self.RefreshButtons()
        event.Skip()

    def OnUpButton(self, event):
        self.MoveValue(self.VariablesGrid.GetGridCursorRow(), -1)
        self.RefreshButtons()
        event.Skip()

    def OnDownButton(self, event):
        self.MoveValue(self.VariablesGrid.GetGridCursorRow(), 1)
        self.RefreshButtons()
        event.Skip()

    def OnVariablesGridCellChange(self, event):
        row, col = event.GetRow(), event.GetCol()
        colname = self.Table.GetColLabelValue(col)
        value = self.Table.GetValue(row, col)
        if colname == "Name":
            if not TestIdentifier(value):
                message = wx.MessageDialog(self, "\"%s\" is not a valid identifier!"%value, "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
                event.Veto()
            elif value.upper() in IEC_KEYWORDS:
                message = wx.MessageDialog(self, "\"%s\" is a keyword. It can't be used!"%value, "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
                event.Veto()
            elif value.upper() in self.PouNames:
                message = wx.MessageDialog(self, "A pou with \"%s\" as name exists!"%value, "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
                event.Veto()
            elif value.upper() in [var["Name"].upper() for var in self.Values if var != self.Table.data[row]]:
                message = wx.MessageDialog(self, "A variable with \"%s\" as name exists in this pou!"%value, "Error", wx.OK|wx.ICON_ERROR)
                message.ShowModal()
                message.Destroy()
                event.Veto()
            else:
                self.SaveValues(False)
                old_value = self.Table.GetOldValue()
                if old_value != "":
                    self.Controler.UpdateCurrentPouEditingUsedVariable(old_value, value)
                self.Controler.BufferProject()
                self.ParentWindow.RefreshTitle()
                self.ParentWindow.RefreshEditMenu()
                self.RefreshViewerVarList()
                self.Viewer.RefreshView()
                event.Skip()
        else:
            self.SaveValues()
            event.Skip()

    def OnVariablesGridCellLeftClick(self, event):
        if event.GetCol() == "#":
            row = event.GetRow()
            var_name = self.Table.GetValueByName(row, "Name")
            var_class = self.Table.GetValueByName(row, "Class")
            var_type = self.Table.GetValueByName(row, "Type")
            data = wx.TextDataObject(str((var_name, var_class, var_type)))
            dragSource = wx.DropSource(self.VariablesGrid)
            dragSource.SetData(data)
            dragSource.DoDragDrop()
        event.Skip()
    
    def OnVariablesGridEditorShown(self, event):
        row, col = event.GetRow(), event.GetCol() 
        if self.Table.GetColLabelValue(col) == "Type":
            type_menu = wx.Menu(title='')
            base_menu = wx.Menu(title='')
            for base_type in self.TypeList:
                new_id = wx.NewId()
                base_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=base_type)
                self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(base_type), id=new_id)
            type_menu.AppendMenu(-1, "Base Types", base_menu, '')
            functionblock_menu = wx.Menu(title='')
            for functionblock_type in self.Controler.GetFunctionBlockTypes():
                new_id = wx.NewId()
                functionblock_menu.Append(help='', id=new_id, kind=wx.ITEM_NORMAL, text=functionblock_type)
                self.Bind(wx.EVT_MENU, self.GetVariableTypeFunction(functionblock_type), id=new_id)
            type_menu.AppendMenu(-1, "Function Block Types", functionblock_menu, '')
            rect = self.VariablesGrid.BlockToDeviceRect((row, col), (row, col))
            self.VariablesGrid.PopupMenuXY(type_menu, rect.x + rect.width, rect.y + self.VariablesGrid.GetColLabelSize())
            event.Veto()
        else:
            event.Skip()
    
    def GetVariableTypeFunction(self, base_type):
        def VariableTypeFunction(event):
            row = self.VariablesGrid.GetGridCursorRow()
            self.Table.SetValueByName(row, "Type", base_type)
            self.SaveValues()
            self.Table.ResetView(self.VariablesGrid)
            event.Skip()
        return VariableTypeFunction
    
    def OnVariablesGridCellLeftClick(self, event):
        if event.GetCol() == 0:
            row = event.GetRow()
            var_name = self.Table.GetValueByName(row, "Name")
            var_class = self.Table.GetValueByName(row, "Class")
            var_type = self.Table.GetValueByName(row, "Type")
            data = wx.TextDataObject(str((var_name, var_class, var_type)))
            dragSource = wx.DropSource(self.VariablesGrid)
            dragSource.SetData(data)
            dragSource.DoDragDrop()
        event.Skip()
    
    def OnVariablesGridSelectCell(self, event):
        wx.CallAfter(self.RefreshButtons)
        event.Skip()

    def MoveValue(self, value_index, move):
        new_index = max(0, min(value_index + move, len(self.Values) - 1))
        if new_index != value_index:
            self.Values.insert(new_index, self.Values.pop(value_index))
            self.RefreshValues()
            self.VariablesGrid.SetGridCursor(new_index, self.VariablesGrid.GetGridCursorCol())
        else:
            self.RefreshValues()
        self.SaveValues()

    def RefreshValues(self):
        if len(self.Table.data) > 0:
            self.VariablesGrid.SetGridCursor(0, 1)
        data = []
        for num, variable in enumerate(self.Values):
            if variable["Class"] in self.ClassList:
                variable["Number"] = num + 1
                data.append(variable)
        self.Table.SetData(data)
        self.Table.ResetView(self.VariablesGrid)

    def SaveValues(self, buffer = True):
        if self.ElementType == "config":
            self.Controler.SetConfigurationGlobalVars(self.ConfigName, self.Values)
        elif self.ElementType == "resource":
            self.Controler.SetConfigurationResourceGlobalVars(self.ConfigName, self.ResourceName, self.Values)
        else:
            if self.ReturnType.IsEnabled():
                self.Controler.SetPouInterfaceReturnType(self.PouName, self.ReturnType.GetStringSelection())
            self.Controler.SetPouInterfaceVars(self.PouName, self.Values)
        if buffer:
            self.Controler.BufferProject()
            self.ParentWindow.RefreshTitle()
            self.ParentWindow.RefreshEditMenu()

#-------------------------------------------------------------------------------
#                               Exception Handler
#-------------------------------------------------------------------------------

Max_Traceback_List_Size = 20

def Display_Exception_Dialog(e_type,e_value,e_tb):
    trcbck_lst = []
    for i,line in enumerate(traceback.extract_tb(e_tb)):
        trcbck = " " + str(i+1) + ". "
        if line[0].find(os.getcwd()) == -1:
            trcbck += "file : " + str(line[0]) + ",   "
        else:
            trcbck += "file : " + str(line[0][len(os.getcwd()):]) + ",   "
        trcbck += "line : " + str(line[1]) + ",   " + "function : " + str(line[2])
        trcbck_lst.append(trcbck)
        
    # Allow clicking....
    cap = wx.Window_GetCapture()
    if cap:
        cap.ReleaseMouse()

    dlg = wx.SingleChoiceDialog(None, 
        """
An error happens.

Click on OK for saving an error report.

Please contact LOLITech at:
+33 (0)3 29 52 95 67
bugs_PLCOpenEditor@lolitech.fr


Error:
""" +
        str(e_type) + " : " + str(e_value), 
        "Error",
        trcbck_lst)
    try:
        res = (dlg.ShowModal() == wx.ID_OK)
    finally:
        dlg.Destroy()

    return res

def Display_Error_Dialog(e_value):
    message = wx.MessageDialog(None, str(e_value), "Error", wx.OK|wx.ICON_ERROR)
    message.ShowModal()
    message.Destroy()

def get_last_traceback(tb):
    while tb.tb_next:
        tb = tb.tb_next
    return tb


def format_namespace(d, indent='    '):
    return '\n'.join(['%s%s: %s' % (indent, k, repr(v)[:10000]) for k, v in d.iteritems()])


ignored_exceptions = [] # a problem with a line in a module is only reported once per session

def AddExceptHook(path, app_version='[No version]'):#, ignored_exceptions=[]):
    
    def handle_exception(e_type, e_value, e_traceback):
        traceback.print_exception(e_type, e_value, e_traceback) # this is very helpful when there's an exception in the rest of this func
        last_tb = get_last_traceback(e_traceback)
        ex = (last_tb.tb_frame.f_code.co_filename, last_tb.tb_frame.f_lineno)
        if str(e_value).startswith("!!!"):
            Display_Error_Dialog(e_value)
        elif ex not in ignored_exceptions:
            result = Display_Exception_Dialog(e_type,e_value,e_traceback)
            if result:
                ignored_exceptions.append(ex)
                info = {
                    'app-title' : wx.GetApp().GetAppName(), # app_title
                    'app-version' : app_version,
                    'wx-version' : wx.VERSION_STRING,
                    'wx-platform' : wx.Platform,
                    'python-version' : platform.python_version(), #sys.version.split()[0],
                    'platform' : platform.platform(),
                    'e-type' : e_type,
                    'e-value' : e_value,
                    'date' : time.ctime(),
                    'cwd' : os.getcwd(),
                    }
                if e_traceback:
                    info['traceback'] = ''.join(traceback.format_tb(e_traceback)) + '%s: %s' % (e_type, e_value)
                    last_tb = get_last_traceback(e_traceback)
                    exception_locals = last_tb.tb_frame.f_locals # the locals at the level of the stack trace where the exception actually occurred
                    info['locals'] = format_namespace(exception_locals)
                    if 'self' in exception_locals:
                        info['self'] = format_namespace(exception_locals['self'].__dict__)
                
                output = open(path+os.sep+"bug_report_"+info['date'].replace(':','-').replace(' ','_')+".txt",'w')
                lst = info.keys()
                lst.sort()
                for a in lst:
                    output.write(a+":\n"+str(info[a])+"\n\n")

    #sys.excepthook = lambda *args: wx.CallAfter(handle_exception, *args)
    sys.excepthook = handle_exception

if __name__ == '__main__':
    def usage():
        print "\nUsage of PLCOpenEditor.py :"
        print "\n   %s [Filepath]\n"%sys.argv[0]

    try:
        opts, args = getopt.getopt(sys.argv[1:], "h", ["help"])
    except getopt.GetoptError:
        # print help information and exit:
        usage()
        sys.exit(2)

    for o, a in opts:
        if o in ("-h", "--help"):
            usage()
            sys.exit()

    fileOpen = None
    if len(args) > 1:
        usage()
        sys.exit()
    elif len(args) == 1:
        fileOpen = args[0]
    
    app = wx.PySimpleApp()
    wx.InitAllImageHandlers()
    
    # Install a exception handle for bug reports
    AddExceptHook(os.getcwd(),__version__)
    
    frame = PLCOpenEditor(None, fileOpen=fileOpen)

    frame.Show()
    app.MainLoop()