// -----------------------------------------------------------------------
// ASCIICatalogueWriter.cc: Writing output catalogues to basic ASCII-format
//                          files (or the screen).
// -----------------------------------------------------------------------
// 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 <duchamp/duchamp.hh>
#include <duchamp/Outputs/ASCIICatalogueWriter.hh>
#include <duchamp/Outputs/CatalogueWriter.hh>
#include <duchamp/Outputs/FileCatalogueWriter.hh>
#include <duchamp/param.hh>
#include <duchamp/fitsHeader.hh>
#include <duchamp/Detection/detection.hh>
#include <duchamp/Outputs/columns.hh>
#include <duchamp/Cubes/cubes.hh>
#include <duchamp/Utils/Statistics.hh>
#include <ios>
#include <iostream>
#include <fstream>
#include <vector>

namespace duchamp {

  ASCIICatalogueWriter::ASCIICatalogueWriter():
    FileCatalogueWriter()
  {
    this->itsStream=0;
  }

  ASCIICatalogueWriter::ASCIICatalogueWriter(std::string name):
    FileCatalogueWriter(name)
  {
    this->itsStream=0;
  }

  ASCIICatalogueWriter::ASCIICatalogueWriter(Catalogues::DESTINATION dest):
    itsDestination(dest)
  {
    this->itsStream=0;
  }

  ASCIICatalogueWriter::ASCIICatalogueWriter(std::string name, Catalogues::DESTINATION dest):
    FileCatalogueWriter(name), itsDestination(dest)
  {
    this->itsStream=0;
  }

  ASCIICatalogueWriter::ASCIICatalogueWriter(const ASCIICatalogueWriter& other)
  {
    this->operator=(other);
  }

  ASCIICatalogueWriter& ASCIICatalogueWriter::operator= (const ASCIICatalogueWriter& other)
  {
    if(this==&other) return *this;
    ((FileCatalogueWriter &) *this) = other;
    this->itsStream = other.itsStream;
    this->itsDestination = other.itsDestination;
    return *this;
  }
  
  bool ASCIICatalogueWriter::openCatalogue(std::ios_base::openmode mode)
  {
    if(this->itsName == "" && this->itsDestination!=Catalogues::SCREEN){
      DUCHAMPERROR("ASCIICatalogueWriter","No catalogue name provided");
      this->itsOpenFlag = false;
    }
    else {
      if(this->itsDestination==Catalogues::SCREEN){
	this->itsStream = &std::cout;
	this->itsOpenFlag = true;
      }
      else{
	FileCatalogueWriter::openCatalogue(mode);
	this->itsStream = &this->itsFileStream;
      }

      if(!this->itsOpenFlag) 
	DUCHAMPERROR("ASCIICatalogueWriter","Could not open file \""<<this->itsName<<"\"");
    }
    return this->itsOpenFlag;
  }

  void ASCIICatalogueWriter::writeHeader()
  {
    if(this->itsOpenFlag){
      *this->itsStream <<"Results of the Duchamp source finder v."<<VERSION<<": ";
      time_t now = time(NULL);
      *this->itsStream << asctime( localtime(&now) );
    }
  }

  void ASCIICatalogueWriter::writeCommandLineEntry(int argc, char *argv[])
  {
    if(this->itsOpenFlag){
      *this->itsStream << "Executing statement: ";
       for(int i=0;i<argc;i++) *this->itsStream << argv[i] << " ";
       *this->itsStream << std::endl;
    }
  }

  void ASCIICatalogueWriter::writeParameters()
  {
    if(this->itsOpenFlag){
      *this->itsStream << *this->itsParam << "\n";
    }
  }

