/
coder.c
98 lines (96 loc) · 2.13 KB
/
coder.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
#include <ctype.h>
#include <string.h>
#include <stdint.h>
#include <math.h>
void put3c(int a,int b,int c,FILE * out){
putc(a,out);
putc(b,out);
putc(c,out);
return;
}
int get_mask(int n){
int mask=0;
for(int i=0;i<n;i++){
mask <<=1;
mask |= 1;
}
mask <<= 8-n;
return mask;
}
int half(int n,int num){
int tmp=0;
tmp |=1;
tmp<<=7-n;
num |=tmp;
return num;
}
int quant (int c,int mask, int bit){
return half(bit,c&mask);
}
int main (int argc,char **argv){
FILE *fp = fopen(argv[1], "rb");
FILE *fp2 = fopen("lower.tga", "wb");
FILE *fp3 = fopen("upper.tga", "wb");
int w, h, mask_r,mask_g,mask_b;
long long r=0,g=0,b=0;
//int r_bit=argv[3][0]-48,g_bit=argv[4][0]-48,b_bit=argv[5][0]-48;
//printf("%d %d %d",r_bit,g_bit,b_bit);
int r_bit=3,g_bit=3,b_bit=2;
r_bit=b_bit=g_bit=(argv[2][0]-48);
float mse,mse_r,mse_g,mse_b,snr,snr_r,snr_g,snr_b;
mse_r=mse_g=mse_b=mse=snr=snr_r=snr_g=snr_b=0;
mask_r=get_mask(r_bit);
mask_g=get_mask(g_bit);
mask_b=get_mask(b_bit);
int header[18];
int header2[18];
for (int i =0; i<18 ; i++){
header[i]=getc(fp);
header2[i]=header[i];
}
//pobieranie nagłówka
w = (header[13]<<8)+header[12];
h = (header[15]<<8)+header[14];
for (int i =0; i<18 ; i++){
putc(header2[i],fp2);
putc(header2[i],fp3);
}
int i=0;
int j=0;
int rgb[3];
int rgb_l[3];
int rgb_u[3],rgb_ur[3];
int nrgb[3];
int prgb[3]={0},prgb_u[3]={0};
for (i=0;i<h;i++){
for( j=0;j < w;j++){
if((j%2)==0){
for(int k=0;k<3;k++){
rgb[k]=getc(fp);
int tmp2=rgb[k];
rgb_l[k]=(tmp2+prgb[k])/2;
int tmp=rgb[k];
rgb_u[k]=((unsigned)(tmp-prgb[k])/2);
rgb_ur[k]=rgb_u[k]-prgb_u[k];
}
for(int l=0;l<3;l++) prgb[l]=rgb[l];
for(int l=0;l<3;l++) prgb_u[l]=rgb_u[l];
nrgb[0]=quant(rgb_l[0],mask_r,r_bit);
nrgb[1]=quant(rgb_l[1],mask_g,g_bit);
nrgb[2]=quant(rgb_l[2],mask_b,b_bit);
put3c(nrgb[0],nrgb[1],nrgb[2],fp2);
put3c(rgb_ur[0],rgb_ur[1],rgb_ur[2],fp2);
}else{
for(int i=0;i<3;i++){
getc(fp);
}
}
}
}
int all=w*h;
return 0;
}