import os

from LPCArch import GetLPCArch, GetLPCProduct, SOM28_modules, GetLPCProductDesc

import util.paths as paths

from plcopen.structures import LOCATIONDATATYPES

from PLCControler import LOCATION_CONFNODE, LOCATION_MODULE, LOCATION_GROUP,\

LOCATION_VAR_INPUT, LOCATION_VAR_OUTPUT, LOCATION_VAR_MEMORY

product = GetLPCProduct()

product_desc = GetLPCProductDesc()

modpath = os.path.split(__file__)[0]

LPCBusSourcePath = os.path.join(modpath,"LPCBus")

def GetLocalCode(fname):

return open(os.path.join(LPCBusSourcePath,fname)).read()

LOCATION_TYPES = {"I": LOCATION_VAR_INPUT,

"Q": LOCATION_VAR_OUTPUT,

"M": LOCATION_VAR_MEMORY}

LOCATION_DIRS = dict([(dir, size) for size, dir in LOCATION_TYPES.items()])

LOCATION_SIZES = {}

for size, types in LOCATIONDATATYPES.items():

for _type in types:

LOCATION_SIZES[_type] = size

def GetModuleChildren(module):

children = []

for child in module["children"]:

if child["type"] == LOCATION_GROUP:

children.extend(child["children"])

else:

children.append(child)

return children

def _GetVariables(module):

variables = []

for child in module["children"]:

if child["type"] in [LOCATION_GROUP, LOCATION_MODULE]:

variables.extend(_GetVariables(child))

else:

variables.append(child)

return variables

def GetLastModuleGroup(module):

group = module

for child in module["children"]:

if child["type"] == LOCATION_GROUP:

group = child

return group["children"]

def GetModuleBySomething(module, something, toks):

for child in GetModuleChildren(module):

if child.get(something) == toks[0]:

if len(toks) > 1:

return GetModuleBySomething(child, something, toks[1:])

return child

return None

def GetModuleVariable(module, location, direction):

for child in GetModuleChildren(module):

if child["location"] == location and child["type"] == LOCATION_TYPES[direction]:

return child

return None

def RemoveModuleChild(module, child):

if child in module["children"]:

module["children"].remove(child)

else:

for group in module["children"]:

if group["type"] == LOCATION_GROUP and child in group["children"]:

group["children"].remove(child)

class LPCBus(object):

def __init__(self):

self.VariableLocationTree = []

self.ResetUsedLocations()

self.Icon = None

def __getitem__(self, key):

if key == "children":

return self.VariableLocationTree

raise KeyError("Only 'children' key is available")

def CTNEnabled(self):

return None

def SetIcon(self, icon):

self.Icon = icon

def _GetChildBySomething(self, something, toks):

return GetModuleBySomething({"children" : self.VariableLocationTree}, something, toks)

def GetBaseTypes(self):

return self.GetCTRoot().GetBaseTypes()

def GetSizeOfType(self, type):

return LOCATION_SIZES[self.GetCTRoot().GetBaseType(type)]

def _GetVariableLocationTree(self, current_location, infos):

if infos["type"] == LOCATION_MODULE:

location = current_location + (infos["IEC_Channel"],)

return {"name": infos["name"],

"type": infos["type"],

"location": ".".join(map(str, location + ("x",))),

"icon": infos["icon"],

"children": [self._GetVariableLocationTree(location, child) for child in infos["children"]]}

elif infos["type"] == LOCATION_GROUP:

return {"name": infos["name"],

"type": infos["type"],

"location": "",

"icon": infos["icon"],

"children": [self._GetVariableLocationTree(current_location, child) for child in infos["children"]]}

else:

size = self.GetSizeOfType(infos["IEC_type"])

location = "%" + LOCATION_DIRS[infos["type"]] + size + ".".join(map(str, current_location + infos["location"]))

return {"name": infos["name"],

"type": infos["type"],

"size": size,

"IEC_type": infos["IEC_type"],

"var_name": infos["name"],

"location": location,

"description": infos["description"],

"children": []}

def GetVariableLocationTree(self):

return {"name": self.BaseParams.getName(),

"type": LOCATION_CONFNODE,

"location": self.GetFullIEC_Channel(),

"icon": self.Icon,

"children": [self._GetVariableLocationTree(self.GetCurrentLocation(), child)

for child in self.VariableLocationTree]}

def CTNTestModified(self):

return False

def CTNMakeDir(self):

pass

def CTNRequestSave(self, from_project_path=None):

return None

def ResetUsedLocations(self):

self.UsedLocations = {}

def _AddUsedLocation(self, parent, location):

num = location.pop(0)

