@xcpcio/core 0.13.0 → 0.15.0

package/src/award.ts

@@ -1,9 +1,9 @@
 export enum MedalType {
-  UNKNOWN,
-  GOLD,
-  SILVER,
-  BRONZE,
-  HONORABLE,
+  UNKNOWN = "Unknown",
+  GOLD = "Gold",
+  SILVER = "Silver",
+  BRONZE = "Bronze",
+  HONORABLE = "Honorable",
 }
 
 export class Award {
@@ -18,4 +18,15 @@ export class Award {
   }
 }
 
+export function isValidMedalType(medal: MedalType): boolean {
+  const validMedalType = [
+    MedalType.GOLD,
+    MedalType.SILVER,
+    MedalType.BRONZE,
+    MedalType.HONORABLE,
+  ];
+
+  return validMedalType.includes(medal);
+}
+
 export type Awards = Map<string, Award[]>;

package/src/contest.ts

@@ -136,6 +136,18 @@ export class Contest {
 
     return Math.round((pass * 100) / total);
   }
+
+  isEnableAwards(group: string): boolean {
+    if (!this.awards) {
+      return false;
+    }
+
+    if (!this.awards.has(group)) {
+      return false;
+    }
+
+    return true;
+  }
 }
 
 export function createContest(contestJSON: IContest): Contest {

package/src/export/cf.ts

@@ -0,0 +1,120 @@
+import { SubmissionStatus } from "@xcpcio/types";
+
+import type { Rank } from "../rank";
+
+import {
+  isAccepted,
+  isNotCalculatedPenaltyStatus,
+  isPending,
+} from "../submission-status";
+
+import dayjs from "../utils/dayjs";
+
+export class CodeforcesGymGhostDATConverter {
+  constructor() {}
+
+  public convert(rank: Rank): string {
+    let res = "";
+
+    res += `@contest "${rank.contest.name}"
+@contlen ${Math.floor(dayjs.duration(rank.contest.endTime.diff(rank.contest.startTime)).asMinutes())}
+@problems ${rank.contest.problems.length}
+@teams ${rank.teams.length + 100}
+@submissions ${rank.submissions.length}
+`;
+
+    rank.contest.problems.forEach((p) => {
+      res += `@p ${p.label},${p.label},20,0\n`;
+    });
+
+    let teamIndex = 1;
+    const teamIdMap = new Map<string, number>();
+    const submissionsIdMap = new Map<string, Map<string, number>>();
+
+    rank.teams.forEach((team) => {
+      let name = team.name;
+
+      if (team.organization) {
+        name = `${team.organization} - ${name}`;
+      }
+
+      if (team.members) {
+        name = `${name} - ${team.membersToString}`;
+      }
+
+      res += `@t ${teamIndex},0,1,${name}\n`;
+      teamIdMap.set(team.id, teamIndex);
+      teamIndex++;
+
+      {
+        const mp = new Map<string, number>();
+        rank.contest.problems.forEach((p) => {
+          mp.set(p.id, 0);
+        });
+        submissionsIdMap.set(team.id, mp);
+      }
+    });
+
+    for (let i = 0; i < 100; i++) {
+      res += `@t ${teamIndex},0,1,Пополнить команду\n`;
+      teamIndex++;
+    }
+
+    rank.getSubmissions().forEach((submission) => {
+      const teamId = submission.teamId;
+      const problemId = submission.problemId;
+      const problem = rank.contest.problemsMap.get(problemId);
+
+      if (!problem) {
+        return;
+      }
+
+      const status = this.submissionStatusToCodeforcesGymDatStatus(submission.status);
+      submissionsIdMap.get(teamId)!.set(problemId, submissionsIdMap.get(teamId)!.get(problemId)! + 1);
+
+      res += `@s ${teamIdMap.get(teamId)},${problem.label},${submissionsIdMap.get(teamId)?.get(problemId)},${submission.timestamp},${status}\n`;
+    });
+
+    return res;
+  }
+
+  private submissionStatusToCodeforcesGymDatStatus(status: SubmissionStatus): string {
+    if (isAccepted(status)) {
+      return "OK";
+    }
+
+    if (status === SubmissionStatus.WRONG_ANSWER) {
+      return "WA";
+    }
+
+    if (status === SubmissionStatus.TIME_LIMIT_EXCEEDED) {
+      return "TL";
+    }
+
+    if (status === SubmissionStatus.MEMORY_LIMIT_EXCEEDED) {
+      return "ML";
+    }
+
+    if (status === SubmissionStatus.OUTPUT_LIMIT_EXCEEDED) {
+      return "IL";
+    }
+
+    if (status === SubmissionStatus.PRESENTATION_ERROR) {
+      return "PE";
+    }
+
+    if (status === SubmissionStatus.RUNTIME_ERROR) {
+      return "RT";
+    }
+
+    if (status === SubmissionStatus.COMPILATION_ERROR || isNotCalculatedPenaltyStatus(status)) {
+      return "CE";
+    }
+
+    if (isPending(status)) {
+      return "PD";
+    }
+
+    return "RJ";
+  }
+}

package/src/export/general-excel.ts

@@ -0,0 +1,184 @@
+import _ from "lodash";
+import * as XLSX from "xlsx-js-style";
+import stringWidth from "string-width";
+
+import { isValidMedalType } from "../award";
+import type { Rank } from "../rank";
+
+export class GeneralExcelConverter {
+  constructor() {}
+
+  public convert(oriRank: Rank): XLSX.WorkBook {
+    const rank = _.cloneDeep(oriRank);
+
+    rank.options.disableFilterTeamsByGroup();
+    rank.options.disableFilterSubmissionByTimestamp();
+
+    const workbook = XLSX.utils.book_new();
+
+    for (const [k, v] of rank.contest.group) {
+      rank.options.setGroup(k);
+      rank.buildRank();
+
+      const sheet = this.convertToSheet(rank);
+      XLSX.utils.book_append_sheet(workbook, sheet, v.names.get(v.defaultLang) as string);
+    }
+
+    return workbook;
+  }
+
+  public convertAndWrite(rank: Rank, filename: string): any {
+    return XLSX.writeFile(
+      this.convert(rank),
+      filename,
+      {
+        compression: true,
+      });
+  }
+
+  private convertToSheet(rank: Rank): XLSX.WorkSheet {
+    const aoa = this.convertToAoa(rank);
+    const sheet = XLSX.utils.aoa_to_sheet(aoa);
+
+    const cols = [];
+    const head = aoa[1];
+    for (let j = 0; j < head.length; j++) {
+      let wch = 10;
+      for (let i = 1; i < aoa.length; i++) {
+        wch = Math.max(wch, stringWidth(aoa[i][j]) + 2);
+      }
+
+      cols.push({
+        wch,
+      });
+    }
+
+    sheet["!cols"] = cols;
+
+    {
+      const mergeRange = { s: { r: 0, c: 0 }, e: { r: 0, c: head.length - 1 } };
+      const merges = [{ s: mergeRange.s, e: mergeRange.e }];
+      sheet["!merges"] = merges;
+    }
+
+    const font = {
+      name: "Arial Unicode MS",
+      bold: false,
+      italic: false,
+      sz: 12,
+    };
+
+    const borderStyle = {
+      style: "thin",
+    };
+
+    const cellStyle = {
+      alignment: {
+        vertical: "center",
+        horizontal: "center",
+      },
+      border: {
+        top: borderStyle,
+        bottom: borderStyle,
+        left: borderStyle,
+        right: borderStyle,
+      },
+      font,
+    };
+
+    for (let i = 1; i < aoa.length; i++) {
+      for (let j = 0; j < aoa[i].length; j++) {
+        const cellAddress = XLSX.utils.encode_cell({ r: i, c: j });
+        const cell = sheet[cellAddress];
+        cell.s = cellStyle;
+      }
+    }
+
+    {
+      const cellAddress = XLSX.utils.encode_cell({ r: 0, c: 0 });
+      const cell = sheet[cellAddress];
+      const titleStyle = _.cloneDeep(cellStyle);
+      titleStyle.font.sz = 28;
+      titleStyle.font.bold = true;
+      cell.s = titleStyle;
+    }
+
+    return sheet;
+  }
+
+  private convertToAoa(rank: Rank): string[][] {
+    const aoa: string[][] = [];
+
+    const enableAwards = rank.contest.isEnableAwards(rank.options.group);
+
+    {
+      aoa.push([rank.contest.name]);
+    }
+
+    {
+      const head: string[] = [];
+      head.push("Rank");
+
+      if (rank.contest.organization) {
+        head.push(`${rank.contest.organization} Rank`);
+        head.push(rank.contest.organization);
+      }
+
+      head.push("Name", "Solved", "Penalty", ...rank.contest.problems.map(p => p.label), "Dict");
+
+      if (enableAwards) {
+        head.push("Medal");
+      }
+
+      aoa.push(head);
+    }
+
+    for (const team of rank.teams) {
+      const arr: string[] = [];
+
+      arr.push(team.rank.toString());
+      if (team.organization) {
+        if (team.organizationRank !== -1) {
+          arr.push(team.organizationRank.toString());
+        } else {
+          arr.push("");
+        }
+
+        arr.push(team.organization);
+      }
+
+      arr.push(team.name, team.solvedProblemNum.toString(), team.penaltyToMinute.toString());
+
+      for (const p of team.problemStatistics) {
+        if (p.isUnSubmitted) {
+          arr.push("-");
+        }
+
+        if (p.isSolved) {
+          arr.push(`+${p.totalCount}(${p.solvedTimestampToMinute})`);
+        }
+
+        if (p.isWrongAnswer) {
+          arr.push(`-${p.failedCount}`);
+        }
+
+        if (p.isPending) {
+          arr.push(`? ${p.failedCount} + ${p.pendingCount}`);
+        }
+      }
+
+      arr.push(`${team.dict}%`);
+
+      if (enableAwards) {
+        const medals = team.awards
+          .filter(a => isValidMedalType(a))
+          .map(a => a.toString());
+        arr.push(medals.join(", "));
+      }
+
+      aoa.push(arr);
+    }
+
+    return aoa;
+  }
+}

package/src/export/index.ts

@@ -0,0 +1,2 @@
+export * from "./cf";
+export * from "./general-excel";

package/src/index.ts

@@ -1,8 +1,8 @@
+export * from "./export";
 export * from "./award";
 export * from "./contest-index";
 export * from "./utils";
 export * from "./contest";
-export * from "./export";
 export * from "./image";
 export * from "./problem";
 export * from "./rank-statistics";

package/src/problem.ts

@@ -189,4 +189,12 @@ export class TeamProblemStatistics {
 
     return Math.floor(this.solvedTimestamp / 60) * 60 + this.failedCount * this.contestPenalty;
   }
+
+  get penaltyToMinute() {
+    return Math.floor(this.penalty / 60);
+  }
+
+  get solvedTimestampToMinute() {
+    return Math.floor(this.solvedTimestamp / 60);
+  }
 }

