Skip to content

Commit

Permalink
Add vote tally visualization to case files.
Browse files Browse the repository at this point in the history
  • Loading branch information
Ubuntu committed Oct 3, 2024
1 parent 5b548e4 commit 6a291e5
Show file tree
Hide file tree
Showing 2 changed files with 67 additions and 69 deletions.
134 changes: 66 additions & 68 deletions ban.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -179,6 +179,69 @@ async function UpdateTrial(cu) {
}
await cu.setGoodStanding(true);
}
const canvas = new Canvas.Canvas(360, 16 + 32 + 16);
const context = canvas.getContext('2d');
context.fillStyle = '#313338'; // Discord grey.
context.fillRect(0, 0, canvas.width, canvas.height);
context.strokeStyle = '#FFFFFF';
context.beginPath();
const halfX = Math.floor(canvas.width / 2) + 0.5;
context.moveTo(halfX, 8);
context.lineTo(halfX, 56);
context.stroke();
const sortedVotes = BanVoteCache.GetSortedVotesForDefendant(cu.commissar_id);
if (voteCount > 0) {
const gap = 2;
const maxWeight = Math.max(yesWeight, noWeight);
const yesPixels = Math.round(yesPercentage * canvas.width) + (gap / 2);
const yesVotes = sortedVotes[1];
let cumulativeYesWeight = 0;
for (const vote of yesVotes) {
const left = Math.floor(yesPixels * cumulativeYesWeight / yesWeight);
cumulativeYesWeight += vote.weight;
const right = Math.floor(yesPixels * cumulativeYesWeight / yesWeight);
let rectangleWidth = right - left - gap;
if (rectangleWidth <= 0) {
rectangleWidth = yesPixels - left - gap;
}
if (rectangleWidth <= 0) {
break;
}
context.fillStyle = vote.color;
context.fillRect(left, 16, rectangleWidth, 32);
}
const noPixels = canvas.width - yesPixels + gap;
const noVotes = sortedVotes[2];
let cumulativeNoWeight = 0;
for (const vote of noVotes) {
const left = Math.floor(noPixels * cumulativeNoWeight / noWeight);
cumulativeNoWeight += vote.weight;
const right = Math.floor(noPixels * cumulativeNoWeight / noWeight);
let rectangleWidth = right - left - gap;
if (rectangleWidth <= 0) {
rectangleWidth = noPixels - left - gap;
}
if (rectangleWidth <= 0) {
break;
}
context.fillStyle = vote.color;
context.fillRect(canvas.width - left - rectangleWidth, 16, rectangleWidth, 32);
}
if (yesWeight > 0 && noWeight > 0) {
context.fillStyle = '#FFFFFF';
context.beginPath();
context.moveTo(yesPixels - 1, 14);
context.lineTo(yesPixels + 5, 8);
context.lineTo(yesPixels - 7, 8);
context.fill();
}
}
const buffer = canvas.toBuffer();
const imageFilename = `case-${cu.commissar_id}.png`;
const voteTallyAttachment = {
attachment: buffer,
name: imageFilename,
};
await cu.setBanVoteStartTime(startTime.format());
const turnout = combinedWeight / availableWeight;
const durationDays = baselineVoteDurationDays * (1 - turnout);
Expand Down Expand Up @@ -220,15 +283,15 @@ async function UpdateTrial(cu) {
await message.edit(trialSummary);
await channel.send({ content: trialSummary });
const dateString = new Date().toISOString().substring(0, 10).replace(/-/g, '');
const attachment = await discordTranscripts.createTranscript(channel, {
const transcriptAttachment = await discordTranscripts.createTranscript(channel, {
filename: `ban-court-${dateString}-${cu.commissar_id}-${roomName}.html`,
poweredBy: false,
saveImages: false,
});
const transcriptChannel = await guild.channels.resolve('1110429964580433920');
await transcriptChannel.send({
content: trialSummary,
files: [attachment],
files: [transcriptAttachment, voteTallyAttachment],
});
try {
await channel.delete();
Expand All @@ -254,74 +317,9 @@ async function UpdateTrial(cu) {
`${voteCount} voters` +
`${threeTicks}`
);
const canvas = new Canvas.Canvas(360, 16 + 32 + 16);
const context = canvas.getContext('2d');
context.fillStyle = '#313338'; // Discord grey.
context.fillRect(0, 0, canvas.width, canvas.height);
context.strokeStyle = '#FFFFFF';
context.beginPath();
const halfX = Math.floor(canvas.width / 2) + 0.5;
context.moveTo(halfX, 8);
context.lineTo(halfX, 56);
context.stroke();
const sortedVotes = BanVoteCache.GetSortedVotesForDefendant(cu.commissar_id);
//console.log('sortedVotes[0]', sortedVotes[0]);
//console.log('sortedVotes[1]', sortedVotes[1]);
//console.log('sortedVotes[2]', sortedVotes[2]);
if (voteCount > 0) {
const gap = 2;
const maxWeight = Math.max(yesWeight, noWeight);
const yesPixels = Math.round(yesPercentage * canvas.width) + (gap / 2);
const yesVotes = sortedVotes[1];
let cumulativeYesWeight = 0;
for (const vote of yesVotes) {
const left = Math.floor(yesPixels * cumulativeYesWeight / yesWeight);
cumulativeYesWeight += vote.weight;
const right = Math.floor(yesPixels * cumulativeYesWeight / yesWeight);
let rectangleWidth = right - left - gap;
if (rectangleWidth <= 0) {
rectangleWidth = yesPixels - left - gap;
}
if (rectangleWidth <= 0) {
break;
}
context.fillStyle = vote.color;
context.fillRect(left, 16, rectangleWidth, 32);
}
const noPixels = canvas.width - yesPixels + gap;
const noVotes = sortedVotes[2];
let cumulativeNoWeight = 0;
for (const vote of noVotes) {
const left = Math.floor(noPixels * cumulativeNoWeight / noWeight);
cumulativeNoWeight += vote.weight;
const right = Math.floor(noPixels * cumulativeNoWeight / noWeight);
let rectangleWidth = right - left - gap;
if (rectangleWidth <= 0) {
rectangleWidth = noPixels - left - gap;
}
if (rectangleWidth <= 0) {
break;
}
context.fillStyle = vote.color;
context.fillRect(canvas.width - left - rectangleWidth, 16, rectangleWidth, 32);
}
if (yesWeight > 0 && noWeight > 0) {
context.fillStyle = '#FFFFFF';
context.beginPath();
context.moveTo(yesPixels - 1, 14);
context.lineTo(yesPixels + 5, 8);
context.lineTo(yesPixels - 7, 8);
context.fill();
}
}
const buffer = canvas.toBuffer();
const imageFilename = `case-${cu.commissar_id}.png`;
await message.edit({
content: trialMessage,
files: [{
attachment: buffer,
name: imageFilename,
}],
files: [voteTallyAttachment],
});
}
}
Expand Down
2 changes: 1 addition & 1 deletion bot-commands.js
View file
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ async function HandlePresidentVoteCommand(discordMessage) {
});
const message = await channel.send('Whoever gets the most votes will be Mr. or Madam President in October 2024.');
await message.react('❤️');
const generalRankUsers = await UserCache.GetTopRankedUsers(19);
const generalRankUsers = await UserCache.GetTopRankedUsers(20);
const candidateNames = [];
for (const user of generalRankUsers) {
if (user.commissar_id === 7) {
Expand Down

0 comments on commit 6a291e5

Please sign in to comment.