/*
 * This file is a part of EpnTAPClient.
 * This program aims to provide EPN-TAP support for software clients, like CASSIS spectrum analyzer.
 * See draft specifications: https://voparis-confluence.obspm.fr/pages/viewpage.action?pageId=559861
 * Copyright (C) 2016 Institut de Recherche en Astrophysique et Planétologie.
 *
 * 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 3 of the License, or (at your option) any later
 * version. This program 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 this program. If not, see
 * <http://www.gnu.org/licenses/>.
 */

package eu.omp.irap.vespa.votable.votabledata;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Logger;

import eu.omp.irap.vespa.epntapclient.votable.model.Data;
import eu.omp.irap.vespa.epntapclient.votable.model.Field;
import eu.omp.irap.vespa.epntapclient.votable.model.Table;
import eu.omp.irap.vespa.votable.controller.CantGetVOTableException;
import eu.omp.irap.vespa.votable.utils.StringJoiner;

/**
 * @author N. Jourdane
 */
public class VOTableDataParser {

	/** The logger for the class */
	private static final Logger logger = Logger.getLogger(VOTableDataParser.class.getName());

	private VOTableData data;

	private Data dataNode;

	private List<Field> fieldsNodes;


	/**
	 * In VOTables, data can be stored in 4 different serializations: BINARY, BINARY2, TABLEDATA and
	 * FITS. This class aims to provide a unique way to get the data table, regardless of its
	 * serialization type. DO NOT PARSE DATA: data parsing requires parseData() call.
	 *
	 * @param table The table on which we want get the data.
	 * @throws IOException If the VOTable data can not be parsed.
	 */
	public VOTableDataParser(Table table) {

		dataNode = table.getDATA();
		fieldsNodes = new ArrayList<>();

		for (Object obj : table.getFIELDOrPARAMOrGROUP()) {
			if (obj.getClass() == Field.class) {
				fieldsNodes.add((Field) obj);
			}
		}
		String[] columnsNames = getColumnsName(fieldsNodes);
		data = new VOTableData(columnsNames);
		logger.info("Columns names:  " + StringJoiner.join(columnsNames));
	}

	private static String[] getColumnsName(List<Field> fieldsNodes) {
		String[] columnsName = new String[fieldsNodes.size()];
		for (int i = 0; i < fieldsNodes.size(); i++) {
			columnsName[i] = fieldsNodes.get(i).getName();
		}
		return columnsName;
	}

	public VOTableData parseData() throws CantGetVOTableException {
		DataParser parser;
		if (dataNode.getBINARY() != null) {
			parser = new BinaryStreamParser(dataNode.getBINARY().getSTREAM(), fieldsNodes);
		} else if (dataNode.getBINARY2() != null) {
			parser = new Binary2StreamParser(dataNode.getBINARY2().getSTREAM(), fieldsNodes);
		} else if (dataNode.getTABLEDATA() != null) {
			parser = new TableDataParser(dataNode.getTABLEDATA(), fieldsNodes);
		} else if (dataNode.getFITS() != null) {
			parser = new FitsParser(dataNode.getFITS(), fieldsNodes);
		} else {
			throw new CantGetVOTableException("Can not locate the data in the VOTable.");
		}
		data.setData(parser.parse());

		return data;
	}

	/**
	 * @return the data stored in the Table.
	 */
	public VOTableData getData() {
		return data;
	}

}