1

import wx

     2

import os

     3

from xml.dom import minidom

     4

import cPickle

     5

     6

from xmlclass import *

     7

     8

from plugger import PlugTemplate, opjimg

     9

from CFileEditor import CFileEditor

    10

from PLCControler import PLCControler, UndoBuffer, LOCATION_PLUGIN, LOCATION_MODULE, LOCATION_GROUP, LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY

    11

    12

CFileClasses = GenerateClassesFromXSD(os.path.join(os.path.dirname(__file__), "cext_xsd.xsd"))

    13

    14

TYPECONVERSION = {"BOOL" : "X", "SINT" : "B", "INT" : "W", "DINT" : "D", "LINT" : "L",

    15

    "USINT" : "B", "UINT" : "W", "UDINT" : "D", "ULINT" : "L", "REAL" : "D", "LREAL" : "L",

    16

    "STRING" : "B", "BYTE" : "B", "WORD" : "W", "DWORD" : "D", "LWORD" : "L", "WSTRING" : "W"}

    17

    18

class _Cfile:

    19

    XSD = """<?xml version="1.0" encoding="ISO-8859-1" ?>

    20

    <xsd:schema xmlns:xsd="http://www.w3.org/2001/XMLSchema">

    21

      <xsd:element name="CExtension">

    22

        <xsd:complexType>

    23

          <xsd:attribute name="CFLAGS" type="xsd:string" use="required"/>

    24

          <xsd:attribute name="LDFLAGS" type="xsd:string" use="required"/>

    25

        </xsd:complexType>

    26

      </xsd:element>

    27

    </xsd:schema>

    28

    """

    29

    EditorType = CFileEditor

    30

    31

    def __init__(self):

    32

        filepath = self.CFileName()

    33

    34

        self._View = None

    35

    36

        self.Buffering = False

    37

        self.CFile = CFileClasses["CFile"]()

    38

        if os.path.isfile(filepath):

    39

            xmlfile = open(filepath, 'r')

    40

            tree = minidom.parse(xmlfile)

    41

            xmlfile.close()

    42

    43

            for child in tree.childNodes:

    44

                if child.nodeType == tree.ELEMENT_NODE and child.nodeName == "CFile":

    45

                    self.CFile.loadXMLTree(child, ["xmlns", "xmlns:xsi", "xsi:schemaLocation"])

    46

                    self.CFileBuffer = UndoBuffer(self.Copy(self.CFile), True)

    47

        else:

    48

            self.CFileBuffer = UndoBuffer(self.Copy(self.CFile), False)

    49

            self.OnPlugSave()

    50

    51

    def CFileName(self):

    52

        return os.path.join(self.PlugPath(), "cfile.xml")

    53

    54

    def GetFilename(self):

    55

        return self.MandatoryParams[1].getName()

    56

    57

    def GetBaseTypes(self):

    58

        return self.GetPlugRoot().GetBaseTypes()

    59

    60

    def GetDataTypes(self, basetypes = False, only_locatables = False):

    61

        return self.GetPlugRoot().GetDataTypes(basetypes=basetypes, only_locatables=only_locatables)

    62

    63

    def GetSizeOfType(self, type):

    64

        return TYPECONVERSION.get(self.GetPlugRoot().GetBaseType(type), None)

    65

    66

    def SetVariables(self, variables):

    67

        self.CFile.variables.setvariable([])

    68

        for var in variables:

    69

            variable = CFileClasses["variables_variable"]()

    70

            variable.setname(var["Name"])

    71

            variable.settype(var["Type"])

    72

            variable.setclass(var["Class"])

    73

            self.CFile.variables.appendvariable(variable)

    74

    75

    def GetVariables(self):

    76

        datas = []

    77

        for var in self.CFile.variables.getvariable():

    78

            datas.append({"Name" : var.getname(), "Type" : var.gettype(), "Class" : var.getclass()})

    79

        return datas

    80

    81

    def GetVariableLocationTree(self):

    82

        '''See PlugTemplate.GetVariableLocationTree() for a description.'''

    83

    84

        current_location = ".".join(map(str, self.GetCurrentLocation()))

    85

    86

        vars = []

    87

        input = memory = output = 0

    88

        for var in self.CFile.variables.getvariable():

    89

            var_size = self.GetSizeOfType(var.gettype())

    90

            var_location = ""

    91

            if var.getclass() == "input":

    92

                var_class = LOCATION_VAR_INPUT

    93

                if var_size is not None:

    94

                    var_location = "%%I%s%s.%d"%(var_size, current_location, input)

    95

                input += 1

    96

            elif var.getclass() == "memory":

    97

                var_class = LOCATION_VAR_INPUT

    98

                if var_size is not None:

    99

                    var_location = "%%M%s%s.%d"%(var_size, current_location, memory)

   100

                memory += 1

   101

            else:

   102

                var_class = LOCATION_VAR_OUTPUT

   103

                if var_size is not None:

   104

                    var_location = "%%Q%s%s.%d"%(var_size, current_location, output)

   105

                output += 1

   106

            vars.append({"name": var.getname(),

   107

                         "type": var_class,

   108

                         "size": var_size,

   109

                         "IEC_type": var.gettype(),

   110

                         "location": var_location,

   111

                         "description": "",

   112

                         "children": []})

   113

   114

        return  {"name": self.BaseParams.getName(),

   115

                "type": LOCATION_PLUGIN,

   116

                "location": self.GetFullIEC_Channel(),

   117

                "children": vars}

   118

   119

    def SetPartText(self, name, text):

   120

        if name == "Includes":

   121

            self.CFile.includes.settext(text)

   122

        elif name == "Globals":

   123

            self.CFile.globals.settext(text)

   124

        elif name == "Init":

   125

            self.CFile.initFunction.settext(text)

   126

        elif name == "CleanUp":

   127

            self.CFile.cleanUpFunction.settext(text)

   128

        elif name == "Retrieve":

   129

            self.CFile.retrieveFunction.settext(text)

   130

        elif name == "Publish":

   131

            self.CFile.publishFunction.settext(text)

   132

   133

    def GetPartText(self, name):

   134

        if name == "Includes":

   135

            return self.CFile.includes.gettext()

   136

        elif name == "Globals":

   137

            return self.CFile.globals.gettext()

   138

        elif name == "Init":

   139

            return self.CFile.initFunction.gettext()

   140

        elif name == "CleanUp":

   141

            return self.CFile.cleanUpFunction.gettext()

   142

        elif name == "Retrieve":

   143

            return self.CFile.retrieveFunction.gettext()

   144

        elif name == "Publish":

   145

            return self.CFile.publishFunction.gettext()

   146

        return ""

   147

   148

    PluginMethods = [

   149

        {"bitmap" : os.path.join("images", "EditCfile"),

   150

         "name" : _("Edit C File"), 

   151

         "tooltip" : _("Edit C File"),

   152

         "method" : "_OpenView"},

   153

    ]

   154

   155

    def PlugTestModified(self):

   156

        return self.ChangesToSave or not self.CFileIsSaved()    

   157

   158

    def OnPlugSave(self):

   159

        filepath = self.CFileName()

   160

   161

        text = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"no\"?>\n"

   162

        extras = {"xmlns":"http://www.w3.org/2001/XMLSchema",

   163

                  "xmlns:xsi":"http://www.w3.org/2001/XMLSchema-instance",

   164

                  "xsi:schemaLocation" : "cext_xsd.xsd"}

   165

        text += self.CFile.generateXMLText("CFile", 0, extras)

   166

   167

        xmlfile = open(filepath,"w")

   168

        xmlfile.write(text.encode("utf-8"))

   169

        xmlfile.close()

   170

   171

        self.MarkCFileAsSaved()

   172

        return True

   173

   174

    def PlugGenerate_C(self, buildpath, locations):

   175

        """

   176

        Generate C code

   177

        @param current_location: Tupple containing plugin IEC location : %I0.0.4.5 => (0,0,4,5)

   178

        @param locations: List of complete variables locations \

   179

            [{"IEC_TYPE" : the IEC type (i.e. "INT", "STRING", ...)

   180

            "NAME" : name of the variable (generally "__IW0_1_2" style)

   181

            "DIR" : direction "Q","I" or "M"

   182

            "SIZE" : size "X", "B", "W", "D", "L"

   183

            "LOC" : tuple of interger for IEC location (0,1,2,...)

   184

            }, ...]

   185

        @return: [(C_file_name, CFLAGS),...] , LDFLAGS_TO_APPEND

   186

        """

   187

        current_location = self.GetCurrentLocation()

   188

        # define a unique name for the generated C file

   189

        location_str = "_".join(map(str, current_location))

   190

   191

        text = "/* Code generated by Beremiz c_ext plugin */\n\n"

   192

   193

        # Adding includes

   194

        text += "/* User includes */\n"

   195

        text += self.CFile.includes.gettext()

   196

        text += "\n"

   197

   198

        text += """/* Beremiz c_ext plugin includes */

   199

#ifdef _WINDOWS_H

   200

  #include "iec_types.h"

   201

#else

   202

  #include "iec_std_lib.h"

   203

#endif

   204

   205

"""

   206

   207

        # Adding variables

   208

        vars = []

   209

        inputs = memories = outputs = 0

   210

        for variable in self.CFile.variables.variable:

   211

            var = {"Name" : variable.getname(), "Type" : variable.gettype()}

   212

            if variable.getclass() == "input":

   213

                var["location"] = "__I%s%s_%d"%(self.GetSizeOfType(var["Type"]), location_str, inputs)

   214

                inputs += 1

   215

            elif variable.getclass() == "memory":

   216

                var["location"] = "__M%s%s_%d"%(self.GetSizeOfType(var["Type"]), location_str, memories)

   217

                memories += 1

   218

            else:

   219

                var["location"] = "__Q%s%s_%d"%(self.GetSizeOfType(var["Type"]), location_str, outputs)

   220

                outputs += 1

   221

            vars.append(var)

   222

        text += "/* Beremiz c_ext plugin user variables definition */\n"

   223

        base_types = self.GetPlugRoot().GetBaseTypes()

   224

        for var in vars:

   225

            if var["Type"] in base_types:

   226

                prefix = "IEC_"

   227

            else:

   228

                prefix = ""

   229

            text += "%s%s beremiz%s;\n"%(prefix, var["Type"], var["location"])

   230

            text += "%s%s *%s = &beremiz%s;\n"%(prefix, var["Type"], var["location"], var["location"])

   231

        text += "/* User variables reference */\n"

   232

        for var in vars:

   233

            text += "#define %s beremiz%s\n"%(var["Name"], var["location"])

   234

        text += "\n"

   235

   236

        # Adding user global variables and routines

   237

        text += "/* User internal user variables and routines */\n"

   238

        text += self.CFile.globals.gettext()

   239

   240

        # Adding Beremiz plugin functions

   241

        text += "/* Beremiz plugin functions */\n"

   242

        text += "int __init_%s(int argc,char **argv)\n{\n"%location_str

   243

        text += self.CFile.initFunction.gettext()

   244

        text += "  return 0;\n"

   245

        text += "\n}\n\n"

   246

   247

        text += "void __cleanup_%s(void)\n{\n"%location_str

   248

        text += self.CFile.cleanUpFunction.gettext()

   249

        text += "\n}\n\n"

   250

   251

        text += "void __retrieve_%s(void)\n{\n"%location_str

   252

        text += self.CFile.retrieveFunction.gettext()

   253

        text += "\n}\n\n"

   254

   255

        text += "void __publish_%s(void)\n{\n"%location_str

   256

        text += self.CFile.publishFunction.gettext()

   257

        text += "\n}\n\n"

   258

   259

        Gen_Cfile_path = os.path.join(buildpath, "CFile_%s.c"%location_str)

   260

        cfile = open(Gen_Cfile_path,'w')

   261

        cfile.write(text)

   262

        cfile.close()

   263

   264

        matiec_flags = '"-I%s"'%os.path.abspath(self.GetPlugRoot().GetIECLibPath())

   265

   266

        return [(Gen_Cfile_path, str(self.CExtension.getCFLAGS() + matiec_flags))],str(self.CExtension.getLDFLAGS()),True

   267

   268

   269

