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

from objdictedit import objdictedit

    11

import canfestival_config

    12

from plugger import PlugTemplate

    13

    14

from gnosis.xml.pickle import *

    15

from gnosis.xml.pickle.util import setParanoia

    16

setParanoia(0)

    17

    18

class _NetworkEdit(networkedit):

    19

    " Overload some of CanFestival Network Editor methods "

    20

    def OnCloseFrame(self, event):

    21

        " Do reset _NodeListPlug.View when closed"

    22

        self._onclose()

    23

        event.Skip()

    24

    25

class _NodeListPlug(NodeList):

    26

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

    27

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

    28

      <xsd:element name="CanFestivalNode">

    29

        <xsd:complexType>

    30

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

    31

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

    32

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

    33

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

    34

        </xsd:complexType>

    35

      </xsd:element>

    36

    </xsd:schema>

    37

    """

    38

    39

    def __init__(self):

    40

        manager = NodeManager()

    41

        # TODO change netname when name change

    42

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

    43

        self.LoadProject(self.PlugPath())

    44

    45

    _View = None

    46

    def _OpenView(self, logger):

    47

        if not self._View:

    48

            def _onclose():

    49

                self._View = None

    50

            def _onsave():

    51

                self.GetPlugRoot().SaveProject()

    52

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

    53

            # TODO redefine BusId when IEC channel change

    54

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

    55

            self._View._onclose = _onclose

    56

            self._View._onsave = _onsave

    57

            self._View.Show()

    58

    59

    def _ShowMasterGenerated(self, logger):

    60

        buildpath = self._getBuildPath()

    61

        # Eventually create build dir

    62

        if not os.path.exists(buildpath):

    63

            logger.write_error("Error: No PLC built\n")

    64

            return

    65

    66

        masterpath = os.path.join(buildpath, "MasterGenerated.od")

    67

        if not os.path.exists(masterpath):

    68

            logger.write_error("Error: No Master generated\n")

    69

            return

    70

    71

        new_dialog = objdictedit(None, [masterpath])

    72

        new_dialog.Show()

    73

    74

    PluginMethods = [

    75

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

    76

         "name" : "Edit network", 

    77

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

    78

         "method" : "_OpenView"},

    79

        {"name" : "Show Master", 

    80

         "tooltip" : "Show Master generated by config_utils",

    81

         "method" : "_ShowMasterGenerated"}

    82

    ]

    83

    84

    def OnPlugClose(self):

    85

        if self._View:

    86

            self._View.Close()

    87

    88

    def PlugTestModified(self):

    89

        return self.ChangesToSave or self.HasChanged()

    90

    91

    def OnPlugSave(self):

    92

        self.SetRoot(self.PlugPath())

    93

        self.SaveProject()

    94

        return True

    95

    96

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

    97

        """

    98

        Generate C code

    99

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

   100

        @param locations: List of complete variables locations \

   101

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

   102

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

   103

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

   104

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

   105

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

   106

            }, ...]

   107

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

   108

        """

   109

        current_location = self.GetCurrentLocation()

   110

        # define a unique name for the generated C file

   111

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

   112

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

   113

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

   114

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

   115

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

   116

        if res :

   117

            raise Exception, res

   118

   119

        file = open(os.path.join(buildpath, "MasterGenerated.od"), "w")

   120

        dump(master, file)

   121

        file.close()

   122

   123

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

   124

   125

class RootClass:

   126

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

   127

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

   128

      <xsd:element name="CanFestivalInstance">

   129

        <xsd:complexType>

   130

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

   131

        </xsd:complexType>

   132

      </xsd:element>

   133

    </xsd:schema>

   134

    """

   135

    PlugChildsTypes = [("CanOpenNode",_NodeListPlug, "CanOpen node")]

   136

    def GetParamsAttributes(self, path = None):

   137

        infos = PlugTemplate.GetParamsAttributes(self, path = None)

   138

        for element in infos:

   139

            if element["name"] == "CanFestivalInstance":                         

   140

                for child in element["children"]:

   141

                    if child["name"] == "CAN_Driver":

   142

                        DLL_LIST= getattr(canfestival_config,"DLL_LIST",None)

   143

                        if DLL_LIST is not None:

   144

                            child["type"] = DLL_LIST

   145

                        return infos    

   146

        return infos

   147

   148

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

   149

   150

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

   151

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

   152

                       "nodes_includes" : "",

   153

                       "board_decls" : "",

   154

                       "nodes_declare" : "",

   155

                       "nodes_init" : "",

   156

                       "nodes_open" : "",

   157

                       "nodes_close" : "",

   158

                       "nodes_send_sync" : "",

   159

                       "nodes_proceed_sync" : ""}

   160

        for child in self.IECSortedChilds():

   161

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

   162

            nodename = "OD_%s" % childlocstr

   163

   164

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

   165

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

   166

                   nodename,

   167

                   child.CanFestivalNode.getCAN_Device(),

   168

                   child.CanFestivalNode.getCAN_Baudrate())

   169

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

   170

                   nodename,

   171

                   child.CanFestivalNode.getNodeId())

   172

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

   173

                   nodename,

   174

                   child.CanFestivalNode.getNodeId())

   175

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

   176

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

   177

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

   178

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

   179

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

   180

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

   181

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

   182

        f = open(cf_main_path,'w')

   183

        f.write(cf_main)

   184

        f.close()

   185

   186

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

   187

   188