imported>Mxn: Fixed off-by-one error

2021-09-01T05:04:01Z

Fixed off-by-one error

New page

--- Example usage:
--- {{#invoke:Medical cases chart/data|externalData|page=COVID-19 cases in Santa Clara County, California.tab|recoveries=hospitalized|cases=totalConfirmedCases}}
--- =p._externalData({datapage="COVID-19 cases in Santa Clara County, California.tab",datarecoveries="hospitalized",datacases="totalConfirmedCases"})

local p = {}
local lang = mw.getContentLanguage()
local english = mw.getLanguage("en")

local function round(x)
return (math.modf(x + (x < 0 and -0.5 or 0.5)))
end

local function formatChange(previous, current)
if not previous or previous == 0 then
return
end
if previous == current then
return "="
end

local change = current / previous * 100 - 100
local sign = change < 0 and "−" or "+"

if (math.abs(change)) >= 10 then
return mw.ustring.format("%s%.0f%%", sign, (math.abs(change)))
end
if (math.abs(change)) < 1 then
return mw.ustring.format("%s%.3f%%", sign, (math.abs(change)))
end
return mw.ustring.format("%s%.2f%%", sign, (math.abs(change)))
end

function p._externalData(args)
local data = mw.ext.data.get(args.datapage)

local dateIndex
local deathsIndex
local recoveriesIndex
local casesIndex
local class4Index
local class5Index
for i, field in ipairs(data.schema.fields) do
if field.name == "date" or field.name == args.datadate then
dateIndex = i
elseif field.name == "deaths" or field.name == args.datadeaths then
deathsIndex = i
elseif field.name == "recoveries" or field.name == args.datarecoveries then
recoveriesIndex = i
elseif field.name == "cases" or field.name == args.datacases then
casesIndex = i
elseif field.name == "class4" or field.name == args.dataclass4 then
class4Index = i
elseif field.name == "class5" or field.name == args.dataclass5 then
class5Index = i
end
end
assert(dateIndex, "Date field not found.")
assert(deathsIndex or not args.datadeaths, "Deaths field not found.")
assert(recoveriesIndex or not args.datarecoveries, "Recoveries field not found.")
assert(casesIndex or not args.datacases, "Cases field not found.")
assert(class4Index or not args.dataclass4, "Class 4 field not found.")
assert(class5Index or not args.dataclass5, "Class 5 field not found.")

-- Restructure the data as tables with keys.
local records = {}
for i, row in ipairs(data.data) do
local record = {
date = row[dateIndex],
deaths = deathsIndex and row[deathsIndex],
recoveries = recoveriesIndex and row[recoveriesIndex],
cases = casesIndex and row[casesIndex],
class4 = class4Index and row[class4Index],
class5 = class5Index and row[class5Index],
options = {},
streak = 1,
}
local prevRecord = records[#records] or {}
if casesIndex and not prevRecord.cases and record.cases and record.cases > 0 then
record.options.firstright1 = "y"
end
if deathsIndex and prevRecord.deaths == 0 and record.deaths and record.deaths > 0 then
record.options.firstright2 = "y"
end
if deathsIndex and (prevRecord.deaths or prevRecord.assumedDeaths) and not record.deaths then
record.assumedDeaths = prevRecord.deaths or prevRecord.assumedDeaths
end
if casesIndex and (prevRecord.cases or prevRecord.assumedCases) and not record.cases then
record.assumedCases = prevRecord.cases or prevRecord.assumedCases
end
if record.deaths == prevRecord.deaths
and record.recoveries == prevRecord.recoveries
and record.cases == prevRecord.cases
and record.class4 == prevRecord.class4
and record.class5 == prevRecord.class5 then
record.streak = prevRecord.streak + 1
end
table.insert(records, record)
end

-- Collapse streaks of identical data.
for i = #records, 1, -1 do
local record = records[i]
if i < #records and record.streak > 3 then
for j = i, i - record.streak + 3, -1 do
table.remove(records, j)
end
i = i - record.streak + 2
record = records[i]
record.options.collapse = "y"
record.options.id = english:formatDate("M", record.date):lower()
record.date = nil
end
end

-- Stringify the data.
local rows = {}
for i, record in ipairs(records) do
local prevRecord = records[i - 1] or {}
local row = {
record.date or "",
tostring(record.deaths or record.assumedDeaths or ""),
tostring(record.recoveries or ""),
tostring(record.cases or record.assumedCases or ""),
tostring(record.class4 or ""),
tostring(record.class5 or ""),
record.cases and lang:formatNum(record.cases) or "",
record.cases and formatChange(prevRecord.cases or prevRecord.assumedCases, record.cases) or "",
record.deaths and lang:formatNum(record.deaths) or "",
record.deaths and formatChange(prevRecord.deaths or prevRecord.assumedDeaths, record.deaths) or "",
}
for k, v in pairs(record.options) do
table.insert(row, string.format("%s=%s", k, v))
end
table.insert(rows, table.concat(row, ";"))
end
return table.concat(rows, "\n")
end

function p.externalData(frame)
local args = {}
for k,v in pairs(frame.args) do
if (v or '') ~= '' then
args['data'..k] = v
end
end
return p._externalData(args)
end
return p

Module:Medical cases chart/data - Revision history

imported>Mxn: Fixed off-by-one error