1

import os

     2

from nodelist import NodeList

     3

from nodemanager import NodeManager

     4

import config_utils, gen_cfile

     5

from networkedit import networkedit

     6

     7

class _NetworkEdit(networkedit):

     8

    " Overload some of CanFestival Network Editor methods "

     9

    def OnCloseFrame(self, event):

    10

        " Do reset _NodeListPlug.View when closed"

    11

        self._onclose()

    12

        event.Skip()

    13

    14

class _NodeListPlug(NodeList):

    15

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

    16

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

    17

      <xsd:element name="CanFestivalNode">

    18

        <xsd:complexType>

    19

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

    20

        </xsd:complexType>

    21

      </xsd:element>

    22

    </xsd:schema>

    23

    """

    24

    25

    def __init__(self):

    26

        manager = NodeManager()

    27

        NodeList.__init__(self, manager)

    28

        self.LoadProject(self.PlugPath())

    29

    30

    _View = None

    31

    def _OpenView(self, logger):

    32

        if not self._View:

    33

            def _onclose():

    34

                self.View = None

    35

            self._View = _NetworkEdit()

    36

            self._View._onclose = _onclose

    37

        return self.View

    38

    PluginMethods = [("NetworkEdit",_OpenView)]

    39

    40

    def OnPlugClose(self):

    41

        if self._View:

    42

            self._View.Close()

    43

    44

    def PlugTestModified(self):

    45

        return self.HasChanged()

    46

    47

    def OnPlugSave(self):

    48

        self.SaveProject()

    49

        return True

    50

    51

    def PlugGenerate_C(self, buildpath, current_location, locations, logger):

    52

        """

    53

        Generate C code

    54

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

    55

        @param locations: List of complete variables locations \

    56

            [(IEC_loc, IEC_Direction IEC_Type, Name)]\

    57

            ex: [((0,0,4,5),'I','X','__IX_0_0_4_5'),...]

    58

        """

    59

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

    60

        Gen_OD_path = os.path.join(buildpath, prefix + "_OD.c" )

    61

        master = config_utils.GenerateConciseDCF(locations, current_location, self)

    62

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

    63

        if res :

    64

            raise Exception, res

    65

    66

        return [(Gen_OD_path,CanFestival_OD_CFLAGS)],""

    67

    68

class RootClass:

    69

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

    70

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

    71

      <xsd:element name="CanFestivalInstance">

    72

        <xsd:complexType>

    73

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

    74

        </xsd:complexType>

    75

      </xsd:element>

    76

    </xsd:schema>

    77

    """

    78

    79

    PlugChildsTypes = [("CanOpenNode",_NodeListPlug)]

    80

    81

    def PlugGenerate_C(self, buildpath, current_location, locations):

    82

        """

    83

        Generate C code

    84

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

    85

        @param locations: List of complete variables locations \

    86

            [(IEC_loc, IEC_Direction IEC_Type, Name)]\

    87

            ex: [((0,0,4,5),'I','X','__IX_0_0_4_5'),...]

    88

        """

    89

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

    90

        Gen_OD_path = os.path.join(buildpath, prefix + "_OD.c" )

    91

        master = config_utils.GenerateConciseDCF(locations, self)

    92

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

    93

        if not res:

    94

             s = str(self.BaseParams.BusId)+"_IN(){}\n"

    95

             s += "CanOpen(str(\""+self.CanFestivalNode.CAN_Device+"\")"

    96

             f = file(filepath, 'a')

    97

             f.write(s)

    98

        else:

    99

             pass # error

   100

   101

        return [],""

   102

   103