simplify validPetPoses out of the server directory
This commit is contained in:
parent
88862f9ce7
commit
453953dded
4 changed files with 84 additions and 6270 deletions
|
|
@ -1,4 +1,87 @@
|
||||||
import getValidPetPoses from "../src/server/getValidPetPoses";
|
import connectToDb from "../src/server/db";
|
||||||
|
import { getPoseFromPetState, normalizeRow } from "../src/server/util";
|
||||||
|
|
||||||
|
export async function getValidPetPoses() {
|
||||||
|
const db = await connectToDb();
|
||||||
|
|
||||||
|
const numSpeciesPromise = getNumSpecies(db);
|
||||||
|
const numColorsPromise = getNumColors(db);
|
||||||
|
const distinctPetStatesPromise = getDistinctPetStates(db);
|
||||||
|
|
||||||
|
const [numSpecies, numColors, distinctPetStates] = await Promise.all([
|
||||||
|
numSpeciesPromise,
|
||||||
|
numColorsPromise,
|
||||||
|
distinctPetStatesPromise,
|
||||||
|
]);
|
||||||
|
|
||||||
|
const poseStrs = new Set();
|
||||||
|
for (const petState of distinctPetStates) {
|
||||||
|
const { speciesId, colorId } = petState;
|
||||||
|
const pose = getPoseFromPetState(petState);
|
||||||
|
const poseStr = `${speciesId}-${colorId}-${pose}`;
|
||||||
|
poseStrs.add(poseStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
function hasPose(speciesId, colorId, pose) {
|
||||||
|
const poseStr = `${speciesId}-${colorId}-${pose}`;
|
||||||
|
return poseStrs.has(poseStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
const numPairs = numSpecies * numColors;
|
||||||
|
const buffer = Buffer.alloc(numPairs + 2);
|
||||||
|
buffer.writeUInt8(numSpecies, 0);
|
||||||
|
buffer.writeUInt8(numColors, 1);
|
||||||
|
|
||||||
|
for (let speciesId = 1; speciesId <= numSpecies; speciesId++) {
|
||||||
|
const speciesIndex = speciesId - 1;
|
||||||
|
for (let colorId = 1; colorId <= numColors; colorId++) {
|
||||||
|
const colorIndex = colorId - 1;
|
||||||
|
|
||||||
|
// We fill in the high bits first, and shift left as we go!
|
||||||
|
let byte = 0;
|
||||||
|
byte += hasPose(speciesId, colorId, "UNKNOWN") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "UNCONVERTED") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "SICK_FEM") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "SAD_FEM") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "HAPPY_FEM") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "SICK_MASC") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "SAD_MASC") ? 1 : 0;
|
||||||
|
byte <<= 1;
|
||||||
|
byte += hasPose(speciesId, colorId, "HAPPY_MASC") ? 1 : 0;
|
||||||
|
|
||||||
|
buffer.writeUInt8(byte, speciesIndex * numColors + colorIndex + 2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return buffer;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getNumSpecies(db) {
|
||||||
|
const [rows, _] = await db.query(`SELECT count(*) FROM species`);
|
||||||
|
return rows[0]["count(*)"];
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getNumColors(db) {
|
||||||
|
const [rows, _] = await db.query(
|
||||||
|
`SELECT count(*) FROM colors WHERE prank = 0`
|
||||||
|
);
|
||||||
|
return rows[0]["count(*)"];
|
||||||
|
}
|
||||||
|
|
||||||
|
async function getDistinctPetStates(db) {
|
||||||
|
const [rows, _] = await db.query(`
|
||||||
|
SELECT DISTINCT species_id, color_id, mood_id, female, unconverted
|
||||||
|
FROM pet_states
|
||||||
|
INNER JOIN pet_types ON pet_types.id = pet_states.pet_type_id
|
||||||
|
WHERE color_id >= 1`);
|
||||||
|
return rows.map(normalizeRow);
|
||||||
|
}
|
||||||
|
|
||||||
export default async (req, res) => {
|
export default async (req, res) => {
|
||||||
const buffer = await getValidPetPoses();
|
const buffer = await getValidPetPoses();
|
||||||
|
|
|
||||||
File diff suppressed because it is too large
Load diff
|
|
@ -1,85 +0,0 @@
|
||||||
import connectToDb from "./db";
|
|
||||||
|
|
||||||
import { getPoseFromPetState, normalizeRow } from "./util";
|
|
||||||
|
|
||||||
export default async function getValidPetPoses() {
|
|
||||||
const db = await connectToDb();
|
|
||||||
|
|
||||||
const numSpeciesPromise = getNumSpecies(db);
|
|
||||||
const numColorsPromise = getNumColors(db);
|
|
||||||
const distinctPetStatesPromise = getDistinctPetStates(db);
|
|
||||||
|
|
||||||
const [numSpecies, numColors, distinctPetStates] = await Promise.all([
|
|
||||||
numSpeciesPromise,
|
|
||||||
numColorsPromise,
|
|
||||||
distinctPetStatesPromise,
|
|
||||||
]);
|
|
||||||
|
|
||||||
const poseStrs = new Set();
|
|
||||||
for (const petState of distinctPetStates) {
|
|
||||||
const { speciesId, colorId } = petState;
|
|
||||||
const pose = getPoseFromPetState(petState);
|
|
||||||
const poseStr = `${speciesId}-${colorId}-${pose}`;
|
|
||||||
poseStrs.add(poseStr);
|
|
||||||
}
|
|
||||||
|
|
||||||
function hasPose(speciesId, colorId, pose) {
|
|
||||||
const poseStr = `${speciesId}-${colorId}-${pose}`;
|
|
||||||
return poseStrs.has(poseStr);
|
|
||||||
}
|
|
||||||
|
|
||||||
const numPairs = numSpecies * numColors;
|
|
||||||
const buffer = Buffer.alloc(numPairs + 2);
|
|
||||||
buffer.writeUInt8(numSpecies, 0);
|
|
||||||
buffer.writeUInt8(numColors, 1);
|
|
||||||
|
|
||||||
for (let speciesId = 1; speciesId <= numSpecies; speciesId++) {
|
|
||||||
const speciesIndex = speciesId - 1;
|
|
||||||
for (let colorId = 1; colorId <= numColors; colorId++) {
|
|
||||||
const colorIndex = colorId - 1;
|
|
||||||
|
|
||||||
// We fill in the high bits first, and shift left as we go!
|
|
||||||
let byte = 0;
|
|
||||||
byte += hasPose(speciesId, colorId, "UNKNOWN") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "UNCONVERTED") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "SICK_FEM") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "SAD_FEM") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "HAPPY_FEM") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "SICK_MASC") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "SAD_MASC") ? 1 : 0;
|
|
||||||
byte <<= 1;
|
|
||||||
byte += hasPose(speciesId, colorId, "HAPPY_MASC") ? 1 : 0;
|
|
||||||
|
|
||||||
buffer.writeUInt8(byte, speciesIndex * numColors + colorIndex + 2);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return buffer;
|
|
||||||
}
|
|
||||||
|
|
||||||
async function getNumSpecies(db) {
|
|
||||||
const [rows, _] = await db.query(`SELECT count(*) FROM species`);
|
|
||||||
return rows[0]["count(*)"];
|
|
||||||
}
|
|
||||||
|
|
||||||
async function getNumColors(db) {
|
|
||||||
const [rows, _] = await db.query(
|
|
||||||
`SELECT count(*) FROM colors WHERE prank = 0`
|
|
||||||
);
|
|
||||||
return rows[0]["count(*)"];
|
|
||||||
}
|
|
||||||
|
|
||||||
async function getDistinctPetStates(db) {
|
|
||||||
const [rows, _] = await db.query(`
|
|
||||||
SELECT DISTINCT species_id, color_id, mood_id, female, unconverted
|
|
||||||
FROM pet_states
|
|
||||||
INNER JOIN pet_types ON pet_types.id = pet_states.pet_type_id
|
|
||||||
WHERE color_id >= 1`);
|
|
||||||
return rows.map(normalizeRow);
|
|
||||||
}
|
|
||||||
|
|
@ -1,17 +0,0 @@
|
||||||
import getValidPetPoses from "./getValidPetPoses";
|
|
||||||
|
|
||||||
describe("getValidPetPoses", () => {
|
|
||||||
it("gets them and writes them to a buffer", async () => {
|
|
||||||
const buffer = await getValidPetPoses();
|
|
||||||
expect(asBinaryString(buffer)).toMatchSnapshot();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
function asBinaryString(buffer) {
|
|
||||||
let str = "";
|
|
||||||
for (let i = 0; i < buffer.length; i++) {
|
|
||||||
const byte = buffer.readUInt8(i);
|
|
||||||
str += byte.toString(2).padStart(8, "0") + "\n";
|
|
||||||
}
|
|
||||||
return str;
|
|
||||||
}
|
|
||||||
Loading…
Reference in a new issue