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

    41

            try:

    42

                python_code, error = PythonParser.LoadXMLString(pythonfile_xml)

    43

                if error is None:

    44

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

    45

                    os.remove(filepath)

    46

                    self.CreateCodeFileBuffer(False)

    47

                    self.OnCTNSave()

    48

            except Exception, exc:

    49

                error = unicode(exc)

    50

    51

            if error is not None:

    52

                self.GetCTRoot().logger.write_error(

    53

                    _("Couldn't import old %s file.") % self.CTNName())

    54

    55

    def CodeFileName(self):

    56

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

    57

    58

    def PythonFileName(self):

    59

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

    60

    61

    PreSectionsTexts = {}

    62

    PostSectionsTexts = {}

    63

    def GetSection(self,section):

    64

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

    65

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

    66

               self.PostSectionsTexts.get(section,"")

    67

    68

    def CTNGenerate_C(self, buildpath, locations):

    69

        # location string for that CTN

    70

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

    71

                                self.GetCurrentLocation()))

    72

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

    73

    74

        pyextname = self.CTNName()

    75

        varinfos = map(lambda variable : {

    76

                    "name": variable.getname(),

    77

                    "desc" : repr(variable.getdesc()),

    78

                    "onchangecode" : '"'+variable.getonchange()+\

    79

                                         "('"+variable.getname()+"')\"" \

    80

                                     if variable.getonchange() else '""',

    81

                    "onchange" : repr(variable.getonchange()) \

    82

                                 if variable.getonchange() else None,

    83

                    "opts" : repr(variable.getopts()),

    84

                    "configname" : configname.upper(),

    85

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

    86

                    "IECtype" : variable.gettype(),

    87

                    "pyextname" :pyextname},

    88

                    self.CodeFile.variables.variable)

    89

        # python side PLC global variables access stub

    90

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

    91

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

    92

    TypeTranslator["%(IECtype)s"]

    93

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

    94

_PySafeGetPLCGlob_%(name)s.restype = None

    95

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

    96

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

    97

_PySafeSetPLCGlob_%(name)s.restype = None

    98

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

    99

_%(pyextname)sGlobalsDesc.append((

   100

    "%(name)s",

   101

    "%(IECtype)s",

   102

    %(desc)s,

   103

    %(onchange)s,

   104

    %(opts)s))

   105

""" % varinfo

   106

      for varinfo in varinfos])

   107

   108

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

   109

        rtcalls = ""

   110

        for section in self.SECTIONS_NAMES:

   111

            if section != "globals":

   112

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

   113

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

   114

                if sectiontext:

   115

                    rtcalls += '    ' + \

   116

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

   117

                else:

   118

                    rtcalls += "    pass\n\n"

   119

   120

        globalsection = self.GetSection("globals")

   121

   122

        PyFileContent = """\

   123

#!/usr/bin/env python

   124

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

   125

## Code generated by Beremiz python mixin confnode

   126

##

   127

   128

## Code for PLC global variable access

   129

from targets.typemapping import TypeTranslator

   130

import ctypes

   131

_%(pyextname)sGlobalsDesc = []

   132

__ext_name__ = "%(pyextname)s"

   133

PLCGlobalsDesc.append(( "%(pyextname)s" , _%(pyextname)sGlobalsDesc ))

   134

%(globalstubs)s

   135

   136

## User code in "global" scope

   137

%(globalsection)s

   138

   139

## Beremiz python runtime calls

   140

%(rtcalls)s

   141

   142

del __ext_name__

   143

   144

""" % locals()

   145

   146

        # write generated content to python file

   147

        runtimefile_path = os.path.join(buildpath,

   148

            "runtime_%s.py"%location_str)

   149

        runtimefile = open(runtimefile_path, 'w')

   150

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

   151

        runtimefile.close()

   152

   153

        # C code for safe global variables access

   154

   155

        vardecfmt = """\

   156

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

   157

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

   158

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

   159

long __%(name)s_rlock = 0;

   160

long __%(name)s_wlock = 0;

   161

int __%(name)s_wbuffer_written = 0;

   162

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

   163

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

   164

    *pvalue = __%(name)s_rbuffer;

   165

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

   166

}

   167

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

   168

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

   169

    __%(name)s_wbuffer = *value;

   170

    __%(name)s_wbuffer_written = 1;

   171

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

   172

}

   173

   174

"""

   175

   176

        vardeconchangefmt = """\

   177

PYTHON_POLL* __%(name)s_notifier;

   178

"""

   179

   180

        varretfmt = """\

   181

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

   182

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

   183

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

   184

            __%(name)s_wbuffer_written = 0;

   185

        }

   186

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

   187

    }

   188

"""

   189

        varpubfmt = """\

   190

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

   191

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

   192

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

   193

    }

   194

"""

   195

   196

        varpubonchangefmt = """\

   197

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

   198

        IEC_%(IECtype)s tmp = __GET_VAR(%(configname)s__%(uppername)s);

   199

        if(__%(name)s_rbuffer != tmp){

   200

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

   201

            PYTHON_POLL_body__(__%(name)s_notifier);

   202

        }

   203

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

   204

    }

   205

"""

   206

        varinitonchangefmt = """\

   207

    __%(name)s_notifier = __GET_GLOBAL_ON%(uppername)sCHANGE();

   208

    __SET_VAR(__%(name)s_notifier->,TRIG,,__BOOL_LITERAL(TRUE));

   209

    __SET_VAR(__%(name)s_notifier->,CODE,,__STRING_LITERAL(%(onchangelen)d,%(onchangecode)s));

   210

"""

   211

        vardec = "\n".join([(vardecfmt + vardeconchangefmt

   212

                             if varinfo["onchange"] else vardecfmt)% varinfo

   213

                            for varinfo in varinfos])

   214

        varret = "\n".join([varretfmt % varinfo for varinfo in varinfos])

   215

        varpub = "\n".join([(varpubonchangefmt if varinfo["onchange"] else

   216

                             varpubfmt) % varinfo

   217

                            for varinfo in varinfos])

   218

        varinit = "\n".join([varinitonchangefmt % dict(

   219

                                onchangelen = len(varinfo["onchangecode"]),**varinfo)

   220

                            for varinfo in varinfos if varinfo["onchange"]])

   221

   222

        # TODO : use config name obtained from model instead of default

   223

        # "config.h". User cannot change config name, but project imported

   224

        # or created in older beremiz vesion could use different name.

   225

        PyCFileContent = """\

   226

/*

   227

 * Code generated by Beremiz py_ext confnode

   228

 * for safe global variables access

   229

 */

   230

#include "iec_types_all.h"

   231

#include "POUS.h"

   232

#include "config.h"

   233

#include "beremiz.h"

   234

   235

/* User variables reference */

   236

%(vardec)s

   237

   238

/* Beremiz confnode functions */

   239

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

   240

%(varinit)s

   241

    return 0;

   242

}

   243

   244

void __cleanup_%(location_str)s(void){

   245

}

   246

   247

void __retrieve_%(location_str)s(void){

   248

%(varret)s

   249

}

   250

   251

void __publish_%(location_str)s(void){

   252

%(varpub)s

   253

}

   254

""" % locals()

   255

   256

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

   257

        pycfile = open(Gen_PyCfile_path,'w')

   258

        pycfile.write(PyCFileContent)

   259

        pycfile.close()

   260

   261

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

   262

            self.GetCTRoot().GetIECLibPath())

   263

   264

        return ([(Gen_PyCfile_path, matiec_flags)],

   265

                "",

   266

                True,

   267

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

   268