NVSharpen.h 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950
  1. // The MIT License(MIT)
  2. //
  3. // Copyright(c) 2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  4. //
  5. // Permission is hereby granted, free of charge, to any person obtaining a copy of
  6. // this software and associated documentation files(the "Software"), to deal in
  7. // the Software without restriction, including without limitation the rights to
  8. // use, copy, modify, merge, publish, distribute, sublicense, and / or sell copies of
  9. // the Software, and to permit persons to whom the Software is furnished to do so,
  10. // subject to the following conditions :
  11. //
  12. // The above copyright notice and this permission notice shall be included in all
  13. // copies or substantial portions of the Software.
  14. //
  15. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
  17. // FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.IN NO EVENT SHALL THE AUTHORS OR
  18. // COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
  19. // IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  20. // CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  21. #pragma once
  22. #include <dxgi1_4.h>
  23. #include <d3d11_3.h>
  24. #include <d3dcompiler.h>
  25. #include <iostream>
  26. #include "DXUtilities.h"
  27. #include "DeviceResources.h"
  28. #include "Utilities.h"
  29. #include "NIS_Config.h"
  30. class NVSharpen {
  31. public:
  32. NVSharpen(DeviceResources& deviceResources, const std::vector<std::string>& shaderPaths);
  33. void update(float sharpness, uint32_t inputWidth, uint32_t inputHeight);
  34. void dispatch(ID3D11ShaderResourceView* const* input, ID3D11UnorderedAccessView* const* output);
  35. private:
  36. DeviceResources& m_deviceResources;
  37. NISConfig m_config;
  38. ComPtr<ID3D11ComputeShader> m_computeShader;
  39. ComPtr<ID3D11Buffer> m_csBuffer;
  40. ComPtr<ID3D11SamplerState> m_LinearClampSampler;
  41. uint32_t m_outputWidth;
  42. uint32_t m_outputHeight;
  43. uint32_t m_blockWidth;
  44. uint32_t m_blockHeight;
  45. };