EDIT

/**

* Copyright (C) 2009 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.io.dwca.out;

import java.io.PrintWriter;

import java.net.URISyntaxException;

import org.apache.commons.lang.StringUtils;

import org.apache.log4j.Logger;

import eu.etaxonomy.cdm.io.dwca.TermUri;

import eu.etaxonomy.cdm.model.common.CdmBase;

import eu.etaxonomy.cdm.model.common.DefinedTerm;

import eu.etaxonomy.cdm.model.common.TimePeriod;

import eu.etaxonomy.cdm.model.description.PresenceAbsenceTerm;

import eu.etaxonomy.cdm.model.location.Country;

import eu.etaxonomy.cdm.model.location.NamedArea;

/**

 * @author a.mueller

 * @date 20.04.2011

 *

 */

public class DwcaDistributionRecord extends DwcaRecordBase implements IDwcaAreaRecord{

	@SuppressWarnings("unused")

	private static final Logger logger = Logger.getLogger(DwcaDistributionRecord.class);

	private DwcaId locationId;

	private String locationIdString;

	private String locality;

	private String countryCode;

	private DefinedTerm lifeStage;

	private PresenceAbsenceTerm occurrenceStatus;

	private String threadStatus;

	private PresenceAbsenceTerm establishmentMeans;

	private String appendixCITES;

	private TimePeriod eventDate;

	private TimePeriod seasonalDate; //startDayOfYear and endDayOfYear

	private String source;

	private String occurrenceRemarks;

	public DwcaDistributionRecord(DwcaMetaDataRecord metaDataRecord, DwcaTaxExportConfigurator config){

		super(metaDataRecord, config);

		locationId = new DwcaId(config);

	}

	@Override

    protected void registerKnownFields(){

		try {

			addKnownField(TermUri.DWC_LOCATION_ID);

			addKnownField(TermUri.DWC_LOCALITY);

			addKnownField(TermUri.DWC_COUNTRY_CODE);

			addKnownField(TermUri.DWC_LIFESTAGE);

			addKnownField(TermUri.IUCN_THREAD_STATUS);

            addKnownField(TermUri.DWC_OCCURRENCE_STATUS);

            addKnownField(TermUri.DWC_ESTABLISHMENT_MEANS);

            addKnownField(TermUri.GBIF_APPENDIX_CITES);

            addKnownField(TermUri.DWC_EVENT_DATE);

            addKnownField(TermUri.DWC_START_DAY_OF_YEAR);

            addKnownField(TermUri.DC_SOURCE);

            addKnownField(TermUri.DWC_OCCURRENCE_REMARKS);

		} catch (URISyntaxException e) {

			throw new RuntimeException(e);

		}

	}

//	@Override

//	public List<String> getHeaderList() {

//		String[] result = new String[]{"coreid",

//				"locationId",

//				"locality",

//				"countryCode",

//				"lifeStage",

//				"occurrenceStatus",

//				"threadStatus",

//				"establishmentMeans",

//				"appendixCITES",

//				"eventDate",

//				"startDayOfYear",

//				"endDayOfYear",

//				"source",

//				"occurrenceRemarks"};

//		return Arrays.asList(result);

//	}

    @Override

    public void writeCsv(DwcaTaxExportState state) {

        try {

            DwcaTaxExportFile table = DwcaTaxExportFile.DISTRIBUTION;

            String[] csvLine = new String[table.getSize()];

            //

            line(state, csvLine, table, TermUri.DC_IDENTIFIER, getUuid());

            if (StringUtils.isNotBlank(locationIdString)){

                line(state, csvLine, table, TermUri.DWC_LOCATION_ID, locationIdString);

            }else{

                line(state, csvLine, table, TermUri.DWC_LOCATION_ID, locationId);

            }

            line(state, csvLine, table, TermUri.DWC_LOCALITY, locality);

            line(state, csvLine, table, TermUri.DWC_COUNTRY_CODE, countryCode);

            line(state, csvLine, table, TermUri.DWC_LIFESTAGE, getLifeStage(lifeStage));

            line(state, csvLine, table, TermUri.IUCN_THREAD_STATUS, getOccurrenceStatus(occurrenceStatus));

            line(state, csvLine, table, TermUri.DWC_OCCURRENCE_STATUS, threadStatus);

            line(state, csvLine, table, TermUri.DWC_ESTABLISHMENT_MEANS, getEstablishmentMeans(establishmentMeans));

            line(state, csvLine, table, TermUri.GBIF_APPENDIX_CITES, appendixCITES);

            line(state, csvLine, table, TermUri.DWC_EVENT_DATE, getTimePeriod(eventDate));

            line(state, csvLine, table, TermUri.DWC_START_DAY_OF_YEAR, getTimePeriodPart(seasonalDate, false));

            line(state, csvLine, table, TermUri.DWC_END_DAY_OF_YEAR, getTimePeriodPart(seasonalDate, true));

            line(state, csvLine, table, TermUri.DC_SOURCE, source);

            line(state, csvLine, table, TermUri.DWC_OCCURRENCE_REMARKS, occurrenceRemarks);

            state.getProcessor().put(table, getId().toString(), csvLine);

        } catch (Exception e) {

            String message = "Unhandled exception when writing distribution record: " + e.getMessage();

            state.getResult().addException(e, message);

        }

    }

