[SHELL32] CDrivesFolder: Implement the eject and disconnect menu items. CORE-13841
[reactos.git] / dll / win32 / windowscodecs / main.c
1 /*
2 * Copyright 2009 Vincent Povirk for CodeWeavers
3 *
4 * This library is free software; you can redistribute it and/or
5 * modify it under the terms of the GNU Lesser General Public
6 * License as published by the Free Software Foundation; either
7 * version 2.1 of the License, or (at your option) any later version.
8 *
9 * This library is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * Lesser General Public License for more details.
13 *
14 * You should have received a copy of the GNU Lesser General Public
15 * License along with this library; if not, write to the Free Software
16 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17 */
18
19 #include "wincodecs_private.h"
20
21 extern BOOL WINAPI WIC_DllMain(HINSTANCE, DWORD, LPVOID) DECLSPEC_HIDDEN;
22
23 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
24 {
25
26 switch (fdwReason)
27 {
28 case DLL_PROCESS_ATTACH:
29 DisableThreadLibraryCalls(hinstDLL);
30 break;
31 }
32
33 return WIC_DllMain(hinstDLL, fdwReason, lpvReserved);
34 }
35
36 HRESULT WINAPI DllCanUnloadNow(void)
37 {
38 return S_FALSE;
39 }
40
41 HRESULT copy_pixels(UINT bpp, const BYTE *srcbuffer,
42 UINT srcwidth, UINT srcheight, INT srcstride,
43 const WICRect *rc, UINT dststride, UINT dstbuffersize, BYTE *dstbuffer)
44 {
45 UINT bytesperrow;
46 UINT row_offset; /* number of bits into the source rows where the data starts */
47 WICRect rect;
48
49 if (!rc)
50 {
51 rect.X = 0;
52 rect.Y = 0;
53 rect.Width = srcwidth;
54 rect.Height = srcheight;
55 rc = ▭
56 }
57 else
58 {
59 if (rc->X < 0 || rc->Y < 0 || rc->X+rc->Width > srcwidth || rc->Y+rc->Height > srcheight)
60 return E_INVALIDARG;
61 }
62
63 bytesperrow = ((bpp * rc->Width)+7)/8;
64
65 if (dststride < bytesperrow)
66 return E_INVALIDARG;
67
68 if ((dststride * (rc->Height-1)) + ((rc->Width * bpp) + 7)/8 > dstbuffersize)
69 return E_INVALIDARG;
70
71 /* if the whole bitmap is copied and the buffer format matches then it's a matter of a single memcpy */
72 if (rc->X == 0 && rc->Y == 0 && rc->Width == srcwidth && rc->Height == srcheight &&
73 srcstride == dststride && srcstride == bytesperrow)
74 {
75 memcpy(dstbuffer, srcbuffer, srcstride * srcheight);
76 return S_OK;
77 }
78
79 row_offset = rc->X * bpp;
80
81 if (row_offset % 8 == 0)
82 {
83 /* everything lines up on a byte boundary */
84 INT row;
85 const BYTE *src;
86 BYTE *dst;
87
88 src = srcbuffer + (row_offset / 8) + srcstride * rc->Y;
89 dst = dstbuffer;
90 for (row=0; row < rc->Height; row++)
91 {
92 memcpy(dst, src, bytesperrow);
93 src += srcstride;
94 dst += dststride;
95 }
96 return S_OK;
97 }
98 else
99 {
100 /* we have to do a weird bitwise copy. eww. */
101 FIXME("cannot reliably copy bitmap data if bpp < 8\n");
102 return E_FAIL;
103 }
104 }
105
106 HRESULT configure_write_source(IWICBitmapFrameEncode *iface,
107 IWICBitmapSource *source, const WICRect *prc,
108 const WICPixelFormatGUID *format,
109 INT width, INT height, double xres, double yres)
110 {
111 HRESULT hr = S_OK;
112
113 if (width == 0 || height == 0)
114 return WINCODEC_ERR_WRONGSTATE;
115
116 if (!format)
117 {
118 WICPixelFormatGUID src_format;
119
120 hr = IWICBitmapSource_GetPixelFormat(source, &src_format);
121 if (FAILED(hr)) return hr;
122
123 hr = IWICBitmapFrameEncode_SetPixelFormat(iface, &src_format);
124 if (FAILED(hr)) return hr;
125 }
126
127 if (xres == 0.0 || yres == 0.0)
128 {
129 hr = IWICBitmapSource_GetResolution(source, &xres, &yres);
130 if (FAILED(hr)) return hr;
131 hr = IWICBitmapFrameEncode_SetResolution(iface, xres, yres);
132 if (FAILED(hr)) return hr;
133 }
134
135 return hr;
136 }
137
138 HRESULT write_source(IWICBitmapFrameEncode *iface,
139 IWICBitmapSource *source, const WICRect *prc,
140 const WICPixelFormatGUID *format, UINT bpp,
141 INT width, INT height)
142 {
143 IWICBitmapSource *converted_source;
144 HRESULT hr=S_OK;
145 WICRect rc;
146 UINT stride;
147 BYTE* pixeldata;
148
149 if (!prc)
150 {
151 UINT src_width, src_height;
152 hr = IWICBitmapSource_GetSize(source, &src_width, &src_height);
153 if (FAILED(hr)) return hr;
154 rc.X = 0;
155 rc.Y = 0;
156 rc.Width = src_width;
157 rc.Height = src_height;
158 prc = &rc;
159 }
160
161 if (prc->Width != width || prc->Height <= 0)
162 return E_INVALIDARG;
163
164 hr = WICConvertBitmapSource(format, source, &converted_source);
165 if (FAILED(hr))
166 {
167 ERR("Failed to convert source, target format %s, %#x\n", debugstr_guid(format), hr);
168 return hr;
169 }
170
171 stride = (bpp * width + 7)/8;
172
173 pixeldata = HeapAlloc(GetProcessHeap(), 0, stride * prc->Height);
174 if (!pixeldata)
175 {
176 IWICBitmapSource_Release(converted_source);
177 return E_OUTOFMEMORY;
178 }
179
180 hr = IWICBitmapSource_CopyPixels(converted_source, prc, stride,
181 stride*prc->Height, pixeldata);
182
183 if (SUCCEEDED(hr))
184 {
185 hr = IWICBitmapFrameEncode_WritePixels(iface, prc->Height, stride,
186 stride*prc->Height, pixeldata);
187 }
188
189 HeapFree(GetProcessHeap(), 0, pixeldata);
190 IWICBitmapSource_Release(converted_source);
191
192 return hr;
193 }
194
195 void reverse_bgr8(UINT bytesperpixel, LPBYTE bits, UINT width, UINT height, INT stride)
196 {
197 UINT x, y;
198 BYTE *pixel, temp;
199
200 for (y=0; y<height; y++)
201 {
202 pixel = bits + stride * y;
203
204 for (x=0; x<width; x++)
205 {
206 temp = pixel[2];
207 pixel[2] = pixel[0];
208 pixel[0] = temp;
209 pixel += bytesperpixel;
210 }
211 }
212 }
213
214 HRESULT get_pixelformat_bpp(const GUID *pixelformat, UINT *bpp)
215 {
216 HRESULT hr;
217 IWICComponentInfo *info;
218 IWICPixelFormatInfo *formatinfo;
219
220 hr = CreateComponentInfo(pixelformat, &info);
221 if (SUCCEEDED(hr))
222 {
223 hr = IWICComponentInfo_QueryInterface(info, &IID_IWICPixelFormatInfo, (void**)&formatinfo);
224
225 if (SUCCEEDED(hr))
226 {
227 hr = IWICPixelFormatInfo_GetBitsPerPixel(formatinfo, bpp);
228
229 IWICPixelFormatInfo_Release(formatinfo);
230 }
231
232 IWICComponentInfo_Release(info);
233 }
234
235 return hr;
236 }