GmlRecordWriter.java

package com.revolsys.record.io.format.gml;

import java.io.Writer;

import javax.xml.namespace.QName;

import com.revolsys.datatype.DataType;
import com.revolsys.geometry.model.BoundingBox;
import com.revolsys.geometry.model.ClockDirection;
import com.revolsys.geometry.model.GeometryFactory;
import com.revolsys.io.AbstractRecordWriter;
import com.revolsys.io.IoConstants;
import com.revolsys.io.PathUtil;
import com.revolsys.record.Record;
import com.revolsys.record.io.format.gml.type.GmlFieldType;
import com.revolsys.record.io.format.gml.type.GmlFieldTypeRegistry;
import com.revolsys.record.io.format.xml.XmlWriter;
import com.revolsys.record.property.RecordProperties;
import com.revolsys.record.schema.FieldDefinition;
import com.revolsys.record.schema.RecordDefinition;

public class GmlRecordWriter extends AbstractRecordWriter {
  public static final void srsName(final XmlWriter out, final GeometryFactory geometryFactory) {
    final int coordinateSystemId = geometryFactory.getCoordinateSystemId();
    if (coordinateSystemId > 0) {
      out.attribute(Gml.SRS_NAME, "EPSG:" + coordinateSystemId);
    }
  }

  private final GmlFieldTypeRegistry fieldTypes = GmlFieldTypeRegistry.INSTANCE;

  private GeometryFactory geometryFactory;

  private final String namespaceUri;

  private boolean opened;

  private final XmlWriter out;

  private QName qualifiedName;

  public GmlRecordWriter(final RecordDefinition recordDefinition, final Writer out) {
    this.out = new XmlWriter(out);
    this.qualifiedName = recordDefinition.getProperty(RecordProperties.QUALIFIED_NAME);
    if (this.qualifiedName == null) {
      this.qualifiedName = new QName(recordDefinition.getName());
    }
    this.namespaceUri = this.qualifiedName.getNamespaceURI();
    this.out.setPrefix(this.qualifiedName);
  }

  private void box(final GeometryFactory geometryFactory, final BoundingBox areaBoundingBox) {
    this.out.startTag(Gml.BOX);
    srsName(this.out, geometryFactory);
    this.out.startTag(Gml.COORDINATES);
    this.out.text(areaBoundingBox.getMinX());
    this.out.text(",");
    this.out.text(areaBoundingBox.getMinY());
    this.out.text(" ");
    this.out.text(areaBoundingBox.getMaxX());
    this.out.text(",");
    this.out.text(areaBoundingBox.getMaxY());
    this.out.endTag(Gml.COORDINATES);
    this.out.endTag(Gml.BOX);
  }

  @Override
  public void close() {
    if (!this.opened) {
      writeHeader();
    }

    writeFooter();
    this.out.close();
  }

  @Override
  public void flush() {
    this.out.flush();
  }

  @Override
  public ClockDirection getPolygonRingDirection() {
    return ClockDirection.COUNTER_CLOCKWISE;
  }

  @Override
  public void setProperty(final String name, final Object value) {
    if (name.equals(IoConstants.GEOMETRY_FACTORY)) {
      this.geometryFactory = (com.revolsys.geometry.model.GeometryFactory)value;
    }
    super.setProperty(name, value);
  }

  @Override
  public void write(final Record record) {
    if (!this.opened) {
      writeHeader();
    }
    this.out.startTag(Gml.FEATURE_MEMBER);
    final RecordDefinition recordDefinition = record.getRecordDefinition();
    QName qualifiedName = recordDefinition.getProperty(RecordProperties.QUALIFIED_NAME);
    if (qualifiedName == null) {
      final String typeName = recordDefinition.getPath();
      final String path = PathUtil.getPath(typeName);
      final String name = PathUtil.getName(typeName);
      qualifiedName = new QName(path, name);
      recordDefinition.setProperty(RecordProperties.QUALIFIED_NAME, qualifiedName);
    }
    this.out.startTag(qualifiedName);

    for (final FieldDefinition fieldDefinition : recordDefinition.getFields()) {
      final String fieldName = fieldDefinition.getName();
      final Object value;
      if (isWriteCodeValues()) {
        value = record.getCodeValue(fieldName);
      } else {
        value = record.getValue(fieldName);
      }
      if (isValueWritable(value)) {
        this.out.startTag(this.namespaceUri, fieldName);
        final DataType dataType = fieldDefinition.getDataType();
        final GmlFieldType fieldType = this.fieldTypes.getFieldType(dataType);
        if (fieldType != null) {
          fieldType.writeValue(this.out, value);
        }
        this.out.endTag();
      }
    }

    this.out.endTag(qualifiedName);
    this.out.endTag(Gml.FEATURE_MEMBER);
  }

  public void writeFooter() {
    this.out.endTag(Gml.FEATURE_COLLECTION);
    this.out.endDocument();
  }

  private void writeHeader() {
    this.out.setIndent(isIndent());
    this.opened = true;
    this.out.startDocument("UTF-8", "1.0");

    this.out.startTag(Gml.FEATURE_COLLECTION);
    if (this.geometryFactory != null) {
      this.out.startTag(Gml.BOUNDED_BY);
      box(this.geometryFactory, this.geometryFactory.getCoordinateSystem().getAreaBoundingBox());
      this.out.endTag(Gml.BOUNDED_BY);
    }
  }

}