From 603155e6a8a961b22af9c3af331e612f6a16178e Mon Sep 17 00:00:00 2001
From: Pregmodder <pregmodder@gmail.com>
Date: Sat, 20 Jun 2020 01:26:25 -0400
Subject: [PATCH] Fertility diet superfetation support

---
 src/interaction/slaveInteract.js | 2 +-
 src/uncategorized/saDiet.tw      | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/interaction/slaveInteract.js b/src/interaction/slaveInteract.js
index 0d22fd72612..06514d42db4 100644
--- a/src/interaction/slaveInteract.js
+++ b/src/interaction/slaveInteract.js
@@ -802,7 +802,7 @@ App.UI.SlaveInteract.diet = function(slave) {
 	if (slave.balls > 0 && V.cumProDiet === 1) {
 		production.push({text: `Cum production`, updateSlave: {diet: "cum production"}});
 	}
-	if (isFertile(slave) && slave.preg === 0 && V.dietFertility === 1) {
+	if (((isFertile(slave) && slave.preg === 0) || (slave.geneticQuirks.superfetation === 2 && canGetPregnant(slave) && V.geneticMappingUpgrade !== 0)) && V.dietFertility === 1) {
 		production.push({text: `Fertility`, updateSlave: {diet: "fertility"}});
 	}
 
diff --git a/src/uncategorized/saDiet.tw b/src/uncategorized/saDiet.tw
index 333f82a7034..b3ad1278579 100644
--- a/src/uncategorized/saDiet.tw
+++ b/src/uncategorized/saDiet.tw
@@ -1016,7 +1016,7 @@
 		<</if>>
 	<</if>>
 <<case "fertility">> /* + ovum and small boosts to energy and attrXY */
-	<<if !isFertile($slaves[$i]) || $slaves[$i].preg != 0>>
+	<<if !isFertile($slaves[$i]) || ($slaves[$i].preg < 0) || ($slaves[$i].preg > 0 && ($slaves[$i].geneticQuirks.superfetation != 2 || $geneticMappingUpgrade < 1))>>
 		<<if $slaves[$i].pregKnown == 0 && $slaves[$i].preg > 0>>
 			The diet is not properly working; tests reveal the reason is a @@.lime;new pregnancy.@@
 			<<set $slaves[$i].pregKnown = 1>>
-- 
GitLab