compute-runtime/unit_tests/mocks/mock_gmm_resource_info.h

113 lines
4.0 KiB
C++

/*
* Copyright (c) 2017, Intel Corporation
*
* Permission is hereby granted, free of charge, to any person obtaining a
* copy of this software and associated documentation files (the "Software"),
* to deal in the Software without restriction, including without limitation
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
* and/or sell copies of the Software, and to permit persons to whom the
* Software is furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included
* in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
* OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
* ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
* OTHER DEALINGS IN THE SOFTWARE.
*/
#pragma once
#include "runtime/gmm_helper/resource_info.h"
#include "gmock/gmock.h"
#if defined(__clang__)
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Winconsistent-missing-override"
#endif
namespace OCLRT {
struct SurfaceFormatInfo;
class MockGmmResourceInfo : public GmmResourceInfo {
public:
~MockGmmResourceInfo();
MockGmmResourceInfo(GMM_RESCREATE_PARAMS *resourceCreateParams);
MockGmmResourceInfo(GMM_RESOURCE_INFO *inputGmmResourceInfo);
size_t getSizeAllocation() override { return size; }
size_t getBaseWidth() override { return static_cast<size_t>(mockResourceCreateParams.BaseWidth); }
size_t getBaseHeight() override { return static_cast<size_t>(mockResourceCreateParams.BaseHeight); }
size_t getBaseDepth() override { return static_cast<size_t>(mockResourceCreateParams.Depth); }
size_t getArraySize() override { return static_cast<size_t>(mockResourceCreateParams.ArraySize); }
size_t getRenderPitch() override { return rowPitch; }
uint32_t getNumSamples() override { return mockResourceCreateParams.MSAA.NumSamples; }
uint32_t getQPitch() override { return qPitch; }
uint32_t getBitsPerPixel() override;
uint32_t getHAlign() override { return 4u; }
uint32_t getVAlign() override { return 4u; }
GMM_TILE_TYPE getTileType() override { return mockResourceCreateParams.Flags.Info.TiledY ? GMM_TILED_Y : GMM_NOT_TILED; }
GMM_RESOURCE_FORMAT getResourceFormat() override { return mockResourceCreateParams.Format; }
GMM_SURFACESTATE_FORMAT getResourceFormatSurfaceState() override { return (GMM_SURFACESTATE_FORMAT)0; }
GMM_RESOURCE_TYPE getResourceType() override { return mockResourceCreateParams.Type; }
GMM_RESOURCE_FLAG *getResourceFlags() override { return &mockResourceCreateParams.Flags; }
MOCK_METHOD1(getOffset, GMM_STATUS(GMM_REQ_OFFSET_INFO &reqOffsetInfo));
MOCK_METHOD1(cpuBlt, uint8_t(GMM_RES_COPY_BLT *resCopyBlt));
void *getSystemMemPointer(uint8_t isD3DDdiAllocation) override { return (void *)mockResourceCreateParams.pExistingSysMem; }
MOCK_METHOD0(getRenderAuxPitchTiles, uint32_t(void));
MOCK_METHOD0(getAuxQPitch, uint32_t(void));
MOCK_METHOD1(getUnifiedAuxSurfaceOffset, uint64_t(GMM_UNIFIED_AUX_TYPE auxType));
GMM_RESOURCE_INFO *peekHandle() const override { return mockResourceInfoHandle; }
GMM_RESOURCE_INFO *mockResourceInfoHandle = (GMM_RESOURCE_INFO *)this;
GMM_RESCREATE_PARAMS mockResourceCreateParams = {};
void overrideReturnedRenderPitch(size_t newPitch) { rowPitch = newPitch; }
void setUnifiedAuxTranslationCapable();
protected:
MockGmmResourceInfo();
void computeRowPitch();
void setupDefaultActions();
void setSurfaceFormat();
const SurfaceFormatInfo *surfaceFormatInfo = nullptr;
size_t size = 0;
size_t rowPitch = 0;
uint32_t qPitch = 0;
};
} // namespace OCLRT
#if defined(__clang__)
#pragma clang diagnostic pop
#endif