-
Notifications
You must be signed in to change notification settings - Fork 1
/
ObjectMergeNode.ts
38 lines (34 loc) · 1.34 KB
/
ObjectMergeNode.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import { DataFrame, DataObject } from '../../data';
import { PushOptions } from '../../graph/options';
import { ObjectProcessingNodeOptions } from '../ObjectProcessingNode';
import { FrameMergeNode } from './FrameMergeNode';
import { MergeShapeOptions } from './MergeShape';
/**
* @category Flow shape
*/
export class ObjectMergeNode<InOut extends DataFrame> extends FrameMergeNode<InOut> {
protected options: ObjectMergeOptions;
constructor(groupFn: (frame: InOut, options?: PushOptions) => any, options?: ObjectMergeOptions) {
super(
(frame: InOut) =>
frame
.getObjects()
.filter((value: DataObject) => this.options.objectFilter(value, frame))
.map((object) => object.uid),
groupFn,
options,
);
this.options.objectFilter = this.options.objectFilter || (() => true);
}
public merge(frames: InOut[]): InOut {
const mergedFrame = super.merge(frames);
mergedFrame
.getObjects()
.filter((object) => !this.options.objectFilter(object, mergedFrame))
.forEach((obj) => {
mergedFrame.removeObject(obj);
});
return mergedFrame;
}
}
export interface ObjectMergeOptions extends MergeShapeOptions, ObjectProcessingNodeOptions {}