passepartout overlay wasnt written into thumbnails
[blender.git] / source / blender / imbuf / intern / thumbs_blend.c
1 /**
2  * $Id:
3  *
4  * ***** BEGIN GPL LICENSE BLOCK *****
5  *
6  * This program is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU General Public License
8  * as published by the Free Software Foundation; either version 2
9  * of the License, or (at your option) any later version. 
10  *
11  * This program is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software Foundation,
18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19  *
20  * Contributor(s): Campbell Barton.
21  *
22  * ***** END GPL LICENSE BLOCK *****
23  */
24
25 #include <string.h>
26
27 #include "zlib.h"
28
29 #include "BKE_utildefines.h"
30 #include "BKE_global.h"
31
32 #include "IMB_imbuf_types.h"
33 #include "IMB_imbuf.h"
34 #include "IMB_thumbs.h"
35
36 #include "MEM_guardedalloc.h"
37
38 /* extracts the thumbnail from between the 'REND' and the 'GLOB'
39  * chunks of the header, dont use typical blend loader because its too slow */
40 ImBuf *IMB_loadblend_thumb(const char *path)
41 {
42         char buf[8];
43         int code= 0;
44         char endian, pointer_size;
45         char endian_switch;
46         int len, im_len, x, y;
47         int *rect= NULL;
48
49         gzFile gzfile;
50         
51         ImBuf *img;
52         
53         /* not necessarily a gzip */
54         gzfile = gzopen(path, "rb");
55
56         if (NULL == gzfile ) {
57                 return NULL;
58         }
59         
60         /* read the blend file header */
61         if(gzread(gzfile, buf, 8) != 8)         goto thumb_error;
62         if(strncmp(buf, "BLENDER", 7))          goto thumb_error;
63         
64         if(buf[7]=='-')                                         pointer_size= 8;
65         else if(buf[7]=='_')                            pointer_size= 4;
66         else                                                            goto thumb_error;
67         
68         /* read the next 4 bytes, only need the first char, ignore the version */
69         /* endian and vertsion (ignored) */
70         if(gzread(gzfile, buf, 4) != 4)         goto thumb_error;
71         
72         if(buf[0]=='V')                                         endian= B_ENDIAN; /* big: PPC */
73         else if(buf[0]=='v')                            endian= L_ENDIAN; /* little: x86 */
74         else                                                            goto thumb_error;
75
76         while(gzread(gzfile, &code, sizeof(int)) == sizeof(int)) {
77                 endian_switch = ((ENDIAN_ORDER != endian)) ? 1 : 0;
78                 
79                 if(gzread(gzfile, buf, sizeof(int)) != sizeof(int))             goto thumb_error;
80                 len = *( (int *)((void *)buf) );
81                 if(endian_switch) SWITCH_INT(len);
82                 
83                 /* finally read the rest of the bhead struct, pointer and 2 ints */
84                 if(gzread(gzfile, buf, pointer_size) != pointer_size)   goto thumb_error;
85                 if(gzread(gzfile, buf, sizeof(int) * 2) != sizeof(int) * 2)     goto thumb_error;
86                 /* we dont actually care whats in the bhead */
87                 
88                 if (code==REND) {
89                         gzseek(gzfile, len, SEEK_CUR); /* skip to the next */
90                 }
91                 else {
92                         break;
93                 }
94         }
95         
96         /* using 'TEST' since new names segfault when loading in old blenders */
97         if(code != TEST)                                        goto thumb_error;
98         
99         if(gzread(gzfile, &x, sizeof(int)) != sizeof(int))              goto thumb_error;
100         if(gzread(gzfile, &y, sizeof(int)) != sizeof(int))              goto thumb_error;
101         len -= sizeof(int) * 2;
102
103         if(endian_switch) { SWITCH_INT(x); SWITCH_INT(y); }
104
105         /* inconsistant image size, quit early */
106         im_len = x * y * sizeof(int);
107         if(im_len != len)                                       goto thumb_error;
108
109         /* finally malloc and read the data */
110         rect= MEM_mallocN(len, "imb_loadblend_thumb");
111
112         if(gzread(gzfile, rect, len) != len) goto thumb_error;
113
114         /* read ok! */
115         gzclose(gzfile);
116
117         img = IMB_allocImBuf(x, y, 32, IB_rect | IB_metadata, 0);
118
119         memcpy(img->rect, rect, im_len);
120
121         MEM_freeN(rect);
122         
123         return img;     
124
125 thumb_error:
126         gzclose(gzfile);
127         if(rect) MEM_freeN(rect);
128         return NULL;
129 }
130
131 /* add a fake passepartout overlay to a byte buffer, use for blend file thumbnails */
132 #define MARGIN 2
133
134 void IMB_overlayblend_thumb(unsigned int *thumb, int width, int height, float aspect)
135 {
136         unsigned char *px= (unsigned char *)thumb;
137         int margin_l = MARGIN;
138         int margin_b = MARGIN;
139         int margin_r = width - MARGIN;
140         int margin_t = height - MARGIN;
141
142         if(aspect < 1.0f) {
143                 margin_l= (int)((width - ((float)width * aspect)) / 2.0f);
144                 margin_l += MARGIN;
145                 CLAMP(margin_l, MARGIN, (width/2));
146                 margin_r = width - margin_l;
147         }
148         else if (aspect > 1.0f) {
149                 margin_b= (int)((height - ((float)height / aspect)) / 2.0f);
150                 margin_b += MARGIN;
151                 CLAMP(margin_b, MARGIN, (height/2));
152                 margin_t = height - margin_b;
153         }
154
155         {       
156                 int x, y;
157                 int hline, vline;
158                 int stride_x= (margin_r - margin_l) - 2;
159                 
160                 for(y=0; y < height; y++) {
161                         for(x=0; x < width; x++, px+=4) {
162                                 if((x > margin_l && x < margin_r) && (y > margin_b && y < margin_t)) {
163                                         /* interior. skip */
164                                         x  += stride_x;
165                                         px += stride_x * 4;
166                                 } else if(      (hline=(((x == margin_l || x == margin_r)) && y >= margin_b && y <= margin_t)) ||
167                                                         (vline=(((y == margin_b || y == margin_t)) && x >= margin_l && x <= margin_r))
168                                 ) {
169                                         /* dashed line */
170                                         if((hline && y % 2) || (vline && x % 2)) {
171                                                 px[0]= px[1]= px[2]= 0;
172                                                 px[3] = 255;
173                                         }
174                                 }
175                                 else {
176                                         /* outside, fill in alpha, like passepartout */
177                                         px[0] *= 0.5f;
178                                         px[1] *= 0.5f;
179                                         px[2] *= 0.5f;
180                                         px[3] = (px[3] * 0.5f) + 96;
181                                 }
182                         }
183                 }
184         }
185 }