/*
Copyright 2000 by Ralph Hartley
This software is licenced under the terms of the
Gnu Public Licence
*/

import java.io.*;
import java.util.*;

import  org.w3c.dom.*;
import  org.apache.xerces.dom.DocumentImpl;
import  org.apache.xerces.dom.DocumentTypeImpl;
import  org.apache.xerces.dom.DOMImplementationImpl;
import  org.w3c.dom.Document;
import  org.apache.xml.serialize.OutputFormat;
import  org.apache.xml.serialize.Serializer;
import  org.apache.xml.serialize.SerializerFactory;
import  org.apache.xml.serialize.XMLSerializer;

public class Cmap2Xml implements Serializable{

  private static final long serialVersionUID = -48513234512356439L;

  TreeMap stations = new TreeMap();
  TreeSet shots = new TreeSet();
  Document doc;
  Element base;

  static String dtdurl = "http://www.psc.cavers/cavexml/cavexml.dtd";

  static Version version = new Version();

  static String unit = "feetdecimal";
  static int style = 1;

  public Cmap2Xml() {
    doc = new DocumentImpl();
// Couldn't figure out how to pu in correct document type. Fix later.
//    DocumentType type = new DocumentTypeImpl(doc,"CaveXML",dtdurl,"");
//    doc.appendChild(type);

    Element root = doc.createElement("CaveSurvey");
    doc.appendChild(root);
    Element prov = doc.createElement("ProvenanceImport");
    prov.setAttribute("format","CMAP");
    prov.setAttribute("converter","Cmap2Xml");
    prov.setAttribute("version",version.getVersion());
    prov.setAttribute("date",getDate());
    root.appendChild(prov);

    Element def1 = doc.createElement("Default");
    def1.setAttribute("element","Distance");
    def1.setAttribute("unit",unit);
    prov.appendChild(def1);

    Element names = doc.createElement("NameSet");
    names.setAttribute("ID","CMAPnames");
    names.setAttribute("program","CMAP");
    def1.appendChild(names);

    Element def2 = doc.createElement("Default");
    def2.setAttribute("element","Station");
    def2.setAttribute("nameset","CMAPnames");
    def1.appendChild(def2);

    base = def2;
  }

  static String getDate() {
    Calendar cal = new GregorianCalendar();
    String year = ""+cal.get(Calendar.YEAR);
    String month = ""+cal.get(Calendar.MONTH);
    String day = ""+cal.get(Calendar.DATE);

    if (month.length()==1) month = "0"+month;
    if (day.length()==1) day = "0"+day;
    return(year+"-"+month+"-"+day);
  }


  public void addFile(File file) throws IOException {
    readCmap(new FileInputStream(file));
  }

  static boolean hasNonWhite(String str) {
    for (int i=0;i<str.length();i++)
      if (!Character.isWhitespace(str.charAt(i))) return(true);
    return(false);
  }

  public void readCmap(InputStream file) throws IOException {
    BufferedReader input = new BufferedReader(new InputStreamReader(file));

//    String filename = file.getName();
    
    for (String line = input.readLine();line!=null;line = input.readLine()) {
      if (hasNonWhite(line) && Character.isWhitespace(line.charAt(0))) {
	String fromname = removeWhite(line.substring(1,7));
	String toname = removeWhite(line.substring(8,14)); 
	String dist = null;
	String az = null;
	String inc = null;
	String shotunit = null;

	try {
	  int len = line.length();
	  String str;
	  if (len<16) dist = null;
	  else {
	    str = line.substring(15,(22<=len?22:len));
	    shotunit = getUnit(str);
	    dist = deThrunize(str);
	  }
	  if (len<24) az = null;
	  else {
	    str = line.substring(23,(30<=len?30:len));
	    az = deThrunize(str);
	  }
	  if (len<37) inc = null;
	  else {
	    str = line.substring(36,(43<=len?43:len));	    
	    inc = deThrunize(str);
	  }
	} catch (Throwable /*NumberFormatException*/ ex) {
	  System.out.println("Trouble reading shot from "+fromname +" to "+toname);
	  System.out.println(line);
	  ex.printStackTrace();
	  continue;
	}

	Station from = addStation(fromname);
	Station to = null;
	if (style == 1) to = addStation(toname);

	Element shotell = doc.createElement("Shot");
	if (style==1) {
	  shotell.setAttribute("from","CMAP"+fromname);
	  shotell.setAttribute("to","CMAP"+toname);
	}
	base.appendChild(shotell);

	if (dist != null) {
	  Element distell = doc.createElement("Distance");
	  if (shotunit!=null) distell.setAttribute("unit",shotunit);
	  distell.setAttribute("value",dist);
	  shotell.appendChild(distell);
	}

	if (az != null) {
	  Element azell = doc.createElement("Azimuth");
	  azell.setAttribute("value",az);
	  shotell.appendChild(azell);
	}

	if (inc != null) {
	  Element incell = doc.createElement("Inclination");
	  incell.setAttribute("value",inc);
	  shotell.appendChild(incell);
	}

	if (style == 2) to = addStation(toname);

	Shot shot = new Shot(from,to,dist,az,inc,shotunit,shotell);
	shots.add(shot);
      }
    }

  }

