123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576 |
- var _util = require("zrender/lib/core/util");
- var each = _util.each;
- var map = _util.map;
- var isFunction = _util.isFunction;
- var createHashMap = _util.createHashMap;
- var noop = _util.noop;
- var _task = require("./task");
- var createTask = _task.createTask;
- var _component = require("../util/component");
- var getUID = _component.getUID;
- var GlobalModel = require("../model/Global");
- var ExtensionAPI = require("../ExtensionAPI");
- var _model = require("../util/model");
- var normalizeToArray = _model.normalizeToArray;
- function Scheduler(ecInstance, api, dataProcessorHandlers, visualHandlers) {
- this.ecInstance = ecInstance;
- this.api = api;
- this.unfinished;
-
-
-
- var dataProcessorHandlers = this._dataProcessorHandlers = dataProcessorHandlers.slice();
- var visualHandlers = this._visualHandlers = visualHandlers.slice();
- this._allHandlers = dataProcessorHandlers.concat(visualHandlers);
-
- this._stageTaskMap = createHashMap();
- }
- var proto = Scheduler.prototype;
- proto.restoreData = function (ecModel, payload) {
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
- ecModel.restoreData(payload);
-
-
-
-
-
-
- this._stageTaskMap.each(function (taskRecord) {
- var overallTask = taskRecord.overallTask;
- overallTask && overallTask.dirty();
- });
- };
- proto.getPerformArgs = function (task, isBlock) {
-
- if (!task.__pipeline) {
- return;
- }
- var pipeline = this._pipelineMap.get(task.__pipeline.id);
- var pCtx = pipeline.context;
- var incremental = !isBlock && pipeline.progressiveEnabled && (!pCtx || pCtx.progressiveRender) && task.__idxInPipeline > pipeline.blockIndex;
- var step = incremental ? pipeline.step : null;
- var modDataCount = pCtx && pCtx.modDataCount;
- var modBy = modDataCount != null ? Math.ceil(modDataCount / step) : null;
- return {
- step: step,
- modBy: modBy,
- modDataCount: modDataCount
- };
- };
- proto.getPipeline = function (pipelineId) {
- return this._pipelineMap.get(pipelineId);
- };
- proto.updateStreamModes = function (seriesModel, view) {
- var pipeline = this._pipelineMap.get(seriesModel.uid);
- var data = seriesModel.getData();
- var dataLen = data.count();
-
-
-
-
- var progressiveRender = pipeline.progressiveEnabled && view.incrementalPrepareRender && dataLen >= pipeline.threshold;
- var large = seriesModel.get('large') && dataLen >= seriesModel.get('largeThreshold');
-
- var modDataCount = seriesModel.get('progressiveChunkMode') === 'mod' ? dataLen : null;
- seriesModel.pipelineContext = pipeline.context = {
- progressiveRender: progressiveRender,
- modDataCount: modDataCount,
- large: large
- };
- };
- proto.restorePipelines = function (ecModel) {
- var scheduler = this;
- var pipelineMap = scheduler._pipelineMap = createHashMap();
- ecModel.eachSeries(function (seriesModel) {
- var progressive = seriesModel.getProgressive();
- var pipelineId = seriesModel.uid;
- pipelineMap.set(pipelineId, {
- id: pipelineId,
- head: null,
- tail: null,
- threshold: seriesModel.getProgressiveThreshold(),
- progressiveEnabled: progressive && !(seriesModel.preventIncremental && seriesModel.preventIncremental()),
- blockIndex: -1,
- step: Math.round(progressive || 700),
- count: 0
- });
- pipe(scheduler, seriesModel, seriesModel.dataTask);
- });
- };
- proto.prepareStageTasks = function () {
- var stageTaskMap = this._stageTaskMap;
- var ecModel = this.ecInstance.getModel();
- var api = this.api;
- each(this._allHandlers, function (handler) {
- var record = stageTaskMap.get(handler.uid) || stageTaskMap.set(handler.uid, []);
- handler.reset && createSeriesStageTask(this, handler, record, ecModel, api);
- handler.overallReset && createOverallStageTask(this, handler, record, ecModel, api);
- }, this);
- };
- proto.prepareView = function (view, model, ecModel, api) {
- var renderTask = view.renderTask;
- var context = renderTask.context;
- context.model = model;
- context.ecModel = ecModel;
- context.api = api;
- renderTask.__block = !view.incrementalPrepareRender;
- pipe(this, model, renderTask);
- };
- proto.performDataProcessorTasks = function (ecModel, payload) {
-
- performStageTasks(this, this._dataProcessorHandlers, ecModel, payload, {
- block: true
- });
- };
- proto.performVisualTasks = function (ecModel, payload, opt) {
- performStageTasks(this, this._visualHandlers, ecModel, payload, opt);
- };
- function performStageTasks(scheduler, stageHandlers, ecModel, payload, opt) {
- opt = opt || {};
- var unfinished;
- each(stageHandlers, function (stageHandler, idx) {
- if (opt.visualType && opt.visualType !== stageHandler.visualType) {
- return;
- }
- var stageHandlerRecord = scheduler._stageTaskMap.get(stageHandler.uid);
- var seriesTaskMap = stageHandlerRecord.seriesTaskMap;
- var overallTask = stageHandlerRecord.overallTask;
- if (overallTask) {
- var overallNeedDirty;
- var agentStubMap = overallTask.agentStubMap;
- agentStubMap.each(function (stub) {
- if (needSetDirty(opt, stub)) {
- stub.dirty();
- overallNeedDirty = true;
- }
- });
- overallNeedDirty && overallTask.dirty();
- updatePayload(overallTask, payload);
- var performArgs = scheduler.getPerformArgs(overallTask, opt.block);
-
-
-
- agentStubMap.each(function (stub) {
- stub.perform(performArgs);
- });
- unfinished |= overallTask.perform(performArgs);
- } else if (seriesTaskMap) {
- seriesTaskMap.each(function (task, pipelineId) {
- if (needSetDirty(opt, task)) {
- task.dirty();
- }
- var performArgs = scheduler.getPerformArgs(task, opt.block);
-
-
-
-
-
-
-
- performArgs.skip = !stageHandler.performRawSeries && ecModel.isSeriesFiltered(task.context.model);
- updatePayload(task, payload);
- unfinished |= task.perform(performArgs);
- });
- }
- });
- function needSetDirty(opt, task) {
- return opt.setDirty && (!opt.dirtyMap || opt.dirtyMap.get(task.__pipeline.id));
- }
- scheduler.unfinished |= unfinished;
- }
- proto.performSeriesTasks = function (ecModel) {
- var unfinished;
- ecModel.eachSeries(function (seriesModel) {
-
- unfinished |= seriesModel.dataTask.perform();
- });
- this.unfinished |= unfinished;
- };
- proto.plan = function () {
-
- this._pipelineMap.each(function (pipeline) {
- var task = pipeline.tail;
- do {
- if (task.__block) {
- pipeline.blockIndex = task.__idxInPipeline;
- break;
- }
- task = task.getUpstream();
- } while (task);
- });
- };
- var updatePayload = proto.updatePayload = function (task, payload) {
- payload !== 'remain' && (task.context.payload = payload);
- };
- function createSeriesStageTask(scheduler, stageHandler, stageHandlerRecord, ecModel, api) {
- var seriesTaskMap = stageHandlerRecord.seriesTaskMap || (stageHandlerRecord.seriesTaskMap = createHashMap());
- var seriesType = stageHandler.seriesType;
- var getTargetSeries = stageHandler.getTargetSeries;
-
-
- if (stageHandler.createOnAllSeries) {
- ecModel.eachRawSeries(create);
- } else if (seriesType) {
- ecModel.eachRawSeriesByType(seriesType, create);
- } else if (getTargetSeries) {
- getTargetSeries(ecModel, api).each(create);
- }
- function create(seriesModel) {
- var pipelineId = seriesModel.uid;
-
- var task = seriesTaskMap.get(pipelineId) || seriesTaskMap.set(pipelineId, createTask({
- plan: seriesTaskPlan,
- reset: seriesTaskReset,
- count: seriesTaskCount
- }));
- task.context = {
- model: seriesModel,
- ecModel: ecModel,
- api: api,
- useClearVisual: stageHandler.isVisual && !stageHandler.isLayout,
- plan: stageHandler.plan,
- reset: stageHandler.reset,
- scheduler: scheduler
- };
- pipe(scheduler, seriesModel, task);
- }
- var pipelineMap = scheduler._pipelineMap;
- seriesTaskMap.each(function (task, pipelineId) {
- if (!pipelineMap.get(pipelineId)) {
- task.dispose();
- seriesTaskMap.removeKey(pipelineId);
- }
- });
- }
- function createOverallStageTask(scheduler, stageHandler, stageHandlerRecord, ecModel, api) {
- var overallTask = stageHandlerRecord.overallTask = stageHandlerRecord.overallTask
- || createTask({
- reset: overallTaskReset
- });
- overallTask.context = {
- ecModel: ecModel,
- api: api,
- overallReset: stageHandler.overallReset,
- scheduler: scheduler
- };
- var agentStubMap = overallTask.agentStubMap = overallTask.agentStubMap || createHashMap();
- var seriesType = stageHandler.seriesType;
- var getTargetSeries = stageHandler.getTargetSeries;
- var overallProgress = true;
- var modifyOutputEnd = stageHandler.modifyOutputEnd;
-
-
-
- if (seriesType) {
- ecModel.eachRawSeriesByType(seriesType, createStub);
- } else if (getTargetSeries) {
- getTargetSeries(ecModel, api).each(createStub);
- }
-
-
-
- else {
- overallProgress = false;
- each(ecModel.getSeries(), createStub);
- }
- function createStub(seriesModel) {
- var pipelineId = seriesModel.uid;
- var stub = agentStubMap.get(pipelineId);
- if (!stub) {
- stub = agentStubMap.set(pipelineId, createTask({
- reset: stubReset,
- onDirty: stubOnDirty
- }));
-
- overallTask.dirty();
- }
- stub.context = {
- model: seriesModel,
- overallProgress: overallProgress,
- modifyOutputEnd: modifyOutputEnd
- };
- stub.agent = overallTask;
- stub.__block = overallProgress;
- pipe(scheduler, seriesModel, stub);
- }
- var pipelineMap = scheduler._pipelineMap;
- agentStubMap.each(function (stub, pipelineId) {
- if (!pipelineMap.get(pipelineId)) {
- stub.dispose();
-
- overallTask.dirty();
- agentStubMap.removeKey(pipelineId);
- }
- });
- }
- function overallTaskReset(context) {
- context.overallReset(context.ecModel, context.api, context.payload);
- }
- function stubReset(context, upstreamContext) {
- return context.overallProgress && stubProgress;
- }
- function stubProgress() {
- this.agent.dirty();
- this.getDownstream().dirty();
- }
- function stubOnDirty() {
- this.agent && this.agent.dirty();
- }
- function seriesTaskPlan(context) {
- return context.plan && context.plan(context.model, context.ecModel, context.api, context.payload);
- }
- function seriesTaskReset(context) {
- if (context.useClearVisual) {
- context.data.clearAllVisual();
- }
- var resetDefines = context.resetDefines = normalizeToArray(context.reset(context.model, context.ecModel, context.api, context.payload));
- return resetDefines.length > 1 ? map(resetDefines, function (v, idx) {
- return makeSeriesTaskProgress(idx);
- }) : singleSeriesTaskProgress;
- }
- var singleSeriesTaskProgress = makeSeriesTaskProgress(0);
- function makeSeriesTaskProgress(resetDefineIdx) {
- return function (params, context) {
- var data = context.data;
- var resetDefine = context.resetDefines[resetDefineIdx];
- if (resetDefine && resetDefine.dataEach) {
- for (var i = params.start; i < params.end; i++) {
- resetDefine.dataEach(data, i);
- }
- } else if (resetDefine && resetDefine.progress) {
- resetDefine.progress(params, data);
- }
- };
- }
- function seriesTaskCount(context) {
- return context.data.count();
- }
- function pipe(scheduler, seriesModel, task) {
- var pipelineId = seriesModel.uid;
- var pipeline = scheduler._pipelineMap.get(pipelineId);
- !pipeline.head && (pipeline.head = task);
- pipeline.tail && pipeline.tail.pipe(task);
- pipeline.tail = task;
- task.__idxInPipeline = pipeline.count++;
- task.__pipeline = pipeline;
- }
- Scheduler.wrapStageHandler = function (stageHandler, visualType) {
- if (isFunction(stageHandler)) {
- stageHandler = {
- overallReset: stageHandler,
- seriesType: detectSeriseType(stageHandler)
- };
- }
- stageHandler.uid = getUID('stageHandler');
- visualType && (stageHandler.visualType = visualType);
- return stageHandler;
- };
- function detectSeriseType(legacyFunc) {
- seriesType = null;
- try {
-
- legacyFunc(ecModelMock, apiMock);
- } catch (e) {}
- return seriesType;
- }
- var ecModelMock = {};
- var apiMock = {};
- var seriesType;
- mockMethods(ecModelMock, GlobalModel);
- mockMethods(apiMock, ExtensionAPI);
- ecModelMock.eachSeriesByType = ecModelMock.eachRawSeriesByType = function (type) {
- seriesType = type;
- };
- ecModelMock.eachComponent = function (cond) {
- if (cond.mainType === 'series' && cond.subType) {
- seriesType = cond.subType;
- }
- };
- function mockMethods(target, Clz) {
-
- for (var name in Clz.prototype) {
-
- target[name] = noop;
- }
-
- }
- var _default = Scheduler;
- module.exports = _default;
|