using System.Numerics;
using System.Runtime.InteropServices;
using System.Security.Cryptography;
using System.Text.Json;
using Vortice.D3DCompiler;
using Vortice.Direct3D;
using Vortice.Direct3D11;
using Vortice.DXGI;
using Vortice.Mathematics;
using XFEEDToolBox.Core.Galaxy;
using XFEEDToolBox.Core.Rendering;
using static Vortice.Direct3D11.D3D11;

const int width=256,height=256;
var output=Path.GetFullPath("artifacts/galaxy-power-activity-ring-probe");Directory.CreateDirectory(output);
void Require(bool condition,string message){if(!condition)throw new InvalidOperationException(message);}
Require(GalaxyOriginalPowerActivityRingGeometry.TryCreate(new(false,9007199254740993,new(100,20,30),20),new(160,20,30),out var actual),"Actual bound rejected");
Require(Math.Abs(actual.UAxisInner.W-19.85f)<1e-5&&actual.Position.Y==20,"Original width/centre changed");
Require(Vector3.Distance(GalaxyOriginalPowerActivityRingGeometry.Vertex(actual,0).World,GalaxyOriginalPowerActivityRingGeometry.Vertex(actual,32).World)<1e-5,"Original strip did not close");
Require(!GalaxyOriginalPowerActivityRingGeometry.TryCreate(new(true,1,null,20),new(0,0,0),out _),"Missing centre was invented");
Require(GalaxyOriginalPowerActivityRingGeometry.Coverage(new(7,10,6,0))==0,"Annulus interior was filled");

D3D11CreateDevice(IntPtr.Zero,DriverType.Warp,DeviceCreationFlags.BgraSupport,[FeatureLevel.Level_11_0],out var device,out _,out var context).CheckError();
using var d=device;using var c=context;
using var texture=device.CreateTexture2D(Format.R32G32B32A32_Float,width,height,mipLevels:1,bindFlags:BindFlags.RenderTarget);
using var target=device.CreateRenderTargetView(texture);
using var staging=device.CreateTexture2D(new Texture2DDescription{Width=width,Height=height,Format=Format.R32G32B32A32_Float,ArraySize=1,MipLevels=1,
    SampleDescription=new(1,0),Usage=ResourceUsage.Staging,CPUAccessFlags=CpuAccessFlags.Read});
using var blend=device.CreateBlendState(new BlendDescription(Blend.One,Blend.InverseSourceAlpha));
using var raster=device.CreateRasterizerState(new RasterizerDescription(CullMode.None,FillMode.Solid));
using var depth=device.CreateDepthStencilState(new DepthStencilDescription(false,DepthWriteMask.Zero,ComparisonFunction.Always));
var shader=Path.GetFullPath("XFEEDToolBox/Assets/GalaxyPowerActivityRing.hlsl");
var translatedVsCode=Compiler.CompileFromFile(shader,"VSOriginalPowerActivityRing","vs_5_0");
var translatedPsCode=Compiler.CompileFromFile(shader,"PSOriginalPowerActivityRing","ps_5_0");
var effects="artifacts/galaxy-query-research/power-outer-sources/effects";
var originalVsCode=File.ReadAllBytes(Path.Combine(effects,"vs_GalaxyMapFocusRing_Forward_0_Win64_SM50.dxbc"));
var originalPsCode=File.ReadAllBytes(Path.Combine(effects,"ps_GalaxyMapFocusRing_Forward_0_Win64_SM50.dxbc"));
using var translatedVs=device.CreateVertexShader(translatedVsCode.Span);using var translatedPs=device.CreatePixelShader(translatedPsCode.Span);
using var originalVs=device.CreateVertexShader(originalVsCode);using var originalPs=device.CreatePixelShader(originalPsCode);
InputElementDescription[] elements=[new("POSITION",0,Format.R32G32B32A32_Float,0,0,InputClassification.PerInstanceData,1),
    new("UAXISINNER",0,Format.R32G32B32A32_Float,16,0,InputClassification.PerInstanceData,1),
    new("VAXISOUTER",0,Format.R32G32B32A32_Float,32,0,InputClassification.PerInstanceData,1),
    new("USERDATA",0,Format.R32G32B32A32_Float,48,0,InputClassification.PerInstanceData,1)];
