# This file is part of Sympathy for Data.
# Copyright (c) 2013, 2017, Combine Control Systems AB
#
# Sympathy for Data is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3 of the License.
#
# Sympathy for Data is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with Sympathy for Data. If not, see <http://www.gnu.org/licenses/>.
from sympathy.api import qt2 as qt_compat
from sympathy.api import node as synode
from sympathy.api import node_helper
from sympathy.api.nodeconfig import Port, Ports, Tag, Tags, adjust
from sympathy.api.exceptions import SyDataError, SyConfigurationError
QtGui = qt_compat.import_module('QtGui')
QtWidgets = qt_compat.import_module('QtWidgets')
COMMON_DOCS = """
The target container in the ADAF is specified in the configuration GUI. If the
timeseries container is chosen it is necessary to specify the column in the
Table which will be the time basis signal in the ADAF. You can also specify
the name of the system and raster containers.
See also :ref:`working_with_adafs` for tips about how to use these conversion
nodes.
"""
def write_table_timeseries_to_adaf(system_name, raster_name, tb_column,
tabledata, adaffile):
if system_name == '':
raise SyConfigurationError('System name can not be left blank.')
if raster_name == '':
raise SyConfigurationError('Raster name can not be left blank.')
if tb_column in tabledata:
tb_group = adaffile.sys
if system_name in tb_group:
system = tb_group[system_name]
else:
system = tb_group.create(system_name)
if raster_name in system:
raster = system[raster_name]
else:
raster = system.create(raster_name)
# Move the table into the raster and remove tb_column from raster
raster.from_table(tabledata, tb_column)
else:
raise SyDataError('The selected time basis column does not exist in '
'the incoming Table')
def write_tabledata_to_adaf(export_to_meta, tablefile, adaffile):
if export_to_meta:
adaffile.meta.from_table(tablefile)
else:
adaffile.res.from_table(tablefile)
def get_parameters(combo_editors=False):
if combo_editors:
editor = synode.editors.combo_editor(edit=True)
else:
editor = None
parameters = synode.parameters()
parameters.set_string(
'export_to_group', label='Target group', value='Meta',
description=('Choose a container in the ADAF as target'),
editor=synode.editors.combo_editor(
options=['Meta', 'Result', 'Time series']))
parameters.set_string(
'system', label='System name',
description='Specify name of the created system in the ADAF',
editor=editor)
parameters.set_string(
'raster', label='Raster name',
description='Specify name of the created raster in the ADAF',
editor=editor)
parameters.set_string(
'tb',
label="Time basis column",
description=('Select a column in the Table which will '
'become the time basis of the new raster'),
editor=synode.editors.combo_editor(filter=True, edit=True))
return parameters
[docs]class Table2ADAF(synode.Node):
__doc__ = COMMON_DOCS
author = "Alexander Busck"
name = 'Table to ADAF'
description = ('Convert a Table into an ADAF, placing its content '
'in the specified container.')
nodeid = 'org.sysess.sympathy.data.table.table2adaf'
icon = 'import_table.svg'
tags = Tags(Tag.DataProcessing.Convert)
related = ['org.sysess.sympathy.data.table.tables2adafs',
'org.sysess.sympathy.data.table.updateadafwithtable',
'org.sysess.sympathy.data.adaf.adaf2table']
parameters = get_parameters()
controllers = synode.controller(
when=synode.field('export_to_group', 'value', value='Time series'),
action=(synode.field('system', 'enabled'),
synode.field('raster', 'enabled'),
synode.field('tb', 'enabled'))
)
inputs = Ports([Port.Table('Input Table', name='port1')])
outputs = Ports([Port.ADAF('ADAF with data in input Table', name='port1')])
def adjust_parameters(self, node_context):
adjust(node_context.parameters['tb'],
node_context.input['port1'])
def execute(self, node_context):
parameters = node_context.parameters
group_name = parameters['export_to_group'].value
tb_column = parameters['tb'].value
system_name = parameters['system'].value
raster_name = parameters['raster'].value
export_to = group_name.lower()
tablefile = node_context.input['port1']
adaffile = node_context.output['port1']
if export_to in ('meta', 'result'):
write_tabledata_to_adaf(export_to == 'meta', tablefile,
adaffile)
else:
write_table_timeseries_to_adaf(system_name, raster_name,
tb_column, tablefile,
adaffile)
[docs]@node_helper.list_node_decorator(['port1'], ['port1'])
class Tables2ADAFs(Table2ADAF):
name = 'Tables to ADAFs'
nodeid = 'org.sysess.sympathy.data.table.tables2adafs'
[docs]class UpdateADAFWithTable(Table2ADAF):
__doc__ = COMMON_DOCS
author = "Erik der Hagopian"
name = 'Update ADAF with Table'
description = ('Replace specified container in an ADAF '
'with the content of a Table.')
nodeid = 'org.sysess.sympathy.data.table.updateadafwithtable'
icon = 'import_table.svg'
tags = Tags(Tag.DataProcessing.Convert)
related = ['org.sysess.sympathy.data.table.updateadafswithtables',
'org.sysess.sympathy.data.table.table2adaf',
'org.sysess.sympathy.data.adaf.adaf2table']
parameters = get_parameters(combo_editors=True)
inputs = Ports([Port.Table('Input Table', name='port1'),
Port.ADAF('Input ADAF', name='port2')])
outputs = Ports([Port.ADAF(
'ADAF updated with data in input Table', name='port1')])
def _systems_rasters(self, port):
if not port.is_valid():
return [], []
systems = list(port.sys.keys())
rasters = []
for system in port.sys.values():
rasters.extend(system.keys())
return systems, rasters
def adjust_parameters(self, node_context):
adjust(node_context.parameters['tb'],
node_context.input['port1'])
systems, rasters = self._systems_rasters(
node_context.input['port2'])
node_context.parameters['system'].adjust(systems)
node_context.parameters['raster'].adjust(rasters)
def execute(self, node_context):
node_context.output['port1'].source(node_context.input['port2'])
super().execute(node_context)
[docs]@node_helper.list_node_decorator(['port1', 'port2'], ['port1'])
class UpdateADAFsWithTables(UpdateADAFWithTable):
name = 'Update ADAFs with Tables'
nodeid = 'org.sysess.sympathy.data.table.updateadafswithtables'
def _systems_rasters(self, port):
if not port.is_valid() or not len(port):
return [], []
return super()._systems_rasters(port[0])