if num not in parent:

parent[num] = {"used": False, "children": {}}

if len(location) > 0:

self._AddUsedLocation(parent[num]["children"], location)

else:

parent[num]["used"] = True

def AddUsedLocation(self, location):

if len(location) > 0:

self._AddUsedLocation(self.UsedLocations, list(location))

def _CheckLocationConflicts(self, parent, location):

num = location.pop(0)

if num not in parent:

return False

if len(location) > 0:

if parent[num]["used"]:

return True

return self._CheckLocationConflicts(parent[num]["children"], location)

elif len(parent[num]["children"]) > 0:

return True

return False

def CheckLocationConflicts(self, location):

if len(location) > 0:

return self._CheckLocationConflicts(self.UsedLocations, list(location))

return False

def CTNGenerate_C(self, buildpath, locations):

"""

Generate C code

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

@param locations: List of complete variables locations \

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

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

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

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

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

}, ...]

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

"""

current_location = self.GetCurrentLocation()

# define a unique name for the generated C file

location_str = "_".join(map(str, current_location))

code_str = {"location_str": location_str,

"var_decl": "",

"declare_code": "",

"init_code": "",

"retrieve_code": "",

"publish_code": "",

}

BusName = self.BaseParams.getName()

bus_code = None

for bus_name, bus_template in product_desc["features"]:

if bus_name == BusName:

bus_code = { section :

header + GetLocalCode("%s_%s.c"%(bus_template,section))

for section, header in [

("decl",GetLocalCode("%s.h" % product_desc["header"])),

("init", ""),

("retrieve", ""),

("publish", ""),

("cleanup", "")]}

break

if bus_code is None:

# Silently ignored, for example MC9 has no Left Bus, but composer always adds one.

print(("No LPCBus named "+BusName+" for "+product))

return [], "", False

for module in GetModuleChildren(self):

if module["init"] != "":

code_str["init_code"] += " %s\n" % module["init"]

# Adding variables

vars = []

self.ResetUsedLocations()

for location in locations:

loc = location["LOC"][len(current_location):]

group = next = self

i = 0

while next is not None and i < len(loc):

next = self._GetChildBySomething("IEC_Channel", loc[:i + 1])

if next is not None:

i += 1

group = next

var_loc = loc[i:]

for variable in GetModuleChildren(group):

if variable["location"] == var_loc and location["DIR"] == LOCATION_DIRS[variable["type"]]:

# if location["DIR"] != LOCATION_DIRS[variable["type"]]:

# raise Exception, "Direction conflict in variable definition"

# if location["IEC_TYPE"] != variable["IEC_type"]:

# raise Exception, "Type conflict in variable definition"

if location["DIR"] == "Q":

if self.CheckLocationConflicts(location["LOC"]):

raise Exception("BYTE and BIT from the same BYTE can't be used together")

self.AddUsedLocation(location["LOC"])

vars.append({"location": location["NAME"],

"Type": variable["IEC_type"],

"Retrieve": variable["retrieve"],

"Publish": variable["publish"],

})

break

base_types = self.GetCTRoot().GetBaseTypes()

for var in vars:

prefix = ""

if var["Type"] in base_types:

prefix = "IEC_"

code_str["var_decl"] += "%s%s beremiz%s;\n"%(prefix, var["Type"], var["location"])

code_str["var_decl"] += "%s%s *%s = &beremiz%s;\n"%(prefix, var["Type"], var["location"], var["location"])

if var["Retrieve"] != "":

code_str["retrieve_code"] += " " + var["Retrieve"] % ("*" + var["location"]) + "\n"

if var["Publish"] != "":

code_str["publish_code"] += " " + var["Publish"] % ("*" + var["location"]) + "\n"

def bcode(section, default):

return bus_code.get(section, default) % code_str

code_str.update({

"bus_decl": bcode("decl", ""),

"bus_init_code": bcode("init", "%(init_code)s"),

"bus_cleanup_code": bcode("cleanup", ""),

"bus_retrieve_code": bcode("retrieve", "%(retrieve_code)s"),

"bus_publish_code": bcode("publish", "%(publish_code)s"),

})

Gen_Module_path = os.path.join(buildpath, "Bus_%s.c"%location_str)

module = open(Gen_Module_path,'w')

module.write(GetLocalCode(product_desc["type"]+".c") % code_str)

module.close()

cflags = '"-I%s" "-I%s" -Wno-unused-function'%(

os.path.abspath(self.GetCTRoot().GetIECLibPath()), # is it still necessary ?

LPCBusSourcePath

)

return [(Gen_Module_path, cflags)], product_desc.get("LDFLAGS", ""), True