#-------------------------------------------------------------------------------

   270

#                      Current Buffering Management Functions

   271

#-------------------------------------------------------------------------------

   272

   273

    """

   274

    Return a copy of the cfile model

   275

    """

   276

    def Copy(self, model):

   277

        return cPickle.loads(cPickle.dumps(model))

   278

   279

    def CreateConfigBuffer(self, saved):

   280

        self.CFileBuffer = UndoBuffer(cPickle.dumps(self.CFile), saved)

   281

   282

    def BufferCFile(self):

   283

        self.CFileBuffer.Buffering(cPickle.dumps(self.CFile))

   284

   285

    def StartBuffering(self):

   286

        self.Buffering = True

   287

   288

    def EndBuffering(self):

   289

        if self.Buffering:

   290

            self.CFileBuffer.Buffering(cPickle.dumps(self.CFile))

   291

            self.Buffering = False

   292

   293

    def MarkCFileAsSaved(self):

   294

        self.EndBuffering()

   295

        self.CFileBuffer.CurrentSaved()

   296

   297

    def CFileIsSaved(self):

   298

        if self.CFileBuffer:

   299

            return self.CFileBuffer.IsCurrentSaved() and not self.Buffering

   300

        else:

   301

            return True

   302

   303

    def LoadPrevious(self):

   304

        self.EndBuffering()

   305

        self.CFile = cPickle.loads(self.CFileBuffer.Previous())

   306

   307

    def LoadNext(self):

   308

        self.CFile = cPickle.loads(self.CFileBuffer.Next())

   309

   310

    def GetBufferState(self):

   311

        first = self.CFileBuffer.IsFirst()

   312

        last = self.CFileBuffer.IsLast()

   313

        return not first, not last

   314

   315

class RootClass:

   316

   317

    PlugChildsTypes = [("C_File",_Cfile, "C file")]

   318

   319

    def PlugGenerate_C(self, buildpath, locations):

   320

        return [],"",False

   321

   322