/*
Copyright 2002 by Ralph Hartley
This software is licenced under the terms of the
Gnu Public Licence
*/
import java.awt.*;
import java.text.*;
import java.awt.geom.*;
import java.util.*;
import java.io.*;
import javax.swing.*;

public class Survey extends Symbol implements Element {

  private static final long serialVersionUID = Version.getSUID();

/* These variables are part of the old implementation they remain for
   backwards compatability */
  Vector stationlist = null;
  Hashtable dict = null;
  Vector shotlist = null;

/* stations replaces the two structures above */
  TreeSet stations = null;
  TreeSet shots = null;

//  File file;

  Survey old = null;
  Survey next = null;

//  public String name = null;

  static public Size shotsize =null;

  private void readObject(java.io.ObjectInputStream stream)
      throws java.io.IOException,java.lang.ClassNotFoundException {
    stream.defaultReadObject();
    if (stations==null)
      stations = new TreeSet(stationlist);
    if (shots==null) 
      shots = new TreeSet(shotlist);

    //This is for backwards compatability - the order has been changed
    //make sure they are  re sorted.
    TreeSet oldstations = stations;
    stations = new TreeSet();
    for (Iterator it=oldstations.iterator();it.hasNext();stations.add(it.next()));
    TreeSet oldshots = shots;
    shots = new TreeSet();
    for (Iterator it=oldshots.iterator();it.hasNext();shots.add(it.next()));
  }

  public static Survey open(Survey older) {
    JFileChooser choose = Persist.current.getFileChooser();
    choose.setDialogTitle("Choose survey file");
    choose.setFileFilter(new FileTypes(FileTypes.SURVEY));
    choose.rescanCurrentDirectory();

    try {
      if (choose.showOpenDialog(null)==choose.APPROVE_OPTION) {
        try {
          if (older == null) older = new Survey();
          older.update(choose.getSelectedFile());
        } catch (IOException e) {
          JOptionPane.showMessageDialog(null,
        				"Could not read survey file: "+choose.getSelectedFile(),
        				"Survey File",
        				JOptionPane.ERROR_MESSAGE);
          return(null);
        }
      }
      else return(null);
    } finally {
      Persist.current.clearChooser();
      Persist.save();
    }

    return(older);
  }

  public Survey() {
    stations = new TreeSet();
    shots = new TreeSet();
    name = "empty survey";
  }

  public Survey(String name,Survey older) {
    stations = new TreeSet();
    shots = new TreeSet();
    this.name = name;
    old = older;
  }

  public Symbol importSym(Carto dest,Carto source) {
    if (dest.survey!=null) return(dest.survey);
    dest.survey = this;
    return(this);
  }

  public void purge(Carto user) {
    if (JOptionPane.showConfirmDialog(null,"Deletes all Stations and shots not in the given file.\n"+
        			      "You should already have done an update.\n"+
        			      "Purging the survey cannot be undone.\n"+
        			      "All previous surveys will become unavailable."
        			      ,"Purge",JOptionPane.OK_CANCEL_OPTION) ==
        JOptionPane.OK_OPTION) {

      Survey res = open(new Survey(name,null));
      if (res==null) {
        JOptionPane.showMessageDialog(null,
        			      "Purge did not take place",
        			      "Purge aborted",
        			      JOptionPane.ERROR_MESSAGE);
        return;
      }

      for (Iterator it = stations.iterator();it.hasNext();) {
        Vertex v = (Vertex)it.next();
        if (res.stations.contains(v)) {
          res.stations.remove(v);
          res.stations.add(v);
        } else {
          String use = user.used(v);
          if (use!=null) {
            if (JOptionPane.showConfirmDialog(null,"Station "+v+" is used\n"+use+
        				      "\nClick OK to purge anyway (risky)."
        				      ,"Station Used",JOptionPane.OK_CANCEL_OPTION) ==
        	JOptionPane.OK_OPTION) {
              user.remove(v);
            } else
              return;
          }
        }
      }
      for (Iterator it = shots.iterator();it.hasNext();) {
        Line line = (Line)it.next();
        if (res.shots.contains(line)) {
          res.shots.remove(line);
          res.shots.add(line);
        }
      }
      stations = res.stations;
      shots = res.shots;
      old = next = null;
    }
  }

