// 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;
}