/[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.5 by joko, Sat Jul 1 12:19:46 2006 UTC revision 1.15 by joko, Sun Jul 2 12:08:10 2006 UTC
# Line 15  Line 15 
15  #define YSIZE 500  #define YSIZE 500
16  #include "algorithm.h"  #include "algorithm.h"
17    
18    BOOL VERBOSE = FALSE;
19    
20    
21  /* BMP Header */  /* BMP Header */
22  unsigned char header[54]={0x42,0x4d,              // signature BM  unsigned char header[54]={0x42,0x4d,              // signature BM
# Line 47  void printErrorAndExit(const char *msg, Line 49  void printErrorAndExit(const char *msg,
49                                            0,                                            0,
50                                            NULL ))                                            NULL ))
51          {          {
52                  fprintf(stderr,"%s : %s\n",msg,lpMsgBuf);                  fprintf(stderr, "%s: %s\n", msg, lpMsgBuf);
53                  LocalFree(lpMsgBuf);                  LocalFree(lpMsgBuf);
54          }          }
55          else          else
56          {          {
57                  fprintf(stderr,"Error at FormatMesage: %d\n",err=GetLastError());                  fprintf(stderr, "Error at FormatMesage: %d\n",err=GetLastError());
58          }          }
59          exit(err);          exit(err);
60  }  }
# Line 66  void write_blank_file(char *filename) { Line 68  void write_blank_file(char *filename) {
68    // open file handle    // open file handle
69    fd = fopen(filename, "wb+");    fd = fopen(filename, "wb+");
70    if (NULL == fd) {    if (NULL == fd) {
71      perror("open");      perror("Error while opening file for writing");
72      exit(1);      exit(1);
73    }    }
74    
# Line 75  void write_blank_file(char *filename) { Line 77  void write_blank_file(char *filename) {
77        
78    // error checking    // error checking
79    if (-1 == len || len != sizeof(header)) {    if (-1 == len || len != sizeof(header)) {
80      perror("write");      perror("Error while writing header to file");
81      exit(2);      exit(2);
82    }    }
83        
# Line 84  void write_blank_file(char *filename) { Line 86  void write_blank_file(char *filename) {
86    for (i = 0; i < img_size; i++) {    for (i = 0; i < img_size; i++) {
87      len = fwrite("\0\0\0", 1, 3, fd);      len = fwrite("\0\0\0", 1, 3, fd);
88      if (-1 == len || len != 3) {      if (-1 == len || len != 3) {
89        perror("write");        perror("Error while writing data to file");
90        exit(4);        exit(4);
91      }      }
92    }    }
# Line 94  void write_blank_file(char *filename) { Line 96  void write_blank_file(char *filename) {
96  }  }
97    
98    
99    // arguments for each thread
100    typedef struct _WorkerArguments {
101      int start_row;
102      int number_of_rows;
103      unsigned char * pBitmap;
104    } WORKERARGS, *PWORKERARGS;
105    
106    // worker thread - main entry function
107    DWORD WINAPI fractal_create_segment (LPVOID lpParam) {
108      
109      // thread stuff
110      int thread_id;
111      PWORKERARGS args;
112      unsigned char *pDataBitmapSegment;
113    
114      // fractal calculation
115      int x, y;
116      char bgr[3];
117    
118      thread_id = GetCurrentThreadId();
119      
120      // get worker arguments
121      args = (PWORKERARGS)lpParam;
122    
123      // 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    
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    BOOL scan_argv(int argc, char *argv[], char opt_name[], char *opt_value) {
161      int i;
162      char * opt_current_name;
163      char * opt_current_value;
164    
165      //printf("searching for: '%s'\n", opt_name);
166      
167      for (i = 1; i < argc; i++) {
168        opt_current_name = argv[i];
169        if (strcmp(opt_current_name, opt_name) == 0) {
170          opt_current_value = argv[i+1];
171          if (opt_current_value != NULL) {
172            strcpy(opt_value, opt_current_value);
173          }
174          return TRUE;
175        }
176      }
177      return FALSE;
178    }
179    
180    
181  int main(int argc, char *argv[]) {  int main(int argc, char *argv[]) {
182    
183    // MMF support    // MMF support
184    DWORD err;    DWORD err;
185    HANDLE hMap, hFile;    HANDLE hMap, hFile;
186    LPVOID pData;    LPVOID pData;
187    unsigned char *pDataBitmap, *pDataBitmapCurrent;    unsigned char *pDataBitmap;
188    
189    // workers    // workers
190    unsigned int workers = 3;    int workers = 5;
191    unsigned int worker_index, worker_rows, worker_startrow;    int worker_index, worker_rows, worker_startrow;
192      HANDLE *worker_handles;
193      PWORKERARGS worker_args;
194    
195      // threads or processes?
196      BOOL use_processes = FALSE;
197      BOOL is_worker_process = FALSE;
198        
199    // fractal calculation    // information for creating processes
200    int x, y;    STARTUPINFO si;
201    char bgr[3];    PROCESS_INFORMATION pi;
202      char szCmdline[65536];
203        
204      // command line stuff
205      char arg_option[1024];
206      char arg_value[1024];
207      char *bmp_filename;
208      char *verbose_option = "";
209        
   // create empty bmp-file (black background)  
   write_blank_file("test.bmp");  
210    
211    /* open file for reading and writing */    // parse command line arguments
212    hFile = CreateFile("test.bmp", GENERIC_WRITE|GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);    if (argc < 2) {
213    if (INVALID_HANDLE_VALUE == hFile) {      fprintf(stderr, "Can not run without arguments!\nPlease specify '-t {number of threads}' or '-p {number of processes}' and an image filename.\n");
214      err = GetLastError();      exit(EXIT_FAILURE);
215      printErrorAndExit("Error at CreateFile",err);    }
216      
217      if (scan_argv(argc, argv, "--verbose", arg_value)) {
218        VERBOSE = TRUE;
219      }
220      
221      if (scan_argv(argc, argv, "--worker", arg_value)) {
222        use_processes = TRUE;
223        is_worker_process = TRUE;
224      
225      } else if (scan_argv(argc, argv, "-p", arg_value)) {
226        if (strlen(arg_value) == 0) {
227          fprintf(stderr, "Please specify number of processes!\n");
228          exit(EXIT_FAILURE);
229        }
230        use_processes = TRUE;
231        is_worker_process = FALSE;
232        workers = atoi(arg_value);
233      
234      } else if (scan_argv(argc, argv, "-t", arg_value)) {
235        if (strlen(arg_value) == 0) {
236          fprintf(stderr, "Please specify number of threads!\n");
237          exit(EXIT_FAILURE);
238        }
239        use_processes = FALSE;
240        is_worker_process = FALSE;
241        workers = atoi(arg_value);
242        
243      }
244      
245      
246      if (VERBOSE && use_processes) {
247        fprintf(stdout, "===================================================== ");
248        if (is_worker_process)
249          fprintf(stdout, "WORKER-PROCESS\n");
250        else
251          fprintf(stdout, "MASTER-PROCESS\n");
252    }    }
253    
254    /* create the file mapping object */    
255    hMap = CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);    // master creates memory mapped file ("empty" image)
256    if (NULL == hMap) {    if (!is_worker_process) {
257      printErrorAndExit("Error at CreateFileMapping", GetLastError());      
258        if (argc < 4) {
259          fprintf(stderr, "Must give filename of image as third argument!\n");
260          exit(EXIT_FAILURE);
261        }
262        
263        bmp_filename = argv[3];
264        
265        // create empty bmp-file (black background)
266        write_blank_file(bmp_filename);
267    
268        // open file for reading and writing
269        hFile = CreateFile(bmp_filename, GENERIC_WRITE|GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
270        if (INVALID_HANDLE_VALUE == hFile) {
271          err = GetLastError();
272          printErrorAndExit("Error at CreateFile",err);
273        }
274    
275        // create the file mapping object
276        hMap = CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, "bmp_fractal");
277        if (NULL == hMap) {
278          printErrorAndExit("Error at CreateFileMapping", GetLastError());
279        }
280      
281      // worker uses existing memory mapped file
282      } else {
283        
284        // open existing mapping object
285        hMap = OpenFileMapping(FILE_MAP_WRITE, FALSE, "bmp_fractal");
286        if (NULL == hMap)
287          printErrorAndExit("Error at OpenFileMapping", GetLastError());
288    }    }
289        
290    /* map the whole file into the process context */    // map the whole file into the process context
291    pData = MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0);    pData = MapViewOfFile(hMap, FILE_MAP_WRITE, 0, 0, 0);
292    if (NULL == pData) {    if (NULL == pData) {
293      printErrorAndExit("Error at MapViewOfFile", GetLastError());      printErrorAndExit("Error at MapViewOfFile", GetLastError());
# Line 138  int main(int argc, char *argv[]) { Line 298  int main(int argc, char *argv[]) {
298    pDataBitmap = (unsigned char *)((INT_PTR)pData + sizeof(header));    pDataBitmap = (unsigned char *)((INT_PTR)pData + sizeof(header));
299        
300    // debugging    // debugging
301    printf("pos. of file: %p\n", pData);    if (VERBOSE) {
302    printf("pos. of bitmap: %p\n", pDataBitmap);      fprintf(stdout, "pos. of file: %p\n", pData);
303        fprintf(stdout, "pos. of bitmap: %p\n", pDataBitmap);
304    // pointer to current pixel    }
   pDataBitmapCurrent = pDataBitmap;  
305    
306      
307      if (use_processes && is_worker_process) {
308      
309        if (VERBOSE)
310          fprintf(stdout, "inside worker-process\n");
311        
312        // get segment information from command line
313        worker_startrow = atoi(argv[2]);
314        worker_rows = atoi(argv[3]);
315        
316        // allocate memory for one worker's arguments
317        if ((worker_args = malloc(sizeof(worker_args[0]))) == NULL)
318          perror("Error while allocating memory for worker arguments via malloc"), exit(1);
319        
320        // assign worker's arguments
321        worker_args[0].start_row = worker_startrow;
322        worker_args[0].number_of_rows = worker_rows;
323        worker_args[0].pBitmap = pDataBitmap;
324    
325        fractal_create_segment(&worker_args[0]);
326    
327        // cleanup mmap-handle
328        if (!CloseHandle(hMap))
329          printErrorAndExit("Error at CloseHandle", GetLastError());
330        
331        return 0;
332      }
333      
334    /*    /*
335    // turn bitmap into white wand    // turn bitmap into white canvas
336    for (offset = 0; offset < 500 * 500 * 3; offset++) {    for (offset = 0; offset < 500 * 500 * 3; offset++) {
337      //pDataBitmap[offset] = 255;      *pDataBitmap = 255;
338      *pDataBitmapCurrent = 255;      pDataBitmap++;
     pDataBitmapCurrent++;  
339    }    }
340      exit(0);
341    */    */
342    
343    // calculate segments of bitmap for worker threads/processes    // allocate memory for bitmap
344      /*
345      if ((pDataBitmap = malloc(XSIZE * YSIZE * 3 * sizeof(pDataBitmap[0]))) == NULL)
346        perror("malloc"), exit(1);
347      */
348      
349      // allocate memory for table of all worker handles
350      if ((worker_handles = malloc(workers * sizeof(worker_handles[0]))) == NULL)
351        perror("Error while allocating memory for worker handles via malloc"), exit(1);
352    
353      // allocate memory for table of all worker arguments
354      if ((worker_args = malloc(workers * sizeof(worker_args[0]))) == NULL)
355        perror("Error while allocating memory for worker arguments via malloc"), exit(1);
356      
357      
358      // calculate bitmap segment length for workers
359    worker_rows = YSIZE / workers;    worker_rows = YSIZE / workers;
360    printf("rows for each worker: %i\n", worker_rows);    if (VERBOSE)
361        fprintf(stdout, "rows for each worker: %i\n", worker_rows);
362      
363      // start workers
364    for (worker_index = 0; worker_index < workers; worker_index++) {    for (worker_index = 0; worker_index < workers; worker_index++) {
365            
366        // debugging: just run single thread
367        //if (worker_index == 1)
368        //  continue;
369        
370      // number of row to start for each worker      // number of row to start for each worker
371      worker_startrow = worker_index * worker_rows;      worker_startrow = worker_index * worker_rows;
372            
373      // recalculate number of rows for last worker if (YSIZE mod workers) != 0      // recalculate number of rows for last worker if (YSIZE mod workers) != 0
374      if (worker_index == workers - 1) {      if (worker_index == workers - 1) {
375        worker_rows = YSIZE - worker_startrow;        worker_rows = YSIZE - worker_startrow;
376        printf("rows for last worker: %i\n", worker_rows);        if (VERBOSE)
377            fprintf(stdout, "rows for last worker: %i\n", worker_rows);
378        }
379        
380        // assign each worker's arguments
381        worker_args[worker_index].start_row = worker_startrow;
382        worker_args[worker_index].number_of_rows = worker_rows;
383        worker_args[worker_index].pBitmap = pDataBitmap;
384        
385        if (!use_processes) {
386        
387          worker_handles[worker_index] = CreateThread(
388            NULL,                          // LPSECURITY_ATTRIBUTES lpThreadAttributes,
389            0,                             // SIZE_T dwStackSize,
390            &fractal_create_segment,                // LPTHREAD_START_ROUTINE lpStartAddress,
391            &worker_args[worker_index],    // LPVOID lpParameter,
392            0,                             // DWORD dwCreationFlags,
393            NULL                           // LPDWORD lpThreadId
394          );
395    
396          if (!worker_handles[worker_index])
397            printErrorAndExit("CreateThread failed", GetLastError());
398          
399        } else {
400    
401          if (VERBOSE)
402            verbose_option = "--verbose";
403          _snprintf(szCmdline, 1023, "%s %s %i %i %s", argv[0], "--worker", worker_startrow, worker_rows, verbose_option);
404          if (VERBOSE)
405            fprintf(stdout, "starting worker process: %s\n", szCmdline);
406          ZeroMemory( &si, sizeof(si) );
407          si.cb = sizeof(si);
408          ZeroMemory( &pi, sizeof(pi) );
409          
410          if (!CreateProcess(
411              NULL,           // No module name (use command line)
412              szCmdline,      // Command line
413              NULL,           // Process handle not inheritable
414              NULL,           // Thread handle not inheritable
415              FALSE,          // Set handle inheritance to FALSE
416              0,              // No creation flags
417              NULL,           // Use parent's environment block
418              NULL,           // Use parent's starting directory
419              &si,            // Pointer to STARTUPINFO structure
420              &pi             // Pointer to PROCESS_INFORMATION structure
421          ))
422            printErrorAndExit("CreateProcess failed", GetLastError());
423        
424          worker_handles[worker_index] = pi.hProcess;
425          
426      }      }
   }  
427    
   // calculate fractal  
   for (y=YSIZE-1; y>=0; y--) {  
           for (x=0; x<XSIZE; x++) {  
             getColorValuesAt(x * (2.0 / XSIZE) - 1.5, y * (2.0 / YSIZE) - 1.0, &bgr[2], &bgr[1], &bgr[0]);  
               
             // debugging  
             //printf("pointer: %p\n", pDataBitmapCurrent);  
               
             // transfer color values to current pixel  
             pDataBitmapCurrent[0] = bgr[0];  
             pDataBitmapCurrent[1] = bgr[1];  
             pDataBitmapCurrent[2] = bgr[2];  
               
             // move pointer to next pixel  
             pDataBitmapCurrent += 3;  
               
           }  
           //no padding required because 1500%4 =0  
428    }    }
429    
430        // wait for all threads
431    /* write the result into the file */    if (VERBOSE)
432        fprintf(stdout, "waiting for workers to finish...\n");
433      if (WaitForMultipleObjects(workers, worker_handles, TRUE, INFINITE) == WAIT_FAILED)
434        printErrorAndExit("Error at WaitForMultipleObjects", GetLastError());
435    
436      // debugging: just run single thread
437      //if (WaitForSingleObject(worker_handles[0], INFINITE) == WAIT_FAILED)
438      //  perror("WaitForSingleObject");
439    
440      // close all worker handles
441      for (worker_index = 0; worker_index < workers; worker_index++)
442        CloseHandle(worker_handles[worker_index]);
443        
444      // write the result into the file
445    if (!FlushViewOfFile(pData, 0)) {    if (!FlushViewOfFile(pData, 0)) {
446      err = GetLastError();      err = GetLastError();
447      printErrorAndExit("Error at UnmapViewOfFile", err);      printErrorAndExit("Error at UnmapViewOfFile", err);
448    }    }
449    
450    /* remove the mapped file */    // remove the mapped file
451    if (!UnmapViewOfFile(pData)) {    if (!UnmapViewOfFile(pData)) {
452      err = GetLastError();      err = GetLastError();
453      printErrorAndExit("Error at UnmapViewOfFile", err);      printErrorAndExit("Error at UnmapViewOfFile", err);
454      exit(err);      exit(err);
455    }    }
456        
457    /* cleanup handles */    // cleanup handles
458    if (!CloseHandle(hMap) || !CloseHandle(hFile) ) {    if (!CloseHandle(hMap) || !CloseHandle(hFile) ) {
459      err = GetLastError();      err = GetLastError();
460      printErrorAndExit("Error at CloseHandle", err);      printErrorAndExit("Error at CloseHandle", err);
461    }    }
462    
463      free(worker_args);
464      free(worker_handles);
465      
466      return 0;
467      
468  }  }
Legend:
Removed from v.1.5  
changed lines
  Added in v.1.15
MailToCvsAdmin">MailToCvsAdmin
 ViewVC Help
Powered by ViewVC 1.1.26 RSS 2.0 feed