/[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.9 by joko, Sun Jul 2 00:24:17 2006 UTC revision 1.10 by joko, Sun Jul 2 01:51:26 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;  #define MASTER -1
19    
20    BOOL VERBOSE = TRUE;
21    
22    
23  /* BMP Header */  /* BMP Header */
# Line 104  typedef struct _WorkerArguments { Line 106  typedef struct _WorkerArguments {
106  } WORKERARGS, *PWORKERARGS;  } WORKERARGS, *PWORKERARGS;
107    
108  // worker thread - main entry function  // worker thread - main entry function
109  DWORD WINAPI fractal_thread (LPVOID lpParam) {  DWORD WINAPI fractal_create_segment (LPVOID lpParam) {
110        
111    // thread stuff    // thread stuff
112    int thread_id;    int thread_id;
# Line 166  int main(int argc, char *argv[]) { Line 168  int main(int argc, char *argv[]) {
168    unsigned char *pDataBitmap;    unsigned char *pDataBitmap;
169    
170    // workers    // workers
171    int workers = 50;    int workers = 5;
172    int worker_index, worker_rows, worker_startrow;    int worker_index, worker_rows, worker_startrow;
173    HANDLE *worker_handles;    HANDLE *worker_handles;
174    PWORKERARGS worker_args;    PWORKERARGS worker_args;
175    
176      // threads or processes?
177      BOOL use_processes = TRUE;
178      int worker_id = MASTER;
179      
180      // information for creating processes
181      STARTUPINFO si;
182      PROCESS_INFORMATION pi;
183      char szCmdline[1024];
184      
185      
186      // "parse" command line arguments
187      if (argc >= 2) {
188        worker_id = atoi(argv[1]);
189      }
190      
191      if (VERBOSE) {
192        printf("==============================================================\n");
193        printf("worker-id: %i\n", worker_id);
194      }
195        
196    // create empty bmp-file (black background)    // create empty bmp-file (black background)
197    write_blank_file("test.bmp");    if (worker_id == MASTER)
198        write_blank_file("test.bmp");
199    
200    /* open file for reading and writing */    if (worker_id == MASTER) {
201    hFile = CreateFile("test.bmp", GENERIC_WRITE|GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);      
202    if (INVALID_HANDLE_VALUE == hFile) {      /* open file for reading and writing */
203      err = GetLastError();      hFile = CreateFile("test.bmp", GENERIC_WRITE|GENERIC_READ, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
204      printErrorAndExit("Error at CreateFile",err);      if (INVALID_HANDLE_VALUE == hFile) {
205    }        err = GetLastError();
206          printErrorAndExit("Error at CreateFile",err);
207        }
208    
209    /* create the file mapping object */      /* create the file mapping object */
210    hMap = CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, NULL);      hMap = CreateFileMapping(hFile, NULL, PAGE_READWRITE, 0, 0, "bmp_fractal");
211    if (NULL == hMap) {      if (NULL == hMap) {
212      printErrorAndExit("Error at CreateFileMapping", GetLastError());        printErrorAndExit("Error at CreateFileMapping", GetLastError());
213        }
214        
215      } else {
216        
217        // open existing mapping object
218        hMap = OpenFileMapping(FILE_MAP_WRITE, FALSE, "bmp_fractal");
219        if (NULL == hMap)
220          printErrorAndExit("Error at OpenFileMapping", GetLastError());
221    }    }
222        
223    /* map the whole file into the process context */    /* map the whole file into the process context */
# Line 204  int main(int argc, char *argv[]) { Line 236  int main(int argc, char *argv[]) {
236      printf("pos. of bitmap: %p\n", pDataBitmap);      printf("pos. of bitmap: %p\n", pDataBitmap);
237    }    }
238    
239      
240      if (use_processes && worker_id != MASTER) {  
241      
242        if (VERBOSE)
243          printf("inside worker-process\n");
244        
245        worker_index = worker_id;
246        
247        worker_startrow = atoi(argv[2]);
248        worker_rows = atoi(argv[3]);
249        
250        // allocate memory for one worker's arguments
251        if ((worker_args = malloc(sizeof(worker_args[0]))) == NULL)
252          perror("malloc"), exit(1);
253        
254        // assign worker's arguments
255        worker_args[0].start_row = worker_startrow;
256        worker_args[0].number_of_rows = worker_rows;
257        worker_args[0].pBitmap = pDataBitmap;
258    
259        fractal_create_segment(&worker_args[0]);
260        
261        return 0;
262      }
263      
264    /*    /*
265    // turn bitmap into white canvas    // turn bitmap into white canvas
266    for (offset = 0; offset < 500 * 500 * 3; offset++) {    for (offset = 0; offset < 500 * 500 * 3; offset++) {
# Line 228  int main(int argc, char *argv[]) { Line 285  int main(int argc, char *argv[]) {
285      perror("malloc"), exit(1);      perror("malloc"), exit(1);
286        
287        
288    // calculate segments of bitmap for worker threads/processes and start them    // calculate bitmap segment length for workers
289    worker_rows = YSIZE / workers;    worker_rows = YSIZE / workers;
290    if (VERBOSE)    if (VERBOSE)
291      printf("rows for each worker: %i\n", worker_rows);      printf("rows for each worker: %i\n", worker_rows);
292      
293      // start workers
294    for (worker_index = 0; worker_index < workers; worker_index++) {    for (worker_index = 0; worker_index < workers; worker_index++) {
295            
296      // debugging: just run single thread      // debugging: just run single thread
# Line 248  int main(int argc, char *argv[]) { Line 307  int main(int argc, char *argv[]) {
307          printf("rows for last worker: %i\n", worker_rows);          printf("rows for last worker: %i\n", worker_rows);
308      }      }
309            
310        // assign each worker's arguments
311      worker_args[worker_index].start_row = worker_startrow;      worker_args[worker_index].start_row = worker_startrow;
312      worker_args[worker_index].number_of_rows = worker_rows;      worker_args[worker_index].number_of_rows = worker_rows;
313      worker_args[worker_index].pBitmap = pDataBitmap;      worker_args[worker_index].pBitmap = pDataBitmap;
314            
315      worker_handles[worker_index] = CreateThread(      if (!use_processes) {
       NULL,                          // LPSECURITY_ATTRIBUTES lpThreadAttributes,  
       0,                             // SIZE_T dwStackSize,  
       &fractal_thread,                // LPTHREAD_START_ROUTINE lpStartAddress,  
       &worker_args[worker_index],    // LPVOID lpParameter,  
       0,                             // DWORD dwCreationFlags,  
       NULL                           // LPDWORD lpThreadId  
     );  
316            
317          worker_handles[worker_index] = CreateThread(
318            NULL,                          // LPSECURITY_ATTRIBUTES lpThreadAttributes,
319            0,                             // SIZE_T dwStackSize,
320            &fractal_create_segment,                // LPTHREAD_START_ROUTINE lpStartAddress,
321            &worker_args[worker_index],    // LPVOID lpParameter,
322            0,                             // DWORD dwCreationFlags,
323            NULL                           // LPDWORD lpThreadId
324          );
325    
326          if (!worker_handles[worker_index])
327            printErrorAndExit("CreateThread failed", GetLastError());
328          
329        } else {
330    
331          _snprintf(szCmdline, 1023, "%s %i %i %i", argv[0], worker_index + 1, worker_startrow, worker_rows);
332          ZeroMemory( &si, sizeof(si) );
333          si.cb = sizeof(si);
334          ZeroMemory( &pi, sizeof(pi) );
335          
336          if (!CreateProcess(
337              NULL,           // No module name (use command line)
338              szCmdline,      // Command line
339              NULL,           // Process handle not inheritable
340              NULL,           // Thread handle not inheritable
341              FALSE,          // Set handle inheritance to FALSE
342              0,              // No creation flags
343              NULL,           // Use parent's environment block
344              NULL,           // Use parent's starting directory
345              &si,            // Pointer to STARTUPINFO structure
346              &pi             // Pointer to PROCESS_INFORMATION structure
347          ))
348            printErrorAndExit("CreateProcess failed", GetLastError());
349        
350          worker_handles[worker_index] = pi.hProcess;
351          
352        }
353    
354    }    }
355    
356    // wait for all threads    // wait for all threads
357    if (VERBOSE)    if (VERBOSE)
358      printf("waiting...\n");      printf("waiting for workers to finish...\n");
359    if (WaitForMultipleObjects(workers, worker_handles, TRUE, INFINITE) == WAIT_FAILED)    if (WaitForMultipleObjects(workers, worker_handles, TRUE, INFINITE) == WAIT_FAILED)
360      perror("WaitForMultipleObjects");      perror("WaitForMultipleObjects");
361    

Legend:
Removed from v.1.9  
changed lines
  Added in v.1.10

MailToCvsAdmin">MailToCvsAdmin
ViewVC Help
Powered by ViewVC 1.1.26 RSS 2.0 feed