Better Anki forecasting for Review Heatmap with Anki Simulator

DECK_ID = 1234567890123 # replace with the deck ID of the deck you need to forecast

from datetime import datetime, timedelta
import review_heatmap
from review_simulator import collection_simulator, review_simulator

class MWStandin:
	pass

def mp_cardsDue(self, start=None, stop=None):
	mw = MWStandin()
	mw.col = self.col
	col_sim = collection_simulator.CollectionSimulator(mw)
	
	conf = self.col.decks.confForDid(DECK_ID)
	
	learningSteps = conf["new"]["delays"]
	lapseSteps = conf["lapse"]["delays"]
	
	dateArray = col_sim.generate_for_deck(
		DECK_ID,
		365, # daysToSimulate
		conf["new"]["perDay"], # newCardsPerDay
		conf["new"]["initialFactor"] / 10.0, # startingEase
		len(learningSteps),
		len(lapseSteps),
		True, # includeOverdueCards
		True, # includeSuspendedNewCards
		0, # newCardsToGenerate
	)
	
	sim = review_simulator.ReviewSimulator(
		dateArray,
		365, # daysToSimulate
		conf["new"]["perDay"], # newCardsPerDay
		conf["rev"]["ivlFct"], # intervalModifier
		conf["rev"]["perDay"], # maxReviewsPerDay
		learningSteps,
		lapseSteps,
		conf["new"]["ints"][0], # graduatingInterval
		conf["lapse"]["mult"], # newLapseInterval
		conf["rev"]["maxIvl"], # maxInterval
		[100] * len(learningSteps), # percentagesCorrectForLearningSteps
		[100] * len(lapseSteps), # percentagesCorrectForLapseSteps
		100, # percentageGoodYoung
		100, # percentageGoodMature
		0,  # Percentage hard is set to 0
		0,  # Percentage easy is set to 0
		self.col.schedVer()
	)
	
	result = sim.simulate()
	result_dt = []
	
	for entry in result:
		if entry['y']:
			dt = datetime.strptime(entry['x'], '%Y-%m-%d')
			dt += timedelta(days=1) # Appears to be necessary for some reason
			result_dt.append([int(dt.timestamp()), -entry['y']]) # Review Heatmap expects negative numbers
	
	return result_dt

review_heatmap.activity.ActivityReporter._cardsDue = mp_cardsDue