1

import os, sys

     2

base_folder = os.path.split(sys.path[0])[0]

     3

sys.path.append(os.path.join(base_folder, "CanFestival-3", "objdictgen"))

     4

CanfestivalIncludePath = os.path.join(base_folder, "CanFestival-3", "include")

     5

CanfestivalLibPath = os.path.join(base_folder, "CanFestival-3", "src")

     6

     7

from nodelist import NodeList

     8

from nodemanager import NodeManager

     9

import config_utils, gen_cfile

    10

from networkedit import networkedit

    11

    12

class _NetworkEdit(networkedit):

    13

    " Overload some of CanFestival Network Editor methods "

    14

    def OnCloseFrame(self, event):

    15

        " Do reset _NodeListPlug.View when closed"

    16

        self._onclose()

    17

        event.Skip()

    18

    19

class _NodeListPlug(NodeList):

    20

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

    21

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

    22

      <xsd:element name="CanFestivalNode">

    23

        <xsd:complexType>

    24

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

    25

          <xsd:attribute name="Sync_TPDOs" type="xsd:boolean" use="required" default="true"/>

    26

        </xsd:complexType>

    27

      </xsd:element>

    28

    </xsd:schema>

    29

    """

    30

    31

    def __init__(self):

    32

        manager = NodeManager()

    33

        # TODO change netname when name change

    34

        NodeList.__init__(self, manager, self.BaseParams.getName())

    35

        self.LoadProject(self.PlugPath())

    36

    37

    _View = None

    38

    def _OpenView(self, logger):

    39

        if not self._View:

    40

            def _onclose():

    41

                self._View = None

    42

            def _onsave():

    43

                self.GetPlugRoot().SaveProject()

    44

            self._View = _NetworkEdit(self.GetPlugRoot().AppFrame, self)

    45

            # TODO redefine BusId when IEC channel change

    46

            self._View.SetBusId(self.GetCurrentLocation())

    47

            self._View._onclose = _onclose

    48

            self._View._onsave = _onsave

    49

            self._View.Show()

    50

    51

    PluginMethods = [("NetworkEdit",_OpenView)]

    52

    53

    def OnPlugClose(self):

    54

        if self._View:

    55

            self._View.Close()

    56

    57

    def PlugTestModified(self):

    58

        return self.HasChanged()

    59

    60

    def OnPlugSave(self):

    61

        self.SetRoot(self.PlugPath())

    62

        self.SaveProject()

    63

        return True

    64

    65

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

    66

        """

    67

        Generate C code

    68

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

    69

        @param locations: List of complete variables locations \

    70

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

    71

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

    72

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

    73

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

    74

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

    75

            }, ...]

    76

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

    77

        """

    78

        current_location = self.GetCurrentLocation()

    79

        # define a unique name for the generated C file

    80

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

    81

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

    82

        # Create a new copy of the model with DCF loaded with PDO mappings for desired location

    83

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

    84

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

    85

        if res :

    86

            raise Exception, res

    87

    88

        return [(Gen_OD_path,"-I"+CanfestivalIncludePath)],""

    89

    90

class RootClass:

    91

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

    92

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

    93

      <xsd:element name="CanFestivalInstance">

    94

        <xsd:complexType>

    95

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

    96

        </xsd:complexType>

    97

      </xsd:element>

    98

    </xsd:schema>

    99

    """

   100

   101

    PlugChildsTypes = [("CanOpenNode",_NodeListPlug)]

   102

   103

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

   104

        return [],"-L"+CanfestivalLibPath+" -lcanfestival"

   105

   106