1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/segmentation_platform/embedder/home_modules/test_utils.h"
#include <string_view>
namespace segmentation_platform::home_modules {
AllCardSignals CreateAllCardSignals(CardSelectionInfo* card,
const std::vector<float>& signal_values) {
CardSignalMap card_signal_map;
std::map<SignalKey, size_t> card_signals;
size_t input_counter = 0;
const auto& card_inputs = card->GetInputs();
for (const auto& key_and_input : card_inputs) {
card_signals[key_and_input.first] = input_counter;
input_counter++;
}
card_signal_map[card->card_name()] = card_signals;
return AllCardSignals(card_signal_map, signal_values);
}
std::vector<std::string> ExtractCardNames(
const std::vector<std::unique_ptr<CardSelectionInfo>>& all_cards) {
std::vector<std::string> card_names;
for (const auto& card : all_cards) {
if (card) {
card_names.push_back(card->card_name());
}
}
return card_names;
}
std::vector<std::string> GetSignalKeys(const CardSignalMap& cardSignalMap,
const char* cardName) {
std::vector<std::string> signalKeys;
auto singleCardSignalMap = cardSignalMap.find(cardName);
if (singleCardSignalMap != cardSignalMap.end()) {
const auto& signalMap = singleCardSignalMap->second;
for (const auto& signalPair : signalMap) {
signalKeys.push_back(signalPair.first);
}
}
return signalKeys;
}
} // namespace segmentation_platform::home_modules
|
