]> hackdaworld.org Git - my-code/api.git/blob - bmp/bmp.c
another bugfix
[my-code/api.git] / bmp / bmp.c
1 /* bmp.c -- bmp write/read api
2  *
3  * author: hackbard@hackdaworld.dyndns.org
4  *
5  */
6
7 #include "bmp.h"
8
9 int bmp_init(t_bmp *bmp,int outfd) {
10
11   dprintf(outfd,"[bmp] initializing bmp api ...\n");
12
13   memset(bmp,0,sizeof(t_bmp));
14   bmp->outfd=outfd;
15
16   return B_SUCCESS;
17 }
18
19 int bmp_shutdown(t_bmp *bmp) {
20
21   if(bmp->map!=NULL) {
22     dprintf(bmp->outfd,"[bmp] free pixmap memory\n");
23     free(bmp->map);
24   }
25
26   dprintf(bmp->outfd,"[bmp] shutdown\n");
27
28   return B_SUCCESS;
29 }
30
31 int bmp_check_header_and_info(t_bmp *bmp) {
32
33   dprintf(bmp->outfd,"[bmp] magic identifier: %c%c\n",
34           bmp->hdr.identifier&0xff,bmp->hdr.identifier>>8);
35
36   if(bmp->info.compression!=0) {
37     dprintf(bmp->outfd,"[bmp] compression not supported\n");
38     return B_NO_SUPPORT;
39   }
40   
41   if(bmp->info.bpp!=24) {
42     dprintf(bmp->outfd,"[bmp] only true color (24bpp) supported\n");
43     return B_NO_SUPPORT;
44   }
45     
46   if(bmp->hdr.offset!=BMP_H_SIZE+BMP_I_SIZE) {
47     dprintf(bmp->outfd,"[bmp] files with %d bytes offset not supported\n",
48             bmp->hdr.offset);
49     return B_NO_SUPPORT;
50   }
51
52   if(bmp->info.size!=BMP_I_SIZE) {
53     dprintf(bmp->outfd,"[bmp] files with %d bytes info size not supported\n",
54             bmp->info.size);
55     return B_NO_SUPPORT;
56   }
57     
58   return B_SUCCESS;
59 }
60
61 int bmp_alloc_map(t_bmp *bmp) {
62
63   int size;
64
65   size=bmp->width*bmp->height*3;
66
67   dprintf(bmp->outfd,"[bmp] alloc map memory (%d bytes)\n",size);
68
69   if((bmp->map=(t_pixel *)malloc(size))==NULL) {
70     dprintf(bmp->outfd,"[bmp] memory map alloc failed\n");
71     return B_E_MEM;
72   }
73
74   return B_SUCCESS;
75 }
76
77 int bmp_write_file(t_bmp *bmp) {
78  
79   int fill,xsize,size;
80   int y;
81   unsigned char buf[3];
82
83   memset(buf,0,3);
84
85   if(!(bmp->mode&WRITE)) {
86     dprintf(bmp->outfd,"[bmp] write mode not specified\n");
87     return B_WRONG_MODE;
88   }
89
90   xsize=bmp->width*3;
91   fill=(4-(xsize%4))%4;
92   size=(xsize+fill)*bmp->height;
93
94   /* construct it */
95   bmp->hdr.identifier='B'|('M'<<8);
96   bmp->hdr.size=size+BMP_H_SIZE+BMP_I_SIZE;
97   bmp->hdr.offset=BMP_H_SIZE+BMP_I_SIZE;
98   bmp->info.size=BMP_I_SIZE;
99   bmp->info.width=bmp->width;
100   bmp->info.height=bmp->height;
101   bmp->info.planes=1;
102   bmp->info.bpp=24;
103   bmp->info.imagesize=size;
104   if(bmp->info.xres==0) bmp->info.xres=2048;
105   if(bmp->info.yres==0) bmp->info.yres=2048;
106   bmp->info.noc=0;
107   bmp->info.ic=0;
108
109   /* write it */
110   if((bmp->fd=open(bmp->file,O_WRONLY|O_CREAT))<0) {
111     dprintf(bmp->outfd,"[bmp] unable to open file %s\n",bmp->file);
112     return B_NO_FILE;
113   }
114
115   if(write(bmp->fd,&(bmp->hdr),BMP_H_SIZE)<BMP_H_SIZE) {
116     dprintf(bmp->outfd,"[bmp] unable to write bmp header\n");
117     return B_E_WRITE_DATA;
118   }
119
120   if(write(bmp->fd,&(bmp->info),BMP_I_SIZE)<BMP_I_SIZE) {
121     dprintf(bmp->outfd,"[bmp] unable to write bmp info\n");
122     return B_E_WRITE_DATA;
123   }
124
125   for(y=0;y<bmp->height;y++) {
126     if(write(bmp->fd,bmp->map+y*bmp->width,xsize)<xsize) {
127       dprintf(bmp->outfd,"[bmp] unable to write image data line %d\n",y);
128       return B_E_WRITE_DATA;
129     }
130     if(write(bmp->fd,buf,fill)<fill) {
131       dprintf(bmp->outfd,"[bmp] unable to write fill bytes\n");
132       return B_E_WRITE_DATA;
133     }
134   }
135
136   close(bmp->fd);
137
138   return B_SUCCESS;
139 }
140
141 int bmp_cut_grab_bottom(t_bmp *dst,t_bmp *src,int dz,unsigned char m) {
142
143   int off;
144
145   dst->width=src->width;
146   dst->height=dz;
147
148   if(dz>src->height) {
149     dprintf(src->outfd,"[bmp] cut region greater than image height\n");
150     return B_E_GEOMETRY;
151   }
152
153   if(bmp_alloc_map(dst)!=B_SUCCESS) {
154     dprintf(dst->outfd,"[bmp] no map memory\n");
155     return B_E_MEM;
156   }
157
158   off=(m==GRAB)?0:(src->height-dz)*src->width;
159   memcpy(dst->map,src->map+off,dz*src->width*sizeof(t_pixel));
160
161   return B_SUCCESS;
162 }
163
164 int bmp_read_file(t_bmp *bmp) {
165
166   unsigned char buf[BMP_H_SIZE+BMP_I_SIZE];
167   int y,xsize;
168   int crop;
169
170   if(!(bmp->mode&READ)) {
171     dprintf(bmp->outfd,"[bmp] read mode not specified\n");
172     return B_WRONG_MODE;
173   }
174
175   if((bmp->fd=open(bmp->file,O_RDONLY))<0) {
176     dprintf(bmp->outfd,"[bmp] unable to open file %s\n",bmp->file);
177     return B_NO_FILE;
178   }
179
180   if(read(bmp->fd,buf,BMP_H_SIZE+BMP_I_SIZE)<BMP_H_SIZE+BMP_I_SIZE) {
181     dprintf(bmp->outfd,"[bmp] error reading bmp header & info\n");
182     return B_NO_HI;
183   }
184
185   memcpy(&(bmp->hdr),buf,BMP_H_SIZE);
186   memcpy(&(bmp->info),buf+BMP_H_SIZE,BMP_I_SIZE);
187
188   if(bmp_check_header_and_info(bmp)!=B_SUCCESS) {
189     dprintf(bmp->outfd,"[bmp] header/info check failed\n");
190     return B_HI_FAIL;
191   }
192
193   bmp->width=bmp->info.width;
194   bmp->height=bmp->info.height;
195
196   bmp->map=(t_pixel *)malloc(bmp->width*bmp->height*sizeof(t_pixel));
197   if(bmp->map==NULL) {
198     dprintf(bmp->outfd,"[bmp] malloc of map memory failed\n");
199     return B_E_MEM;
200   }
201
202   crop=bmp->info.imagesize/bmp->height-bmp->width*(bmp->info.bpp/8);
203   xsize=(bmp->info.bpp/8)*bmp->width;
204
205   for(y=0;y<bmp->height;y++) {
206     if(read(bmp->fd,bmp->map+y*bmp->width,xsize)<xsize) {
207       dprintf(bmp->outfd,"[bmp] reading image data of line %d failed\n",y);
208       return B_E_READ_DATA;
209     }
210     if(read(bmp->fd,buf,crop)<crop) {
211       dprintf(bmp->outfd,"[bmp] failed reading rest of line\n");
212       return B_E_READ_DATA;
213     }
214   }
215     
216   close(bmp->fd);
217
218   return B_SUCCESS;
219 }