1/* 2 * Copyright (C) 2022 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16import { CHART_OFFSET_LEFT, MAX_COUNT, QueryEnum, TraficEnum } from '../utils/QueryEnum'; 17import { getThreadPool } from '../../SqlLite'; 18import { TraceRow } from '../../../component/trace/base/TraceRow'; 19import { XpowerGpuFreqCountStruct } from '../../ui-worker/ProcedureWorkerXpowerGpuFreqCount'; 20import { XpowerGpuFreqStruct } from '../../ui-worker/ProcedureWorkerXpowerGpuFreq'; 21import { Utils } from '../../../component/trace/base/Utils'; 22 23export function xpowerGpuFreqCountDataSender( 24 row: TraceRow<XpowerGpuFreqCountStruct>, 25 args?: unknown 26): Promise<XpowerGpuFreqCountStruct[]> { 27 let trafic: number = TraficEnum.Memory; 28 let width = row.clientWidth - CHART_OFFSET_LEFT; 29 if (trafic === TraficEnum.SharedArrayBuffer && !row.sharedArrayBuffers) { 30 row.sharedArrayBuffers = { 31 value: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 32 startNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 33 dur: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 34 }; 35 } 36 return new Promise((resolve, reject): void => { 37 getThreadPool(row.traceId).submitProto( 38 QueryEnum.XpowerGpuFreqCountData, 39 { 40 startNS: TraceRow.range?.startNS || 0, 41 endNS: TraceRow.range?.endNS || 0, 42 totalNS: TraceRow.range?.totalNS || 0, 43 recordStartNS: Utils.getInstance().getRecordStartNS(row.traceId), 44 recordEndNS: Utils.getInstance().getRecordEndNS(row.traceId), 45 // @ts-ignore 46 queryAll: args && args.queryAll, 47 // @ts-ignore 48 selectStartNS: args ? args.startNS : 0, 49 // @ts-ignore 50 selectEndNS: args ? args.endNS : 0, 51 // @ts-ignore 52 selectTotalNS: args ? args.endNS - args.startNS : 0, 53 t: Date.now(), 54 width: width, 55 trafic: trafic, 56 sharedArrayBuffers: row.sharedArrayBuffers, 57 }, 58 (res: unknown, len: number, transfer: boolean): void => { 59 resolve(gpuFreqCountArrayBufferHandler(transfer ? res : row.sharedArrayBuffers, len)); 60 } 61 ); 62 }); 63} 64 65function gpuFreqCountArrayBufferHandler(buffers: unknown, len: number): XpowerGpuFreqCountStruct[] { 66 let outArr: XpowerGpuFreqCountStruct[] = []; 67 // @ts-ignore 68 let value = new Float64Array(buffers.value); 69 // @ts-ignore 70 let startNS = new Float64Array(buffers.startNS); 71 // @ts-ignore 72 let dur = new Float64Array(buffers.dur); 73 for (let i = 0; i < len; i++) { 74 outArr.push({ 75 value: value[i], 76 startNS: startNS[i], 77 dur: dur[i], 78 } as unknown as XpowerGpuFreqCountStruct); 79 } 80 return outArr; 81} 82 83export function xpowerGpuFreqDataSender( 84 row: TraceRow<XpowerGpuFreqStruct>, 85 args?: unknown 86): Promise<XpowerGpuFreqStruct[]> { 87 let trafic: number = TraficEnum.Memory; 88 let width = row.clientWidth - CHART_OFFSET_LEFT; 89 if (trafic === TraficEnum.SharedArrayBuffer && !row.sharedArrayBuffers) { 90 row.sharedArrayBuffers = { 91 startNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 92 dur: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 93 runTime: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 94 idleTime: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 95 frequency: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT), 96 }; 97 } 98 return new Promise((resolve, reject): void => { 99 getThreadPool(row.traceId).submitProto( 100 QueryEnum.XpowerGpuFreqData, 101 { 102 startNS: TraceRow.range?.startNS || 0, 103 endNS: TraceRow.range?.endNS || 0, 104 totalNS: TraceRow.range?.totalNS || 0, 105 recordStartNS: Utils.getInstance().getRecordStartNS(row.traceId), 106 recordEndNS: Utils.getInstance().getRecordEndNS(row.traceId), 107 // @ts-ignore 108 queryAll: args && args.queryAll, 109 // @ts-ignore 110 selectStartNS: args ? args.startNS : 0, 111 // @ts-ignore 112 selectEndNS: args ? args.endNS : 0, 113 // @ts-ignore 114 selectTotalNS: args ? args.endNS - args.startNS : 0, 115 t: Date.now(), 116 width: width, 117 trafic: trafic, 118 sharedArrayBuffers: row.sharedArrayBuffers, 119 }, 120 (res: unknown, len: number, transfer: boolean): void => { 121 resolve(gpuFreqArrayBufferHandler(transfer ? res : row.sharedArrayBuffers, len)); 122 } 123 ); 124 }); 125} 126 127function gpuFreqArrayBufferHandler(buffers: unknown, len: number): XpowerGpuFreqStruct[] { 128 let outArr: XpowerGpuFreqStruct[] = []; 129 // @ts-ignore 130 let startNS = new Float64Array(buffers.startNS); 131 // @ts-ignore 132 let dur = new Float64Array(buffers.dur); 133 // @ts-ignore 134 let runTime = new Float64Array(buffers.runTime); 135 // @ts-ignore 136 let idleTime = new Float64Array(buffers.idleTime); 137 // @ts-ignore 138 let frequency = new Float64Array(buffers.frequency); 139 for (let i = 0; i < len; i++) { 140 outArr.push({ 141 startNS: startNS[i], 142 dur: dur[i], 143 runTime: runTime[i], 144 idleTime: idleTime[i], 145 frequency: frequency[i], 146 } as unknown as XpowerGpuFreqStruct); 147 } 148 return outArr; 149} 150