aboutsummaryrefslogtreecommitdiff
path: root/react-frontend/src/components/Visualization.js
blob: d3747386e4bbbd0d0f6ea7914d6c266ebc7e7433 (plain)
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
// JS module imports
import { useCallback, useEffect, useMemo, useRef, useState } from "react";
import uuid from 'react-uuid';
import Graph from 'vis-react';

// CSS imports
import '../css/Canvas.css';


/**
 * This function renders and mantains thhe canvas.
 * @param {Object} props The props for the canvas.
 * @returns {import("react").HtmlHTMLAttributes} The canvas to be retured.
 */
function Visualization(props) {
  const [graphState, setGraphState] = useState({
    nodes: [],
    edges: []
  });

  const options = {
    autoResize: true,
    edges: {
      color: "#ffffff"
    }
  };

  const events = {
    selectNode: event => {
      console.log(event);
      //props.setSelectedId(event.nodes[0]);
    }
  }

  const getNodes = () => {
    let nodes = [];
    const maxScore = props.data[0].suspicionScore;
    const interval = maxScore / 4;

    props.data.forEach(hub => {
      if (hub.followers) {
        let colorVal = '#f6f7d4';
        const score = hub.suspicionScore;

        if(score > (maxScore - interval)){
          colorVal = '#d92027'
        }
        if(score <= (maxScore - interval) && score > (maxScore - interval*2)){
          colorVal = '#f37121'
        }
        if(score <= (maxScore - interval*2) && score > (maxScore - interval*3)){
          colorVal = '#fdca40'
        }
        nodes.push({
          id: hub.id,
          autoResize: true,
          label: hub.name,
          labelHighlightBold: true,
          shape: "dot",
          value: hub.suspicionScore*1000,
          color: {
            background: colorVal,
            border: '#2b2e4a',
            highlight:{
              background: '#29bb89',
              border: '#fdca40'
            }
          },
          font: {
            color: '#9fd8df',
            size: 20,
          }
        });
      }
    });
    return nodes;
  }

  const getEdges = () => {
    let edges = []
    props.data.forEach(hub => {
      hub.followers.forEach(follower => {
        edges.push({
          from: follower.id,
          to: hub.id,
          dashes: false,
          color:{
            opacity: 0.7,
            highlight:'#fdca40',
          }
        });
      });
    });
    return edges;
  }

  // Hooks to update graph state when data changes
  useMemo(() => setGraphState({nodes: getNodes(), edges: getEdges()}), [props.data]);

  return (
    <div className="Map-canvas">
      <Graph
        key={uuid()} 
        graph={graphState}
        options={options}
        events={events}>
      </Graph>
    </div>
  );
}

export default Visualization;