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

export class MemoryAbilityRender extends Render {
  renderMainThread(
    req: {
      context: CanvasRenderingContext2D;
      useCache: boolean;
      type: string;
      maxMemoryByte: number;
      maxMemoryByteName: string;
    },
    memoryAbilityRow: TraceRow<MemoryAbilityMonitorStruct>
  ): void {
    let memoryAbilityList = memoryAbilityRow.dataList;
    let memoryAbilityFilter = memoryAbilityRow.dataListCache;
    dataFilterHandler(memoryAbilityList, memoryAbilityFilter, {
      startKey: 'startNS',
      durKey: 'dur',
      startNS: TraceRow.range?.startNS ?? 0,
      endNS: TraceRow.range?.endNS ?? 0,
      totalNS: TraceRow.range?.totalNS ?? 0,
      frame: memoryAbilityRow.frame,
      paddingTop: 5,
      useCache: req.useCache || !(TraceRow.range?.refresh ?? false),
    });
    drawLoadingFrame(req.context, memoryAbilityRow.dataListCache, memoryAbilityRow);
    req.context.beginPath();
    let find = false;
    for (let re of memoryAbilityFilter) {
      MemoryAbilityMonitorStruct.draw(req.context, re, req.maxMemoryByte, memoryAbilityRow.isHover);
      if (
        memoryAbilityRow.isHover &&
        re.frame &&
        isFrameContainPoint(re.frame, memoryAbilityRow.hoverX, memoryAbilityRow.hoverY)
      ) {
        MemoryAbilityMonitorStruct.hoverMemoryAbilityStruct = re;
        find = true;
      }
    }

    if (!find && memoryAbilityRow.isHover) {
      MemoryAbilityMonitorStruct.hoverMemoryAbilityStruct = undefined;
    }
    req.context.closePath();
    let textMetrics = req.context.measureText(req.maxMemoryByteName);
    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(req.maxMemoryByteName, 4, 5 + 9);
  }
}

export function memoryAbility(
  memoryAbilityList: Array<MemoryAbilityMonitorStruct>,
  res: Array<MemoryAbilityMonitorStruct>,
  startNS: number,
  endNS: number,
  totalNS: number,
  frame: Rect,
  use: boolean
): void {
  if (use && res.length > 0) {
    for (let i = 0; i < res.length; i++) {
      let memoryAbilityItem = res[i] as MemoryAbilityMonitorStruct;
      if (
        (memoryAbilityItem.startNS || 0) + (memoryAbilityItem.dur || 0) > startNS &&
        (memoryAbilityItem.startNS || 0) < endNS
      ) {
        MemoryAbilityMonitorStruct.setMemoryFrame(memoryAbilityItem, 5, startNS, endNS, totalNS, frame);
      } else {
        memoryAbilityItem.frame = undefined;
      }
    }
    return;
  }
  res.length = 0;
  setMemoryAbility(memoryAbilityList, res, startNS, endNS, totalNS, frame);
}
function setMemoryAbility(
  list: Array<MemoryAbilityMonitorStruct>,
  res: Array<MemoryAbilityMonitorStruct>,
  startNS: number,
  endNS: number,
  totalNS: number,
  frame: Rect
): void {
  if (list) {
    for (let index = 0; index < list.length; index++) {
      let item = list[index] as MemoryAbilityMonitorStruct;
      item.dur =
        index === list.length - 1
          ? (endNS || 0) - (item.startNS || 0)
          : // @ts-ignore
            (list[index + 1].startNS || 0) - (item.startNS || 0);
      if ((item.startNS || 0) + (item.dur || 0) > startNS && (item.startNS || 0) < endNS) {
        MemoryAbilityMonitorStruct.setMemoryFrame(item, 5, startNS, endNS, totalNS, frame);
        if (
          !(
            index > 0 &&
            // @ts-ignore
            (list[index - 1].frame.x || 0) === (item.frame.x || 0) &&
            // @ts-ignore
            (list[index - 1].frame.width || 0) === (item.frame.width || 0)
          )
        ) {
          res.push(item);
        }
      }
    }
  }
}

