compute-runtime/unit_tests/mocks/mock_aub_subcapture_manager.h

75 lines
2.0 KiB
C++

/*
* Copyright (C) 2018-2019 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "runtime/command_stream/aub_subcapture.h"
using namespace NEO;
class AubSubCaptureManagerMock : public AubSubCaptureManager {
public:
using AubSubCaptureManager::AubSubCaptureManager;
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;
}
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 = "";
};