  public void update(File file) throws java.io.IOException {
    if (old == null)
      old = new Survey();
    else
      old = new Survey(name,old);

    old.next = this;

    this.name = file.getName()+" "+DateFormat.getDateTimeInstance(DateFormat.SHORT,DateFormat.SHORT).format(new Date());

    if (file.getName().endsWith(".XYZ")) {     // cmap output
      Vector seqlist = new Vector();
      String line;
      StringTokenizer tokens;
      BufferedReader input = new BufferedReader(new InputStreamReader(new FileInputStream(file)));

//      shots = new TreeSet();
      
      do { line = input.readLine();
      } while (line.length()>0 && line.charAt(0)!='-');
      for (tokens = new StringTokenizer(input.readLine());
           tokens.countTokens()==6;
           tokens = new StringTokenizer(input.readLine())) {
        String lab=tokens.nextToken();
        Vertex p = new Vertex(new Vect(Double.valueOf(tokens.nextToken()).doubleValue(),
        			       Double.valueOf(tokens.nextToken()).doubleValue(),
        			       Double.valueOf(tokens.nextToken()).doubleValue()),
        		      lab);
        seqlist.addElement(addStation(p,old));
        addShot(
          new Line((Vertex)seqlist.elementAt(Integer.valueOf(tokens.nextToken()).intValue()-1),
        	   (Vertex)seqlist.elementAt(Integer.valueOf(tokens.nextToken()).intValue()-1)));
      }
      input.close();
    }
    else if (file.getName().endsWith(".plt")||file.getName().endsWith(".PLT")) {     // compass output
      String line;
      StringTokenizer tokens;
      BufferedReader input = new BufferedReader(new InputStreamReader(new FileInputStream(file)));
      Vertex currentPos = null;
      double northing = 0;

//      shots = new TreeSet();
      
      for (line = input.readLine();line!=null;line = input.readLine()) {
        tokens = new StringTokenizer(line);
        String command = tokens.nextToken();
        if (command.charAt(0)=='D' ||command.charAt(0)=='M') {
          northing = Double.valueOf(tokens.nextToken()).doubleValue();
          Vect pos = new Vect(Double.valueOf(tokens.nextToken()).doubleValue(),
        		      northing,
        		      Double.valueOf(tokens.nextToken()).doubleValue());
          String lab=tokens.nextToken().substring(1);
          Vertex newpos = addStation(new Vertex(pos,lab),old);
//          addStation(newpos,old);
          if (command.charAt(0)=='D')
            addShot(new Line(currentPos,newpos));
          currentPos = newpos;
        }
      }
      input.close();
    }

    else if (file.getName().endsWith(".3D")||file.getName().endsWith(".3d")) {  // Survex output

      Input3d input = new Reader3d().Reader3d(new FileInputStream(file));
      
      ErrorLog.log("Code for reading Survex files is not finished");
      
    }
  }

  Vertex addStation(Vertex v,Survey undo) {
    if (stations.contains(v)) {
      Vertex oldsta = (Vertex)stations.tailSet(v).first();
      if (!v.position.equals(oldsta.position)) {
        undo.stations.add(new Vertex(oldsta));
        oldsta.position.copyVal(v.position);
      }
      return(oldsta);
    }
    else {
      stations.add(v);
      return(v);
    }
  }

  public Vertex getStation(Vertex old) {
    if (stations.contains(old))
      return((Vertex)stations.tailSet(old).first());
    else {
      stations.add(old);
      return(old);
    }
  }

  void undo() {
    if (old==null) return;
    Survey mynext = new Survey(name,this);
    mynext.next = next;
    if (next!=null) next.old = mynext;
    next = mynext;
    
    Survey edits = old;
    old = old.old;
    if (old!=null) old.next = this;

    this.name = edits.name;
    for (Iterator it=edits.stations.iterator();it.hasNext();)
      addStation((Vertex)it.next(),next);
  }

  void redo() {
    if (next==null) return;
    Survey myold = new Survey(name,old);
    myold.next = this;
    if (old!=null) old.next = myold;
    old = myold;
    
    Survey edits = next;
    next = next.next;
    if (next!=null) next.old = this;

    this.name = edits.name;
    for (Iterator it=edits.stations.iterator();it.hasNext();)
      addStation((Vertex)it.next(),old);
  }

  void addShot(Line s) {
    if (shots.contains(s)) shots.remove(s);
    shots.add(s);
  }

  public String getName() {
    return(name);
  }

  public void draw(Graphics g, AffineTransform trans) {
    for (Iterator it=stations.iterator();it.hasNext();)
      ((Vertex)it.next()).mapdraw(g,trans);
    for (Iterator it=shots.iterator();it.hasNext();)
      ((Line)it.next()).mapdraw(g,trans);
  }

/*
  public Vertex get(String name) {
    if (name!=null)
      return((Vertex)(dict.get(name)));
    else return(null);
  }
*/
  
  public int shotnum() {
    return shots.size();
  }
/*
  public Enumeration stationNames( ) {
    return dict.keys();
  }
*/
  public int vertexnum() {
    return stations.size();
  }

  public Rectangle2D getBounds(AffineTransform trans) {
    if (stations.size()==0) return(null);

    Point2D.Double pos = new Point2D.Double();
    trans.transform(new Point2D.Double(((Vertex)stations.first()).position.X[0],
        			       ((Vertex)stations.first()).position.X[1]),pos);
    Point2D.Double min = (Point2D.Double)pos.clone();
    Point2D.Double max = (Point2D.Double)pos.clone();

    for (Iterator it = stations.iterator(); it.hasNext();) {
      Vertex vert = (Vertex)it.next();
      trans.transform(new Point2D.Double(vert.position.X[0],vert.position.X[1]),pos);
      if (pos.x<min.x) min.x = pos.x;
      if (pos.y<min.y) min.y = pos.y;
      if (pos.x>max.x) max.x = pos.x;
      if (pos.y>max.y) max.y = pos.y;
    }
    return(new Rectangle2D.Double(min.x,min.y,max.x-min.x,max.y-min.y));
  }

  public void paint(View view) {
    if (!view.visible.isMember(this)) return;

    if (shotsize==null)
      shotsize = (Size)FileDefaultable.lookup(Size.class,Size.SHOTLINE);

    Stroke oldstroke = view.draw.getStroke();

    view.draw.setStroke(new BasicStroke((float)shotsize.getSize(view.mapscale)));

    Vertex.paintAll(stations.iterator(),view);

    for (Iterator e = shots.iterator();e.hasNext();)
      ((Line)e.next()).paint(view);

    view.draw.setStroke(oldstroke);
  }
}