#!/usr/bin/env python

# -*- coding: utf-8 -*-

from __future__ import absolute_import

import os

from ConfigTreeNode import ConfigTreeNode

from canfestival import RootClass as CanOpenRootClass

from canfestival.canfestival import _SlaveCTN, _NodeListCTN, NodeManager

from canfestival.NetworkEditor import NetworkEditor

from canfestival.SlaveEditor import SlaveEditor

from canfestival import canfestival

DEFAULT_SETTINGS = {

"CAN_Baudrate": "125K",

"Slave_NodeId": 2,

"Master_NodeId": 1,

}

class LPCSlaveEditor(SlaveEditor):

# SHOW_BASE_PARAMS = False

pass

class LPCCanOpenSlave(_SlaveCTN):

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

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

<xsd:element name="CanFestivalSlaveNode">

<xsd:complexType>

<xsd:attribute name="CAN_Baudrate" type="xsd:string" use="optional" default="%(CAN_Baudrate)s"/>

<xsd:attribute name="NodeId" type="xsd:integer" use="optional" default="%(Slave_NodeId)d"/>

<xsd:attribute name="Sync_Align" type="xsd:integer" use="optional" default="0"/>

<xsd:attribute name="Sync_Align_Ratio" use="optional" default="50">

<xsd:simpleType>

<xsd:restriction base="xsd:integer">

<xsd:minInclusive value="1"/>

<xsd:maxInclusive value="99"/>

</xsd:restriction>

</xsd:simpleType>

</xsd:attribute>

</xsd:complexType>

</xsd:element>

</xsd:schema>

""" % DEFAULT_SETTINGS

EditorType = LPCSlaveEditor

def __init__(self):

# TODO change netname when name change

NodeManager.__init__(self)

odfilepath = self.GetSlaveODPath()

if (os.path.isfile(odfilepath)):

self.OpenFileInCurrent(odfilepath)

else:

self.CreateNewNode("SlaveNode", # Name - will be changed at build time

0x00, # NodeID - will be changed at build time

"slave", # Type

"", # description

"None", # profile

"", # prfile filepath

"heartbeat", # NMT

[]) # options

self.OnCTNSave()

def GetCanDevice(self):

return str(self.BaseParams.getIEC_Channel())

ConfNodeMethods = [

{"bitmap": "NetworkEdit",

"name": _("Edit slave"),

"tooltip": _("Edit CanOpen slave with ObjdictEdit"),

"method": "_OpenView"},

] + _SlaveCTN.ConfNodeMethods

class LPCNetworkEditor(NetworkEditor):

# SHOW_BASE_PARAMS = False

pass

class LPCCanOpenMaster(_NodeListCTN):

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

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

<xsd:element name="CanFestivalNode">

<xsd:complexType>

<xsd:attribute name="CAN_Baudrate" type="xsd:string" use="optional" default="%(CAN_Baudrate)s"/>

<xsd:attribute name="NodeId" type="xsd:integer" use="optional" default="%(Master_NodeId)d"/>

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

</xsd:complexType>

</xsd:element>

</xsd:schema>

""" % DEFAULT_SETTINGS

EditorType = LPCNetworkEditor

def GetCanDevice(self):

return str(self.BaseParams.getIEC_Channel())

ConfNodeMethods = [

{"bitmap": "NetworkEdit",

"name": _("Edit network"),

"tooltip": _("Edit CanOpen Network with NetworkEdit"),

"method": "_OpenView"},

] + _NodeListCTN.ConfNodeMethods

class LPCCanOpen(CanOpenRootClass):

XSD = None

CTNChildrenTypes = [("CanOpenNode", LPCCanOpenMaster, "CanOpen Master"),

("CanOpenSlave", LPCCanOpenSlave, "CanOpen Slave")]

def GetCanDriver(self):

return None

def LoadChildren(self):

ConfigTreeNode.LoadChildren(self)

if self.GetChildByName("Master") is None:

master = self.CTNAddChild("Master", "CanOpenNode", 0)

# master.BaseParams.setEnabled(False)

master.CTNRequestSave()

if self.GetChildByName("Slave") is None:

slave = self.CTNAddChild("Slave", "CanOpenSlave", 1)

# slave.BaseParams.setEnabled(False)

slave.CTNRequestSave()