// -----------------------------------------------------------------------
// existingDetections.cc: Functions to read in previously-made
//                         detections from an existing log file
// -----------------------------------------------------------------------
// 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 <fstream>
#include <sstream>
#include <string>

#include <duchamp/Cubes/cubes.hh>
#include <duchamp/param.hh>
#include <duchamp/PixelMap/Scan.hh>
#include <duchamp/Detection/detection.hh>

using namespace PixelInfo;

namespace duchamp 
{

  int Cube::getExistingDetections()
  {

    std::ifstream logfile(this->par.getLogFile().c_str());
  
    if(!logfile.is_open()){
      std::stringstream errmsg;
      errmsg << "Unable to open log file " << this->par.getLogFile();
      duchampError("getExistingDetections",errmsg.str());
      return FAILURE;
    }

    std::cout << "Reading from logfile : " << this->par.getLogFile() << "\n";

    std::string temp,filename;
    std::stringstream ss;

    // first check filename, just to be sure
    while(getline(logfile,temp), temp.substr(0,11)!="Image to be"){}
    ss.str(temp);
    ss >> temp >> temp >> temp >> temp>> temp >> filename;
    if(filename != this->par.getFullImageFile()){
      std::stringstream errmsg;
      errmsg << "The image file given the log file (" << filename
	     <<") is different to that in the parameter file (" << this->par.getFullImageFile()<<").";
      duchampError("getExistingDetections", errmsg.str());
      return FAILURE;
    }

    // read down until first Detection # line
    while(getline(logfile,temp), temp.substr(0,9)!="Threshold"){}
    float threshold,middle,spread;
    bool robust;
    logfile >> threshold >> middle >> spread >> robust;
    std::cout << "Detection threshold used was " << threshold << "\n";
    this->Stats.setRobust(robust);
    this->Stats.setThreshold(threshold);
    this->Stats.setMiddle(middle);
    this->Stats.setSpread(spread);
    getline(logfile,temp);
    getline(logfile,temp);
    int x1,x2, ypix, zpix;
    while(!logfile.eof()){
      Detection obj;
      while(getline(logfile,temp), temp.substr(0,3)!="---"){
	for(uint i=0;i<temp.size();i++)
	  if(temp[i]=='-' || temp[i]==',') temp[i] = ' ';
	std::stringstream ss;
	ss.str(temp);
	ss >> x1 >> x2 >> ypix >> zpix;
	Scan scn(ypix,x1,x2-x1+1);
	obj.pixels().addScan(scn,zpix);
      }
      obj.setOffsets(this->par);
      obj.calcParams();
      if(obj.getSize()>0) this->addObject(obj);
      getline(logfile,temp); // reads next line -- should be Detection #...
      if(temp.substr(0,11)!="Detection #"){
	// if it is, then read two lines to finish off the file. This should trigger the eof flag above.
	getline(logfile,temp);
	getline(logfile,temp);
      }
    }

    std::cout<<"Final object count = "<<this->objectList->size()<<std::endl; 
    
    return SUCCESS;

  }


}