1

import os

     2

from xml.dom import minidom

     3

import cPickle

     4

     5

from xmlclass import *

     6

     7

from PLCControler import UndoBuffer

     8

from editors.CodeFileEditor import SECTIONS_NAMES

     9

    10

CodeFileClasses = GenerateClassesFromXSD(os.path.join(os.path.dirname(__file__), "code_file.xsd"))

    11

    12

class CodeFile:

    13

    14

    def __init__(self):

    15

        filepath = self.CodeFileName()

    16

    17

        self.CodeFile = CodeFileClasses["CodeFile"]()

    18

        if os.path.isfile(filepath):

    19

            xmlfile = open(filepath, 'r')

    20

            tree = minidom.parse(xmlfile)

    21

            xmlfile.close()

    22

    23

            for child in tree.childNodes:

    25

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

    26

                    self.CreateCodeFileBuffer(True)

    27

        else:

    28

            self.CreateCodeFileBuffer(False)

    29

            self.OnCTNSave()

    30

    31

    def GetBaseTypes(self):

    32

        return self.GetCTRoot().GetBaseTypes()

    33

    34

    def GetDataTypes(self, basetypes = False):

    35

        return self.GetCTRoot().GetDataTypes(basetypes=basetypes)

    36

    37

    def SetVariables(self, variables):

    38

        self.CodeFile.variables.setvariable([])

    39

        for var in variables:

    40

            variable = CodeFileClasses["variables_variable"]()

    41

            variable.setname(var["Name"])

    42

            variable.settype(var["Type"])

    43

            variable.setinitial(var["Initial"])

    44

            self.CodeFile.variables.appendvariable(variable)

    45

    46

    def GetVariables(self):

    47

        datas = []

    48

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

    49

            datas.append({"Name" : var.getname(), "Type" : var.gettype(), "Initial" : var.getinitial()})

    50

        return datas

    51

    52

    def SetTextParts(self, parts):

    53

        for section, code_object in zip(

    54

            SECTIONS_NAMES,

    55

            [self.CodeFile.includes,

    56

             self.CodeFile.globals,

    57

             self.CodeFile.initFunction,

    58

             self.CodeFile.cleanUpFunction,

    59

             self.CodeFile.retrieveFunction,

    60

             self.CodeFile.publishFunction]):

    61

            code_object.settext(parts[section])

    62

    63

    def GetTextParts(self):

    64

        parts = {}

    65

        for section, code_object in zip(

    66

            SECTIONS_NAMES,

    67

            [self.CodeFile.includes,

    68

             self.CodeFile.globals,

    69

             self.CodeFile.initFunction,

    70

             self.CodeFile.cleanUpFunction,

    71

             self.CodeFile.retrieveFunction,

    72

             self.CodeFile.publishFunction]):

    73

            parts[section] = code_object.gettext()

    74

        return parts

    75

    76

    def CTNTestModified(self):

    77

        return self.ChangesToSave or not self.CodeFileIsSaved()    

    78

    79

    def OnCTNSave(self, from_project_path=None):

    80

        filepath = self.CodeFileName()

    81

    82

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

    83

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

    84

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

    86

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

    87

    88

        xmlfile = open(filepath,"w")

    89

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

    90

        xmlfile.close()

    91

    92

        self.MarkCodeFileAsSaved()

    93

        return True

    94

    95

    def CTNGlobalInstances(self):

    96

        current_location = self.GetCurrentLocation()

    97

        return [(variable.getname(),

    98

                 variable.gettype(),

    99

                 variable.getinitial())

   100

                for variable in self.CodeFile.variables.variable]

   101

   102

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

   103

#                      Current Buffering Management Functions

   104

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

   105

   106

    """

   107

    Return a copy of the codefile model

   108

    """

   109

    def Copy(self, model):

   110

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

   111

   112

    def CreateCodeFileBuffer(self, saved):

   113

        self.Buffering = False

   114

        self.CodeFileBuffer = UndoBuffer(cPickle.dumps(self.CodeFile), saved)

   115

   116

    def BufferCodeFile(self):

   117

        self.CodeFileBuffer.Buffering(cPickle.dumps(self.CodeFile))

   118

   119

    def StartBuffering(self):

   120

        self.Buffering = True

   121

   122

    def EndBuffering(self):

   123

        if self.Buffering:

   124

            self.CodeFileBuffer.Buffering(cPickle.dumps(self.CodeFile))

   125

            self.Buffering = False

   126

   127

    def MarkCodeFileAsSaved(self):

   128

        self.EndBuffering()

   129

        self.CodeFileBuffer.CurrentSaved()

   130

   131

    def CodeFileIsSaved(self):

   132

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

   133

   134

    def LoadPrevious(self):

   135

        self.EndBuffering()

   136

        self.CodeFile = cPickle.loads(self.CodeFileBuffer.Previous())

   137

   138

    def LoadNext(self):

   139

        self.CodeFile = cPickle.loads(self.CodeFileBuffer.Next())

   140

   141

    def GetBufferState(self):

   142

        first = self.CodeFileBuffer.IsFirst() and not self.Buffering

   143

        last = self.CodeFileBuffer.IsLast()

   144

        return not first, not last

   145