package/src/team.ts

@@ -101,6 +101,8 @@ export class Team {
     this.submissions = [];
 
     this.placeChartPoints = [];
+
+    this.awards = [];
   }
 
   get penaltyToMinute() {

package/src/export.ts

@@ -1,116 +0,0 @@
-import { SubmissionStatus } from "@xcpcio/types";
-
-import type { Rank } from "./rank";
-
-import {
-  isAccepted,
-  isNotCalculatedPenaltyStatus,
-  isPending,
-} from "./submission-status";
-
-import dayjs from "./utils/dayjs";
-
-function submissionStatusToCodeforcesGymDatStatus(status: SubmissionStatus): string {
-  if (isAccepted(status)) {
-    return "OK";
-  }
-
-  if (status === SubmissionStatus.WRONG_ANSWER) {
-    return "WA";
-  }
-
-  if (status === SubmissionStatus.TIME_LIMIT_EXCEEDED) {
-    return "TL";
-  }
-
-  if (status === SubmissionStatus.MEMORY_LIMIT_EXCEEDED) {
-    return "ML";
-  }
-
-  if (status === SubmissionStatus.OUTPUT_LIMIT_EXCEEDED) {
-    return "IL";
-  }
-
-  if (status === SubmissionStatus.PRESENTATION_ERROR) {
-    return "PE";
-  }
-
-  if (status === SubmissionStatus.RUNTIME_ERROR) {
-    return "RT";
-  }
-
-  if (status === SubmissionStatus.COMPILATION_ERROR || isNotCalculatedPenaltyStatus(status)) {
-    return "CE";
-  }
-
-  if (isPending(status)) {
-    return "PD";
-  }
-
-  return "RJ";
-}
-
-export function rankToCodeforcesGymDAT(rank: Rank) {
-  let res = "";
-
-  res += `@contest "${rank.contest.name}"
-@contlen ${Math.floor(dayjs.duration(rank.contest.endTime.diff(rank.contest.startTime)).asMinutes())}
-@problems ${rank.contest.problems.length}
-@teams ${rank.teams.length + 100}
-@submissions ${rank.submissions.length}
-`;
-
-  rank.contest.problems.forEach((p) => {
-    res += `@p ${p.label},${p.label},20,0\n`;
-  });
-
-  let teamIndex = 1;
-  const teamIdMap = new Map<string, number>();
-  const submissionsIdMap = new Map<string, Map<string, number>>();
-
-  rank.teams.forEach((team) => {
-    let name = team.name;
-
-    if (team.organization) {
-      name = `${team.organization} - ${name}`;
-    }
-
-    if (team.members) {
-      name = `${name} - ${team.membersToString}`;
-    }
-
-    res += `@t ${teamIndex},0,1,${name}\n`;
-    teamIdMap.set(team.id, teamIndex);
-    teamIndex++;
-
-    {
-      const mp = new Map<string, number>();
-      rank.contest.problems.forEach((p) => {
-        mp.set(p.id, 0);
-      });
-      submissionsIdMap.set(team.id, mp);
-    }
-  });
-
-  for (let i = 0; i < 100; i++) {
-    res += `@t ${teamIndex},0,1,Пополнить команду\n`;
-    teamIndex++;
-  }
-
-  rank.getSubmissions().forEach((submission) => {
-    const teamId = submission.teamId;
-    const problemId = submission.problemId;
-    const problem = rank.contest.problemsMap.get(problemId);
-
-    if (!problem) {
-      return;
-    }
-
-    const status = submissionStatusToCodeforcesGymDatStatus(submission.status);
-    submissionsIdMap.get(teamId)!.set(problemId, submissionsIdMap.get(teamId)!.get(problemId)! + 1);
-
-    res += `@s ${teamIdMap.get(teamId)},${problem.label},${submissionsIdMap.get(teamId)?.get(problemId)},${submission.timestamp},${status}\n`;
-  });
-
-  return res;
-}