Add a configurable LOD bias

Author: TellowKrinkle

impl.cpp

@@ -834,11 +834,28 @@ class DeviceWrapper final : public ID3D11Device, IDXGIDevice1 {
 
   HRESULT STDMETHODCALLTYPE CreateSamplerState(const D3D11_SAMPLER_DESC* pSamplerDesc, ID3D11SamplerState** ppSamplerState) override {
     D3D11_SAMPLER_DESC desc = *pSamplerDesc;
+    bool modified = false;
     if (config.anisotropy && desc.Filter == D3D11_FILTER_MIN_MAG_MIP_LINEAR) {
       desc.Filter = D3D11_FILTER_ANISOTROPIC;
       desc.MaxAnisotropy = config.anisotropy;
+      modified = true;
     }
-    return dev->CreateSamplerState(&desc, ppSamplerState);
+    float bias;
+    if (config.lodBias < 0)
+        bias = std::max(config.lodBias, config.lodBias + desc.MipLODBias);
+    else
+        bias = std::min(config.lodBias, config.lodBias + desc.MipLODBias);
+    if (bias != desc.MipLODBias) {
+        desc.MipLODBias = bias;
+        modified = true;
+    }
+    HRESULT hr = dev->CreateSamplerState(&desc, ppSamplerState);
+    if (SUCCEEDED(hr)) {
+      ID3D11SamplerState* orig;
+      if (SUCCEEDED(dev->CreateSamplerState(pSamplerDesc, &orig)))
+        (*ppSamplerState)->SetPrivateDataInterface(IID_OriginalShader, orig);
+    }
+    return hr;
   }
 
   HRESULT STDMETHODCALLTYPE CreateShaderResourceView(ID3D11Resource* pResource, const D3D11_SHADER_RESOURCE_VIEW_DESC* pDesc, ID3D11ShaderResourceView** ppSRView) override {
@@ -1039,7 +1056,10 @@ class ContextWrapper final : public ID3D11DeviceContext {
   ID3D11BlendState* requestedBlend = nullptr;
   ID3D11PixelShader* requestedPS = nullptr;
   D3D11_MAPPED_SUBRESOURCE lastMap;
+  void* tmpMemory = nullptr;
+  UINT tmpMemorySize = 0;
   UINT scissorWidth, scissorHeight, viewportWidth, viewportHeight;
+  bool shouldUseOldShaders = false;
   bool blendStateSupportsA2C = false;
   bool psSupportsA2C = false;
   bool rtsizeDirty = false;
@@ -1097,7 +1117,6 @@ class ContextWrapper final : public ID3D11DeviceContext {
 
   // ID3D11DeviceContext
   void VSSetConstantBuffers(UINT StartSlot, UINT NumBuffers, ID3D11Buffer* const* ppConstantBuffers) override { ctx->VSSetConstantBuffers(StartSlot, NumBuffers, ppConstantBuffers); }
-  void PSSetSamplers(UINT StartSlot, UINT NumSamplers, ID3D11SamplerState* const* ppSamplers) override { ctx->PSSetSamplers(StartSlot, NumSamplers, ppSamplers); }
   void VSSetShader(ID3D11VertexShader* pVertexShader, ID3D11ClassInstance* const* ppClassInstances, UINT NumClassInstances) override { ctx->VSSetShader(pVertexShader, ppClassInstances, NumClassInstances); }
   void PSSetConstantBuffers(UINT StartSlot, UINT NumBuffers, ID3D11Buffer* const* ppConstantBuffers) override { ctx->PSSetConstantBuffers(StartSlot, NumBuffers, ppConstantBuffers); }
   void IASetInputLayout(ID3D11InputLayout* pInputLayout) override { ctx->IASetInputLayout(pInputLayout); }
@@ -1345,6 +1364,31 @@ class ContextWrapper final : public ID3D11DeviceContext {
     ctx->DrawIndexedInstancedIndirect(pBufferForArgs, AlignedByteOffsetForArgs);
   }
 
+  void PSSetSamplers(UINT StartSlot, UINT NumSamplers, ID3D11SamplerState* const* ppSamplers) override {
+    bool replacedSamplers = false;
+    if (shouldUseOldShaders) {
+      for (UINT i = 0; i < NumSamplers; i++) {
+        ID3D11SamplerState* orig;
+        UINT size = sizeof(orig);
+        if (SUCCEEDED(ppSamplers[i]->GetPrivateData(IID_OriginalShader, &size, &orig))) {
+          orig->Release(); // Still being held by sampler private data
+          if (!replacedSamplers) {
+            replacedSamplers = true;
+            SIZE_T size = sizeof(*ppSamplers) * NumSamplers;
+            if (tmpMemorySize < size) {
+              tmpMemory = realloc(tmpMemory, size);
+              tmpMemorySize = size;
+            }
+            memcpy(tmpMemory, ppSamplers, size);
+            ppSamplers = static_cast<ID3D11SamplerState**>(tmpMemory);
+          }
+          const_cast<ID3D11SamplerState**>(ppSamplers)[i] = orig;
+        }
+      }
+    }
+    ctx->PSSetSamplers(StartSlot, NumSamplers, ppSamplers);
+  }
+
   void PSSetShaderResources(UINT StartSlot, UINT NumViews, ID3D11ShaderResourceView* const* ppShaderResourceViews) override {
     for (UINT i = 0; i < NumViews; i++) {
       if (ppShaderResourceViews[i]) {
@@ -1360,7 +1404,7 @@ class ContextWrapper final : public ID3D11DeviceContext {
   void PSSetShader(ID3D11PixelShader* pPixelShader, ID3D11ClassInstance* const* ppClassInstances, UINT NumClassInstances) override {
     requestedPS = pPixelShader;
     ID3D11PixelShader* a2c = nullptr;
-    bool useOriginalShaders = shouldUseOldShaders();
+    bool useOriginalShaders = shouldUseOldShaders;
     bool oldUseA2C = ShouldUseA2C();
     if (pPixelShader && !useOriginalShaders) {
       UINT size = sizeof(a2c);
@@ -1410,6 +1454,8 @@ class ContextWrapper final : public ID3D11DeviceContext {
   }
 
   void ClearDepthStencilView(ID3D11DepthStencilView* pDepthStencilView, UINT ClearFlags, FLOAT Depth, UINT8 Stencil) override {
+    // Pick something that happens fairly infrequently to update this so we don't spam XInputGetState
+    shouldUseOldShaders = ::atfix::shouldUseOldShaders();
     if (ID3D11DepthStencilView* msaa = getMSAADSV(pDepthStencilView)) {
       ctx->ClearDepthStencilView(msaa, ClearFlags, Depth, Stencil);
       msaa->Release();

impl.h

@@ -14,6 +14,7 @@ extern Log log;
 extern struct Config {
 	DWORD msaaSamples;
 	DWORD anisotropy;
+	float lodBias;
 	bool ssaaCharacters;
 	bool sampleRateAlpha;
 	bool ssaaAll;

main.cpp

@@ -79,6 +79,8 @@ D3D11Proc loadSystemD3D11() {
                       "[Other]\n"
                       "; Set to 1-16 to enable anisotropic filtering\n"
                       "AF = 0\n"
+                      "; Apply an additional bias to texture LODs (negative makes things less blurry and more shimmery, positive does the opposite)\n"
+                      "LODBias = 0\n"
                       "; Allow toggling shader enhancements by holding BACK / SELECT (will not toggle MSAA but will toggle most other things)\n"
                       "EnhancementToggle = 0\n";
     HANDLE file = CreateFileA("atfix.ini", GENERIC_WRITE, 0, nullptr, CREATE_NEW, 0, nullptr);
@@ -92,23 +94,26 @@ D3D11Proc loadSystemD3D11() {
     char SampleRateAlpha[8];
     char FullSSAA[8];
     char AF[8];
+    char LODBias[8];
     char UseShaderToggle[8];
     bool ok = true;
     GetPrivateProfileStringA("MSAA", "NumSamples", "8", NumSamples, sizeof(NumSamples), ".\\atfix.ini");
     GetPrivateProfileStringA("MSAA", "CharacterSSAA", "1", CharacterSSAA, sizeof(CharacterSSAA), ".\\atfix.ini");
     GetPrivateProfileStringA("MSAA", "AlphaSSAA", "1", SampleRateAlpha, sizeof(SampleRateAlpha), ".\\atfix.ini");
     GetPrivateProfileStringA("MSAA", "FullSSAA", "0", FullSSAA, sizeof(FullSSAA), ".\\atfix.ini");
     GetPrivateProfileStringA("Other", "AF", "0", AF, sizeof(AF), ".\\atfix.ini");
+    GetPrivateProfileStringA("Other", "LODBias", "0", LODBias, sizeof(LODBias), ".\\atfix.ini");
     GetPrivateProfileStringA("Other", "EnhancementToggle", "0", UseShaderToggle, sizeof(UseShaderToggle), ".\\atfix.ini");
     config.msaaSamples = atoi(NumSamples);
     config.ssaaCharacters = atoi(CharacterSSAA);
     config.sampleRateAlpha = atoi(SampleRateAlpha);
     config.ssaaAll = atoi(FullSSAA);
     config.anisotropy = atoi(AF);
+    config.lodBias = atof(LODBias);
     config.allowShaderToggle = atoi(UseShaderToggle);
     if (config.msaaSamples < 1)
       config.msaaSamples = 1;
-    log("Loaded config, ", config.msaaSamples, " samples, ", config.ssaaCharacters, " characterSSAA, ", config.sampleRateAlpha, " sampleRateAlpha, ", config.ssaaAll, " fullSSAA, ", config.anisotropy, " AF, ", config.allowShaderToggle, " enhancementToggle ");
+    log("Loaded config, ", config.msaaSamples, " samples, ", config.ssaaCharacters, " characterSSAA, ", config.sampleRateAlpha, " sampleRateAlpha, ", config.ssaaAll, " fullSSAA, ", config.anisotropy, " AF, ", config.lodBias, " lodBias, ", config.allowShaderToggle, " enhancementToggle ");
   }
 
   d3d11Proc.D3D11CreateDevice = reinterpret_cast<PFN_D3D11CreateDevice>(