using var layout=device.CreateInputLayout(elements,translatedVsCode.Span);
// Deliberately wide synthetic annulus samples the PS over many pixels. It is
// only a shader fixture; it is never inserted into the game/source catalogue.
var instance=new GalaxyOriginalPowerActivityRing(new(0,0,0,1),new(1,0,0,7),new(0,0,1,10),new(2,1,.5f,0));
using var instances=device.CreateBuffer<GalaxyOriginalPowerActivityRing>(new[]{instance},BindFlags.VertexBuffer);
var camera=GalaxyCamera.Create(new(0,0,0),width,height,0,1.25,7);
var eye=camera.Eye;var right=camera.Right;var up=camera.Up;var forward=camera.Forward;
var original0=new Vector4[12];original0[9]=new((float)right.X,(float)right.Y,(float)right.Z,-(float)(right.X*eye.X+right.Y*eye.Y+right.Z*eye.Z));
original0[10]=new((float)up.X,(float)up.Y,(float)up.Z,-(float)(up.X*eye.X+up.Y*eye.Y+up.Z*eye.Z));
original0[11]=new((float)forward.X,(float)forward.Y,(float)forward.Z,-(float)(forward.X*eye.X+forward.Y*eye.Y+forward.Z*eye.Z));
var original1=new Vector4[274];original1[90]=new(0,1,0,0);original1[270]=new((float)(camera.FocalLength/(width*.5)),0,0,0);
original1[271]=new(0,(float)(camera.FocalLength/(height*.5)),0,0);original1[272]=new(0,0,0,1);original1[273]=new(0,0,(float)camera.Near,0);
float Int(int value)=>BitConverter.Int32BitsToSingle(value);
Vector4[] original2=[new(Int(1),Int(2),0,Int(0)),new(-20,20,GalaxyOriginalPowerActivityRingGeometry.RadialFeather,GalaxyOriginalPowerActivityRingGeometry.SolidRatio),
    new(GalaxyOriginalPowerActivityRingGeometry.SegmentFeather,200,Int(16),0)];
using var cb0=device.CreateBuffer<Vector4>(original0,BindFlags.ConstantBuffer);using var cb1=device.CreateBuffer<Vector4>(original1,BindFlags.ConstantBuffer);
using var cb2=device.CreateBuffer<Vector4>(original2,BindFlags.ConstantBuffer);
var translated=new Frame(new((float)eye.X,(float)eye.Y,(float)eye.Z,(float)camera.FocalLength),
    new((float)right.X,(float)right.Y,(float)right.Z,width),new((float)up.X,(float)up.Y,(float)up.Z,height),
    new((float)forward.X,(float)forward.Y,(float)forward.Z,(float)camera.Near),new(16,.8f,.3f,.85f),new(200,1,1,0));
using var cb11=device.CreateBuffer((uint)Marshal.SizeOf<Frame>(),BindFlags.ConstantBuffer);context.UpdateSubresource(in translated,cb11);
var exposure=new[]{new Vector4(8),new Vector4(8)};ID3D11Texture2D exposureTexture;
unsafe{fixed(Vector4* p=exposure)exposureTexture=device.CreateTexture2D(new Texture2DDescription{Width=2,Height=1,Format=Format.R32G32B32A32_Float,ArraySize=1,MipLevels=1,
    SampleDescription=new(1,0),Usage=ResourceUsage.Immutable,BindFlags=BindFlags.ShaderResource},[new SubresourceData((nint)p,32,32)]);}