export class MemoryAbilityMonitorStruct extends BaseStruct {
  static maxMemoryByte: number = 0;
  static maxMemoryByteName: string = '0 MB';
  static hoverMemoryAbilityStruct: MemoryAbilityMonitorStruct | undefined;
  static selectMemoryAbilityStruct: MemoryAbilityMonitorStruct | undefined;
  cpu: number | undefined;
  value: number | undefined;
  startNS: number | undefined;
  dur: number | undefined;

  static draw(
    memoryAbilityContext2D: CanvasRenderingContext2D,
    memoryAbilityData: MemoryAbilityMonitorStruct,
    maxMemoryByte: number,
    isHover: boolean
  ): void {
    if (memoryAbilityData.frame) {
      let width = memoryAbilityData.frame.width || 0;
      let index = 2;
      memoryAbilityContext2D.fillStyle = ColorUtils.colorForTid(index);
      memoryAbilityContext2D.strokeStyle = ColorUtils.colorForTid(index);
      let drawHeight: number = Math.floor(
        ((memoryAbilityData.value || 0) * (memoryAbilityData.frame.height || 0) * 1.0) / maxMemoryByte
      );
      let y = memoryAbilityData.frame.y + memoryAbilityData.frame.height - drawHeight + 4;
      if (memoryAbilityData.startNS === MemoryAbilityMonitorStruct.hoverMemoryAbilityStruct?.startNS && isHover) {
        memoryAbilityContext2D.lineWidth = 1;
        memoryAbilityContext2D.globalAlpha = 0.6;
        memoryAbilityContext2D.fillRect(memoryAbilityData.frame.x, y, width, drawHeight);
        memoryAbilityContext2D.beginPath();
        memoryAbilityContext2D.arc(memoryAbilityData.frame.x, y, 3, 0, 2 * Math.PI, true);
        memoryAbilityContext2D.fill();
        memoryAbilityContext2D.globalAlpha = 1.0;
        memoryAbilityContext2D.stroke();
        memoryAbilityContext2D.beginPath();
        memoryAbilityContext2D.moveTo(memoryAbilityData.frame.x + 3, y);
        memoryAbilityContext2D.lineWidth = 3;
        memoryAbilityContext2D.lineTo(memoryAbilityData.frame.x + width, y);
        memoryAbilityContext2D.stroke();
      } else {
        memoryAbilityContext2D.globalAlpha = 0.6;
        memoryAbilityContext2D.lineWidth = 1;
        let drawHeight: number = Math.floor(
          ((memoryAbilityData.value || 0) * (memoryAbilityData.frame.height || 0)) / maxMemoryByte
        );
        memoryAbilityContext2D.fillRect(memoryAbilityData.frame.x, y, width, drawHeight);
      }
    }
    memoryAbilityContext2D.globalAlpha = 1.0;
    memoryAbilityContext2D.lineWidth = 1;
  }

  static setMemoryFrame(
    memoryNode: MemoryAbilityMonitorStruct,
    padding: number,
    startNS: number,
    endNS: number,
    totalNS: number,
    frame: Rect
  ): void {
    let memoryStartPointX: number;
    let memoryEndPointX: number;

    if ((memoryNode.startNS || 0) < startNS) {
      memoryStartPointX = 0;
    } else {
      memoryStartPointX = ns2x(memoryNode.startNS || 0, startNS, endNS, totalNS, frame);
    }
    if ((memoryNode.startNS || 0) + (memoryNode.dur || 0) > endNS) {
      memoryEndPointX = frame.width;
    } else {
      memoryEndPointX = ns2x((memoryNode.startNS || 0) + (memoryNode.dur || 0), startNS, endNS, totalNS, frame);
    }
    let frameWidth: number = memoryEndPointX - memoryStartPointX <= 1 ? 1 : memoryEndPointX - memoryStartPointX;
    if (!memoryNode.frame) {
      memoryNode.frame = new Rect(0, 0, 0, 0);
    }
    memoryNode.frame.x = Math.floor(memoryStartPointX);
    memoryNode.frame.y = frame.y + padding;
    memoryNode.frame.width = Math.ceil(frameWidth);
    memoryNode.frame.height = Math.floor(frame.height - padding * 2);
  }
}

const textPadding = 2;