1// Copyright (c) 2021 Huawei Device Co., Ltd.
2// Licensed under the Apache License, Version 2.0 (the "License");
3// you may not use this file except in compliance with the License.
4// You may obtain a copy of the License at
5//
6//     http://www.apache.org/licenses/LICENSE-2.0
7//
8// Unless required by applicable law or agreed to in writing, software
9// distributed under the License is distributed on an "AS IS" BASIS,
10// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11// See the License for the specific language governing permissions and
12// limitations under the License.
13
14import { CHART_OFFSET_LEFT, MAX_COUNT, QueryEnum, TraficEnum } from '../utils/QueryEnum';
15import { getThreadPool } from '../../SqlLite';
16import { TraceRow } from '../../../component/trace/base/TraceRow';
17import { CpuStateStruct } from '../../ui-worker/cpu/ProcedureWorkerCpuState';
18import { Utils } from '../../../component/trace/base/Utils';
19
20export function cpuStateSender(filterId: number, row: TraceRow<CpuStateStruct>): Promise<CpuStateStruct[]> {
21  let trafic: number = TraficEnum.Memory;
22  let width = row.clientWidth - CHART_OFFSET_LEFT;
23  if (trafic === TraficEnum.SharedArrayBuffer && !row.sharedArrayBuffers) {
24    row.sharedArrayBuffers = {
25      value: new SharedArrayBuffer(Uint32Array.BYTES_PER_ELEMENT * MAX_COUNT),
26      dur: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
27      startTs: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
28      height: new SharedArrayBuffer(Uint8Array.BYTES_PER_ELEMENT * MAX_COUNT),
29    };
30  }
31  return new Promise((resolve, reject): void => {
32    getThreadPool(row.traceId).submitProto(
33      QueryEnum.CpuStateData,
34      {
35        startNS: TraceRow.range?.startNS || 0,
36        endNS: TraceRow.range?.endNS || 0,
37        recordStartNS: Utils.getInstance().getRecordStartNS(row.traceId),
38        recordEndNS: Utils.getInstance().getRecordEndNS(row.traceId),
39        width: width,
40        trafic: trafic,
41        sharedArrayBuffers: row.sharedArrayBuffers,
42        filterId: filterId,
43      },
44      (res: unknown, len: number, transfer: boolean): void => {
45        resolve(arrayBufferHandler(transfer ? res : row.sharedArrayBuffers, len));
46      }
47    );
48  });
49}
50
51function arrayBufferHandler(res: unknown, len: number): CpuStateStruct[] {
52  let outArr: CpuStateStruct[] = [];
53  // @ts-ignore
54  let startTs = new Float64Array(res.startTs);
55  // @ts-ignore
56  let dur = new Float64Array(res.dur);
57  // @ts-ignore
58  let value = new Uint32Array(res.value);
59  // @ts-ignore
60  let height = new Uint8Array(res.value);
61  for (let i = 0; i < len; i++) {
62    outArr.push({
63      value: value[i],
64      dur: dur[i],
65      height: height[i],
66      startTs: startTs[i],
67    } as unknown as CpuStateStruct);
68  }
69  return outArr;
70}
71