Object.defineProperties(Array.prototype, {
    count: {
        value: function (query) {
            /* 
               Counts number of occurrences of query in array, an integer >= 0 
               Uses the javascript == notion of equality.
            */
            var count = 0;
            for(let i = 0; i < this.length; i++)
                if(this[i] == query)
                    count++;
            return count;
        }
    }
});

const mean = arr => arr.reduce( ( p, c ) => p + c, 0 ) / arr.length;

tables = $(".table-div");

tables.each((n, table) => {
	console.log(table)

	rows = $(table).find(".worker-row");
	console.log(rows)

	workers = {};

	rows.each((n, tr) => {

		//console.log(tr);

		worker_td = $(tr).find("td:first");

		worker = worker_td.attr("data-worker");

		shift_tds = $(tr).find(".rota-day");
		shifts = []

		shift_tds.each((n, td) => {
			shifts.push($(td).attr("data-shift"));
		})

		shifts_unique = new Set(shifts);
		shifts_unique.delete("");
		//console.log(shifts_unique);

		weekends_worked = 0;

		weeks = shift_tds.length / 7;

		whole_weekends_worked = 0;

		for(var i = 1; i <= weeks; i++) {
			//console.log($(tr).find(`[data-week='${i}'][data-day='Sat']`));
			if($(tr).find(`[data-week='${i}'][data-day='Sat']`).attr("data-shift") != "" || $(tr).find(`[data-week='${i}'][data-day='Sun']`).attr("data-shift") != "") { weekends_worked = weekends_worked + 1; }
			if($(tr).find(`[data-week='${i}'][data-day='Sat']`).attr("data-shift") != "" && $(tr).find(`[data-week='${i}'][data-day='Sun']`).attr("data-shift") != "") { whole_weekends_worked = whole_weekends_worked + 1; }
		}

		shift_counts = {};

		total_shifts = 0;
		days_lost = 0;
		shifts_unique.forEach(el => {
			//console.log(el);
			shift_counts[el] = shifts.count(el);
			total_shifts = total_shifts + shifts.count(el);

            // 5 days for weekday nights
			if(el == "night_weekday") { days_lost = days_lost + shifts.count(el) / 4 * 5 }
            // 4 days for weekend nights
			if(el == "night_weekend") {
				days_lost = days_lost + shifts.count(el) / 3 * 4
				whole_weekends_worked = whole_weekends_worked - shifts.count(el) / 3
			}
            // 0.5 for twilights
			if(el.includes("twilight")) { 
                twilight_days_lost = shifts.count(el) / 2
                days_lost = days_lost + twilight_days_lost

            }

		});

        // Also 1 day for a whole weekend
		days_lost = days_lost + whole_weekends_worked
		console.log(worker, whole_weekends_worked);

		//days_lost = days_lost + whole_weekends_worked;
		//console.log(shift_counts);

		workers[worker] = {
			"site": worker_td.attr("data-site"),
			"nwds": worker_td.attr("data-nwds"),
			"fte": worker_td.attr("data-fte_adj"),
			"end_date": worker_td.attr("data-end_date"),
			"shifts": shifts,
			"shift_types": shifts_unique,
			"weekends_worked": weekends_worked,
			"shift_counts": shift_counts,
			"days_lost": days_lost,
			//"shifts"
		}

		sc = JSON.stringify(shift_counts);

		worker_td.after(`<div class='worker-summary'><span>Total shifts: ${total_shifts}, </span><span>Weekends: ${weekends_worked}, </span><span>Training days lost: ${days_lost}, </span><span class=''>${sc}</span></div>`)

	})

	sites = {};
	sites_days_lost = {}
	for (const worker in workers) {
		//console.log(worker, workers[worker]);
		w = workers[worker];
		if (w.fte == 100) { 
			//console.log(w.shift_types);
			for (let shift of w.shift_types) {
				//console.log(shift);
				if (!(w.site in sites)) {
					sites[w.site] = {};
				
				}
				if (!(shift in sites[w.site])) {
					sites[w.site][shift] = [w.shift_counts[shift]];
				} else {
					sites[w.site][shift].push(w.shift_counts[shift]);
				}
			}
			if (!(w.site in sites_days_lost)) {
				sites_days_lost[w.site] = [w.days_lost];
			
			} else {
				sites_days_lost[w.site].push(w.days_lost);
			}
		}
	}

	// Calculate site shift summaries
	$(table).append("<div class='site-summary'>Average shifts for a 100% FTE trainee</div>");

	for (const site in sites) {
		console.log(site, $("div"));
		$(table).find(".site-summary").append(`<div class='site site-${site}'><span class='site-name'>${site}</span></div>`);
		
		for (const shift in sites[site]) {
			console.log(sites[site][shift]);
			average = mean(sites[site][shift])
			$(table).find(`.site-${site}`).append(`<div>${shift}: ${average.toFixed(2)}</div>`);
		}
		
		days_lost_avg = mean(sites_days_lost[site]);
		$(table).find(`.site-${site}`).append(`<div class='training-days-lost'>Training days lost: ${days_lost_avg.toFixed(2)}</div>`);
	}


	$(table).before($("<button>Toggle Summary</button>").click(() => {
		$(table).find(".rota-day").toggle();
		$(table).find(".worker-summary").toggle();

	}))
});