mad2hfits.c

/*******************************************************************************
*   M3:  mad2hfits.c                                                  *
*                                                                              *
*   Version 1.0 May 2004                                                       *
*                                                                              *
*   Copyright (C) 2004  C.M. Cantalupo                                         *
*                                                                              *
*   This program is free software; you can redistribute it and/or modify       *
*   it under the terms of the GNU General Public License as published by       *
*   the Free Software Foundation; either version 2 of the License, or          *
*   (at your option) any later version.                                        *
*                                                                              *
*   This program is distributed in the hope that it will be useful,            *
*   but WITHOUT ANY WARRANTY; without even the implied warranty of             *
*   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the              *
*   GNU General Public License for more details.                               *
*                                                                              *
*   You should have received a copy of the GNU General Public License          *
*   along with this program; if not, write to the Free Software                *
*   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA  *
*                                                                              *
*******************************************************************************/



#include <stdlib.h>
#include <stdio.h>
#include <math.h>
#include "M3_system.h"
#include "M3_error.h"
#include "hpic.h"



typedef struct
{
  int32_t pixel;
  double value;
} M3_MapEl;


int main( int argc, char **argv )
{
  int32_t header[2];
  int32_t header2[2];
  char inFileNameI[256];
  char inFileNameQ[256];
  char inFileNameU[256];
  char outFileName[256];
  M3_MapEl *data;
  FILE *fid;
  int64_t i, j;
  int nside;
  int isPolar;
  hpic_float *hpicMap[3] = {0};
  hpic_keys *hpicKeys;

  if( argc != 3 && argc != 5 )
  {
    fprintf( stderr, "Usage:  %s  { inFile | inFileI inFileQ inFileU } outFile\n\n", argv[0] );
    return(0);
  }

  if( argc == 3 )
  {
    strcpy( inFileNameI, argv[1]);
    strcpy( outFileName, argv[2]);
    isPolar = 0;
  }
  else
  {
    strcpy( inFileNameI, argv[1] );
    strcpy( inFileNameQ, argv[2] );
    strcpy( inFileNameU, argv[3] );
    strcpy( outFileName, argv[4] );
    isPolar = 1;
  }


  hpicKeys = hpic_keys_alloc();

  hpic_keys_iadd( hpicKeys, "POLAR", isPolar, "Pixelisation included (True/False)");
  hpic_keys_sadd( hpicKeys, "TTYPE1", "SIGNAL0 ", "label for field   1");
  
  if( isPolar )
  {
    hpic_keys_sadd( hpicKeys, "TTYPE2", "SIGNAL1 ", "label for field   2");
    hpic_keys_sadd( hpicKeys, "TTYPE3", "SIGNAL2 ", "label for field   3");
  }
  

  fid = fopen( inFileNameI, "r");
  M3_ErrorCheck( -1, inFileNameI, fid, M3_EFLAG_FOPEN_READ);

  fread( header, sizeof(int32_t), 2, fid);

  nside = sqrt(header[0]/12);

  data = (M3_MapEl *)malloc(sizeof(M3_MapEl)*header[1]);
  M3_ErrorCheck(-1, "main", data, M3_EFLAG_MALLOC_FSTRING );

  for( j = 0; j < 1 + 2*isPolar; j++ )
  {
    fread( data, sizeof(M3_MapEl), header[1], fid);
    if( getenv("MAD2HFITS_NEST") )
      hpicMap[j] = hpic_float_alloc( nside, HPIC_NEST, HPIC_COORD_G, HPIC_STND );
    else
      hpicMap[j] = hpic_float_alloc( nside, HPIC_RING, HPIC_COORD_G, HPIC_STND );
    

    for( i = 0; i < header[1]; i++ )
      hpic_float_set( hpicMap[j], data[i].pixel, data[i].value );

    fclose(fid);
    fid = NULL;

    if( isPolar && j == 0 )
    {
      fid = fopen( inFileNameQ, "r");
      M3_ErrorCheck(-1, inFileNameQ, fid, M3_EFLAG_MALLOC_FSTRING );
    }
    else if( isPolar && j == 1 )
    {
      fid = fopen( inFileNameU, "r");
      M3_ErrorCheck(-1, inFileNameQ, fid, M3_EFLAG_MALLOC_FSTRING );
    }

    if(fid)
    {
      fread( header2, sizeof(int32_t), 2, fid );
      M3_ErrorCheck(-1, "Number of pixels in class does not match between files.", header[0] == header2[0], M3_EFLAG_DEFAULT );
      if( header2[1] > header[1] )
      {
        data = (M3_MapEl *)realloc(data, sizeof(M3_MapEl)*header2[1] );
        M3_ErrorCheck(-1, "main", data, M3_EFLAG_MALLOC_FSTRING );
        header[1] = header2[1];
      }
    }
  }

  if( isPolar )
  {
    hpic_cmb_write_fullTQU( outFileName, hpicMap[0], hpicMap[1], hpicMap[2], "MADCAP3", "mad2hfits", hpicKeys );
  }
  else
  {
    hpic_cmb_write_full( outFileName, hpicMap[0], "MADCAP3", "mad2hfits", hpicKeys );
  }


  for( j = 0; j < 3; j++ )
    if( hpicMap[j] )
      hpic_float_free( hpicMap[j] );
  free(data);

  hpic_keys_free( hpicKeys );

}

    

---