andrej@1511: #!/usr/bin/env python
andrej@1511: # -*- coding: utf-8 -*-
andrej@1511:
andrej@1511: # This file is part of Beremiz, a Integrated Development Environment for
andrej@1511: # programming IEC 61131-3 automates supporting plcopen standard and CanFestival.
andrej@1511: #
andrej@1511: # Copyright (C) 2007: Edouard TISSERANT and Laurent BESSARD
andrej@1680: # Copyright (C) 2017: Andrey Skvortsov
andrej@1511: #
andrej@1511: # See COPYING file for copyrights details.
andrej@1511: #
andrej@1511: # This program is free software; you can redistribute it and/or
andrej@1511: # modify it under the terms of the GNU General Public License
andrej@1511: # as published by the Free Software Foundation; either version 2
andrej@1511: # of the License, or (at your option) any later version.
andrej@1511: #
andrej@1511: # This program is distributed in the hope that it will be useful,
andrej@1511: # but WITHOUT ANY WARRANTY; without even the implied warranty of
andrej@1511: # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
andrej@1511: # GNU General Public License for more details.
andrej@1511: #
andrej@1511: # You should have received a copy of the GNU General Public License
andrej@1511: # along with this program; if not, write to the Free Software
andrej@1511: # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
andrej@1511:
andrej@1853:
kinsamanka@3750:
laurent@366: import os
Edouard@728: from POULibrary import POULibrary
andrej@1853: from py_ext.PythonFileCTNMixin import PythonFileCTNMixin
andrej@1680: import util.paths as paths
laurent@366:
Edouard@3707: pyext_python_lib_code = """
Edouard@3707:
Edouard@3707: import csv
Edouard@3707: from collections import OrderedDict
Edouard@3707:
Edouard@3707: csv_int_files = {}
Edouard@3707: def CSVRdInt(fname, rowidx, colidx):
Edouard@3707: \"\"\"
Edouard@3707: Return value at row/column pointed by integer indexes
Edouard@3707: Assumes data starts at first row and first column, no headers.
Edouard@3707: \"\"\"
Edouard@3707: global csv_int_files
Edouard@3707: data = csv_int_files.get(fname, None)
Edouard@3707: if data is None:
Edouard@3707: data = list()
Edouard@3707: try:
Edouard@3707: csvfile = open(fname, 'rb')
Edouard@3707: except IOError:
Edouard@3707: return "#FILE_NOT_FOUND"
Edouard@3707: try:
Edouard@3707: dialect = csv.Sniffer().sniff(csvfile.read(1024))
Edouard@3707: csvfile.seek(0)
Edouard@3707: reader = csv.reader(csvfile, dialect)
Edouard@3707: for row in reader:
Edouard@3707: data.append(row)
Edouard@3707: except csv.Error:
Edouard@3707: return "#CSV_ERROR"
Edouard@3707: finally:
Edouard@3707: csvfile.close()
Edouard@3707: csv_int_files[fname] = data
Edouard@3707:
Edouard@3707: try:
Edouard@3707: row = data[rowidx]
Edouard@3707: except IndexError:
Edouard@3707: return "#ROW_NOT_FOUND"
Edouard@3707:
Edouard@3707: try:
Edouard@3707: return row[colidx]
Edouard@3707: except IndexError:
Edouard@3707: return "#COL_NOT_FOUND"
Edouard@3707:
Edouard@3707:
Edouard@3707: csv_str_files = {}
Edouard@3707: def CSVRdStr(fname, rowname, colname):
Edouard@3707: \"\"\"
Edouard@3707: Return value at row/column pointed by a pair of names as string
Edouard@3707: Assumes first row is column headers and first column is row name.
Edouard@3707: \"\"\"
Edouard@3707: global csv_str_files
Edouard@3707: entry = csv_str_files.get(fname, None)
Edouard@3707: if entry is None:
Edouard@3707: data = dict()
Edouard@3707: try:
Edouard@3707: csvfile = open(fname, 'rb')
Edouard@3707: except IOError:
Edouard@3707: return "#FILE_NOT_FOUND"
Edouard@3707: try:
Edouard@3707: dialect = csv.Sniffer().sniff(csvfile.read(1024))
Edouard@3707: csvfile.seek(0)
Edouard@3707: reader = csv.reader(csvfile, dialect)
Edouard@3707: headers = dict([(name, index) for index, name in enumerate(reader.next()[1:])])
Edouard@3707: for row in reader:
Edouard@3707: data[row[0]] = row[1:]
Edouard@3707: except csv.Error:
Edouard@3707: return "#CSV_ERROR"
Edouard@3707: finally:
Edouard@3707: csvfile.close()
Edouard@3707: csv_str_files[fname] = (headers, data)
Edouard@3707: else:
Edouard@3707: headers, data = entry
Edouard@3707:
Edouard@3707: try:
Edouard@3707: row = data[rowname]
Edouard@3707: except KeyError:
Edouard@3707: return "#ROW_NOT_FOUND"
Edouard@3707:
Edouard@3707: try:
Edouard@3707: colidx = headers[colname]
Edouard@3707: except KeyError:
Edouard@3707: return "#COL_NOT_FOUND"
Edouard@3707:
Edouard@3707: try:
Edouard@3707: return row[colidx]
Edouard@3707: except IndexError:
Edouard@3707: return "#COL_NOT_FOUND"
Edouard@3707:
Edouard@3707: def pyext_csv_reload():
Edouard@3707: global csv_int_files, csv_str_files
Edouard@3707: csv_int_files.clear()
Edouard@3707: csv_str_files.clear()
Edouard@3707:
Edouard@3707: """
andrej@1736:
Edouard@728: class PythonLibrary(POULibrary):
Edouard@728: def GetLibraryPath(self):
andrej@1730: return paths.AbsNeighbourFile(__file__, "pous.xml")
laurent@366:
Edouard@728: def Generate_C(self, buildpath, varlist, IECCFLAGS):
andrej@1730:
andrej@1680: plc_python_filepath = paths.AbsNeighbourFile(__file__, "plc_python.c")
laurent@366: plc_python_file = open(plc_python_filepath, 'r')
laurent@366: plc_python_code = plc_python_file.read()
laurent@366: plc_python_file.close()
laurent@366: python_eval_fb_list = []
Edouard@728: for v in varlist:
Edouard@1132: if v["vartype"] == "FB" and v["type"] in ["PYTHON_EVAL",
Edouard@1132: "PYTHON_POLL"]:
laurent@366: python_eval_fb_list.append(v)
laurent@366: python_eval_fb_count = max(1, len(python_eval_fb_list))
andrej@1730:
laurent@366: # prepare python code
Edouard@1132: plc_python_code = plc_python_code % {
andrej@1746: "python_eval_fb_count": python_eval_fb_count}
andrej@1730:
Edouard@728: Gen_Pythonfile_path = os.path.join(buildpath, "py_ext.c")
andrej@1740: pythonfile = open(Gen_Pythonfile_path, 'w')
laurent@366: pythonfile.write(plc_python_code)
laurent@366: pythonfile.close()
andrej@1730:
Edouard@3707: runtimefile_path = os.path.join(buildpath, "runtime_00_pyext.py")
Edouard@3707: runtimefile = open(runtimefile_path, 'w')
Edouard@3707: runtimefile.write(pyext_python_lib_code)
Edouard@3707: runtimefile.close()
Edouard@3707: return ((["py_ext"], [(Gen_Pythonfile_path, IECCFLAGS)], True), "",
Edouard@3707: ("runtime_00_pyext.py", open(runtimefile_path, "rb")))
Edouard@3707:
Edouard@728:
andrej@1736:
Edouard@728: class PythonFile(PythonFileCTNMixin):
andrej@1730:
laurent@781: def GetIconName(self):
laurent@781: return "Pyfile"