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

     5

from nodelist import NodeList

     6

from nodemanager import NodeManager

     7

import config_utils, gen_cfile

     8

from networkedit import networkedit

     9

    10

class _NetworkEdit(networkedit):

    11

    " Overload some of CanFestival Network Editor methods "

    12

    def OnCloseFrame(self, event):

    13

        " Do reset _NodeListPlug.View when closed"

    14

        self._onclose()

    15

        event.Skip()

    16

    17

class _NodeListPlug(NodeList):

    18

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

    19

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

    20

      <xsd:element name="CanFestivalNode">

    21

        <xsd:complexType>

    22

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

    23

        </xsd:complexType>

    24

      </xsd:element>

    25

    </xsd:schema>

    26

    """

    27

    28

    def __init__(self):

    29

        manager = NodeManager()

    30

        NodeList.__init__(self, manager)

    31

        self.LoadProject(self.PlugPath())

    32

    33

    _View = None

    34

    def _OpenView(self, logger):

    35

        if not self._View:

    36

            def _onclose():

    37

                self.View = None

    38

            self._View = _NetworkEdit()

    39

            self._View._onclose = _onclose

    40

        return self.View

    41

    PluginMethods = [("NetworkEdit",_OpenView)]

    42

    43

    def OnPlugClose(self):

    44

        if self._View:

    45

            self._View.Close()

    46

    47

    def PlugTestModified(self):

    48

        return self.HasChanged()

    49

    50

    def OnPlugSave(self):

    51

        self.SaveProject()

    52

        return True

    53

    54

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

    55

        """

    56

        Generate C code

    57

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

    58

        @param locations: List of complete variables locations \

    59

            [(IEC_loc, IEC_Direction IEC_Type, Name)]\

    60

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

    61

        """

    62

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

    63

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

    64

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

    65

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

    66

        if res :

    67

            raise Exception, res

    68

    69

        return [(Gen_OD_path,CanFestival_OD_CFLAGS)],""

    70

    71

class RootClass:

    72

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

    73

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

    74

      <xsd:element name="CanFestivalInstance">

    75

        <xsd:complexType>

    76

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

    77

        </xsd:complexType>

    78

      </xsd:element>

    79

    </xsd:schema>

    80

    """

    81

    82

    PlugChildsTypes = [("CanOpenNode",_NodeListPlug)]

    83

    84

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

    85

        """

    86

        Generate C code

    87

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

    88

        @param locations: List of complete variables locations \

    89

            [(IEC_loc, IEC_Direction IEC_Type, Name)]\

    90

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

    91

        """

    92

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

    93

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

    94

        master = config_utils.GenerateConciseDCF(locations, self)

    95

        res = gen_cfile.GenerateFile(Gen_OD_path, master)

    96

        if not res:

    97

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

    98

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

    99

             f = file(filepath, 'a')

   100

             f.write(s)

   101

        else:

   102

             pass # error

   103

   104

        return [],""

   105

   106