1

import os, sys

     2

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

     3

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

     4

sys.path.append(os.path.join(CanFestivalPath, "objdictgen"))

     5

     6

from nodelist import NodeList

     7

from nodemanager import NodeManager

     8

import config_utils, gen_cfile

     9

from networkedit import networkedit

    10

import canfestival_config

    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="CAN_Baudrate" type="xsd:string" use="required" />

    26

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

    27

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

    28

        </xsd:complexType>

    29

      </xsd:element>

    30

    </xsd:schema>

    31

    """

    32

    33

    def __init__(self):

    34

        manager = NodeManager()

    35

        # TODO change netname when name change

    36

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

    37

        self.LoadProject(self.PlugPath())

    38

    39

    _View = None

    40

    def _OpenView(self, logger):

    41

        if not self._View:

    42

            def _onclose():

    43

                self._View = None

    44

            def _onsave():

    45

                self.GetPlugRoot().SaveProject()

    46

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

    47

            # TODO redefine BusId when IEC channel change

    48

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

    49

            self._View._onclose = _onclose

    50

            self._View._onsave = _onsave

    51

            self._View.Show()

    52

    53

    PluginMethods = [

    54

        {"bitmap" : os.path.join("images", "NetworkEdit.png"),

    55

         "name" : "Edit network", 

    56

         "tooltip" : "Edit CanOpen Network with NetworkEdit",

    57

         "method" : _OpenView}

    58

    ]

    59

    60

    def OnPlugClose(self):

    61

        if self._View:

    62

            self._View.Close()

    63

    64

    def PlugTestModified(self):

    65

        return self.HasChanged()

    66

    67

    def OnPlugSave(self):

    68

        self.SetRoot(self.PlugPath())

    69

        self.SaveProject()

    70

        return True

    71

    72

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

    73

        """

    74

        Generate C code

    75

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

    76

        @param locations: List of complete variables locations \

    77

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

    78

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

    79

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

    80

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

    81

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

    82

            }, ...]

    83

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

    84

        """

    85

        current_location = self.GetCurrentLocation()

    86

        # define a unique name for the generated C file

    87

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

    88

        Gen_OD_path = os.path.join(buildpath, "OD_%s.c"%prefix )

    89

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

    90

        master = config_utils.GenerateConciseDCF(locations, current_location, self, self.CanFestivalNode.getSync_TPDOs(),"OD_%s"%prefix)

    91

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

    92

        if res :

    93

            raise Exception, res

    94

    95

        return [(Gen_OD_path,canfestival_config.getCFLAGS(CanFestivalPath))],"",False

    96

    97

class RootClass:

    98

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

    99

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

   100

      <xsd:element name="CanFestivalInstance">

   101

        <xsd:complexType>

   102

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

   103

        </xsd:complexType>

   104

      </xsd:element>

   105

    </xsd:schema>

   106

    """

   107

   108

    PlugChildsTypes = [("CanOpenNode",_NodeListPlug)]

   109

   110

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

   111

   112

        format_dict = {"locstr" : "_".join(map(str,self.GetCurrentLocation())),

   113

                       "candriver" : self.CanFestivalInstance.getCAN_Driver(),

   114

                       "nodes_includes" : "",

   115

                       "board_decls" : "",

   116

                       "nodes_declare" : "",

   117

                       "nodes_init" : "",

   118

                       "nodes_open" : "",

   119

                       "nodes_close" : "",

   120

                       "nodes_send_sync" : "",

   121

                       "nodes_proceed_sync" : ""}

   122

        for child in self.IECSortedChilds():

   123

            childlocstr = "_".join(map(str,child.GetCurrentLocation()))

   124

            nodename = "OD_%s" % childlocstr

   125

   126

            format_dict["nodes_includes"] += '#include "%s.h"\n'%(nodename)

   127

            format_dict["board_decls"] += 'BOARD_DECL(%s, "%s", "%s")\n'%(

   128

                   nodename,

   129

                   child.CanFestivalNode.getCAN_Device(),

   130

                   child.CanFestivalNode.getCAN_Baudrate())

   131

            format_dict["nodes_declare"] += 'NODE_DECLARE(%s, %s)\n    '%(

   132

                   nodename,

   133

                   child.CanFestivalNode.getNodeId())

   134

            format_dict["nodes_init"] += 'NODE_INIT(%s, %s)\n    '%(

   135

                   nodename,

   136

                   child.CanFestivalNode.getNodeId())

   137

            format_dict["nodes_open"] += 'NODE_OPEN(%s)\n    '%(nodename)

   138

            format_dict["nodes_close"] += 'NODE_CLOSE(%s)\n    '%(nodename)

   139

            format_dict["nodes_send_sync"] += 'NODE_SEND_SYNC(%s)\n    '%(nodename)

   140

            format_dict["nodes_proceed_sync"] += 'NODE_PROCEED_SYNC(%s)\n    '%(nodename)

   141

        filename = os.path.join(os.path.split(__file__)[0],"cf_runtime.c")

   142

        cf_main = open(filename).read() % format_dict

   143

        cf_main_path = os.path.join(buildpath, "CF_%(locstr)s.c"%format_dict)

   144

        f = open(cf_main_path,'w')

   145

        f.write(cf_main)

   146

        f.close()

   147

   148

        return [(cf_main_path, canfestival_config.getCFLAGS(CanFestivalPath))],canfestival_config.getLDFLAGS(CanFestivalPath), True

   149

   150