harmony 鸿蒙CPU Intensive Task Development

2023-10-30
浏览 (364)

CPU Intensive Task Development

CPU intensive tasks occupy lots of system computing resources for a long period of time, during which other events of the thread are blocked. Example CPU intensive tasks are image processing, video encoding, and data analysis.

OpenHarmony uses multithread concurrency to process CPU intensive tasks. This improves CPU utilization and application response speed.

If a task does not need to occupy a background thread for a long time (3 minutes), you are advised to use TaskPool. Otherwise, use Worker. The following uses histogram processing and a time-consuming model prediction task in the background as examples.

Using TaskPool to Process Histograms

Implement the logic of image processing.
Segment the data, and initiate associated task scheduling through task groups.

Create a task group, call addTask() to add tasks, call execute() to execute the tasks in the task group, and set a high priority for the task group. After all the tasks in the task group are complete, the histogram processing result is returned simultaneously.

Summarize and process the result arrays.

import taskpool from '@ohos.taskpool';

@Concurrent
function imageProcessing(dataSlice: ArrayBuffer): ArrayBuffer {
  // Step 1: Perform specific image processing operations and other time-consuming operations.
  return dataSlice;
}

function histogramStatistic(pixelBuffer: ArrayBuffer): void {
  // Step 2: Perform concurrent scheduling for data in three segments.
  let number: number = pixelBuffer.byteLength / 3;
  let buffer1: ArrayBuffer = pixelBuffer.slice(0, number);
  let buffer2: ArrayBuffer = pixelBuffer.slice(number, number * 2);
  let buffer3: ArrayBuffer = pixelBuffer.slice(number * 2);

  let group: taskpool.TaskGroup = new taskpool.TaskGroup();
  group.addTask(imageProcessing, buffer1);
  group.addTask(imageProcessing, buffer2);
  group.addTask(imageProcessing, buffer3);

  taskpool.execute(group, taskpool.Priority.HIGH).then((ret: ArrayBuffer[]) => {
    // Step 3: Summarize and process the result arrays.
  })
}

@Entry
@Component
struct Index {
  @State message: string = 'Hello World'

  build() {
    Row() {
      Column() {
        Text(this.message)
          .fontSize(50)
          .fontWeight(FontWeight.Bold)
          .onClick(() => {
            let buffer: ArrayBuffer = new ArrayBuffer(24);
            histogramStatistic(buffer);
          })
      }
      .width('100%')
    }
    .height('100%')
  }
}

Using Worker for Time-Consuming Data Analysis

The following uses the training of a region-specific house price prediction model as an example. This model can be used to predict house prices in the region based on the house area and number of rooms. The model needs to run for a long time, and the prediction will use the previous running result. Due to these considerations, Worker is used for the development.

Add the worker creation template provided on DevEco Studio to your project, and name it MyWorker.

newWorker

In the main thread, call constructor() of ThreadWorker to create a Worker object. The calling thread is the host thread.

import worker from '@ohos.worker';


const workerInstance: worker.ThreadWorker = new worker.ThreadWorker('entry/ets/workers/MyWorker.ts');

In the host thread, call onmessage() to receive messages from the worker thread, and call postMessage() to send messages to the worker thread.

For example, the host thread sends training and prediction messages to the worker thread, and receives messages sent back by the worker thread.

```ts
import worker  from '@ohos.worker';

const workerInstance: worker.ThreadWorker = new worker.ThreadWorker('entry/ets/workers/MyWorker.ts');

// Receive the result of the worker thread.
workerInstance.onmessage = (() => {
 console.info('MyWorker.ts onmessage');
 // Perform time-consuming operations in the worker thread.
})

workerInstance.onerror = (() => {
 // Receive error information of the worker thread.
})

// Send a training message to the worker thread.
workerInstance.postMessage({ 'type': 0 });
// Send a prediction message to the worker thread.
workerInstance.postMessage({ 'type': 1, 'value': [90, 5] });
```

Bind the Worker object in the MyWorker.ts file. The calling thread is the worker thread.

   import worker, { ThreadWorkerGlobalScope, MessageEvents, ErrorEvent } from '@ohos.worker';

   let workerPort: ThreadWorkerGlobalScope = worker.workerPort;

In the worker thread, call onmessage() to receive messages sent by the host thread, and call postMessage() to send messages to the host thread.

For example, the prediction model and its training process are defined in the worker thread, and messages are exchanged with the main thread.

import worker, { ThreadWorkerGlobalScope, MessageEvents, ErrorEvent } from '@ohos.worker';
let workerPort: ThreadWorkerGlobalScope = worker.workerPort;
// Define the training model and result.
let result: Array<number>;
// Define the prediction function.
function predict(x: number): number {
 return result[x];
}
// Define the optimizer training process.
function optimize(): void {
 result = [];
}
// onmessage logic of the worker thread.
workerPort.onmessage = (e: MessageEvents): void => {
 // Perform operations based on the type of data to transmit.
 switch (e.data.type as number) {
  case 0:
  // Perform training.
   optimize();
  // Send a training success message to the main thread after training is complete.
   workerPort.postMessage({ type: 'message', value: 'train success.' });
   break;
  case 1:
  // Execute the prediction.
   const output: number = predict(e.data.value as number);
  // Send the prediction result to the main thread.
   workerPort.postMessage({ type: 'predict', value: output });
   break;
  default:
   workerPort.postMessage({ type: 'message', value: 'send message is invalid' });
   break;
 }
}

After the task is completed in the worker thread, destroy the worker thread. The worker thread can be destroyed by itself or the host thread. Then, call onexit() in the host thread to define the processing logic after the worker thread is destroyed.
```
// After the worker thread is destroyed, execute the onexit() callback.
workerInstance.onexit = (): void => {
 console.info("main thread terminate");
}
```
Method 1: In the host thread, call terminate() to destroy the worker thread and stop the worker thread from receiving messages.
```
// Destroy the worker thread.
workerInstance.terminate();
```
Method 2: In the worker thread, call close() to destroy the worker thread and stop the worker thread from receiving messages.
```
// Destroy the worker thread.
workerPort.close();
```