76 lines
2.1 KiB
C++
76 lines
2.1 KiB
C++
/*
|
|
* Copyright (C) 2018-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/aub/aub_subcapture.h"
|
|
|
|
using namespace NEO;
|
|
|
|
class AubSubCaptureManagerMock : public AubSubCaptureManager {
|
|
public:
|
|
AubSubCaptureManagerMock(const std::string &fileName, AubSubCaptureCommon &subCaptureCommon)
|
|
: AubSubCaptureManager(fileName, subCaptureCommon, "") {}
|
|
|
|
void setSubCaptureIsActive(bool on) {
|
|
subCaptureIsActive = on;
|
|
}
|
|
bool isSubCaptureActive() const {
|
|
return subCaptureIsActive;
|
|
}
|
|
void setSubCaptureWasActiveInPreviousEnqueue(bool on) {
|
|
subCaptureWasActiveInPreviousEnqueue = on;
|
|
}
|
|
bool wasSubCaptureActiveInPreviousEnqueue() const {
|
|
return subCaptureWasActiveInPreviousEnqueue;
|
|
}
|
|
void setKernelCurrentIndex(uint32_t index) {
|
|
kernelCurrentIdx = index;
|
|
}
|
|
uint32_t getKernelCurrentIndex() const {
|
|
return kernelCurrentIdx;
|
|
}
|
|
bool getUseToggleFileName() const {
|
|
return useToggleFileName;
|
|
}
|
|
const std::string &getInitialFileName() const {
|
|
return initialFileName;
|
|
}
|
|
const std::string &getCurrentFileName() const {
|
|
return currentFileName;
|
|
}
|
|
SettingsReader *getSettingsReader() const {
|
|
return settingsReader.get();
|
|
}
|
|
void setSubCaptureToggleActive(bool on) {
|
|
isToggledOn = on;
|
|
}
|
|
bool isSubCaptureToggleActive() const override {
|
|
return isToggledOn;
|
|
}
|
|
void setToggleFileName(const std::string &fileName) {
|
|
toggleFileName = fileName;
|
|
}
|
|
std::string getToggleFileName() const override {
|
|
return toggleFileName;
|
|
}
|
|
|
|
[[nodiscard]] std::unique_lock<std::mutex> lock() const override {
|
|
isLocked = true;
|
|
return std::unique_lock<std::mutex>{mutex};
|
|
}
|
|
|
|
using AubSubCaptureManager::generateFilterFileName;
|
|
using AubSubCaptureManager::generateToggleFileName;
|
|
using AubSubCaptureManager::getAubCaptureFileName;
|
|
|
|
mutable bool isLocked = false;
|
|
|
|
protected:
|
|
bool isToggledOn = false;
|
|
std::string toggleFileName = "";
|
|
};
|