mirror of
https://bitbucket.org/wisemapping/wisemapping-frontend.git
synced 2024-12-28 14:13:49 +01:00
501 lines
20 KiB
JavaScript
501 lines
20 KiB
JavaScript
/* eslint-disable no-console */
|
|
/*
|
|
* Copyright [2021] [wisemapping]
|
|
*
|
|
* Licensed under WiseMapping Public License, Version 1.0 (the "License").
|
|
* It is basically the Apache License, Version 2.0 (the "License") plus the
|
|
* "powered by wisemapping" text requirement on every single page;
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the license at
|
|
*
|
|
* http://www.wisemapping.org/license
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
import $ from 'jquery';
|
|
import { $assert } from '@wisemapping/core-js';
|
|
import TestSuite from './TestSuite';
|
|
import LayoutManager from '../../../src/components/layout/LayoutManager';
|
|
|
|
class BalancedTestSuite extends TestSuite {
|
|
constructor() {
|
|
$('#balancedTest').css('display', 'block');
|
|
super();
|
|
this.testBalanced();
|
|
this.testBalancedPredict();
|
|
this.testBalancedNodeDragPredict();
|
|
}
|
|
|
|
testBalanced() {
|
|
console.log('testBalanced:');
|
|
const position = { x: 0, y: 0 };
|
|
const plotsize = { width: 1000, height: 200 };
|
|
const manager = new LayoutManager(0, TestSuite.ROOT_NODE_SIZE);
|
|
|
|
manager.addNode(1, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 1, 0);
|
|
manager.layout();
|
|
manager.plot('testBalanced1', plotsize);
|
|
|
|
manager.addNode(2, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 2, 1);
|
|
manager.layout();
|
|
manager.plot('testBalanced2', plotsize);
|
|
|
|
manager.addNode(3, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 3, 2);
|
|
manager.layout();
|
|
manager.plot('testBalanced3', plotsize);
|
|
|
|
manager.addNode(4, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 4, 3);
|
|
manager.layout();
|
|
manager.plot('testBalanced4', plotsize);
|
|
|
|
manager.addNode(5, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 5, 4);
|
|
manager.layout();
|
|
manager.plot('testBalanced5', plotsize);
|
|
|
|
manager.addNode(6, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 6, 5);
|
|
manager.layout();
|
|
manager.plot('testBalanced6', plotsize);
|
|
|
|
manager.addNode(7, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(8, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(9, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(3, 7, 0);
|
|
manager.connectNode(7, 8, 0);
|
|
manager.connectNode(7, 9, 1);
|
|
manager.layout();
|
|
manager.plot('testBalanced7', plotsize);
|
|
|
|
manager.addNode(10, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(11, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(12, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(6, 10, 0);
|
|
manager.connectNode(10, 11, 0);
|
|
manager.connectNode(10, 12, 1);
|
|
manager.layout();
|
|
manager.plot('testBalanced8', plotsize);
|
|
|
|
manager.addNode(13, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 13, 4);
|
|
manager.layout();
|
|
manager.plot('testBalanced9', { width: 1000, height: 400 });
|
|
|
|
// Check orders have shifted accordingly
|
|
$assert(manager.find(5).getOrder() == 6, 'Node 5 should have order 6');
|
|
|
|
manager.addNode(14, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 14, 5);
|
|
manager.layout();
|
|
manager.plot('testBalanced10', { width: 1000, height: 400 });
|
|
|
|
// Check orders have shifted accordingly
|
|
$assert(manager.find(6).getOrder() == 7, 'Node 6 should have order 7');
|
|
|
|
manager.addNode(15, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 15, 4);
|
|
manager.layout();
|
|
manager.plot('testBalanced11', { width: 1000, height: 400 });
|
|
|
|
// Check orders have shifted accordingly
|
|
$assert(manager.find(13).getOrder() == 6, 'Node 13 should have order 6');
|
|
$assert(manager.find(5).getOrder() == 8, 'Node 5 should have order 8');
|
|
|
|
manager.addNode(16, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 16, 25);
|
|
manager.layout();
|
|
manager.plot('testBalanced12', { width: 1000, height: 400 });
|
|
|
|
// Check orders have shifted accordingly
|
|
$assert(manager.find(16).getOrder() == 9, 'Node 16 should have order 9');
|
|
|
|
manager.addNode(17, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(18, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(19, TestSuite.NODE_SIZE, position);
|
|
manager.connectNode(0, 17, 11);
|
|
manager.connectNode(0, 18, 13);
|
|
manager.connectNode(0, 19, 10);
|
|
manager.layout();
|
|
manager.plot('testBalanced13', { width: 1000, height: 400 });
|
|
|
|
// Check that everything is ok
|
|
$assert(
|
|
manager.find(1).getPosition().x > manager.find(0).getPosition().x,
|
|
'even order nodes must be at right of central topic',
|
|
);
|
|
$assert(
|
|
manager.find(3).getPosition().x > manager.find(0).getPosition().x,
|
|
'even order nodes must be at right of central topic',
|
|
);
|
|
$assert(
|
|
manager.find(5).getPosition().x > manager.find(0).getPosition().x,
|
|
'even order nodes must be at right of central topic',
|
|
);
|
|
$assert(
|
|
manager.find(2).getPosition().x < manager.find(0).getPosition().x,
|
|
'odd order nodes must be at right of central topic',
|
|
);
|
|
$assert(
|
|
manager.find(4).getPosition().x < manager.find(0).getPosition().x,
|
|
'odd order nodes must be at right of central topic',
|
|
);
|
|
$assert(
|
|
manager.find(6).getPosition().x < manager.find(0).getPosition().x,
|
|
'odd order nodes must be at right of central topic',
|
|
);
|
|
$assert(
|
|
manager.find(7).getPosition().x > manager.find(3).getPosition().x,
|
|
'children of 1st level even order nodes must be to the right',
|
|
);
|
|
$assert(
|
|
manager.find(8).getPosition().x > manager.find(7).getPosition().x,
|
|
'children of 1st level even order nodes must be to the right',
|
|
);
|
|
$assert(
|
|
manager.find(9).getPosition().x > manager.find(7).getPosition().x,
|
|
'children of 1st level even order nodes must be to the right',
|
|
);
|
|
$assert(
|
|
manager.find(10).getPosition().x < manager.find(6).getPosition().x,
|
|
'children of 1st level odd order nodes must be to the left',
|
|
);
|
|
$assert(
|
|
manager.find(11).getPosition().x < manager.find(10).getPosition().x,
|
|
'children of 1st level odd order nodes must be to the left',
|
|
);
|
|
$assert(
|
|
manager.find(12).getPosition().x < manager.find(10).getPosition().x,
|
|
'children of 1st level odd order nodes must be to the left',
|
|
);
|
|
|
|
console.log('OK!\n\n');
|
|
}
|
|
|
|
testBalancedPredict() {
|
|
console.log('testBalancedPredict:');
|
|
const position = { x: 0, y: 0 };
|
|
const manager = new LayoutManager(0, TestSuite.ROOT_NODE_SIZE);
|
|
|
|
manager.addNode(1, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(2, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(3, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(4, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(5, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(7, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(8, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(9, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(10, TestSuite.NODE_SIZE, position);
|
|
manager.addNode(11, TestSuite.NODE_SIZE, position);
|
|
|
|
manager.connectNode(0, 1, 0);
|
|
manager.connectNode(0, 2, 1);
|
|
manager.connectNode(0, 3, 2);
|
|
manager.connectNode(0, 4, 3);
|
|
manager.connectNode(0, 5, 4);
|
|
manager.connectNode(4, 7, 0);
|
|
manager.connectNode(4, 8, 1);
|
|
manager.connectNode(8, 9, 0);
|
|
manager.connectNode(3, 10, 0);
|
|
manager.connectNode(3, 11, 1);
|
|
|
|
manager.layout();
|
|
|
|
// Graph 1
|
|
const graph1 = manager.plot('testBalancedPredict1', { width: 1000, height: 400 });
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (165, -70):');
|
|
const prediction1a = manager.predict(0, null, { x: 165, y: -70 });
|
|
this._plotPrediction(graph1, prediction1a);
|
|
$assert(
|
|
prediction1a.position.y < manager.find(1).getPosition().y
|
|
&& prediction1a.position.x == manager.find(1).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction1a.order == 0, 'Prediction order should be 0');
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (165, -10):');
|
|
const prediction1b = manager.predict(0, null, { x: 165, y: -10 });
|
|
this._plotPrediction(graph1, prediction1b);
|
|
$assert(
|
|
prediction1b.position.y > manager.find(1).getPosition().y
|
|
&& prediction1b.position.y < manager.find(3).getPosition().y
|
|
&& prediction1b.position.x == manager.find(1).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction1b.order == 2, 'Prediction order should be 2');
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (145, 15):');
|
|
const prediction1c = manager.predict(0, null, { x: 145, y: 15 });
|
|
this._plotPrediction(graph1, prediction1c);
|
|
$assert(
|
|
prediction1c.position.y > manager.find(3).getPosition().y
|
|
&& prediction1c.position.y < manager.find(5).getPosition().y
|
|
&& prediction1c.position.x == manager.find(3).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction1c.order == 4, 'Prediction order should be 4');
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (145, 70):');
|
|
const prediction1d = manager.predict(0, null, { x: 145, y: 70 });
|
|
this._plotPrediction(graph1, prediction1d);
|
|
$assert(
|
|
prediction1d.position.y > manager.find(5).getPosition().y
|
|
&& prediction1d.position.x == manager.find(5).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction1d.order == 6, 'Prediction order should be 6');
|
|
|
|
// Graph 2
|
|
const graph2 = manager.plot('testBalancedPredict2', { width: 1000, height: 400 });
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (-145, -50):');
|
|
const prediction2a = manager.predict(0, null, { x: -145, y: -50 });
|
|
this._plotPrediction(graph2, prediction2a);
|
|
$assert(
|
|
prediction2a.position.y < manager.find(2).getPosition().y
|
|
&& prediction2a.position.x == manager.find(2).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction2a.order == 1, 'Prediction order should be 1');
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (-145, -10):');
|
|
const prediction2b = manager.predict(0, null, { x: -145, y: -10 });
|
|
this._plotPrediction(graph2, prediction2b);
|
|
$assert(
|
|
prediction2b.position.y > manager.find(2).getPosition().y
|
|
&& prediction2b.position.y < manager.find(4).getPosition().y
|
|
&& prediction2b.position.x == manager.find(2).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction2b.order == 3, 'Prediction order should be 3');
|
|
|
|
console.log('\tAdded as child of node 0 and dropped at (-145, 40):');
|
|
const prediction2c = manager.predict(0, null, { x: -145, y: 400 });
|
|
this._plotPrediction(graph2, prediction2c);
|
|
$assert(
|
|
prediction2c.position.y > manager.find(4).getPosition().y
|
|
&& prediction2c.position.x == manager.find(4).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction2c.order == 5, 'Prediction order should be 5');
|
|
|
|
// Graph 3
|
|
console.log('\tPredict nodes added with no position:');
|
|
const graph3 = manager.plot('testBalancedPredict3', { width: 1000, height: 400 });
|
|
const prediction3 = manager.predict(0, null, null);
|
|
this._plotPrediction(graph3, prediction3);
|
|
$assert(
|
|
prediction3.position.y > manager.find(4).getPosition().y
|
|
&& prediction3.position.x == manager.find(4).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction3.order == 5, 'Prediction order should be 5');
|
|
|
|
console.log('\tPredict nodes added with no position:');
|
|
manager.addNode(6, TestSuite.NODE_SIZE, prediction3.position);
|
|
manager.connectNode(0, 6, prediction3.order);
|
|
manager.layout();
|
|
const graph4 = manager.plot('testBalancedPredict4', { width: 1000, height: 400 });
|
|
const prediction4 = manager.predict(0, null, null);
|
|
this._plotPrediction(graph4, prediction4);
|
|
$assert(
|
|
prediction4.position.y > manager.find(5).getPosition().y
|
|
&& prediction4.position.x == manager.find(5).getPosition().x,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction4.order == 6, 'Prediction order should be 6');
|
|
|
|
console.log('\tPredict nodes added only a root node:');
|
|
manager.removeNode(1).removeNode(2).removeNode(3).removeNode(4)
|
|
.removeNode(5);
|
|
manager.layout();
|
|
const graph5 = manager.plot('testBalancedPredict5', { width: 1000, height: 400 });
|
|
const prediction5a = manager.predict(0, null, null);
|
|
const prediction5b = manager.predict(0, null, { x: 40, y: 100 });
|
|
this._plotPrediction(graph5, prediction5a);
|
|
this._plotPrediction(graph5, prediction5b);
|
|
$assert(
|
|
prediction5a.position.x > manager.find(0).getPosition().x
|
|
&& prediction5a.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction5a.order == 0, 'Prediction order should be 0');
|
|
$assert(
|
|
prediction5a.position.x == prediction5b.position.x
|
|
&& prediction5a.position.y == prediction5b.position.y,
|
|
'Both predictions should be the same',
|
|
);
|
|
$assert(prediction5a.order == prediction5b.order, 'Both predictions should be the same');
|
|
|
|
console.log('OK!\n\n');
|
|
}
|
|
|
|
testBalancedNodeDragPredict() {
|
|
console.log('testBalancedNodeDragPredict:');
|
|
const position = { x: 0, y: 0 };
|
|
const manager = new LayoutManager(0, TestSuite.ROOT_NODE_SIZE);
|
|
|
|
// Graph 1
|
|
manager.addNode(1, TestSuite.NODE_SIZE, position).connectNode(0, 1, 0);
|
|
manager.layout();
|
|
const graph1 = manager.plot('testBalancedNodeDragPredict1', { width: 800, height: 400 });
|
|
|
|
const prediction1a = manager.predict(0, 1, { x: 50, y: 50 });
|
|
this._plotPrediction(graph1, prediction1a);
|
|
$assert(
|
|
prediction1a.position.x == manager.find(1).getPosition().x
|
|
&& prediction1a.position.y == manager.find(1).getPosition().y,
|
|
'Prediction position should be the same as node 1',
|
|
);
|
|
$assert(
|
|
prediction1a.order == manager.find(1).getOrder(),
|
|
'Prediction order should be the same as node 1',
|
|
);
|
|
|
|
const prediction1b = manager.predict(0, 1, { x: 50, y: -50 });
|
|
this._plotPrediction(graph1, prediction1b);
|
|
$assert(
|
|
prediction1b.position.x == manager.find(1).getPosition().x
|
|
&& prediction1b.position.y == manager.find(1).getPosition().y,
|
|
'Prediction position should be the same as node 1',
|
|
);
|
|
$assert(
|
|
prediction1b.order == manager.find(1).getOrder(),
|
|
'Prediction order should be the same as node 1',
|
|
);
|
|
|
|
const prediction1c = manager.predict(0, 1, { x: -50, y: 50 });
|
|
this._plotPrediction(graph1, prediction1c);
|
|
$assert(
|
|
prediction1c.position.x < manager.find(0).getPosition().x
|
|
&& prediction1c.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction1c.order == 1, 'Prediction order should be the same as node 1');
|
|
|
|
const prediction1d = manager.predict(0, 1, { x: -50, y: -50 });
|
|
this._plotPrediction(graph1, prediction1d);
|
|
$assert(
|
|
prediction1d.position.x < manager.find(0).getPosition().x
|
|
&& prediction1d.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction1d.order == 1, 'Prediction order should be the same as node 1');
|
|
|
|
// Graph 2
|
|
manager.disconnectNode(1);
|
|
manager.connectNode(0, 1, 1);
|
|
manager.layout();
|
|
const graph2 = manager.plot('testBalancedNodeDragPredict2', { width: 800, height: 400 });
|
|
|
|
const prediction2a = manager.predict(0, 1, { x: 50, y: 50 });
|
|
this._plotPrediction(graph2, prediction2a);
|
|
$assert(
|
|
prediction2a.position.x > manager.find(0).getPosition().x
|
|
&& prediction2a.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is positioned incorrectly',
|
|
);
|
|
$assert(prediction2a.order == 0, 'Prediction order should be 0');
|
|
|
|
const prediction2b = manager.predict(0, 1, { x: 50, y: -50 });
|
|
this._plotPrediction(graph2, prediction2b);
|
|
$assert(
|
|
prediction2b.position.x > manager.find(0).getPosition().x
|
|
&& prediction2b.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is positioned incorrectly',
|
|
);
|
|
$assert(prediction2b.order == 0, 'Prediction order should be 0');
|
|
|
|
const prediction2c = manager.predict(0, 1, { x: -50, y: 50 });
|
|
this._plotPrediction(graph2, prediction2c);
|
|
$assert(
|
|
prediction2c.position.x == manager.find(1).getPosition().x
|
|
&& prediction2c.position.y == manager.find(1).getPosition().y,
|
|
'Prediction position should be the same as node 1',
|
|
);
|
|
$assert(
|
|
prediction2c.order == manager.find(1).getOrder(),
|
|
'Prediction order should be the same as node 1',
|
|
);
|
|
|
|
const prediction2d = manager.predict(0, 1, { x: -50, y: -50 });
|
|
this._plotPrediction(graph2, prediction2d);
|
|
$assert(
|
|
prediction2d.position.x == manager.find(1).getPosition().x
|
|
&& prediction2d.position.y == manager.find(1).getPosition().y,
|
|
'Prediction position should be the same as node 1',
|
|
);
|
|
$assert(
|
|
prediction2d.order == manager.find(1).getOrder(),
|
|
'Prediction order should be the same as node 1',
|
|
);
|
|
|
|
// Graph 3
|
|
manager.disconnectNode(1);
|
|
manager.connectNode(0, 1, 0);
|
|
manager.addNode(2, TestSuite.NODE_SIZE, position).connectNode(0, 2, 2);
|
|
manager.layout();
|
|
const graph3 = manager.plot('testBalancedNodeDragPredict3', { width: 800, height: 400 });
|
|
|
|
const prediction3a = manager.predict(0, 1, { x: 50, y: 50 });
|
|
this._plotPrediction(graph3, prediction3a);
|
|
$assert(
|
|
prediction3a.position.x == manager.find(2).getPosition().x
|
|
&& prediction3a.position.y > manager.find(2).getPosition().y,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction3a.order == 4, 'Prediction order should be 4');
|
|
|
|
const prediction3b = manager.predict(0, 1, { x: 50, y: -50 });
|
|
this._plotPrediction(graph3, prediction3b);
|
|
$assert(
|
|
prediction3b.position.x == manager.find(1).getPosition().x
|
|
&& prediction3b.position.y == manager.find(1).getPosition().y
|
|
&& prediction3b.order == manager.find(1).getOrder(),
|
|
'Prediction should be the exact same as dragged node',
|
|
);
|
|
|
|
const prediction3c = manager.predict(0, 1, { x: -50, y: 50 });
|
|
this._plotPrediction(graph3, prediction3c);
|
|
$assert(
|
|
prediction3c.position.x < manager.find(0).getPosition().x
|
|
&& prediction3c.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction3c.order == 1, 'Prediction order should be 1');
|
|
|
|
const prediction3d = manager.predict(0, 1, { x: -50, y: -50 });
|
|
this._plotPrediction(graph3, prediction3d);
|
|
$assert(
|
|
prediction3d.position.x < manager.find(0).getPosition().x
|
|
&& prediction3d.position.y == manager.find(0).getPosition().y,
|
|
'Prediction is incorrectly positioned',
|
|
);
|
|
$assert(prediction3d.order == 1, 'Prediction order should be 1');
|
|
|
|
const prediction3e = manager.predict(0, 1, { x: 50, y: 0 });
|
|
this._plotPrediction(graph3, prediction3e);
|
|
$assert(
|
|
prediction3e.position.x == manager.find(1).getPosition().x
|
|
&& prediction3e.position.y == manager.find(1).getPosition().y,
|
|
'Prediction position should be the same as node 1',
|
|
);
|
|
$assert(
|
|
prediction3e.order == manager.find(1).getOrder(),
|
|
'Prediction order should be the same as node 1',
|
|
);
|
|
|
|
console.log('OK!\n\n');
|
|
}
|
|
}
|
|
|
|
export default BalancedTestSuite;
|