mirror of https://github.com/MISP/MISP
224 lines
9.2 KiB
PHP
224 lines
9.2 KiB
PHP
<?php
|
|
|
|
class CsseCovidTrendsWidget
|
|
{
|
|
public $title = 'CSSE Covid-19 trends';
|
|
public $render = 'MultiLineChart';
|
|
public $width = 4;
|
|
public $height = 5;
|
|
public $params = array(
|
|
'event_info' => 'Substring included in the info field of relevant CSSE COVID-19 events.',
|
|
'type' => 'Type of data used for the widget - confirmed (default), death, recovered, mortality, active.',
|
|
'insight' => 'Insight type - raw (default), growth, percent.',
|
|
'countries' => 'List of countries to be included (using the names used by the reports, such as Belgium, US, Germany).',
|
|
'timeframe' => 'Timeframe for events taken into account in days (going back from now, using the date field, default 10).'
|
|
);
|
|
public $description = 'Widget showing line charts for the evolution of the various case types.';
|
|
public $placeholder =
|
|
'{
|
|
"event_info": "%CSSE COVID-19 daily report%",
|
|
"type": "confirmed",
|
|
"insight": "raw",
|
|
"countries": ["Luxembourg", "Germany", "Belgium", "France"],
|
|
"timeframe": 20
|
|
}';
|
|
|
|
private $__countryAliases = array(
|
|
'Mainland China' => 'China',
|
|
'Korea, South' => 'South Korea'
|
|
);
|
|
|
|
public $cacheLifetime = 600;
|
|
public $autoRefreshDelay = false;
|
|
|
|
private $__countries = array();
|
|
|
|
public function handler($user, $options = array())
|
|
{
|
|
$this->Event = ClassRegistry::init('Event');
|
|
if (!isset($options['insight']) || !isset($this->__insightFunctions[$options['insight']])) {
|
|
$options['Insight'] = 'calculate_growth_rate';
|
|
}
|
|
if (empty($options['timeframe'])) {
|
|
$options['timeframe'] = 10;
|
|
}
|
|
if (empty($options['countries'])) {
|
|
$options['countries'] = array("Luxembourg", "Germany", "Belgium", "France");
|
|
}
|
|
if (empty($options['insight'])) {
|
|
$options['insight'] = 'raw';
|
|
}
|
|
$event_info_condition = empty($options['event_info']) ? '%CSSE COVID-19 daily report%' : $options['event_info'];
|
|
$params = array(
|
|
'eventinfo' => $event_info_condition,
|
|
'order' => 'date desc',
|
|
'date' => (empty($options['timeframe']) ? 10 : $options['timeframe']) . 'd'
|
|
);
|
|
$eventIds = $this->Event->filterEventIds($user, $params);
|
|
$eventIds = array_reverse(array_values($eventIds));
|
|
$data = array();
|
|
if (empty($options['type'])) {
|
|
$options['type'] = 'confirmed';
|
|
}
|
|
if (!empty($eventIds)) {
|
|
$previous = false;
|
|
foreach ($eventIds as $eventId) {
|
|
$params = array('eventid' => $eventId);
|
|
$event = $this->Event->fetchEvent($user, $params);
|
|
if (!empty($event)) {
|
|
$data[$event[0]['Event']['date']] = $this->__handleEvent($event[0], $options, $previous);
|
|
}
|
|
$previous = $data[$event[0]['Event']['date']];
|
|
}
|
|
}
|
|
$startDate = date('Y-m-d', strtotime('-' . intval($options['timeframe']) . ' days'));
|
|
//$data = call_user_func_array((array($this, $this->__insightFunctions[$options['Insight']]), array($startDate));
|
|
$data = array('data' => $data);
|
|
$data['insight'] = empty($options['insight']) ? 'raw' : $options['insight'];
|
|
foreach ($data['data'] as $date => $day) {
|
|
$data['data'][$date]['date'] = $date;
|
|
foreach ($this->__countries as $country => $temp) {
|
|
if (empty($data['data'][$date][$country][$data['insight']])) {
|
|
$data['data'][$date][$country][$data['insight']] = 0;
|
|
}
|
|
}
|
|
}
|
|
$data['data'] = array_values($data['data']);
|
|
$formulaData = array(
|
|
'insight' => array(
|
|
'raw' => '',
|
|
'growth' => 'daily increase in ',
|
|
'percent' => 'percentage wise daily increase in '
|
|
),
|
|
'type' => array(
|
|
'confirmed' => 'confirmed cases',
|
|
'death' => 'mortalities',
|
|
'recovered' => 'recoveries',
|
|
'mortality' => 'mortality rate',
|
|
'active' => 'active cases'
|
|
)
|
|
);
|
|
$data['formula'] = sprintf(
|
|
'%s%s',
|
|
(isset($options['insight']) && !empty($formulaData[$options['insight']])) ?
|
|
$formulaData['insight'][$options['insight']] :
|
|
$formulaData['insight']['raw'],
|
|
(isset($options['type']) && !empty($formulaData['type'][$options['type']])) ?
|
|
$formulaData['type'][$options['type']] :
|
|
$formulaData['type']['confirmed']
|
|
);
|
|
$data['formula'] = ucfirst($data['formula']);
|
|
foreach ($data['data'] as &$day) {
|
|
foreach ($day as $key => &$countryData) {
|
|
if ($key !== 'date') {
|
|
$countryData = $countryData[$options['insight']];
|
|
}
|
|
}
|
|
}
|
|
return $data;
|
|
}
|
|
|
|
private function __handleEvent($event, $options, $previous)
|
|
{
|
|
$data = array();
|
|
if (!empty($event['Object'])) {
|
|
$data = $this->__handleObjects($data, $event['Object'], $options, $previous);
|
|
}
|
|
$data['date'] = $event['Event']['date'];
|
|
return $data;
|
|
}
|
|
|
|
private function __handleObjects($data, $objects, $options, $previous)
|
|
{
|
|
foreach ($objects as $object) {
|
|
if ($object['name'] === 'covid19-csse-daily-report') {
|
|
$temp = $this->__interpretObject($object, $previous);
|
|
$data = $this->__rearrangeResults($data, $temp, $options, $previous);
|
|
}
|
|
}
|
|
if ($options['type'] === 'mortality') {
|
|
foreach ($data as $k => $v) {
|
|
$data[$k]['mortality'] = round(100 * (empty($v['death']) ? 0 : $v['death']) / $v['confirmed'], 2);
|
|
}
|
|
}
|
|
if (!empty($options['insight']) && $options['insight'] !== 'raw') {
|
|
if ($options['insight'] == 'growth') {
|
|
foreach ($data as $k => $countryData) {
|
|
foreach ($countryData as $type => &$value) {
|
|
if (!isset($previous[$k][$type])) {
|
|
$previous[$k][$type] = $data[$k][$type];
|
|
}
|
|
$data[$k]['growth'] = $data[$k][$type] - $previous[$k][$type];
|
|
}
|
|
}
|
|
} else if ($options['insight'] == 'percent') {
|
|
foreach ($data as $k => $countryData) {
|
|
foreach ($countryData as $type => &$value) {
|
|
if (empty($previous[$k][$type])) {
|
|
$previous[$k][$type] = $data[$k][$type];
|
|
}
|
|
if (!empty($previous[$k][$type])) {
|
|
$data[$k]['percent'] = 100 * ($data[$k][$type] - $previous[$k][$type]) / $previous[$k][$type];
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
foreach ($data as $k => &$countryData) {
|
|
$data[$k]['raw'] = $data[$k][$options['type']];
|
|
}
|
|
}
|
|
return $data;
|
|
}
|
|
|
|
private function __rearrangeResults($data, $temp, $options, $previous)
|
|
{
|
|
$country = $temp['country-region'];
|
|
if (!in_array($country, $options['countries'])) {
|
|
return $data;
|
|
}
|
|
$this->__countries[$country] = 1;
|
|
if ($options['type'] === 'mortality') {
|
|
foreach (array('confirmed', 'death') as $type) {
|
|
if (!empty($temp[$type])) {
|
|
$data[$country][$type] = (empty($data[$country][$type]) ? $temp[$type] : ($data[$country][$type] + $temp[$type]));
|
|
}
|
|
}
|
|
} else if ($options['type'] === 'active') {
|
|
if (empty($data[$country]['active'])) {
|
|
$data[$country]['active'] = 0;
|
|
}
|
|
$data[$country]['active'] =
|
|
$data[$country]['active'] +
|
|
(empty($temp['confirmed']) ? 0 : $temp['confirmed']) -
|
|
(empty($temp['death']) ? 0 : $temp['death']) -
|
|
(empty($temp['recovered']) ? 0 : $temp['recovered']);
|
|
} else {
|
|
$type = $options['type'];
|
|
if (!empty($temp[$type])) {
|
|
$data[$country][$type] = (empty($data[$country][$type]) ? $temp[$type] : ($data[$country][$type] + $temp[$type]));
|
|
}
|
|
}
|
|
return $data;
|
|
}
|
|
|
|
private function __interpretObject($object, $previous)
|
|
{
|
|
$temp = array();
|
|
$validFields = array('country-region', 'confirmed', 'death', 'recovered');
|
|
foreach ($object['Attribute'] as $attribute) {
|
|
if (in_array($attribute['object_relation'], $validFields)) {
|
|
if ($attribute['object_relation'] !== 'country-region') {
|
|
$attribute['value'] = intval($attribute['value']);
|
|
} else {
|
|
if (isset($this->__countryAliases[$attribute['value']])) {
|
|
$attribute['value'] = $this->__countryAliases[$attribute['value']];
|
|
}
|
|
}
|
|
$temp[$attribute['object_relation']] = $attribute['value'];
|
|
}
|
|
}
|
|
return $temp;
|
|
}
|
|
}
|