You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
98 lines
2.5 KiB
JavaScript
98 lines
2.5 KiB
JavaScript
// SARS-CoV-2-Viz
|
|
// Animated COVID case count visualization
|
|
// Copyright 2022 Edward L. Platt <ed@elplatt.com>
|
|
|
|
const covidUrl = "https://raw.githubusercontent.com/CSSEGISandData/COVID-19/master/csse_covid_19_data/csse_covid_19_time_series/time_series_covid19_confirmed_US.csv";
|
|
|
|
// Parse covid tsv
|
|
function parseCovidData(tsv, onProgress) {
|
|
let rows = tsv.split(/[\r\n]+/);
|
|
let keys = rows[0].split("\t");
|
|
let dates = keys.slice(11);
|
|
|
|
// Initialize data array
|
|
let dataForDay = [];
|
|
for (const date of dates) {
|
|
dataForDay.push([]);
|
|
}
|
|
|
|
// Iterate through non-header rows
|
|
const rowCount = rows.length - 1;
|
|
let percent = 0;
|
|
for (const [index, row] of rows.slice(1).entries()) {
|
|
const rowData = tsvRowToJSON(keys, row);
|
|
|
|
// Only update progress when it has actually changed
|
|
let newPercent = Math.floor(100 * index / rowCount);
|
|
if (newPercent > percent) {
|
|
percent = newPercent;
|
|
onProgress(`${percent}%`);
|
|
}
|
|
|
|
let FIPS = rowData.FIPS;
|
|
if (typeof(FIPS) === "undefined") {
|
|
continue;
|
|
}
|
|
FIPS = FIPS.slice(0, FIPS.indexOf(".") || FIPS.length).padStart(5, "0");
|
|
|
|
let lastSeven = [];
|
|
for (const [index, date] of dates.entries()) {
|
|
|
|
const newCases = index == 0 ?
|
|
parseInt(rowData[date]) :
|
|
parseInt(rowData[date])
|
|
- parseInt(rowData[dates[index - 1]]);
|
|
|
|
// Update array for running average
|
|
if (lastSeven.length == 7) {
|
|
lastSeven.shift();
|
|
}
|
|
lastSeven.push(newCases);
|
|
const mean = Math.round(
|
|
lastSeven.reduce((x, y) => x + y) / lastSeven.length);
|
|
|
|
dataForDay[index].push({
|
|
FIPS: FIPS,
|
|
date: date,
|
|
count: newCases,
|
|
sevenDayMean: mean
|
|
});
|
|
}
|
|
}
|
|
onProgress("100%");
|
|
|
|
// Calculate maximum case count for each day
|
|
let maxSoFar = 0;
|
|
for (const [day, date] of dates.entries()) {
|
|
|
|
let dayMax = Math.max(
|
|
...dataForDay[day].map(
|
|
d => d.sevenDayMean));
|
|
|
|
maxSoFar = Math.max(maxSoFar, dayMax);
|
|
|
|
// Add data to each county for current day
|
|
for (let [countyIndex, d] of dataForDay[day].entries()) {
|
|
d.dayMax = dayMax;
|
|
d.maxSoFar = maxSoFar;
|
|
}
|
|
}
|
|
|
|
return dataForDay.map(data => {
|
|
const entries = new Map(data.map(d => [d.FIPS, d]));
|
|
return Object.fromEntries(entries);
|
|
});
|
|
}
|
|
|
|
async function getData(onProgress) {
|
|
const content = await fetchWithProgress(covidUrl, onProgress);
|
|
return content;
|
|
}
|
|
|
|
async function parseData(content, onProgress) {
|
|
let tsv = csvToTSV(content);
|
|
let dataForDay = parseCovidData(tsv, onProgress);
|
|
return dataForDay;
|
|
}
|
|
|