  void ASCIICatalogueWriter::writeStats()
  {
    if(this->itsOpenFlag){

      *this->itsStream<<"--------------------\n";
      *this->itsStream<<"Summary of statistics:\n";
      *this->itsStream<<"Detection threshold = " << this->itsStats->getThreshold()
		      <<" " << this->itsHead->getFluxUnits();
      if(this->itsParam->getFlagFDR())
	*this->itsStream<<" (or S/N=" << this->itsStats->getThresholdSNR()<<")";
      if(this->itsParam->getFlagSmooth()) *this->itsStream << " in smoothed cube.";
      if(!this->itsParam->getFlagUserThreshold())
	*this->itsStream<<"\nNoise level = " << this->itsStats->getMiddle()
			<<", Noise spread = " << this->itsStats->getSpread();
      if(this->itsParam->getFlagSmooth()) *this->itsStream  <<" in smoothed cube.";
      
	// // calculate the stats for the original array, so that we can
	// // quote S/N values correctly.
	// this->itsParam->setFlagSmooth(false);
	// bool verb=this->itsParam->isVerbose();
	// bool fdrflag=this->itsParam->getFlagFDR();
	// this->itsParam->setVerbosity(false);
	// this->itsParam->setFlagFDR(false);
	// this->setCubeStats();
	// this->itsParam->setVerbosity(verb);
	// this->itsParam->setFlagFDR(fdrflag);
	// this->itsParam->setFlagSmooth(true);
      
	// *this->itsStream << "\nNoise properties for the original cube are:";
      // }
     
      // if(!this->itsParam->getFlagUserThreshold())
      // 	*this->itsStream<<"\nNoise level = " << this->itsStats->getMiddle()
      // 			<<", Noise spread = " << this->itsStats->getSpread();

      if(this->itsParam->getFlagGrowth()){
	Statistics::StatsContainer<float> growthStats = *this->itsStats;
	if(this->itsParam->getFlagUserGrowthThreshold())
	  growthStats.setThreshold(this->itsParam->getGrowthThreshold());
	else
	  growthStats.setThresholdSNR(this->itsParam->getGrowthCut());
	growthStats.setUseFDR(false);
	*this->itsStream<<"\nDetections grown down to threshold of " 
			<< growthStats.getThreshold() << " " 
			<< this->itsHead->getFluxUnits();
      }

      if(!this->itsParam->getFlagUserThreshold())
	*this->itsStream << "\nFull stats:\n" << *this->itsStats;
      else
	*this->itsStream << "\n\nNot calculating full stats since threshold was provided directly.\n";

      *this->itsStream<<"--------------------\n";
 
      *this->itsStream<<"Total number of detections = "<<this->itsObjectList->size()<<"\n";
      *this->itsStream<<"--------------------\n";

    }
  }

  void ASCIICatalogueWriter::writeTableHeader()
  {
    if(this->itsOpenFlag){
      outputTableHeader(*this->itsStream,*this->itsColumnSpecification,this->itsDestination,this->itsHead->isWCS());
    }
  }

  void ASCIICatalogueWriter::writeEntry(Detection *object)
  {
    if(this->itsOpenFlag){
      object->printTableRow(*this->itsStream, *this->itsColumnSpecification, this->itsDestination);
    }
  }

  void ASCIICatalogueWriter::writeCubeSummary()
  {
    if(this->itsOpenFlag){
      
      *this->itsStream << "=-=-=-=-=-=-=-\nCube summary\n=-=-=-=-=-=-=-\n";

      *this->itsStream<<this->itsCubeDim[0];
      for(int i=1;i<3;i++) *this->itsStream<<"x"<<this->itsCubeDim[i];
      *this->itsStream<<std::endl;

      *this->itsStream<<"Threshold\tmiddle\tspread\trobust\n" << this->itsStats->getThreshold() << "\t";
      if(this->itsParam->getFlagUserThreshold())
	*this->itsStream << "0.0000\t" << this->itsStats->getThreshold() << "\t"; 
      else
	*this->itsStream << this->itsStats->getMiddle() << " " << this->itsStats->getSpread() << "\t";
      *this->itsStream << this->itsStats->getRobust()<<"\n";

      *this->itsStream<<this->itsObjectList->size()<<" detections:\n--------------\n";
      std::vector<Detection>::iterator obj;
      for(obj=this->itsObjectList->begin();obj<this->itsObjectList->end();obj++){
	*this->itsStream << "Detection #" << obj->getID()<<std::endl;
	Detection *newobj = new Detection(*obj);
	newobj->addOffsets();
	*this->itsStream<<*newobj;
	delete newobj;
      }
      *this->itsStream<<"--------------\n";
    }

  }


  bool ASCIICatalogueWriter::closeCatalogue()
  {
    bool returnval=true;
    if(this->itsDestination!=Catalogues::SCREEN){
      FileCatalogueWriter::closeCatalogue();
    }
    return returnval;
  }


}