// -----------------------------------------------------------------------
// CasaAnnotationWriter.hh: Class for writing results to CASA annotation files.
// -----------------------------------------------------------------------
// 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
// -----------------------------------------------------------------------
#ifndef DUCHAMP_CASA_ANNOTATION_WRITER_H_
#define DUCHAMP_CASA_ANNOTATION_WRITER_H_
#include <duchamp/Outputs/AnnotationWriter.hh>
#include <duchamp/Outputs/CatalogueWriter.hh>
#include <duchamp/Outputs/FileCatalogueWriter.hh>
#include <duchamp/Detection/detection.hh>
#include <ios>
#include <iostream>
#include <fstream>
#include <string>

namespace duchamp {

  class CasaAnnotationWriter : public AnnotationWriter
  {
  public:
    /// @brief Default constructor
    CasaAnnotationWriter();
    /// @brief Constructor with given filename
    CasaAnnotationWriter(std::string name);
    /// @brief Copy constructor
    CasaAnnotationWriter(const CasaAnnotationWriter& other);
    /// @brief Copy operator
    CasaAnnotationWriter& operator= (const CasaAnnotationWriter& other);
    /// @brief Default destructor
    virtual ~CasaAnnotationWriter(){};

    /// @brief Reimplement writeHeader to ensure the first line is correct.
    void writeHeader();

    /// @brief Write the global properties for the file (colour, WCS info, ...)
    void writeTableHeader();

    /// @brief Reimplement writeEntry to write the object's bounding box as a region prior to the annotation
    void writeEntry(Detection *object);

    std::string spatialUnits(){return itsSpatialUnits;};
    void setSpatialUnits(std::string units){itsSpatialUnits=units;};

    /// @brief Annotate with a text string
    void text(double x, double y, std::string text);
    /// @brief Annotate with a single line
    void line(double x1, double x2, double y1, double y2);
    /// @brief Annotate with a circle
    void circle(double x, double y, double r);
    /// @brief Annotate with a box
    void box(double x1, double x2, double y1, double y2, std::string label="");
   /// @brief Annotate with an ellipse
    void ellipse(double x, double y, double r1, double r2, double angle);
    /// @brief Annotate with a series of lines connecting points
    void joinTheDots(std::vector<double> x, std::vector<double> y);

  protected:
    std::string itsSpatialUnits;

  };

}

#endif