/*
 * 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 { BaseStruct, isFrameContainPoint, ns2x, Rect, Render } from './ProcedureWorkerCommon';
import { TraceRow } from '../../component/trace/base/TraceRow';

export class FpsRender extends Render {
  renderMainThread(
    req: {
      context: CanvasRenderingContext2D;
      useCache: boolean;
      type: string;
    },
    row: TraceRow<FpsStruct>
  ): void {
    let fpsList = row.dataList;
    let fpsFilter = row.dataListCache;
    fps(
      fpsList,
      fpsFilter,
      TraceRow.range!.startNS || 0,
      TraceRow.range!.endNS || 0,
      TraceRow.range!.totalNS || 0,
      row.frame,
      req.useCache || !TraceRow.range!.refresh
    );
    req.context.beginPath();
    let fpsFind = false;
    for (let re of fpsFilter) {
      FpsStruct.draw(req.context, re);
      if (row.isHover && re.frame && isFrameContainPoint(re.frame, row.hoverX, row.hoverY)) {
        FpsStruct.hoverFpsStruct = re;
        fpsFind = true;
      }
    }
    if (!fpsFind && row.isHover) {
      FpsStruct.hoverFpsStruct = undefined;
    }
    req.context.closePath();
    let maxFps = FpsStruct.maxFps + 'FPS';
    let textMetrics = req.context.measureText(maxFps);
    req.context.globalAlpha = 0.8;
    req.context.fillStyle = '#f0f0f0';
    req.context.fillRect(0, 5, textMetrics.width + 8, 18);
    req.context.globalAlpha = 1;
    req.context.fillStyle = '#333';
    req.context.textBaseline = 'middle';
    req.context.fillText(maxFps, 4, 5 + 9);
  }
}

export function fps(
  list: Array<FpsStruct>,
  res: Array<FpsStruct>,
  startNS: number,
  endNS: number,
  totalNS: number,
  frame: Rect,
  use: boolean
): void {
  if (use && res.length > 0) {
    res.forEach((it) => FpsStruct.setFrame(it, 5, startNS, endNS, totalNS, frame));
    return;
  }
  FpsStruct.maxFps = 0;
  res.length = 0;
  if (list) {
    for (let i = 0, len = list.length; i < len; i++) {
      let it = list[i];
      if ((it.fps || 0) > FpsStruct.maxFps) {
        FpsStruct.maxFps = it.fps || 0;
      }
      if (i === list.length - 1) {
        it.dur = endNS - (it.startNS || 0);
      } else {
        it.dur = (list[i + 1].startNS || 0) - (it.startNS || 0);
      }
      if ((it.startNS || 0) + (it.dur || 0) > startNS && (it.startNS || 0) < endNS) {
        FpsStruct.setFrame(list[i], 5, startNS, endNS, totalNS, frame);
        setFPSFilter(list, i, res);
      }
    }
  }
}

function setFPSFilter(list: Array<FpsStruct>, i: number, res: Array<FpsStruct>): void {
  if (
    i > 0 &&
    (list[i - 1].frame?.x || 0) === (list[i].frame?.x || 0) &&
    (list[i - 1].frame?.width || 0) === (list[i].frame?.width || 0)
  ) {
  } else {
    res.push(list[i]);
  }
}

export class FpsStruct extends BaseStruct {
  static maxFps: number = 0;
  static maxFpsName: string = '0 FPS';
  static hoverFpsStruct: FpsStruct | undefined;
  static selectFpsStruct: FpsStruct | undefined;
  fps: number | undefined;
  startNS: number | undefined = 0;
  dur: number | undefined; //自补充，数据库没有返回

  static draw(fpsContext: CanvasRenderingContext2D, data: FpsStruct): void {
    if (data.frame) {
      let width = data.frame.width || 0;
      fpsContext.fillStyle = '#535da6';
      fpsContext.strokeStyle = '#535da6';
      if (data === FpsStruct.hoverFpsStruct) {
        fpsContext.lineWidth = 1;
        fpsContext.globalAlpha = 0.6;
        let drawHeight: number = ((data.fps || 0) * (data.frame.height || 0) * 1.0) / FpsStruct.maxFps;
        fpsContext.fillRect(data.frame.x, data.frame.y + data.frame.height - drawHeight, width, drawHeight);
        fpsContext.beginPath();
        fpsContext.arc(data.frame.x, data.frame.y + data.frame.height - drawHeight, 3, 0, 2 * Math.PI, true);
        fpsContext.fill();
        fpsContext.globalAlpha = 1.0;
        fpsContext.stroke();
        fpsContext.beginPath();
        fpsContext.moveTo(data.frame.x + 3, data.frame.y + data.frame.height - drawHeight);
        fpsContext.lineWidth = 3;
        fpsContext.lineTo(data.frame.x + width, data.frame.y + data.frame.height - drawHeight);
        fpsContext.stroke();
      } else {
        fpsContext.globalAlpha = 0.6;
        fpsContext.lineWidth = 1;
        let drawHeight: number = ((data.fps || 0) * (data.frame.height || 0) * 1.0) / FpsStruct.maxFps;
        fpsContext.fillRect(data.frame.x, data.frame.y + data.frame.height - drawHeight, width, drawHeight);
      }
    }
    fpsContext.globalAlpha = 1.0;
    fpsContext.lineWidth = 1;
  }

  static setFrame(
    fpsNode: FpsStruct,
    padding: number,
    startNS: number,
    endNS: number,
    totalNS: number,
    frame: Rect
  ): void {
    let fpsLeftPointX: number;
    let fpsRightPointX: number;
    if ((fpsNode.startNS || 0) < startNS) {
      fpsLeftPointX = 0;
    } else {
      fpsLeftPointX = ns2x(fpsNode.startNS || 0, startNS, endNS, totalNS, frame);
    }
    if ((fpsNode.startNS || 0) + (fpsNode.dur || 0) > endNS) {
      fpsRightPointX = frame.width;
    } else {
      fpsRightPointX = ns2x((fpsNode.startNS || 0) + (fpsNode.dur || 0), startNS, endNS, totalNS, frame);
    }
    let getV: number = fpsRightPointX - fpsLeftPointX <= 1 ? 1 : fpsRightPointX - fpsLeftPointX;
    let rectangle: Rect = new Rect(
      Math.floor(fpsLeftPointX),
      Math.ceil(frame.y + padding),
      Math.ceil(getV),
      Math.floor(frame.height - padding * 2)
    );
    fpsNode.frame = rectangle;
  }
}

const textPadding = 2;