HTMLParserScheduler.h [plain text]
#ifndef HTMLParserScheduler_h
#define HTMLParserScheduler_h
#include <limits.h>
#include "NestingLevelIncrementer.h"
#include "Timer.h"
#include <wtf/CurrentTime.h>
#include <wtf/PassOwnPtr.h>
#include "WebCoreThread.h"
namespace WebCore {
class HTMLDocumentParser;
class PumpSession : public NestingLevelIncrementer {
public:
PumpSession(unsigned& nestingLevel)
: NestingLevelIncrementer(nestingLevel)
, processedTokens(INT_MAX)
, startTime(0)
, needsYield(false)
{
}
int processedTokens;
double startTime;
bool needsYield;
};
class HTMLParserScheduler {
WTF_MAKE_NONCOPYABLE(HTMLParserScheduler); WTF_MAKE_FAST_ALLOCATED;
public:
static PassOwnPtr<HTMLParserScheduler> create(HTMLDocumentParser* parser)
{
return adoptPtr(new HTMLParserScheduler(parser));
}
~HTMLParserScheduler();
void checkForYieldBeforeToken(PumpSession& session)
{
if (WebThreadShouldYield())
session.needsYield = true;
if (session.processedTokens > m_parserChunkSize) {
if (!session.startTime)
session.startTime = currentTime();
session.processedTokens = 0;
double elapsedTime = currentTime() - session.startTime;
if (elapsedTime > m_parserTimeLimit)
session.needsYield = true;
}
++session.processedTokens;
}
void checkForYieldBeforeScript(PumpSession&);
void scheduleForResume();
bool isScheduledForResume() const { return m_isSuspendedWithActiveTimer || m_continueNextChunkTimer.isActive(); }
void suspend();
void resume();
private:
HTMLParserScheduler(HTMLDocumentParser*);
void continueNextChunkTimerFired(Timer<HTMLParserScheduler>*);
HTMLDocumentParser* m_parser;
double m_parserTimeLimit;
int m_parserChunkSize;
Timer<HTMLParserScheduler> m_continueNextChunkTimer;
bool m_isSuspendedWithActiveTimer;
};
}
#endif