1 /*
2  * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved.
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 
16 #include "dynamic_frame_table.h"
17 
18 #include <cmath>
19 
20 namespace SysTuning {
21 namespace TraceStreamer {
22 enum class Index : int32_t { ID = 0, X, Y, WIDTH, HEIGHT, ALPHA, NAME, END_TIME };
DynamicFrameTable(const TraceDataCache *dataCache)23 DynamicFrameTable::DynamicFrameTable(const TraceDataCache *dataCache) : TableBase(dataCache)
24 {
25     tableColumn_.push_back(TableBase::ColumnInfo("id", "INTEGER"));
26     tableColumn_.push_back(TableBase::ColumnInfo("x", "INTEGER"));
27     tableColumn_.push_back(TableBase::ColumnInfo("y", "INTEGER"));
28     tableColumn_.push_back(TableBase::ColumnInfo("width", "INTEGER"));
29     tableColumn_.push_back(TableBase::ColumnInfo("height", "INTEGER"));
30     tableColumn_.push_back(TableBase::ColumnInfo("alpha", "TEXT"));
31     tableColumn_.push_back(TableBase::ColumnInfo("name", "TEXT"));
32     tableColumn_.push_back(TableBase::ColumnInfo("end_time", "INTEGER"));
33     tablePriKey_.push_back("id");
34 }
35 
~DynamicFrameTable()36 DynamicFrameTable::~DynamicFrameTable() {}
37 
CreateCursor()38 std::unique_ptr<TableBase::Cursor> DynamicFrameTable::CreateCursor()
39 {
40     return std::make_unique<Cursor>(dataCache_, this);
41 }
42 
Cursor(const TraceDataCache *dataCache, TableBase *table)43 DynamicFrameTable::Cursor::Cursor(const TraceDataCache *dataCache, TableBase *table)
44     : TableBase::Cursor(dataCache, table, static_cast<uint32_t>(dataCache->GetConstDynamicFrame().Size())),
45       dynamicFrameObj_(dataCache->GetConstDynamicFrame())
46 {
47 }
48 
~Cursor()49 DynamicFrameTable::Cursor::~Cursor() {}
Column(int32_t col) const50 int32_t DynamicFrameTable::Cursor::Column(int32_t col) const
51 {
52     switch (static_cast<Index>(col)) {
53         case Index::ID:
54             sqlite3_result_int64(context_, static_cast<sqlite3_int64>(dynamicFrameObj_.IdsData()[CurrentRow()]));
55             break;
56         case Index::X: {
57             sqlite3_result_int(context_, static_cast<int32_t>(dynamicFrameObj_.Xs()[CurrentRow()]));
58             break;
59         }
60         case Index::Y: {
61             sqlite3_result_int(context_, static_cast<int32_t>(dynamicFrameObj_.Ys()[CurrentRow()]));
62             break;
63         }
64         case Index::WIDTH: {
65             sqlite3_result_int(context_, static_cast<int32_t>(dynamicFrameObj_.Widths()[CurrentRow()]));
66             break;
67         }
68         case Index::HEIGHT: {
69             sqlite3_result_int(context_, static_cast<int32_t>(dynamicFrameObj_.Heights()[CurrentRow()]));
70             break;
71         }
72         case Index::ALPHA: {
73             if (dynamicFrameObj_.Alphas()[CurrentRow()] != INVALID_UINT64) {
74                 const std::string &str =
75                     dataCache_->GetDataFromDict(static_cast<size_t>(dynamicFrameObj_.Alphas()[CurrentRow()]));
76                 sqlite3_result_text(context_, str.c_str(), STR_DEFAULT_LEN, nullptr);
77             }
78             break;
79         }
80         case Index::NAME: {
81             if (dynamicFrameObj_.Names()[CurrentRow()] != INVALID_UINT64) {
82                 const std::string &str =
83                     dataCache_->GetDataFromDict(static_cast<size_t>(dynamicFrameObj_.Names()[CurrentRow()]));
84                 sqlite3_result_text(context_, str.c_str(), STR_DEFAULT_LEN, nullptr);
85             }
86             break;
87         }
88         case Index::END_TIME:
89             if (dynamicFrameObj_.EndTimes()[CurrentRow()] != INVALID_TIME) {
90                 sqlite3_result_int64(context_, static_cast<sqlite3_int64>(dynamicFrameObj_.EndTimes()[CurrentRow()]));
91             }
92             break;
93         default:
94             TS_LOGF("Unregistered column : %d", col);
95             break;
96     }
97     return SQLITE_OK;
98 }
99 } // namespace TraceStreamer
100 } // namespace SysTuning
101