  Station last = null;

  public Station addStation(String name) {
    Station res = null;

    if (stations.containsKey(name))
      res = (Station)stations.get(name);

    Element sta = doc.createElement("Station");
    if (res==null) {
      sta.setAttribute("name",name);
      sta.setAttribute("ID","CMAP"+name);
    }
    else sta.setAttribute("reference","CMAP"+name);

    if (res==null || (style==2 && res!=last)) base.appendChild(sta);

    if (res==null) {
      res = new Station(name,sta);
      stations.put(name,res);
    }

    last = res;

    return(res);
  }


  public void printStats(PrintStream out) {
    double length = 0;
    for (Iterator it=shots.iterator();it.hasNext();)
      length += ((Shot)it.next()).dist;

    out.println("Stations = "+stations.size()+" Shots = "+shots.size());
    out.println(" length = "+length+" average shot = "+length/shots.size());
  }


  static String removeWhite(String str) {
    StringBuffer buf = new StringBuffer(str);
    for (int i = 0;i<buf.length();i++)
      if (Character.isWhitespace(buf.charAt(i))) {
	buf.deleteCharAt(i);
	i--;
      }
    return(buf.toString());
  }

  static String getUnit(String str) {
    for (int i = 0;i<str.length();i++) {
      char cur = str.charAt(i);
      if ('f'==cur || 'F'==cur) return("feetdecimal");
      else if ('m'==cur || 'M'==cur) return("meters");
      else if ('y'==cur || 'Y'==cur) return("yards");
    }
    return(null);
  }

  static String deThrunize(String str) {
    str = removeWhite(str);
    
    if (str.length()==0) return(null);
    boolean dotseen = false;

    for (int i = 0;i<str.length();i++) {
      char cur = str.charAt(i);
      if ('f'==cur || 'F'==cur || 'm'==cur || 'M'==cur || 'y'==cur || 'Y'==cur) {
	str = str.substring(0,i) + (dotseen?"":".") + (i<(str.length()-2)?str.substring(i+1,str.length()):"");
	dotseen=true;
	i--;
      }
      else if (cur =='.') dotseen = true;
    }

    try {
      Double.parseDouble(str);
    } catch (NumberFormatException ex) {
      return(null);
    }
    return(str);
  }

  public void writeXML(OutputStream out) throws java.io.IOException {
    OutputFormat    format  = new OutputFormat( doc, "UTF-8", true);   //Serialize DOM
    XMLSerializer    serial = new XMLSerializer(out, format );
    serial.asDOMSerializer();                            // As a DOM Serializer
    
    serial.serialize( doc);
//.getDocumentElement() );
  }

  public static void showHelp() {
    System.out.println("java Cmap2Xml filename [ meters | feet | style1 | style2 ]*");
  }

  public static void main(String args[]) {

    Cmap2Xml writer = new Cmap2Xml();
    File file = null;

    if (args.length<1) {
      showHelp();
      return;
    }

    for (int i=1;i<args.length;i++) {
      if (args[i].equals("meters")) unit = "meters";
      else if (args[i].equals("feet")) unit = "feetdecimal";
      else if (args[i].equals("style1")) style = 1;
      else if (args[i].equals("style2")) style = 2;
      else showHelp();
    }

    try {
      file = new File(args[0]+".dat");
      if (file.exists())
	writer.addFile(file);
      else {
	file = new File(args[0]+".DAT");
	if (file.exists())
	  writer.addFile(file);
	else ErrorLog.log("couldnt find "+args[0]+".dat or "+args[0]+".DAT");
      }
    } catch (IOException ex) {
      ErrorLog.exception(ex,"Error reading input file "+file);
    }

    OutputStream outstr = null;
    try {
      outstr = new FileOutputStream(new File(args[0].toLowerCase()+".xml"));
    } catch (IOException ex) {
      ErrorLog.exception(ex,"Error creating file "+args[0].toLowerCase()+".xml");
    }

    try {
      writer.writeXML(outstr);
    } catch (IOException ex) {
      ErrorLog.exception(ex,"Error writing output file "+args[0].toLowerCase()+".xml");
      ex.printStackTrace();
    }
  }
}