61
62    HWND window = CreateWindowExW(0, wndClass.lpszClassName, L"PrivateDataTest", WS_OVERLAPPEDWINDOW|WS_VISIBLE, CW_USEDEFAULT, CW_USEDEFAULT, 800, 600, nullptr, nullptr, wndClass.hInstance, nullptr);
63
64    IDirect3D9* d3d = Direct3DCreate9(D3D_SDK_VERSION);
65
66    D3DPRESENT_PARAMETERS presentParameters {};
67    presentParameters.SwapEffect = D3DSWAPEFFECT_DISCARD;
68    presentParameters.Windowed = TRUE;
69    IDirect3DDevice9* d3dDevice;
70    d3d->CreateDevice(D3DADAPTER_DEFAULT, D3DDEVTYPE_HAL, window, D3DCREATE_HARDWARE_VERTEXPROCESSING, &presentParameters, &d3dDevice);
71
72    IDirect3DSurface9* backBuffer;
73    d3dDevice->GetBackBuffer(0, 0, D3DBACKBUFFER_TYPE_MONO, &backBuffer);
74
75
76    // D3DSPD_IUNKNOWN docs state:
77    // The data at pData is a pointer to an IUnknown interface. SizeOfData must be set to the size of a pointer to IUnknown, that is, sizeof(IUnknown*).
78    // Direct3D automatically callsIUnknown through pData when the private data is destroyed. Private data will be destroyed by a subsequent call
79    // to IDirect3DResource9::SetPrivateData with the same GUID, a subsequent call to IDirect3DResource9::FreePrivateData, or when the IDirect3D9 object is released.
80    {
81        std::cout << "Test #1: Create a private data interface, get it, free it, check reference counters and the destructor call" << std::endl << std::endl;
82
83        bDestructorCalled = false;
84        TestObject* object = new TestObject;
85
86        HRESULT hr = backBuffer->SetPrivateData(TEST_DATA_GUID, object, sizeof(object), D3DSPD_IUNKNOWN);
87        assert(SUCCEEDED(hr));
88
89        object->AddRef();
90        ULONG ref1 = object->Release();
91        assert(ref1 == 2); // One from us, one from private data. dgVoodoo 2.83.2 does NOT fail here
92
93        IUnknown* storedObject;
94        DWORD cbSize = sizeof(storedObject);
95        hr = backBuffer->GetPrivateData(TEST_DATA_GUID, &storedObject, &cbSize);
96        assert(SUCCEEDED(hr));
97        assert(object == storedObject); // dgVoodoo 2.83.2 fails here
98
99        ref1 = storedObject->Release();
100        assert(ref1 == 2); // One from us, one from private data
101
102        object->Release();
103        hr = backBuffer->FreePrivateData(TEST_DATA_GUID);
104        assert(bDestructorCalled);
105    }
106
107    {
108        std::cout << "Test #2: Create a private data interface on a dummy surface, release that surface, check reference counters and the destructor call" << std::endl << std::endl;
109
110        bDestructorCalled = false;
111        TestObject* object = new TestObject;
112
113        HRESULT hr = backBuffer->SetPrivateData(TEST_DATA_GUID, object, sizeof(object), D3DSPD_IUNKNOWN);
114        assert(SUCCEEDED(hr));
115
116        char buffer[] = { 1, 2, 3, 4, 5, 6 };
117        hr = backBuffer->SetPrivateData(TEST_DATA_GUID, buffer, sizeof(buffer), 0);
118        assert(SUCCEEDED(hr));
119
120        object->Release();
121        assert(bDestructorCalled);
122    }
123
124    {
125        std::cout << "Test #3: Create a private data interface, overwrite it with plain data, check reference counters and the destructor call" << std::endl << std::endl;
126    
127        bDestructorCalled = false;
128        IDirect3DSurface9* testSurface;
129        HRESULT hr = d3dDevice->CreateOffscreenPlainSurface(4, 4, D3DFMT_A8R8G8B8, D3DPOOL_SYSTEMMEM, &testSurface, nullptr);
130        assert(SUCCEEDED(hr));
131
132        TestObject* object = new TestObject;
133
134        hr = testSurface->SetPrivateData(TEST_DATA_GUID, object, sizeof(object), D3DSPD_IUNKNOWN);
135        assert(SUCCEEDED(hr));
136
137        ULONG ref1 = object->Release();
138        assert(ref1 == 1); // One from private data
139
140        testSurface->Release();
141        assert(bDestructorCalled);
142    }
143
144    return 0;
145}