From 7aa66f438f4c7854fa9bacb4c15584920e41c278 Mon Sep 17 00:00:00 2001
From: syuilo <syuilotan@yahoo.co.jp>
Date: Sun, 5 May 2019 20:12:35 +0900
Subject: Resolve #4853

---
 src/misc/aiscript/evaluator.ts | 21 +++++++++++++++++++++
 src/misc/aiscript/index.ts     |  1 +
 2 files changed, 22 insertions(+)

(limited to 'src/misc/aiscript')

diff --git a/src/misc/aiscript/evaluator.ts b/src/misc/aiscript/evaluator.ts
index 78b6acf35c..5e511576c8 100644
--- a/src/misc/aiscript/evaluator.ts
+++ b/src/misc/aiscript/evaluator.ts
@@ -178,6 +178,27 @@ export class ASEvaluator {
 			seedRandom: (seed: any, probability: number) => Math.floor(seedrandom(seed)() * 100) < probability,
 			seedRannum: (seed: any, min: number, max: number) => min + Math.floor(seedrandom(seed)() * (max - min + 1)),
 			seedRandomPick: (seed: any, list: any[]) => list[Math.floor(seedrandom(seed)() * list.length)],
+			dailyRPWPM: (list: string[]) => {
+				const xs = [];
+				let totalFactor = 0;
+				for (const x of list) {
+					const parts = x.split(' ');
+					const factor = parseInt(parts.pop()!, 10);
+					const text = parts.join(' ');
+					totalFactor += factor;
+					xs.push({ factor, text });
+				}
+				const r = seedrandom(`${day}:${block.id}`)() * totalFactor;
+				let stackedFactor = 0;
+				for (const x of xs) {
+					if (r >= stackedFactor && r <= x.factor) {
+						return x.text;
+					} else {
+						stackedFactor += x.factor;
+					}
+				}
+				return xs[0].text;
+			},
 		};
 
 		const fnName = block.type;
diff --git a/src/misc/aiscript/index.ts b/src/misc/aiscript/index.ts
index 1ce09b7bdb..61a6b7b139 100644
--- a/src/misc/aiscript/index.ts
+++ b/src/misc/aiscript/index.ts
@@ -81,6 +81,7 @@ export const funcDefs: Record<string, { in: any[]; out: any; category: string; i
 	randomPick:      { in: [0],                            out: 0,             category: 'random',     icon: faDice, },
 	dailyRandomPick: { in: [0],                            out: 0,             category: 'random',     icon: faDice, },
 	seedRandomPick:  { in: [null, 0],                      out: 0,             category: 'random',     icon: faDice, },
+	dailyRPWPM:      { in: ['stringArray'],                out: 'string',      category: 'random',     icon: faDice, }, // dailyRandomPickWithProbabilityMapping
 };
 
 export const literalDefs: Record<string, { out: any; category: string; icon: any; }> = {
-- 
cgit v1.2.3-freya