def merge_period(period_idx, period):
if ent_name in ent_names1:
start, stop = input1_rows.get(period, (0, 0))
input1_array = table1.read(start, stop)
else:
input1_array = None
if ent_name in ent_names2:
start, stop = input2_rows.get(period, (0, 0))
input2_array = table2.read(start, stop)
else:
input2_array = None
if ent_name in ent_names1 and ent_name in ent_names2:
if 'id' in input1_array.dtype.names:
assert 'id' in input2_array.dtype.names
output_array, _ = merge_arrays(input1_array, input2_array)
else:
output_array = merge_array_records(input1_array,
input2_array)
elif ent_name in ent_names1:
output_array = input1_array
elif ent_name in ent_names2:
output_array = input2_array
else:
raise Exception("this shouldn't have happened")
output_table.append(output_array)
output_table.flush()
def load_period_data(self, period):
if self.lag_fields:
#TODO: use ColumnArray here
#XXX: do we need np.empty? (but watch for alias problems)
self.array_lag = np.empty(len(self.array),
dtype=np.dtype(self.lag_fields))
for field, _ in self.lag_fields:
self.array_lag[field] = self.array[field]
rows = self.input_rows.get(period)
if rows is None:
# nothing needs to be done in that case
return
start, stop = rows
# It would be nice to use ColumnArray.from_table and adapt merge_arrays
# to produce a ColumnArray in all cases, but it is not a huge priority
# for now
input_array = self.input_table.read(start, stop)
self.array, self.id_to_rownum = \
merge_arrays(self.array, input_array, result_fields='array1')
# this can happen, depending on the layout of columns in input_array,
# but the usual case (in retro) is that self.array is a superset of
# input_array, in which case merge_arrays returns a ColumnArray
if not isinstance(self.array, ColumnArray):
self.array = ColumnArray(self.array)
def load_period_data(self, period):
if self.lag_fields:
# TODO: use ColumnArray here
# XXX: do we need np.empty? (but watch for alias problems)
self.array_lag = np.empty(len(self.array),
dtype=np.dtype(self.lag_fields))
for field, _ in self.lag_fields:
self.array_lag[field] = self.array[field]
# if not self.indexed_input_table.has_period(period):
# # nothing needs to be done in that case
# return
#
# input_array = self.indexed_input_table.read(period)
rows = self.input_rows.get(period)
if rows is None:
# nothing needs to be done in that case
return
start, stop = rows
# It would be nice to use ColumnArray.from_table and adapt merge_arrays
# to produce a ColumnArray in all cases, but it is not a huge priority
# for now
input_array = self.input_table.read(start, stop)
self.array, self.id_to_rownum = \
merge_arrays(self.array, input_array, result_fields='array1',
default_values=self.fields.default_values)
# this can happen, depending on the layout of columns in input_array,
# but the usual case (in retro) is that self.array is a superset of
# input_array, in which case merge_arrays returns a ColumnArray
if not isinstance(self.array, ColumnArray):
self.array = ColumnArray(self.array)
