1

import os, re

     2

from lxml import etree

     3

     4

from xmlclass import GenerateParserFromXSD

     5

     6

from CodeFileTreeNode import CodeFile

     7

from PythonEditor import PythonEditor

     8

     9

class PythonFileCTNMixin(CodeFile):

    10

    11

    CODEFILE_NAME = "PyFile"

    12

    SECTIONS_NAMES = [

    13

        "globals",

    14

        "init",

    15

        "cleanup",

    16

        "start",

    17

        "stop"]

    18

    EditorType = PythonEditor

    19

    20

    def __init__(self):

    21

        CodeFile.__init__(self)

    22

    23

        filepath = self.PythonFileName()

    24

    25

        if os.path.isfile(filepath):

    26

            PythonParser = GenerateParserFromXSD(

    27

                os.path.join(os.path.dirname(__file__), "py_ext_xsd.xsd")) 

    28

    29

            xmlfile = open(filepath, 'r')

    30

            pythonfile_xml = xmlfile.read()

    31

            xmlfile.close()

    32

    33

            pythonfile_xml = pythonfile_xml.replace(

    34

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

    35

                'xmlns:xhtml="http://www.w3.org/1999/xhtml"')

    36

            for cre, repl in [

    37

                (re.compile("(?<!<xhtml:p>)(?:<!\[CDATA\[)"), "<xhtml:p><![CDATA["),

    38

                (re.compile("(?:]]>)(?!</xhtml:p>)"), "]]></xhtml:p>")]:

    39

                pythonfile_xml = cre.sub(repl, pythonfile_xml)

    40

            python_code = etree.fromstring(pythonfile_xml, PythonParser)

    41

    42

            self.CodeFile.globals.setanyText(python_code.getanyText())

    43

            os.remove(filepath)

    44

            self.CreateCodeFileBuffer(False)

    45

            self.OnCTNSave()

    46

    47

    def CodeFileName(self):

    48

        return os.path.join(self.CTNPath(), "pyfile.xml")

    49

    50

    def PythonFileName(self):

    51

        return os.path.join(self.CTNPath(), "py_ext.xml")

    52

    53

    PreSectionsTexts = {}

    54

    PostSectionsTexts = {}

    55

    def GetSection(self,section):

    56

        return self.PreSectionsTexts.get(section,"") + "\n" + \

    57

               getattr(self.CodeFile, section).getanyText() + "\n" + \

    58

               self.PostSectionsTexts.get(section,"")

    59

    60

    61

    def CTNGenerate_C(self, buildpath, locations):

    62

        # location string for that CTN 

    63

        location_str = "_".join(map(lambda x:str(x), 

    64

                                self.GetCurrentLocation()))

    65

        configname = self.GetCTRoot().GetProjectConfigNames()[0]

    66

    67

    68

        # python side PLC global variables access stub

    69

        globalstubs = "\n".join(["""\

    70

_%(name)s_ctype, _%(name)s_unpack, _%(name)s_pack = \\

    71

    TypeTranslator["%(IECtype)s"]

    72

_PySafeGetPLCGlob_%(name)s = PLCBinary.__SafeGetPLCGlob_%(name)s

    73

_PySafeGetPLCGlob_%(name)s.restype = None

    74

_PySafeGetPLCGlob_%(name)s.argtypes = [ctypes.POINTER(_%(name)s_ctype)] 

    75

_PySafeSetPLCGlob_%(name)s = PLCBinary.__SafeSetPLCGlob_%(name)s

    76

_PySafeSetPLCGlob_%(name)s.restype = None

    77

_PySafeSetPLCGlob_%(name)s.argtypes = [ctypes.POINTER(_%(name)s_ctype)]

    78

""" % { "name": variable.getname(),

    79

        "configname": configname.upper(),

    80

        "uppername": variable.getname().upper(),

    81

        "IECtype": variable.gettype()}

    82

            for variable in self.CodeFile.variables.variable])

    83

    84

        # Runtime calls (start, stop, init, and cleanup)

    85

        rtcalls = ""

    86

        for section in self.SECTIONS_NAMES:

    87

            if section != "globals":

    88

                rtcalls += "def _runtime_%s_%s():\n" % (location_str, section)

    89

                sectiontext = self.GetSection(section).strip()

    90

                if sectiontext:

    91

                    rtcalls += '    ' + \

    92

                        sectiontext.replace('\n', '\n    ')+"\n\n"

    93

                else:

    94

                    rtcalls += "    pass\n\n"

    95

    96

        globalsection = self.GetSection("globals")      

    97

    98

        PyFileContent = """\

    99

#!/usr/bin/env python

   100

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

   101

## Code generated by Beremiz python mixin confnode

   102

##        

   103

   104

## Code for PLC global variable access

   105

from targets.typemapping import TypeTranslator

   106

import ctypes 

   107

%(globalstubs)s

   108

   109

## User code in "global" scope

   110

%(globalsection)s

   111

   112

## Beremiz python runtime calls

   113

%(rtcalls)s

   114

   115

""" % locals()

   116

   117

        # write generated content to python file

   118

        runtimefile_path = os.path.join(buildpath, 

   119

            "runtime_%s.py"%location_str)

   120

        runtimefile = open(runtimefile_path, 'w')

   121

        runtimefile.write(PyFileContent.encode('utf-8'))

   122

        runtimefile.close()

   123

   124

        # C code for safe global variables access

   125

   126

        vardecfmt = """\

   127

extern  __IEC_%(IECtype)s_t %(configname)s__%(uppername)s;

   128

IEC_%(IECtype)s __%(name)s_rbuffer = __INIT_%(IECtype)s;

   129

IEC_%(IECtype)s __%(name)s_wbuffer;

   130

long __%(name)s_rlock = 0;

   131

long __%(name)s_wlock = 0;

   132

int __%(name)s_wbuffer_written = 0;

   133

void __SafeGetPLCGlob_%(name)s(IEC_%(IECtype)s *pvalue){

   134

    while(AtomicCompareExchange(&__%(name)s_rlock, 0, 1));

   135

    *pvalue = __%(name)s_rbuffer;

   136

    AtomicCompareExchange((long*)&__%(name)s_rlock, 1, 0);

   137

}

   138

void __SafeSetPLCGlob_%(name)s(IEC_%(IECtype)s *value){

   139

    while(AtomicCompareExchange(&__%(name)s_wlock, 0, 1));

   140

    __%(name)s_wbuffer = *value;

   141

    __%(name)s_wbuffer_written = 1;

   142

    AtomicCompareExchange((long*)&__%(name)s_wlock, 1, 0);

   143

}

   144

   145

"""

   146

        varretfmt = """\

   147

    if(!AtomicCompareExchange(&__%(name)s_wlock, 0, 1)){

   148

        if(__%(name)s_wbuffer_written == 1){

   149

            %(configname)s__%(uppername)s.value = __%(name)s_wbuffer;

   150

            __%(name)s_wbuffer_written = 0;

   151

        }

   152

        AtomicCompareExchange((long*)&__%(name)s_wlock, 1, 0);

   153

    }

   154

""" 

   155

        varpubfmt = """\

   156

    if(!AtomicCompareExchange(&__%(name)s_rlock, 0, 1)){

   157

        __%(name)s_rbuffer = %(configname)s__%(uppername)s.value;

   158

        AtomicCompareExchange((long*)&__%(name)s_rlock, 1, 0);

   159

    }

   160

""" 

   161

   162

        var_str = map("\n".join, zip(*[

   163

            map(lambda f : f % varinfo,

   164

                (vardecfmt, varretfmt, varpubfmt))

   165

                for varinfo in map(lambda variable : {

   166

                    "name": variable.getname(),

   167

                    "configname": configname.upper(),

   168

                    "uppername": variable.getname().upper(),

   169

                    "IECtype": variable.gettype()},

   170

                    self.CodeFile.variables.variable)]))

   171

        if len(var_str) > 0:

   172

            vardec, varret, varpub = var_str

   173

        else:

   174

            vardec = varret = varpub = ""

   175

   176

        PyCFileContent = """\

   177

/* 

   178

 * Code generated by Beremiz py_ext confnode 

   179

 * for safe global variables access

   180

 */

   181

#include "iec_types_all.h"

   182

#include "beremiz.h"

   183

   184

/* User variables reference */

   185

%(vardec)s

   186

   187

/* Beremiz confnode functions */

   188

int __init_%(location_str)s(int argc,char **argv){

   189

    return 0;

   190

}

   191

   192

void __cleanup_%(location_str)s(void){

   193

}

   194

   195

void __retrieve_%(location_str)s(void){

   196

%(varret)s

   197

}

   198

   199

void __publish_%(location_str)s(void){

   200

%(varpub)s

   201

}

   202

""" % locals()

   203

   204

        Gen_PyCfile_path = os.path.join(buildpath, "PyCFile_%s.c"%location_str)

   205

        pycfile = open(Gen_PyCfile_path,'w')

   206

        pycfile.write(PyCFileContent)

   207

        pycfile.close()

   208

   209

        matiec_flags = '"-I%s"'%os.path.abspath(

   210

            self.GetCTRoot().GetIECLibPath())

   211

   212

        return ([(Gen_PyCfile_path, matiec_flags)],

   213

                "",

   214

                True,

   215

                ("runtime_%s.py"%location_str, file(runtimefile_path,"rb")))

   216