/[cvs]/joko/Uni/BSArch/04/bmp_fractal.c
ViewVC logotype

Diff of /joko/Uni/BSArch/04/bmp_fractal.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.1 by joko, Fri Jun 30 20:39:33 2006 UTC revision 1.13 by joko, Sun Jul 2 10:45:02 2006 UTC
# Line 4  Line 4 
4   *   *
5   * Uebung 4.4   * Uebung 4.4
6   */   */
7    
8    // $Id$
9    
10    #include <windows.h>
11  #include <stdio.h>  #include <stdio.h>
12  #include <errno.h>  #include <errno.h>
13    
# Line 11  Line 15 
15  #define YSIZE 500  #define YSIZE 500
16  #include "algorithm.h"  #include "algorithm.h"
17    
18  /* BMP Header  BOOL VERBOSE = FALSE;
19    
20    
21    /* BMP Header */
22  unsigned char header[54]={0x42,0x4d,              // signature BM  unsigned char header[54]={0x42,0x4d,              // signature BM
23                              0xe6,0x71,0x0b,0x0,   // filesize 750054                              0xe6,0x71,0x0b,0x0,   // filesize 750054
24                              0x0,0x0,0x0,0x0,      // reserved                              0x0,0x0,0x0,0x0,      // reserved
# Line 28  unsigned char header[54]={0x42,0x4d, Line 35  unsigned char header[54]={0x42,0x4d,
35                              0x0,0x0,0x0,0x0,      // number of colortables                                  0x0,0x0,0x0,0x0,      // number of colortables    
36                              0x0,0x0,0x0,0x0       // number of important colors                              0x0,0x0,0x0,0x0       // number of important colors
37                          };                          };
 */  
