# This file is part of Sympathy for Data.
# Copyright (c) 2013, Combine Control Systems AB
#
# SYMPATHY FOR DATA COMMERCIAL LICENSE
# You should have received a link to the License with Sympathy for Data.
import numpy as np
from sympathy.api import node as synode
from sympathy.api import node_helper
from sympathy.api.nodeconfig import Port, Ports, Tag, Tags
[docs]
class ADAFStructureTable(synode.Node):
"""
Takes the structure of an ADAF (Type, System, Raster, and Parameter) and
outputs it to a Table.
"""
author = 'Erik der Hagopian'
icon = 'adaf2adafs.svg'
name = 'ADAF structure to Table'
description = 'Create a Table from the structure of an ADAF.'
nodeid = 'org.sysess.sympathy.data.adaf.adafstructuretable'
tags = Tags(Tag.DataProcessing.Index)
related = [
'org.sysess.sympathy.data.adaf.adafstructuretable',
'org.sysess.sympathy.data.adaf.adafsstructuretables',
'org.sysess.sympathy.data.adaf.selectcolumnsadafwithtable',
]
inputs = Ports([Port.ADAF('Input ADAF', name='port1')])
outputs = Ports([Port.Table('ADAF structure as Table', name='port1')])
@staticmethod
def run(input_file, output_file):
def adaf_iterator(adaf_file):
info = adaf_file.info()
meta = info['meta']
yield ('Metadata', '', '', [col for col in meta['columns']])
res = info['res']
yield ('Result', '', '', [col for col in res['columns']])
for system_key, system in info['sys'].items():
for raster_key, raster in system.items():
yield ('Timeseries', system_key, raster_key,
[signal for signal in raster['signals']])
type_column = []
system_column = []
raster_column = []
parameter_column = []
index_column = []
output_info = [('Type', type_column, 'U'),
('System', system_column, 'U'),
('Raster', raster_column, 'U'),
('Parameter', parameter_column, 'U'),
('Index', index_column, 'i')]
for i, (type_key, system_key, raster_key,
parameter_keys) in enumerate(adaf_iterator(input_file)):
number_of_keys = len(parameter_keys)
type_column.extend([type_key] * number_of_keys)
system_column.extend([system_key] * number_of_keys)
raster_column.extend([raster_key] * number_of_keys)
parameter_column.extend(parameter_keys)
index_column.extend([i] * number_of_keys)
for name, column, kind in output_info:
output_file.set_column_from_array(
name, np.array(column, dtype=kind))
def execute(self, node_context):
input_file = node_context.input[0]
output_file = node_context.output[0]
self.run(input_file, output_file)
[docs]
@node_helper.list_node_decorator(['port1'], ['port1'])
class ADAFsStructureTables(ADAFStructureTable):
name = 'ADAFs structure to Tables'
nodeid = 'org.sysess.sympathy.data.adaf.adafsstructuretables'