using var et=exposureTexture;using var exposureView=device.CreateShaderResourceView(et);
float[] Draw(bool source,int repeats=1,bool ownedExposure=false)
{
    var selected=translated with{Output=new(200,1,1,ownedExposure?1:0)};context.UpdateSubresource(in selected,cb11);
    context.OMSetRenderTargets(target);context.ClearRenderTargetView(target,new Color4(0,0,0,0));context.RSSetViewport(new Viewport(width,height));
    context.RSSetState(raster);context.OMSetDepthStencilState(depth);context.OMSetBlendState(blend);context.IASetInputLayout(layout);
    context.IASetPrimitiveTopology(PrimitiveTopology.TriangleStrip);context.IASetVertexBuffer(0,instances,64);
    context.VSSetShader(source?originalVs:translatedVs);context.PSSetShader(source?originalPs:translatedPs);
    context.VSSetConstantBuffer(0,cb0);context.VSSetConstantBuffer(1,cb1);context.VSSetConstantBuffer(2,cb2);
    context.PSSetConstantBuffer(1,cb1);context.PSSetConstantBuffer(2,cb2);context.PSSetShaderResource(0,exposureView);
    context.PSSetShaderResource(20,ownedExposure?exposureView:null!);
    context.VSSetConstantBuffer(11,cb11);context.PSSetConstantBuffer(11,cb11);
    for(var i=0;i<repeats;i++)context.DrawInstanced(34,1,0,0);
    context.PSSetShaderResource(20,null!);
    context.OMSetRenderTargets(Array.Empty<ID3D11RenderTargetView>());context.CopyResource(staging,texture);
    var mapped=context.Map(staging,0,MapMode.Read);var values=new float[width*height*4];
    for(var row=0;row<height;row++)Marshal.Copy(mapped.DataPointer+(int)(row*mapped.RowPitch),values,row*width*4,width*4);
    context.Unmap(staging,0);return values;
}
var source=Draw(true);var port=Draw(false);var doubled=Draw(false,2);var ownedPort=Draw(false,1,true);
Require(source.All(float.IsFinite)&&port.All(float.IsFinite),"Non-finite shader output");
var maximum=source.Zip(port,(a,b)=>Math.Abs(a-b)).Max();var mean=source.Zip(port,(a,b)=>Math.Abs(a-b)).Average();
var lit=source.Where((_,i)=>i%4==0).Count(v=>v>1e-6);Require(lit>100,"Original shader did not produce annulus samples");
Require(maximum<.01&&mean<.0001,"Original/translated FocusRing shader mismatch: "+maximum+" / "+mean);
Require(port.Where((_,i)=>i%4==3).All(v=>v==0),"Original range alpha0 changed");
Require(port.Zip(doubled,(a,b)=>Math.Abs(b-2*a)).Max()<1e-6,"Original additive blend changed");
Require(source.Zip(ownedPort,(a,b)=>Math.Abs(a-b)).Max()<.01,"Owned exposure texture did not match original texel1/type2 sampling");
XFEEDToolBox.Rendering.GalaxyGraphics.ValidateRingProduction(device,context,shader,camera,instance,target,texture,staging);
var report=new{passed=true,gameAccess=false,sourceDxbcGpuCompared=true,primitive="TriangleStrip",vertexCount=34,litPixels=lit,maximumError=maximum,meanError=mean,
    explicitSyntheticEmissiveRescaling=1,originalRuntimeEmissiveResolved=false,productionUnknownSourceMustRemainHidden=true,
    productionPartialSourceLinked=true,productionUnknownGateTested=true,productionOwnedManualExposureTested=true,
    shaderSha256=Convert.ToHexStringLower(SHA256.HashData(File.ReadAllBytes(shader))),originalVsSha256=Convert.ToHexStringLower(SHA256.HashData(originalVsCode)),
    originalPsSha256=Convert.ToHexStringLower(SHA256.HashData(originalPsCode))};
File.WriteAllText(Path.Combine(output,"result.json"),JsonSerializer.Serialize(report,new JsonSerializerOptions{WriteIndented=true}));Console.WriteLine(JsonSerializer.Serialize(report));
[StructLayout(LayoutKind.Sequential)] readonly record struct Frame(Vector4 EyeFocal,Vector4 RightWidth,Vector4 UpHeight,Vector4 ForwardNear,Vector4 Material,Vector4 Output);

