/*
 * Copyright (C) 2022 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { SpSystemTrace } from '../SpSystemTrace';
import { info } from '../../../log/Log';
import { TraceRow } from '../trace/base/TraceRow';
import { procedurePool } from '../../database/Procedure';
import { CpuRender, CpuStruct } from '../../database/ui-worker/cpu/ProcedureWorkerCPU';
import { renders } from '../../database/ui-worker/ProcedureWorker';
import { Utils } from '../trace/base/Utils';
import { cpuDataSender } from '../../database/data-trafic/CpuDataSender';
import { queryCpuCount, queryCpuMax, queryCpuSchedSlice } from '../../database/sql/Cpu.sql';
import { rowThreadHandler } from './SpChartManager';
import { SpStatisticsHttpUtil } from '../../../statistics/util/SpStatisticsHttpUtil';

export class SpCpuChart {
  private trace: SpSystemTrace;

  constructor(trace: SpSystemTrace) {
    this.trace = trace;
  }

  private cpuSupplierFrame(traceRow: TraceRow<CpuStruct>, cpuId: number): void {
    traceRow.supplierFrame = async (): Promise<CpuStruct[]> => {
      const res = await cpuDataSender(cpuId, traceRow);
      const filterList = SpSystemTrace.keyPathList.filter((item) => {
        return item.cpu === cpuId;
      });
      res.push(...filterList);
      res.forEach((it, i, arr) => {
        let p = Utils.getInstance().getProcessMap().get(it.processId!);
        let t = Utils.getInstance().getThreadMap().get(it.tid!);
        let slice = Utils.getInstance().getSchedSliceMap().get(`${it.id}-${it.startTime}`);
        if (slice) {
          it.end_state = slice.endState;
          it.priority = slice.priority;
        }
        it.processName = p;
        it.processCmdLine = p;
        it.name = t;
        it.type = 'thread';
      });
      return res;
    };
  }

  private cpuThreadHandler(traceRow: TraceRow<CpuStruct>, i1: number): void {
    traceRow.onThreadHandler = (useCache: boolean, buf: ArrayBuffer | undefined | null): void => {
      let context: CanvasRenderingContext2D;
      if (traceRow.currentContext) {
        context = traceRow.currentContext;
      } else {
        context = traceRow.collect ? this.trace.canvasFavoritePanelCtx! : this.trace.canvasPanelCtx!;
      }
      traceRow.canvasSave(context);
      (renders['cpu-data'] as CpuRender).renderMainThread(
        {
          ctx: context,
          useCache: useCache,
          type: `cpu-data-${i1}`,
          translateY: traceRow.translateY,
        },
        traceRow
      );
      traceRow.canvasRestore(context, this.trace);
    };
  }

  // @ts-ignore
  async init(cpuDataCount?: Map<number, number>, parentRow?: TraceRow<unknown>, traceId?: string): Promise<void> {
    let CpuStartTime = new Date().getTime();
    let array = await queryCpuMax(traceId);
    let cpuCountResult = await queryCpuCount(traceId);
    if (cpuCountResult && cpuCountResult.length > 0 && cpuCountResult[0]) {
      // @ts-ignore
      Utils.getInstance().setWinCpuCount(cpuCountResult[0].cpuCount, traceId);
    } else {
      Utils.getInstance().setWinCpuCount(0, traceId);
    }
    let cpuSchedSlice = await queryCpuSchedSlice(traceId);
    this.initSchedSliceData(cpuSchedSlice, traceId);
    info('Cpu trace row data size is: ', array.length);
    if (array && array.length > 0 && array[0]) {
      // 有cpu泳道，统计ftrace插件，cpu插件
      let requestBody = {
        eventData: {
          plugin: ['ftrace-plugin', 'cpu-plugin']
        }
      };
      SpStatisticsHttpUtil.recordPluginUsage(requestBody);
      //@ts-ignore
      let cpuMax = array[0].cpu + 1;
      Utils.getInstance().setCpuCount(cpuMax, traceId);
      for (let i1 = 0; i1 < cpuMax; i1++) {
        if (cpuDataCount && (cpuDataCount.get(i1) || 0) > 0) {
          let traceRow = this.createCpuRow(i1, traceId);
          if (parentRow) {
            parentRow.addChildTraceRow(traceRow);
          } else {
            this.trace.rowsEL?.appendChild(traceRow);
          }
        }
      }
    }
    let CpuDurTime = new Date().getTime() - CpuStartTime;
    info('The time to load the Cpu data is: ', CpuDurTime);
  }

  createCpuRow(cpuId: number, traceId?: string): TraceRow<CpuStruct> {
    let traceRow = TraceRow.skeleton<CpuStruct>(traceId);
    traceRow.rowId = `${cpuId}`;
    traceRow.rowType = TraceRow.ROW_TYPE_CPU;
    traceRow.rowParentId = '';
    traceRow.style.height = '30px';
    traceRow.name = `Cpu ${cpuId}`;
    traceRow.favoriteChangeHandler = this.trace.favoriteChangeHandler;
    traceRow.selectChangeHandler = this.trace.selectChangeHandler;
    traceRow.supplierFrame = async (): Promise<CpuStruct[]> => {
      let res = await cpuDataSender(cpuId, traceRow, traceId);
      const filterList = SpSystemTrace.keyPathList.filter((item): boolean => {
        return item.cpu === cpuId;
      });
      res.push(...filterList);
      res.forEach((it, i, arr): void => {
        let p = Utils.getInstance().getProcessMap(traceId).get(it.processId!);
        let t = Utils.getInstance().getThreadMap(traceId).get(it.tid!);
        let slice = Utils.getInstance().getSchedSliceMap(traceId).get(`${it.id}-${it.startTime}`);
        if (slice) {
          it.end_state = slice.endState;
          it.priority = slice.priority;
        }
        it.processName = p;
        it.processCmdLine = p;
        it.name = t;
        it.type = 'thread';
      });
      return res;
    };
    traceRow.focusHandler = (): void => {
      this.trace?.displayTip(
        traceRow,
        CpuStruct.hoverCpuStruct,
        `<span>P：${CpuStruct.hoverCpuStruct?.processName || 'Process'} [${CpuStruct.hoverCpuStruct?.processId
        }]</span><span>T：${CpuStruct.hoverCpuStruct?.name} [${CpuStruct.hoverCpuStruct?.tid}] [Prio:${CpuStruct.hoverCpuStruct?.priority || 0
        }]</span>`
      );
    };
    traceRow.findHoverStruct = (): void => {
      CpuStruct.hoverCpuStruct = traceRow.getHoverStruct();
    };
    traceRow.onThreadHandler = rowThreadHandler<CpuRender>('cpu-data', 'ctx', {
      type: `cpu-data-${cpuId}`,
      translateY: traceRow.translateY,
    }, traceRow, this.trace);
    return traceRow;
  }

  initProcessThreadStateData = async (progress: Function): Promise<void> => {
    let time = new Date().getTime();
    progress('StateProcessThread', 93);
    procedurePool.submitWithName('logic0', 'spt-init', {}, undefined, (res: unknown) => { });
    let durTime = new Date().getTime() - time;
    info('The time to load the first ProcessThreadState data is: ', durTime);
  };

  initCpuIdle0Data = async (progress: Function): Promise<void> => {
    let time = new Date().getTime();
    progress('CPU Idle', 94);
    procedurePool.submitWithName(
      'logic0',
      'scheduling-getCpuIdle0',
      {
        // @ts-ignore
        endTs: (window as unknown).recordEndNS, // @ts-ignore
        total: (window as unknown).totalNS,
      },
      undefined,
      (res: unknown) => { }
    );
    let durTime = new Date().getTime() - time;
    info('The time to load the first CPU Idle0 data is: ', durTime);
  };

  initSchedSliceData(arr: unknown[], traceId?: string): void {
    Utils.getInstance().getSchedSliceMap(traceId).clear();
    arr.forEach((value) => {
      Utils.getInstance().getSchedSliceMap(traceId). // @ts-ignore
        set(`${value.itid}-${value.ts}`, { endState: value.endState, priority: value.priority });
    });
  }

  initSchedulingPTData = async (progress: Function): Promise<void> => {
    let time = new Date().getTime();
    progress('CPU Idle', 94);
    procedurePool.submitWithName('logic0', 'scheduling-getProcessAndThread', {}, undefined, (res: unknown) => { });
    let durTime = new Date().getTime() - time;
    info('The time to load the first CPU Idle0 data is: ', durTime);
  };

  initSchedulingFreqData = async (progress: Function): Promise<void> => {
    let time = new Date().getTime();
    progress('CPU Scheduling Freq', 94);
    procedurePool.submitWithName('logic0', 'scheduling-initFreqData', {}, undefined, (res: unknown) => { });
    let durTime = new Date().getTime() - time;
    info('The time to load the first CPU Idle0 data is: ', durTime);
  };
}