38    
39  int main(int argc, char *argv[])  
40  {  void printErrorAndExit(const char *msg, DWORD err) {
41      FILE *fd;          LPSTR lpMsgBuf;
42      int len,x,y;          if(FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER|
43      char bgr[3];                                    FORMAT_MESSAGE_FROM_SYSTEM |
44      short svalue;                                            FORMAT_MESSAGE_IGNORE_INSERTS,
45      int   lvalue;                                                  NULL,
46      unsigned char header[54],*ptr=&header[0];                                                  err,
47                                                    MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language
48      fd=fopen("test.bmp","wb+");                                            (LPTSTR) &lpMsgBuf,
49      if(NULL==fd)                                            0,
50      {                                            NULL ))
51          perror("open"); exit(1);          {
52      }                  fprintf(stdout, "%s: %s\n", msg, lpMsgBuf);
53  /* Der folgende Abschnitt dient nur zur information und kann durch                  LocalFree(lpMsgBuf);
54      schreiben der auskommentierten Variable header ersetzt werden */              }
55      svalue=0x4d42;          else
56      memcpy(ptr,&svalue,2);//signatur          {
57      ptr+=2;                  fprintf(stdout, "Error at FormatMesage: %d\n",err=GetLastError());
58      lvalue=XSIZE*YSIZE*3+54;          }
59      memcpy(ptr,&lvalue,4); //filesize          exit(err);
60      ptr+=4;  }
61      lvalue=0;  
62      memcpy(ptr,&lvalue,4);//reserved  
63      ptr+=4;  void write_blank_file(char *filename) {
64      lvalue=54;  
65      memcpy(ptr,&lvalue,4);//image offset    FILE *fd;
66      ptr+=4;    int len, i, img_size;
67      lvalue=40;  
68      memcpy(ptr,&lvalue,4);//size of header follows    // open file handle
69      ptr+=4;    fd = fopen(filename, "wb+");
70      lvalue=XSIZE;    if (NULL == fd) {
71      memcpy(ptr,&lvalue,4);//with of image      perror("open");
72      ptr+=4;      exit(1);
73      lvalue=YSIZE;    }
74      memcpy(ptr,&lvalue,4); //height of image  
75      ptr+=4;    // write bmp header to file
76      svalue=1;    len = fwrite(header, 1, sizeof(header), fd);
77      memcpy(ptr,&svalue,2); //number of planes    
78      ptr+=2;    // error checking
79      svalue=24;    if (-1 == len || len != sizeof(header)) {
80      memcpy(ptr,&svalue,2); //number of pixel      perror("write");
81      ptr+=2;      exit(2);
82      lvalue=0; //compression    }
83      memcpy(ptr,&lvalue,4); //compression    
84      ptr+=4;    // write three null-bytes for each pixel to file to create a black picture
85      lvalue=XSIZE*YSIZE*3;    img_size = XSIZE * YSIZE;
86      memcpy(ptr,&lvalue,4); //size of image    for (i = 0; i < img_size; i++) {
87      ptr+=4;      len = fwrite("\0\0\0", 1, 3, fd);
88      lvalue=0;      if (-1 == len || len != 3) {
89      memcpy(ptr,&lvalue,4); //xres          perror("write");
90      ptr+=4;        exit(4);
91      lvalue=0;      }
92      memcpy(ptr,&lvalue,4); //yres    }
93      ptr+=4;    
94      lvalue=0;    // close file handle
95      memcpy(ptr,&lvalue,4); //number of colortables    fclose(fd);
96      ptr+=4;  }
97      lvalue=0;  
98      memcpy(ptr,&lvalue,4); //number of important colors  
99      ptr+=4;  // arguments for each thread
100  /* Ende Information */  typedef struct _WorkerArguments {
101      int start_row;
102      len=fwrite(header,1,sizeof(header),fd); //write header    int number_of_rows;
103          unsigned char * pBitmap;
104      if(-1==len || len!=sizeof(header))  } WORKERARGS, *PWORKERARGS;
105      {  
106          perror("write");  // worker thread - main entry function
107          exit(2);  DWORD WINAPI fractal_create_segment (LPVOID lpParam) {
108      }    
109          // thread stuff
110      for(y=YSIZE-1;y>=0;y--)    int thread_id;
111      {    PWORKERARGS args;
112              for(x=0;x<XSIZE;x++)    unsigned char *pDataBitmapSegment;
113              {  
114                  getColorValuesAt(x * (2.0 / XSIZE) - 1.5, y * (2.0 / YSIZE) - 1.0,&bgr[2],&bgr[1],&bgr[0]);    // fractal calculation
115                      int x, y;
116                  len=fwrite(bgr,1,3,fd);    char bgr[3];
117                  if(-1==len || len!=3)  
118                  {    thread_id = GetCurrentThreadId();
119                      perror("write");    
120                      exit(4);    // get worker arguments
121                  }    args = (PWORKERARGS)lpParam;
122              }  
123              /*no padding required because 1500%4 =0*/    // calculate pointer to beginning of segment
124      }            pDataBitmapSegment = (unsigned char *)((INT_PTR)args->pBitmap + (YSIZE - (args->start_row + args->number_of_rows)) * 3 * XSIZE);
125      fclose(fd);  
126      // debugging
127      if (VERBOSE) {
128        fprintf(stdout, "----------------------------------------------\n");
129        fprintf(stdout, "thread_id: %i\n", thread_id);
130        fprintf(stdout, "arg.start_row: %i\n", args->start_row);
131        fprintf(stdout, "arg.number_of_rows: %i\n", args->number_of_rows);
132        fprintf(stdout, "segment_start: %p\n", pDataBitmapSegment);
133      }
134      
135      // calculate fractal
136      for (y = (args->start_row + args->number_of_rows) - 1; y >= args->start_row; y--) {
137        //fprintf(stdout, "calc: thread=%i; y=%i            limits: %i,%i    p: %p\n", thread_id, y, args->start_row, args->number_of_rows, pDataBitmapSegment);
138        for (x = 0; x < XSIZE; x++) {
139          getColorValuesAt(x * (2.0 / XSIZE) - 1.5, y * (2.0 / YSIZE) - 1.0, &bgr[2], &bgr[1], &bgr[0]);
140          
141          // transfer color values to current pixel
142          pDataBitmapSegment[0] = bgr[0];
143          pDataBitmapSegment[1] = bgr[1];
144          pDataBitmapSegment[2] = bgr[2];
145          
146          // move pointer to next pixel
147          pDataBitmapSegment += 3;
148          
149        }
150        //no padding required because 1500%4 =0 ???
151      }
152    
153      if (VERBOSE)
154        fprintf(stdout, "thread finished: %i\n", thread_id);
155      return 0;
156    
157    }
158    
159    BOOL scan_argv(int argc, char *argv[], char opt_name[], char *opt_value[]) {
160      int i;
161      char * opt_current_name;
162      char * opt_current_value;
163      
164      //printf("searching for: '%s'\n", opt_name);
165      
166      for (i = 1; i < argc; i++) {
167        opt_current_name = argv[i];
168        if (strcmp(opt_current_name, opt_name) == 0) {
169          //printf("opt: %s\n", opt_current_name);
170          opt_current_value = argv[i+1];
171          //printf("opt-current-value: %s\n", opt_current_value);
172          if (opt_current_value != NULL) {
173            //*opt_value = *argv[i+1];
174            *opt_value = *opt_current_value;
175            //printf("opt-value: %s\n", opt_value);
176          }
177          return TRUE;
178        }
179      }
180      return FALSE;
181    }
182    
183    
184    int main(int argc, char *argv[]) {
185    
186      // MMF support
187      DWORD err;
188      HANDLE hMap, hFile;
189      LPVOID pData;
190      unsigned char *pDataBitmap;
191    
192      // workers
193      int workers = 5;
194      int worker_index, worker_rows, worker_startrow;
195      HANDLE *worker_handles;
196      PWORKERARGS worker_args;
197    
198      // threads or processes?
199      BOOL use_processes = FALSE;
200      BOOL is_worker_process = FALSE;
201      
202      // information for creating processes
203      STARTUPINFO si;
204      PROCESS_INFORMATION pi;
205      char szCmdline[65536];
206      
207      // command line stuff
208      char arg_option[1024];
209      char *arg_value[1024];
210      //arg_value = malloc(1024);
211      
212      VERBOSE = FALSE;
213      
214      // "parse" command line arguments
215      /*
216      if (argc >= 2) {
217        if (strcmp(argv[1], "--worker") == 0) {
218          is_worker_process = TRUE;
219        }
220      }
221      */
222      
223      //scan_argv(argc, argv, "--worker", &arg_value);
224      //*arg_value = '\0';
225      // parse command line arguments
226      if (scan_argv(argc, argv, "--worker", arg_value)) {
227        use_processes = TRUE;
228        is_worker_process = TRUE;
229      
230      } else if (scan_argv(argc, argv, "-p", arg_value)) {
231        use_processes = TRUE;
232        is_worker_process = FALSE;
233        workers = atoi(arg_value);
234      
235      } else if (scan_argv(argc, argv, "-t", arg_value)) {
236        use_processes = FALSE;
237        is_worker_process = FALSE;
238        workers = atoi(arg_value);
239            
240      }
241      //printf("value: %s\n", arg_value);
242      //exit(0);
243      
244      
245      if (VERBOSE && use_processes) {
246        fprintf(stdout, "===================================================== ");
247        if (is_worker_process)
248          fprintf(stdout, "WORKER-PROCESS\n");
249        else
250          fprintf(stdout, "MASTER-PROCESS\n");
251      }
252      
253      // create empty bmp-file (black background)
254      if (!is_worker_process)
255        write_blank_file("test.bmp");
256    
257      // master creates memory mapped file
258      if (!is_worker_process) {
259        
260        // open file for reading and writing
261        hFile = CreateFile("test.bmp", GENERIC_WRITE|GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
262        if (INVALID_HANDLE_VALUE == hFile) {
263          err = GetLastError();
264          printErrorAndExit("Error at CreateFile",err);
265        }
266    
267        // create the file mapping object
268        hMap = CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, "bmp_fractal");
269        if (NULL == hMap) {
270          printErrorAndExit("Error at CreateFileMapping", GetLastError());
271        }
272      
273      // worker uses existing memory mapped file
274      } else {
275        
276        // open existing mapping object
277        hMap = OpenFileMapping(FILE_MAP_WRITE, FALSE, "bmp_fractal");
278        if (NULL == hMap)
279          printErrorAndExit("Error at OpenFileMapping", GetLastError());
280      }
281      
282      // map the whole file into the process context
283      pData = MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0);
284      if (NULL == pData) {
285        printErrorAndExit("Error at MapViewOfFile", GetLastError());
286      }
287    
288      
289      // calculate pointer to beginning of bitmap
290      pDataBitmap = (unsigned char *)((INT_PTR)pData + sizeof(header));
291      
292      // debugging
293      if (VERBOSE) {
294        fprintf(stdout, "pos. of file: %p\n", pData);
295        fprintf(stdout, "pos. of bitmap: %p\n", pDataBitmap);
296      }
297    
298      
299      if (use_processes && is_worker_process) {
300      
301        if (VERBOSE)
302          fprintf(stdout, "inside worker-process\n");
303        
304        // get segment information from command line
305        worker_startrow = atoi(argv[2]);
306        worker_rows = atoi(argv[3]);
307        
308        // allocate memory for one worker's arguments
309        if ((worker_args = malloc(sizeof(worker_args[0]))) == NULL)
310          perror("malloc"), exit(1);
311        
312        // assign worker's arguments
313        worker_args[0].start_row = worker_startrow;
314        worker_args[0].number_of_rows = worker_rows;
315        worker_args[0].pBitmap = pDataBitmap;
316    
317        fractal_create_segment(&worker_args[0]);
318    
319        // cleanup mmap-handle
320        if (!CloseHandle(hMap))
321          printErrorAndExit("Error at CloseHandle", GetLastError());
322        
323        return 0;
324      }
325      
326      /*
327      // turn bitmap into white canvas
328      for (offset = 0; offset < 500 * 500 * 3; offset++) {
329        *pDataBitmap = 255;
330        pDataBitmap++;
331      }
332      exit(0);
333      */
334    
335      // allocate memory for bitmap
336      /*
337      if ((pDataBitmap = malloc(XSIZE * YSIZE * 3 * sizeof(pDataBitmap[0]))) == NULL)
338        perror("malloc"), exit(1);
339      */
340      
341      // allocate memory for table of all worker handles
342      if ((worker_handles = malloc(workers * sizeof(worker_handles[0]))) == NULL)
343        perror("malloc"), exit(1);
344    
345      // allocate memory for table of all worker arguments
346      if ((worker_args = malloc(workers * sizeof(worker_args[0]))) == NULL)
347        perror("malloc"), exit(1);
348      
349      
350      // calculate bitmap segment length for workers
351      worker_rows = YSIZE / workers;
352      if (VERBOSE)
353        fprintf(stdout, "rows for each worker: %i\n", worker_rows);
354      
355      // start workers
356      for (worker_index = 0; worker_index < workers; worker_index++) {
357        
358        // debugging: just run single thread
359        //if (worker_index == 1)
360        //  continue;
361        
362        // number of row to start for each worker
363        worker_startrow = worker_index * worker_rows;
364        
365        // recalculate number of rows for last worker if (YSIZE mod workers) != 0
366        if (worker_index == workers - 1) {
367          worker_rows = YSIZE - worker_startrow;
368          if (VERBOSE)
369            fprintf(stdout, "rows for last worker: %i\n", worker_rows);
370        }
371        
372        // assign each worker's arguments
373        worker_args[worker_index].start_row = worker_startrow;
374        worker_args[worker_index].number_of_rows = worker_rows;
375        worker_args[worker_index].pBitmap = pDataBitmap;
376        
377        if (!use_processes) {
378        
379          worker_handles[worker_index] = CreateThread(
380            NULL,                          // LPSECURITY_ATTRIBUTES lpThreadAttributes,
381            0,                             // SIZE_T dwStackSize,
382            &fractal_create_segment,                // LPTHREAD_START_ROUTINE lpStartAddress,
383            &worker_args[worker_index],    // LPVOID lpParameter,
384            0,                             // DWORD dwCreationFlags,
385            NULL                           // LPDWORD lpThreadId
386          );
387    
388          if (!worker_handles[worker_index])
389            printErrorAndExit("CreateThread failed", GetLastError());
390          
391        } else {
392    
393          _snprintf(szCmdline, 1023, "%s %s %i %i", argv[0], "--worker", worker_startrow, worker_rows);
394          if (VERBOSE)
395            fprintf(stdout, "starting worker process: %s\n", szCmdline);
396          ZeroMemory( &si, sizeof(si) );
397          si.cb = sizeof(si);
398          ZeroMemory( &pi, sizeof(pi) );
399          
400          if (!CreateProcess(
401              NULL,           // No module name (use command line)
402              szCmdline,      // Command line
403              NULL,           // Process handle not inheritable
404              NULL,           // Thread handle not inheritable
405              FALSE,          // Set handle inheritance to FALSE
406              0,              // No creation flags
407              NULL,           // Use parent's environment block
408              NULL,           // Use parent's starting directory
409              &si,            // Pointer to STARTUPINFO structure
410              &pi             // Pointer to PROCESS_INFORMATION structure
411          ))
412            printErrorAndExit("CreateProcess failed", GetLastError());
413        
414          worker_handles[worker_index] = pi.hProcess;
415          
416        }
417    
418      }
419    
420      // wait for all threads
421      if (VERBOSE)
422        fprintf(stdout, "waiting for workers to finish...\n");
423      if (WaitForMultipleObjects(workers, worker_handles, TRUE, INFINITE) == WAIT_FAILED)
424        perror("WaitForMultipleObjects");
425    
426      // debugging: just run single thread
427      //if (WaitForSingleObject(worker_handles[0], INFINITE) == WAIT_FAILED)
428      //  perror("WaitForSingleObject");
429    
430      // close all worker handles
431      for (worker_index = 0; worker_index < workers; worker_index++)
432        CloseHandle(worker_handles[worker_index]);
433        
434      // write the result into the file
435      if (!FlushViewOfFile(pData, 0)) {
436        err = GetLastError();
437        printErrorAndExit("Error at UnmapViewOfFile", err);
438      }
439    
440      // remove the mapped file
441      if (!UnmapViewOfFile(pData)) {
442        err = GetLastError();
443        printErrorAndExit("Error at UnmapViewOfFile", err);
444        exit(err);
445      }
446      
447      // cleanup handles
448      if (!CloseHandle(hMap) || !CloseHandle(hFile) ) {
449        err = GetLastError();
450        printErrorAndExit("Error at CloseHandle", err);
451      }
452    
453      free(worker_args);
454      free(worker_handles);
455      
456      return 0;
457      
458  }  }
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.13
MailToCvsAdmin">MailToCvsAdmin
 ViewVC Help
Powered by ViewVC 1.1.26 RSS 2.0 feed