namespace XFEEDToolBox.Rendering
{
    // Isolated source-linked host for the actual independent rendering partial.
    // No App/control/OS window is created; only the WARP fixture is used.
    internal sealed partial class GalaxyGraphics
    {
        private ID3D11Device device=null!;
        private ID3D11DeviceContext context=null!;
        private int DrawCalls;
        private void Upload<T>(ID3D11Buffer buffer,ReadOnlySpan<T> values)where T:unmanaged
        {
            var mapped=context.Map(buffer,0,MapMode.WriteDiscard);
            unsafe{values.CopyTo(new Span<T>((void*)mapped.DataPointer,values.Length));}
            context.Unmap(buffer,0);
        }
        internal static void ValidateRingProduction(ID3D11Device device,ID3D11DeviceContext context,string shader,GalaxyCamera camera,GalaxyOriginalPowerActivityRing ring,
            ID3D11RenderTargetView target,ID3D11Texture2D texture,ID3D11Texture2D staging)
        {
            var graphics=new GalaxyGraphics{device=device,context=context};graphics.InitializeOriginalPowerActivityRing(shader);
            try
            {
                graphics.SetOriginalPowerActivityRing(ring,null);graphics.DrawOriginalPowerActivityRing(camera,1);
                if(graphics.PreparedPowerActivityRings!=1||!graphics.PowerActivityRingSourceUnknown||graphics.SubmittedPowerActivityRings!=0)
                    throw new InvalidOperationException("Unknown source was rendered by the actual production partial");
                graphics.SetOriginalPowerActivityRing(ring,null,true);graphics.DrawOriginalPowerActivityRing(camera,1);
                if(!graphics.PowerActivityRingSourceDefaultKnown||graphics.PowerActivityRingSourceUnknown||graphics.SubmittedPowerActivityRings!=0)
                    throw new InvalidOperationException("Known bound original zero default was confused with unknown or made visible");
                graphics.SetOriginalPowerActivityRing(ring,1);graphics.DrawOriginalPowerActivityRing(camera,1);
                if(graphics.SubmittedPowerActivityRings!=1||graphics.PowerActivityRingSourceUnknown)throw new InvalidOperationException("Explicit fixture source was not drawn");
                graphics.SetOriginalPowerActivityRing(null,1);graphics.DrawOriginalPowerActivityRing(camera,1);
                if(graphics.SubmittedPowerActivityRings!=0)throw new InvalidOperationException("Missing bound was rendered");
                graphics.InitializeOriginalExposure(shader);
                if(graphics.OriginalExposureState!=GalaxyOriginalExposureState.DefaultCreated)throw new InvalidOperationException("Default creation was not recorded");
                var description=graphics.originalExposureTexture!.Description;description.BindFlags=BindFlags.None;
                description.Usage=ResourceUsage.Staging;description.CPUAccessFlags=CpuAccessFlags.Read;
                using var readback=device.CreateTexture2D(description);
                float[] ReadExposure()
                {
                    context.CopyResource(readback,graphics.originalExposureTexture);var mapped=context.Map(readback,0,MapMode.Read);
                    var values=new float[6];Marshal.Copy(mapped.DataPointer,values,0,6);context.Unmap(readback,0);return values;
                }
                if(ReadExposure().Any(v=>v!=0))throw new InvalidOperationException("Original R32 default was not all zero");
                graphics.SetOriginalPowerActivityRing(ring,null);graphics.DrawOriginalPowerActivityRing(camera,1);
                if(!graphics.PowerActivityRingSourceDefaultKnown||graphics.PowerActivityRingSourceUnknown||graphics.SubmittedPowerActivityRings!=0)
                    throw new InvalidOperationException("Owned original zero default was not distinguished");
                if(!graphics.ComputeOriginalManualExposure()||graphics.OriginalExposureState!=GalaxyOriginalExposureState.ManualComputed)
                    throw new InvalidOperationException("Actual original ManualCS did not finish");
                var values=ReadExposure();var expected=graphics.OriginalManualExposureResult!.Value;
                if(Math.Abs(values[0]-expected.LinearAdaptation)>.00002||Math.Abs(values[1]-expected.CompensatedLuminance)>.00002||values.Skip(2).Any(v=>v!=0))
                    throw new InvalidOperationException("Production original ManualCS R32 output differs from exact controls/reference");
                context.OMSetRenderTargets(target);context.ClearRenderTargetView(target,new Color4(0,0,0,0));
                graphics.DrawOriginalPowerActivityRing(camera,1);
                if(!graphics.PowerActivityRingUsesToolManualExposure||graphics.PowerActivityRingSourceUnknown||graphics.SubmittedPowerActivityRings!=1)
                    throw new InvalidOperationException("Computed tool exposure did not reach actual ring texture binding");
                context.OMSetRenderTargets(Array.Empty<ID3D11RenderTargetView>());context.CopyResource(staging,texture);
                var rendered=context.Map(staging,0,MapMode.Read);var row=new float[(int)camera.Width*4];var lit=0;
                for(var y=0;y<(int)camera.Height;y++)
                {
                    Marshal.Copy(rendered.DataPointer+(int)(y*rendered.RowPitch),row,0,row.Length);
                    for(var x=0;x<row.Length;x+=4)if(float.IsFinite(row[x])&&row[x]>1e-6)lit++;
                }
                context.Unmap(staging,0);if(lit<100)throw new InvalidOperationException("Production computed-exposure ring had no actual pixels");
                graphics.ComputeOriginalManualExposure();if(graphics.OriginalExposureDispatches!=1)throw new InvalidOperationException("Unchanged manual controls were recomputed needlessly");
                graphics.SetOriginalPowerActivityRing(ring,float.NaN);graphics.DrawOriginalPowerActivityRing(camera,1);
                if(!graphics.PowerActivityRingSourceUnknown||graphics.SubmittedPowerActivityRings!=0)throw new InvalidOperationException("Invalid explicit source silently fell back");
                graphics.DisposeOriginalExposure();if(graphics.OriginalExposureState!=GalaxyOriginalExposureState.Uninitialized)throw new InvalidOperationException("Disposed exposure remained ready");
            }
            finally{graphics.DisposeOriginalExposure();graphics.DisposeOriginalPowerActivityRing();}
        }
    }
}