46 lines
1.4 KiB
TypeScript
46 lines
1.4 KiB
TypeScript
|
import type {ColumnDataSource} from "@bokehjs/models/sources/column_data_source"
|
||
|
|
import {isNumber} from "@bokehjs/core/util/types"
|
||
|
|
|
||
|
|
export function transform_cds_to_records(cds: ColumnDataSource, addId: boolean = false, start: number = 0): any {
|
||
|
|
const data: any = []
|
||
|
|
const columns = cds.columns()
|
||
|
|
const cdsLength = cds.get_length()
|
||
|
|
if (columns.length === 0 || cdsLength === null) {
|
||
|
|
return []
|
||
|
|
}
|
||
|
|
|
||
|
|
for (let i = start; i < cdsLength; i++) {
|
||
|
|
const item: any = {}
|
||
|
|
for (const column of columns) {
|
||
|
|
const array: any = cds.get_array(column)
|
||
|
|
const shape = (array[0] == null || array[0].shape == null) ? null : array[0].shape
|
||
|
|
if (shape != null && shape.length > 1 && isNumber(shape[0])) {
|
||
|
|
item[column] = array.slice(i*shape[1], i*shape[1]+shape[1])
|
||
|
|
} else if (array.length != cdsLength && (array.length % cdsLength === 0)) {
|
||
|
|
const offset = array.length / cdsLength
|
||
|
|
item[column] = array.slice(i*offset, i*offset+offset)
|
||
|
|
} else {
|
||
|
|
item[column] = array[i]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (addId) {
|
||
|
|
item._index = i
|
||
|
|
}
|
||
|
|
data.push(item)
|
||
|
|
}
|
||
|
|
return data
|
||
|
|
}
|
||
|
|
|
||
|
|
export function dict_to_records(data: any, index: boolean=true): any[] {
|
||
|
|
const records: any[] = []
|
||
|
|
for (let i = 0; i < data.index.length; i++) {
|
||
|
|
const record: any = {}
|
||
|
|
for (const col of data) {
|
||
|
|
if (index || col !== "index") {
|
||
|
|
record[col] = data[col][i]
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return records
|
||
|
|
}
|