ace4dbe07a93a591edbf3a4b2e741888c6aa4ea0
[vuplus_openvuplus_3.0] / meta-openvuplus / recipes-vuplus / aio-grab / aio-grab / aio-grab_vuplus_8394e1ba98c73d7c56d3235ef16cce253dc597cd.patch
1 diff --git a/.gitignore b/.gitignore
2 index ccfb515..a6f8d11 100644
3 --- a/.gitignore
4 +++ b/.gitignore
5 @@ -16,3 +16,4 @@ install-sh
6  *.o
7  missing
8  stamp-h1
9 +aio-grab_vuplus_*
10 diff --git a/main.c b/main.c
11 index 190ff5f..ef186a8 100644
12 --- a/main.c
13 +++ b/main.c
14 @@ -1,5 +1,5 @@
15  /*
16 -AiO Dreambox Screengrabber v0.83a
17 +AiO Screengrabber v0.83a
18  
19  written 2006 - 2009 by Seddi
20  Contact: seddi@ihad.tv / http://www.ihad.tv
21 @@ -38,6 +38,7 @@ Feel free to use the code for your own projects. See LICENSE file for details.
22  #include <sys/stat.h>
23  #include <sys/types.h>
24  #include <unistd.h>
25 +#include <ctype.h>
26  
27  #include <linux/fb.h>
28  
29 @@ -101,10 +102,24 @@ static void combine(unsigned char *output,
30                     const unsigned char *video, const unsigned char *osd,
31                     unsigned int xres, unsigned int yres);
32  
33 +#define IS_VUPLUS 1
34 +#if IS_VUPLUS
35 +enum {UNKNOWN,PALLAS,VULCAN,XILLEON,BRCM7401,BRCM7400,BRCM7405,BRCM7335,BRCM7325,BRCM7346,BRCM7425};
36 +char *stb_name[]={"unknown","Pallas","Vulcan","Xilleon","Brcm7401","Brcm7400","Brcm7405","Brcm7335","Brcm7325","Brcm7346","Brcm7425\/7241"};
37 +#else
38  enum {UNKNOWN,PALLAS,VULCAN,XILLEON,BRCM7401,BRCM7400,BRCM7405};
39  static const char *stb_name[]={"unknown","Pallas","Vulcan","Xilleon","Brcm7401","Brcm7400","Brcm7405"};
40 +#endif
41  static int stb_type=UNKNOWN;
42  
43 +char* upcase(char* mixedstr)
44 +{
45 +       size_t j;
46 +       for (j=0; j< strlen(mixedstr); ++j)
47 +               mixedstr[j]=toupper(mixedstr[j]);
48 +       return mixedstr;
49 +}
50 +
51  static const char *file_getline(const char *filename)
52  {
53         static char *line = NULL;
54 @@ -184,10 +199,9 @@ static int file_scanf_lines(const char *filename, const char *fmt, ...)
55  }
56  
57  // main program
58 -
59  int main(int argc, char **argv) {
60  
61 -       printf("AiO Dreambox Screengrabber " PACKAGE_VERSION "\n\n");
62 +       printf("AiO Screengrabber " PACKAGE_VERSION "\n\n");
63  
64         unsigned int xres_v = 0,yres_v = 0,xres_o,yres_o,xres,yres,aspect,width;
65         int c,osd_only,video_only,use_osd_res,use_png,use_jpg,jpg_quality,no_aspect,use_letterbox;
66 @@ -205,11 +219,31 @@ int main(int argc, char **argv) {
67         char filename[256] = { "/tmp/screenshot.bmp" };
68  
69         // detect STB
70 -       const char *line = file_getline("/proc/fb");
71 +       char *line = (char*)file_getline("/proc/fb");
72         if (line == NULL)
73                 return 1;
74  
75         if (strstr(line, "bcmfb")) {
76 +#if IS_VUPLUS
77 +               if((line=(char*)file_getline("/proc/stb/info/chipset")) == NULL)
78 +                       return 1;
79 +               if (strstr(upcase(line),"7335"))
80 +                       stb_type=BRCM7335;
81 +               else if (strstr(upcase(line),"7325"))
82 +                       stb_type=BRCM7325;
83 +               else if (strstr(upcase(line),"7405"))
84 +                       stb_type=BRCM7405;
85 +               else if (strstr(upcase(line),"7356"))
86 +                       stb_type=BRCM7346;
87 +               else if (strstr(upcase(line),"7346"))
88 +                       stb_type=BRCM7346;
89 +               else if (strstr(upcase(line),"7424"))
90 +                       stb_type=BRCM7425;
91 +               else if (strstr(upcase(line),"7425"))
92 +                       stb_type=BRCM7425;
93 +               else if (strstr(upcase(line),"7241"))
94 +                       stb_type=BRCM7425;
95 +#else
96                 line = file_getline("/proc/stb/info/model");
97                 if (line == NULL)
98                         return 1;
99 @@ -221,6 +255,7 @@ int main(int argc, char **argv) {
100                          !strcmp(line, "dm800se") ||
101                          !strcmp(line, "dm7020hd"))
102                         stb_type = BRCM7405;
103 +#endif
104         } else if (strstr(line, "xilleonfb")) {
105                 stb_type = XILLEON;
106         } else if (strstr(line, "Pallas FB")) {
107 @@ -563,16 +598,34 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
108                 perror("/dev/mem");
109                 return false;
110         }
111 -
112 +#if IS_VUPLUS
113 +       if (stb_type == BRCM7401 || stb_type == BRCM7400 || stb_type == BRCM7405 || stb_type == BRCM7335 || stb_type == BRCM7325 || stb_type == BRCM7346 || stb_type == BRCM7425)
114 +#else
115         if (stb_type == BRCM7401 || stb_type == BRCM7400 || stb_type == BRCM7405)
116 +#endif
117         {
118                 // grab brcm7401 pic from decoder memory
119 +#if IS_VUPLUS
120 +               off_t offset_addr = 0x10600000;
121 +               switch(stb_type) {
122 +               case BRCM7401:
123 +               case BRCM7400:
124 +               case BRCM7405:
125 +               case BRCM7335:
126 +               case BRCM7325: offset_addr = 0x10100000; break;
127 +               }
128 +               unsigned char *memory = mmap(0, 100, PROT_READ, MAP_SHARED, mem_fd, offset_addr);
129 +               if (memory == MAP_FAILED) {
130 +                       perror("mmap");
131 +                       return false;
132 +               }
133 +#else
134                 unsigned char *memory = mmap(0, 100, PROT_READ, MAP_SHARED, mem_fd, 0x10100000);
135                 if (memory == MAP_FAILED) {
136                         perror("mmap");
137                         return false;
138                 }
139 -
140 +#endif
141                 unsigned char data[100];
142  
143                 unsigned int adr,adr2,ofs,ofs2,offset/*,vert_start,vert_end*/;
144 @@ -582,12 +635,30 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
145                 memcpy(data,memory,100); 
146                 //vert_start=data[0x1B]<<8|data[0x1A];
147                 //vert_end=data[0x19]<<8|data[0x18];
148 +#if IS_VUPLUS
149 +               if(stb_type == BRCM7401 || stb_type == BRCM7400 || stb_type == BRCM7405 || stb_type == BRCM7335 || stb_type == BRCM7325) {
150 +                       stride=data[0x15]<<8|data[0x14];        
151 +                       ofs=(data[0x28]<<8|data[0x27])>>4;
152 +                       ofs2=(data[0x2c]<<8|data[0x2b])>>4;
153 +                       adr=(data[0x1f]<<24|data[0x1e]<<16|data[0x1d]<<8|data[0x1c])&0xFFFFFF00;
154 +                       adr2=(data[0x23]<<24|data[0x22]<<16|data[0x21]<<8|data[0x20])&0xFFFFFF00;
155 +                       offset=adr2-adr;
156 +               } else {
157 +                       stride=data[0x15]<<8|data[0x14];
158 +                       ofs=(data[0x3c]<<8|data[0x3b])>>4;
159 +                       ofs2=(data[0x40]<<8|data[0x3f])>>4;
160 +                       adr=(data[0x1f]<<24|data[0x1e]<<16|data[0x1d]<<8|data[0x1c])&0xFFFFFF00;
161 +                       adr2=(data[0x37]<<24|data[0x36]<<16|data[0x35]<<8|data[0x34])&0xFFFFFF00;
162 +                       offset=adr2-adr;
163 +               }
164 +#else
165                 stride=data[0x15]<<8|data[0x14];        
166                 ofs=(data[0x28]<<8|data[0x27])>>4;
167                 ofs2=(data[0x2c]<<8|data[0x2b])>>4;
168                 adr=(data[0x1f]<<24|data[0x1e]<<16|data[0x1d]<<8|data[0x1c])&0xFFFFFF00;
169                 adr2=(data[0x23]<<24|data[0x22]<<16|data[0x21]<<8|data[0x20])&0xFFFFFF00;
170                 offset=adr2-adr;
171 +#endif
172                 
173                 munmap(memory, 100);
174  
175 @@ -609,7 +680,11 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
176                 assert(chroma);
177  
178                 // grabbing luma & chroma plane from the decoder memory
179 +#if IS_VUPLUS
180 +               if (stb_type == BRCM7401 || stb_type == BRCM7405 || stb_type == BRCM7325 || stb_type == BRCM7346 || stb_type == BRCM7425) {
181 +#else
182                 if (stb_type == BRCM7401 || stb_type == BRCM7405) {
183 +#endif
184                         // on dm800/dm500hd we have direct access to the decoder memory
185                         memory = mmap(0, offset + stride*(ofs2+64), PROT_READ, MAP_SHARED, mem_fd, adr);
186                         if (memory == MAP_FAILED) {
187 @@ -620,7 +695,11 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
188                         usleep(50000);  // we try to get a full picture, its not possible to get a sync from the decoder so we use a delay
189                                         // and hope we get a good timing. dont ask me why, but every DM800 i tested so far produced a good
190                                         // result with a 50ms delay
191 +#if IS_VUPLUS
192 +               } else if (stb_type == BRCM7400 || stb_type == BRCM7335) {
193 +#else
194                 } else if (stb_type == BRCM7400) { 
195 +#endif
196                         // on dm8000 we have to use dma, so dont change anything here until you really know what you are doing !
197                         
198                         unsigned int i = 0;
199 @@ -639,7 +718,11 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
200                         }
201  
202                         volatile unsigned long *mem_dma;
203 +#if IS_VUPLUS
204 +                       mem_dma = mmap(0, 0x1000, PROT_READ|PROT_WRITE, MAP_SHARED, mem_fd, (stb_type==BRCM7400)?0x10c02000:0x10c01000);
205 +#else
206                         mem_dma = mmap(0, 0x1000, PROT_READ|PROT_WRITE, MAP_SHARED, mem_fd, 0x10c02000);
207 +#endif
208                         if (mem_dma == MAP_FAILED) {
209                                 perror("mmap");
210                                 return false;
211 @@ -675,8 +758,11 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
212                 unsigned int t = 0, t2 = 0, dat1 = 0;
213                 unsigned int chr_luma_stride = 0x40;
214                 unsigned int sw;
215 -
216 +#if IS_VUPLUS
217 +               if (stb_type == BRCM7405 || stb_type == BRCM7325 || stb_type == BRCM7346 || stb_type == BRCM7425)
218 +#else
219                 if (stb_type == BRCM7405)
220 +#endif
221                         chr_luma_stride *= 2;
222  
223                 xsub=chr_luma_stride;
224 @@ -710,10 +796,17 @@ static bool getvideo(unsigned char *video, unsigned int *xres, unsigned int *yre
225                                 dat1+=stride;
226                         }
227                 }
228 -
229 +#if IS_VUPLUS
230 +               if (stb_type == BRCM7401 || stb_type == BRCM7405 || stb_type == BRCM7325 || stb_type == BRCM7346 || stb_type == BRCM7425) {
231 +#else
232                 if (stb_type == BRCM7401 || stb_type == BRCM7405)
233 +#endif
234                         munmap(memory, offset + stride * (ofs2 + 64));
235 +#if IS_VUPLUS
236 +               } else if (stb_type == BRCM7400 || stb_type == BRCM7335) {
237 +#else
238                 else if (stb_type == BRCM7400) {
239 +#endif
240                         memory -= 0x1000;
241                         munmap(memory, DMA_BLOCKSIZE + 0x1000);
242                 }