avlmap/src/components/common/utils.js

Avail Map by Avail Labs

const { scaleOrdinal } = require("d3-scale");
const { format } = require("d3-format");

// const LimitedAttributes = {
// 	num_events: "Occurances",
// 	property_damage: "Property Damage $"
// }

const sheldusAttributes = {
	num_events: "Occurances",
	property_damage: "Property Damage $",
	crop_damage: "Crop Damage $",
	injuries: "Injuries",
	fatalities: "Fatalities"
}

const increase = 100
const getColors = num => {
  const colors = [];
  let r = 0, g = increase, b = 0;
  for (let i = 0; i < num; ++i) {

    colors.push(`rgb(${ r },${ g },${ b })`);

    b += increase;

    if (b > 255) {
      b %= 255;
      g += increase;
    }

    if (g > 255) {
      g %= 255;
      r += increase;
    }

    if (r > 255) {
      r %= 255;
    }
  }
  return colors;
}

const increase2 = 85
const getColors2 = num => {
  const colors = [];
  let r = 0, g = increase2, b = 0;
  for (let i = 0; i < num; ++i) {

    colors.push(`rgb(${ r },${ g },${ b })`);

    b += increase2;

    if (b > 255) {
      b = 0;
      g += increase2;
    }

    if (g > 255) {
      g = 0;
      r += increase2;
    }

    if (r > 255) {
      r = 0;
    }
  }
  return colors;
}

const D3_CATEGORY20 = [
  "#1f77b4",
  "#aec7e8",
  "#ff7f0e",
  "#ffbb78",
  "#2ca02c",
  "#98df8a",
  "#d62728",
  "#ff9896",
  "#9467bd",
  "#c5b0d5",
  "#8c564b",
  "#c49c94",
  "#e377c2",
  "#f7b6d2",
  "#7f7f7f",
  "#c7c7c7",
  "#bcbd22",
  "#dbdb8d",
  "#17becf",
  "#9edae5"
];

const getHazardName = hazardid =>
	hazardid === "winterweat"
		? "Winter Weather"
		: hazardid.split("")
			.map((d, i) => i === 0 ? d.toUpperCase() : d)
			.join("")

function fnum(x) {
	if(isNaN(x)) return x;

	if(x < 9999) {
		const frmt = format(",.0f")
		return frmt(x);
	}

	if(x < 1000000) {
		const frmt = format(",.0f")
		return frmt(x/1000) + "K";
	}
	if( x < 10000000) {
		const frmt = format(",.2f")
		return frmt(x/1000000) + "M";
	}

	if(x < 1000000000) {
		const frmt = format(",.1f")
		return frmt(x/1000000) + "M";
	}

	if(x < 1000000000000) {
		const frmt = format(",.1f")
		return frmt(x/1000000000) + "B";
	}

	return "1T+";
}

module.exports = {

	getHazardName,
	fnum,
	getColors,
	getColors2,

	getColorScale: domain =>
		scaleOrdinal()
			.domain(domain)
			.range(D3_CATEGORY20),

	processSheldus : (data,key) => {
		let yearly = {
			id: sheldusAttributes[key],
			"color": "#047bf8",
			data: Object.keys(data).reduce((total,year) => {
				total.push({
						"x": +year,
			        	"y": data[year][key] || 0
			     })
				return total
			}, [])
		}
		let fiveYear = {
			id: '5 Year Avg ' + sheldusAttributes[key],
			"color": "#e65252",
			data: Object.keys(data).reduce((total,year) => {
				let avgTotal = 0
				let count = 0
				for(let i=4; i >= 0; i-- ) {
					if(data[year-i] && data[year-i][key]){
						avgTotal += data[year-i][key]
					}
					count += 1 
				}
				let avg = !isNaN(avgTotal / count)  > 0 
					? (avgTotal / count)
					: 0 
				total.push({
						"x": +year,
			        	"y": +(avg.toFixed(2))
			     })
				return total
			}, [])

		}
		return [yearly,fiveYear]
	},

	processDataForBarChart: (rawData, geoids, lossType="property_damage", hazard=null) => {
// console.log("<processDataForBarChart>",rawData)
		const data = {}, keys = {};
		for (const geoid in rawData) {
			if (!geoids.includes(geoid)) continue;
			for (const hazardid in rawData[geoid]) {
				if (hazard && (hazardid != hazard)) continue;

				if (!(hazardid in keys)) {
					keys[hazardid] = true;
				}
				for (const year in rawData[geoid][hazardid]) {
					if (!(year in data)) {
						data[year] = { year };
					}
					if (!(hazardid in data[year])) {
						data[year][hazardid] = 0;
					}
					const value = data[year][hazardid] + +rawData[geoid][hazardid][year][lossType];
					data[year][hazardid] = value;
				}
			}
		}
		return { data: Object.values(data), keys: Object.keys(keys) };
	},

	processSheldus5year : (data, key, type) => {
		type = type ? type : 'avg'
		return Object.keys(data).reduce((total,year) => {
			let avgTotal = 0
			let count = 0
			for(let i=4; i >= 0; i-- ) {
				if(data[year-i] && data[year-i][key]){
					avgTotal += data[year-i][key]
				}
				count += 1 
			}
			let avg = !isNaN(avgTotal / count) && count > 0 
				? (avgTotal / count)
				: 0 
			total[year] = type === 'avg' ?  +(avg.toFixed(2)) : +(avgTotal.toFixed(2))
			return total
		}, {})
	},
	
	sumData : (data, key, len) => {
		return Object.keys(data).reduce((total,year) => {
			let avgTotal = 0
			let count = 0
			for(let i=(len-1); i >= 0; i-- ) {
				if(data[year-i] && data[year-i][key]){
					avgTotal += data[year-i][key]
				}
				count += 1 
			}
			let avg = !isNaN(avgTotal / count) && count > 0 
				? (avgTotal / count)
				: 0 
			total[year] = +(avgTotal.toFixed(2))
			return total
		}, {})
		
	},

	total : (data,key) => {
		// this is a stupid hack
		// to remove crud keys that falcor adds to object
		let numYears = Object.keys(data)
			.filter(d => d.length === 4)
			.length

		if(numYears <= 0) return ''  

		let total =  Object.keys(data).reduce((total,year) => {
			if(data[year][key] && !isNaN(+data[year][key])) {
				total += data[year][key]
			}
			return total
		}, 0)
		return fnum(total)
	},

	avg : (data,key) => {
		// this is a stupid hack
		// to remove crud keys that falcor adds to object
		let numYears = Object.keys(data)
			.filter(d => d.length === 4)
			.length

		if(numYears <= 0) return ''
		

		let total =  Object.keys(data).reduce((total,year) => {
			if(data[year][key] && !isNaN(+data[year][key])) {
				total += data[year][key]
			}
			return total
		}, 0)

			
		
		return fnum( total / numYears )
	},

	avgData : (data,key, len) => {
		return Object.keys(data).reduce((total,year) => {
			let avgTotal = 0
			let count = 0
			for(let i=len-1; i >= 0; i-- ) {
				if(data[year-i] && data[year-i][key]){
					avgTotal += data[year-i][key]
				}
				count += 1 
			}
			let avg = !isNaN(avgTotal / count) && count > 0 
				? (avgTotal / count)
				: 0 
			total[year] = +(avg.toFixed(2)) 
			return total
		}, {})
		
	}

}