    @Override

    public void write(DwcaTaxExportState state, PrintWriter writer) {

        if(writer == null){

            writeCsv(state);

            return;

        }

		printId(getUuid(), writer, IS_FIRST, "coreid");

		if (StringUtils.isNotBlank(locationIdString)){

			print(locationIdString, writer, IS_NOT_FIRST, TermUri.DWC_LOCATION_ID);

		}else{

			print(locationId, writer, IS_NOT_FIRST, TermUri.DWC_LOCATION_ID);

		}

		print(locality, writer, IS_NOT_FIRST, TermUri.DWC_LOCALITY);

		print(countryCode, writer, IS_NOT_FIRST, TermUri.DWC_COUNTRY_CODE);

		print(getLifeStage(lifeStage), writer, IS_NOT_FIRST, TermUri.DWC_LIFESTAGE);

		print(getOccurrenceStatus(occurrenceStatus), writer, IS_NOT_FIRST, TermUri.DWC_OCCURRENCE_STATUS);

		print(threadStatus, writer, IS_NOT_FIRST, TermUri.IUCN_THREAD_STATUS);

		print(getEstablishmentMeans(establishmentMeans), writer, IS_NOT_FIRST, TermUri.DWC_ESTABLISHMENT_MEANS);

		print(appendixCITES, writer, IS_NOT_FIRST, TermUri.GBIF_APPENDIX_CITES);

		print(getTimePeriod(eventDate),writer, IS_NOT_FIRST, TermUri.DWC_EVENT_DATE);

		print(getTimePeriodPart(seasonalDate, false),writer, IS_NOT_FIRST, TermUri.DWC_START_DAY_OF_YEAR);

		//TODO

		print(getTimePeriodPart(seasonalDate, true),writer, IS_NOT_FIRST, TermUri.DWC_END_DAY_OF_YEAR);

		print(source, writer, IS_NOT_FIRST, TermUri.DC_SOURCE);

		print(occurrenceRemarks, writer, IS_NOT_FIRST, TermUri.DWC_OCCURRENCE_REMARKS);

		writer.println();

	}

	public String getSource() {

		return source;

	}

	public void setSource(String source) {

		this.source = source;

	}

	public String getLocality() {

		return locality;

	}

	@Override

    public void setLocality(String locality) {

		this.locality = locality;

	}

	public String getCountryCode() {

		return countryCode;

	}

	@Override

    public void setCountryCode(String countryCode) {

		this.countryCode = countryCode;

	}

	public DefinedTerm getLifeStage() {

		return lifeStage;

	}

	public void setLifeStage(DefinedTerm lifeStage) {

		this.lifeStage = lifeStage;

	}

	public String getLocationId() {

		return locationId.getId();

	}

	@Override

    public void setLocationId(NamedArea area) {

		if (area.getVocabulary().getUuid().equals(NamedArea.uuidTdwgAreaVocabulary)){

			String locationId = "TDWG:" + area.getIdInVocabulary();

			this.locationIdString = locationId;

		}else if (area.isInstanceOf(Country.class)){

			Country country = CdmBase.deproxy(area, Country.class);

			String locationId = "ISO3166:" + country.getIso3166_A2();

			this.locationIdString = locationId;

		}else{

			this.locationId.setId(area);

		}

	}

	public PresenceAbsenceTerm getOccurrenceStatus() {

		return occurrenceStatus;

	}

	public void setOccurrenceStatus(PresenceAbsenceTerm occurrenceStatus) {

		this.occurrenceStatus = occurrenceStatus;

	}

	public String getThreadStatus() {

		return threadStatus;

	}

	public void setThreadStatus(String threadStatus) {

		this.threadStatus = threadStatus;

	}

	public PresenceAbsenceTerm getEstablishmentMeans() {

		return establishmentMeans;

	}

	public void setEstablishmentMeans(PresenceAbsenceTerm establishmentMeans) {

		this.establishmentMeans = establishmentMeans;

	}

	public String getAppendixCITES() {

		return appendixCITES;

	}

	public void setAppendixCITES(String appendixCITES) {

		this.appendixCITES = appendixCITES;

	}

	public TimePeriod getEventDate() {

		return eventDate;

	}

	public void setEventDate(TimePeriod eventDate) {

		this.eventDate = eventDate;

	}

	public TimePeriod getSeasonalDate() {

		return seasonalDate;

	}

	public void setSeasonalDate(TimePeriod seasonalDate) {

		this.seasonalDate = seasonalDate;

	}

	public String getOccurrenceRemarks() {

		return occurrenceRemarks;

	}

	public void setOccurrenceRemarks(String occurrenceRemarks) {

		this.occurrenceRemarks = occurrenceRemarks;

	}

}