From f71d835e9c1cebc41e2dd4ecfc377475548937e3 Mon Sep 17 00:00:00 2001
From: Emanuele De Rubeis <ederubei@login07.leonardo.local>
Date: Mon, 11 Dec 2023 14:29:00 +0100
Subject: [PATCH] Write data

---
 fourier_transform.c | 51 ++++++++++++++++++++++++++++++++++++---------
 1 file changed, 41 insertions(+), 10 deletions(-)

diff --git a/fourier_transform.c b/fourier_transform.c
index 11dab41..709593b 100644
--- a/fourier_transform.c
+++ b/fourier_transform.c
@@ -215,7 +215,7 @@ void write_fftw_data(){
 
   timing_wt.phase += CPU_TIME_wt - start;
   
- #ifdef WRITE_IMAGE
+#ifdef WRITE_IMAGE
 
   double start_image = CPU_TIME_wt;
   
@@ -276,18 +276,49 @@ void write_fftw_data(){
 
  #endif //FITSIO
 
-  file.pFilereal = fopen (out.fftfile2,"ab");
-  file.pFileimg = fopen (out.fftfile3,"ab");
+  for (int isector=0; isector<size; isector++)
+    {
+
+      MPI_Barrier(MPI_COMM_WORLD);
+      
+      if(isector == rank)
+	{
+
+	  printf("%d writing\n",isector);
+
+	 #ifdef FITSIO
+
+	  fpixel[0] = 1;
+	  fpixel[1] = isector*yaxis+1;
+	  lpixel[0] = xaxis;
+	  lpixel[1] = (isector+1)*yaxis;
 
-  uint global_index = rank*(xaxis*yaxis)*sizeof(double);
+	  status = 0;
+	  fits_open_image(&fptreal, testfitsreal, READWRITE, &status);
+	  fits_write_subset(fptreal, TDOUBLE, fpixel, lpixel, image_real, &status);
+	  fits_close_file(fptreal, &status);
 
-  fseek(file.pFilereal, global_index, SEEK_SET);
-  fwrite(image_real, xaxis*yaxis, sizeof(double), file.pFilereal);
-  fseek(file.pFileimg, global_index, SEEK_SET);
-  fwrite(image_imag, xaxis*yaxis, sizeof(double), file.pFileimg);
+	  status = 0;
+	  fits_open_image(&fptrimg, testfitsimag, READWRITE, &status);
+	  fits_write_subset(fptrimg, TDOUBLE, fpixel, lpixel, image_imag, &status);
+	  fits_close_file(fptrimg, &status);
 
-  fclose(file.pFilereal);
-  fclose(file.pFileimg);
+	 #endif //FITSIO
+
+	  file.pFilereal = fopen (out.fftfile2,"ab");
+	  file.pFileimg = fopen (out.fftfile3,"ab");
+
+	  uint global_index = isector*(xaxis*yaxis)*sizeof(double);
+
+	  fseek(file.pFilereal, global_index, SEEK_SET);
+	  fwrite(image_real, xaxis*yaxis, sizeof(double), file.pFilereal);
+	  fseek(file.pFileimg, global_index, SEEK_SET);
+	  fwrite(image_imag, xaxis*yaxis, sizeof(double), file.pFileimg);
+
+	  fclose(file.pFilereal);
+	  fclose(file.pFileimg);
+	}
+    }
  
   MPI_Barrier(MPI_COMM_WORLD);
 
-- 
GitLab