// -----------------------------------------------------------------------
// DuchampBeam.cc: Handling of a beam in the Duchamp context, with Param and 
//                 FITS I/O interfaces
// -----------------------------------------------------------------------
// Copyright (C) 2006, Matthew Whiting, ATNF
//
// 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.
//
// Duchamp 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 Duchamp; if not, write to the Free Software Foundation,
// Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA
//
// Correspondence concerning Duchamp may be directed to:
//    Internet email: Matthew.Whiting [at] atnf.csiro.au
//    Postal address: Dr. Matthew Whiting
//                    Australia Telescope National Facility, CSIRO
//                    PO Box 76
//                    Epping NSW 1710
//                    AUSTRALIA
// -----------------------------------------------------------------------
#include <iostream>
#include <string.h>
#include <duchamp/duchamp.hh>
#include <duchamp/FitsIO/DuchampBeam.hh>
#include <duchamp/FitsIO/Beam.hh>
#include <duchamp/param.hh>
#include <sstream>

namespace duchamp 
{

  DuchampBeam::DuchampBeam():
    Beam()
  {
    this->itsOrigin = EMPTY;
    this->itsPixelScale = 1.;
  }

  DuchampBeam::DuchampBeam(float maj, float min, float pa):
    Beam(maj,min,pa)
  {
    this->itsOrigin  = EMPTY;
    this->itsPixelScale = 1.;
  }

  DuchampBeam::DuchampBeam(const Beam &b):
    Beam(b)
  {
    this->itsOrigin = EMPTY;
    this->itsPixelScale = 1.;
  }

  DuchampBeam::DuchampBeam(const DuchampBeam &b):
    Beam(b)
  {
    operator=(b);
  }

  DuchampBeam& DuchampBeam::operator=(const DuchampBeam &b)
  {
    ((Beam &) *this) = b;
    this->itsOrigin = b.itsOrigin;		
    this->itsPixelScale = b.itsPixelScale;
    return *this;
  }
    
  void DuchampBeam::define(float maj, float min, float pa, BEAM_ORIGIN origin)
  {
    this->Beam::define(maj,min,pa);
    this->itsOrigin = origin;
    this->itsPixelScale = 1.;
  }
  
  void DuchampBeam::setFWHM(float fwhm, BEAM_ORIGIN origin)
  {
    this->Beam::setFWHM(fwhm);
    this->itsOrigin = origin;
    this->itsPixelScale = 1.;
  }

  void DuchampBeam::setArea(float area, BEAM_ORIGIN origin)
  {
    this->Beam::setArea(area);
    this->itsOrigin = origin;
  }

  void DuchampBeam::empty()
  {
    this->itsMaj = this->itsMin = this->itsPA = this->itsArea = 0.;
    this->itsOrigin = EMPTY;
  }

  std::string DuchampBeam::originString()
  {
    std::string output;
    switch(this->itsOrigin)
      {
      case HEADER:
	output="HEADER";
	break;
      case PARAM:
	output="PARAM";
	break;
      case EMPTY:
	output="EMPTY";
	break;
      default:
	output="ERROR";
	break;
      }
    return output;

  }

  void DuchampBeam::define(Param &par, bool warn)
  {
    std::string paramName;
    bool doWarning=warn;
    if(par.getBeamFWHM()>0.){
      this->setFWHM(par.getBeamFWHM(),PARAM);
      par.setBeamSize(this->itsArea);
      paramName = "beamFWHM";
      par.setBeamAsUsed(*this);
    }
    else if(par.getBeamSize()>0.){
      this->setArea(par.getBeamSize(),PARAM);
      paramName = "beamArea";
      par.setBeamAsUsed(*this);
    }
    else{
      this->empty();
      doWarning = false;
    }

    if(doWarning){
      DUCHAMPWARN("Beam definition","Header beam keywords not present. Using parameter "<< paramName <<" to determine size of beam.");
    }
  }
    

  void DuchampBeam::readFromFITS(fitsfile *fptr, Param &par, float pixelScale) // read from file a la headerIO.cc
  {
    char *comment = new char[80];
    std::string keyword[3]={"BMAJ","BMIN","BPA"};
    float bmaj,bmin,bpa;

    // Read the Keywords. If they are present, read the
    //   others, and calculate the beam size.
    // If it is not, give warning and set beam size to nominal value.
    int bstatus[3]={0,0,0};
    fits_read_key(fptr, TFLOAT, (char *)keyword[0].c_str(), &bmaj, comment, &bstatus[0]);
    fits_read_key(fptr, TFLOAT, (char *)keyword[1].c_str(), &bmin, comment, &bstatus[1]);
    fits_read_key(fptr, TFLOAT, (char *)keyword[2].c_str(), &bpa, comment,  &bstatus[2]);

    if(bstatus[0]||bstatus[1]||bstatus[2]){ // error
      this->define(par);
    }
    else{ // all keywords present
      this->define(bmaj/pixelScale, bmin/pixelScale, bpa, HEADER);
      par.setBeamAsUsed(*this);
      this->itsPixelScale = pixelScale;
    }
    delete [] comment;
  }

  
  void DuchampBeam::writeToFITS(fitsfile *fptr) // write to file, but only if itsOrigin==HEADER. Use cfitsio commands directly.
  {
   
    if(this->itsOrigin == HEADER){
      char *keyword = new char[FLEN_KEYWORD];
      int status = 0;
      strcpy(keyword,"BMAJ");
      float val=this->itsMaj*this->itsPixelScale;
      if (fits_update_key(fptr, TFLOAT, keyword, &val, NULL, &status)){
	DUCHAMPERROR("Writing beam","Error writing beam info:");
	fits_report_error(stderr, status);
      }
      status = 0;
      strcpy(keyword,"BMIN");
      val=this->itsMin*this->itsPixelScale;
      if (fits_update_key(fptr, TFLOAT, keyword, &val, NULL, &status)){
	DUCHAMPERROR("Writing beam","Error writing beam info:");
	fits_report_error(stderr, status);
      }
      status = 0;
      strcpy(keyword,"BPA");
      val=this->itsPA;
      if (fits_update_key(fptr, TFLOAT, keyword, &val, NULL, &status)){
	DUCHAMPERROR("Writing beam","Error writing beam info:");
	fits_report_error(stderr, status);
      }
    }
  }



}