-
Notifications
You must be signed in to change notification settings - Fork 6
/
union.js
91 lines (74 loc) · 1.68 KB
/
union.js
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
import joint from 'jointjs';
import DefaultShape from '../../../../core/graph/DefaultShape';
import NodeTemplate from '../../../../core/graph/NodeTemplate';
import * as config from '../../config';
const NAME = 'Union';
const NODE_TYPE = 'spark.union';
const MODEL = DefaultShape.extend({
defaults: joint.util.deepSupplement({
type: NODE_TYPE,
attrs: {
text : { text: NAME },
rect : {
fill: config.RDD_NODES_FILL
}
},
dfGui: {
description: NAME,
},
ports: {
items: [
{
id: 'in1',
group: 'in'
},
{
id: 'in2',
group: 'in'
},
{
id: 'out',
group: 'out'
}
],
}
}, DefaultShape.prototype.defaults)
});
if(!joint.shapes['spark']) joint.shapes['spark'] = {};
joint.shapes['spark']['union'] = MODEL;
export default class Union extends NodeTemplate{
static getType(){
return NODE_TYPE;
}
static getName(){
return NAME;
}
static getModel(){
return MODEL.bind(joint);
}
static isNodeHidden(){
return false;
}
static generateCode(parameters, lang, prevNodes){
let output = this.getCodePrefix(lang);
for(let [index, node] of prevNodes.entries()){
output += node.variable + (index < prevNodes.length - 1 ? ', ' : '');
}
return 'sc.' + output + this.getCodeSuffix(lang);
}
static getCodePrefix(lang){
return "union([";
}
static getCodeSuffix(lang){
return "])";
}
static getCodeParameters(lang){
return [];
}
static getOutputDataType(langId){
return 'rdd';
}
static isInputDataTypeValid(dataType, langId){
return dataType == 'rdd';
}
}