johncurcio / ETL4LODPlus

public class TurtleGeneratorStepMeta extends BaseStepMeta implements StepMetaInterface {

	// Fields for serialization

	public enum Field {

		DATA_ROOT_NODE, VERSION,

		// Table Dimensions

		MAP_TABLE, MAP_TABLE1, MAP_TABLE2, MAP_TABLE3,

		// TABLE 1

		MAP_TABLE_DIMENSIONS_FIELD_NAME, MAP_TABLE_LABELS_FIELD_NAME, MAP_TABLE_URI_FIELD_NAME, MAP_TABLE_URI_TYPE_FIELD_NAME,

		// TABLE 2

		MAP_TABLE_MEASURE_FIELD_NAME, MAP_TABLE_MEASURE_LABEL_FIELD_NAME, MAP_TABLE_MEASURE_URI_FIELD_NAME, MAP_TABLE_MEASURE_URI_TYPE_FIELD_NAME,

		// TABLE 3

		PREFIXES,

		// TABLE 4

		OUTPUT_UNITY_FIELD_NAME,

		// TABLE 5

		MAP_TABLE_HIERARCHY_FIELD_NAME, MAP_TABLE_HIERARCHY_DE_FIELD_NAME, MAP_TABLE_HIERARCHY_LABEL_FIELD_NAME, MAP_TABLE_HIERARCHY_PARA_FIELD_NAME,

	}

	// Campos do step

	// Table 1

	private DataTable<String> mapTable;

	// Table 2

	private DataTable<String> mapTable1;

	// Table 3

	List<List<String>> prefixes;

	// Table 4

	private String unity;

	// Table 5

	private DataTable<String> mapTable2;

	public TurtleGeneratorStepMeta() {

		setDefault();

	}

	// TODO Validar todos os campos para dar feedback ao usuario!

	@Override

	public void check(List<CheckResultInterface> remarks, TransMeta transMeta, StepMeta stepMeta, RowMetaInterface prev,

			String[] input, String[] output, RowMetaInterface info) {

		CheckResultInterface ok = new CheckResult(CheckResult.TYPE_RESULT_OK, "", stepMeta);

		remarks.add(ok);

	}

	@Override

	public StepInterface getStep(StepMeta stepMeta, StepDataInterface stepDataInterface, int copyNr,

			TransMeta transMeta, Trans trans) {

		return new TurtleGeneratorStep(stepMeta, stepDataInterface, copyNr, transMeta, trans);

	}

	@Override

	public StepDataInterface getStepData() {

		return new TurtleGeneratorStepData();

	}

	@Override

	public String getDialogClassName() {

		return TurtleGeneratorStepDialog.class.getName();

	}

	// Carregar campos a partir do XML de um .ktr

	@Override

	public void loadXML(Node stepDomNode, List<DatabaseMeta> databases, Map<String, Counter> sequenceCounters)

			throws KettleXMLException {

		try {

			XStream xs = new XStream(new DomDriver());

			xs.alias("DataTable", DataTable.class);

			xs.registerConverter(new DataTableConverter());

			mapTable = (DataTable<String>) xs.fromXML(XMLHandler.getTagValue(stepDomNode, Field.MAP_TABLE.name()));

			mapTable1 = (DataTable<String>) xs.fromXML(XMLHandler.getTagValue(stepDomNode, Field.MAP_TABLE1.name()));

			mapTable2 = (DataTable<String>) xs.fromXML(XMLHandler.getTagValue(stepDomNode, Field.MAP_TABLE2.name()));

			prefixes = (List<List<String>>) xs.fromXML(XMLHandler.getTagValue(stepDomNode, Field.PREFIXES.name()));

			unity = (String) xs.fromXML(XMLHandler.getTagValue(stepDomNode, Field.OUTPUT_UNITY_FIELD_NAME.name()));

		} catch (Throwable e) {

			e.printStackTrace();

		}

	}

	// Gerar XML para salvar um .ktr

	@Override

	public String getXML() throws KettleException {

		XStream xs = new XStream();

		xs.alias("DataTable", DataTable.class);

		xs.registerConverter(new DataTableConverter());

		StringBuilder xml = new StringBuilder();

		xml.append(XMLHandler.addTagValue(Field.MAP_TABLE.name(), xs.toXML(mapTable)));

		xml.append(XMLHandler.addTagValue(Field.MAP_TABLE1.name(), xs.toXML(mapTable1)));

		xml.append(XMLHandler.addTagValue(Field.MAP_TABLE2.name(), xs.toXML(mapTable2)));

		xml.append(XMLHandler.addTagValue(Field.PREFIXES.name(), xs.toXML(prefixes)));

		xml.append(XMLHandler.addTagValue(Field.OUTPUT_UNITY_FIELD_NAME.name(), xs.toXML(unity)));

		return xml.toString();

	}

	// Carregar campos a partir do repositorio

	@Override

	public void readRep(Repository repository, ObjectId stepIdInRepository, List<DatabaseMeta> databases,

			Map<String, Counter> sequenceCounters) throws KettleException {

		try {

			int version = (int) repository.getStepAttributeInteger(stepIdInRepository, Field.VERSION.name());

			switch (version) {

			case 1:

				int nrLines = (int) repository.getStepAttributeInteger(stepIdInRepository, "nr_lines");

				// TABLE 1

				mapTable = new DataTable<String>(Field.MAP_TABLE.name(), Field.MAP_TABLE_DIMENSIONS_FIELD_NAME.name(),

						Field.MAP_TABLE_LABELS_FIELD_NAME.name(), Field.MAP_TABLE_URI_FIELD_NAME.name(),

						Field.MAP_TABLE_URI_TYPE_FIELD_NAME.name());

				String[] fields = mapTable.getHeader().toArray(new String[0]);

					int nrfields = fields.length;

					String[] line = new String[nrfields];

					for (int f = 0; f < nrfields; f++) {

						line[f] = repository.getStepAttributeString(stepIdInRepository, i, fields[f]);

					}

					mapTable.add(line);

				}

				// TABLE 2

				mapTable = new DataTable<String>(Field.MAP_TABLE1.name(), Field.MAP_TABLE_MEASURE_FIELD_NAME.name(),

						Field.MAP_TABLE_MEASURE_LABEL_FIELD_NAME.name(), Field.MAP_TABLE_MEASURE_URI_FIELD_NAME.name(),

						Field.MAP_TABLE_MEASURE_URI_TYPE_FIELD_NAME.name());

				fields = mapTable1.getHeader().toArray(new String[0]);

					int nrfields = fields.length;

					String[] line = new String[nrfields];

					for (int f = 0; f < nrfields; f++) {

						line[f] = repository.getStepAttributeString(stepIdInRepository, i, fields[f]);

					}

					mapTable1.add(line);

				}

				// Table 4

				unity = repository.getStepAttributeString(stepIdInRepository, Field.OUTPUT_UNITY_FIELD_NAME.name());

				// Table 5

				mapTable = new DataTable<String>(Field.MAP_TABLE2.name(), Field.MAP_TABLE_HIERARCHY_FIELD_NAME.name(),

						Field.MAP_TABLE_HIERARCHY_DE_FIELD_NAME.name(),

						Field.MAP_TABLE_HIERARCHY_LABEL_FIELD_NAME.name(),

						Field.MAP_TABLE_HIERARCHY_PARA_FIELD_NAME.name());

				fields = mapTable2.getHeader().toArray(new String[0]);

					int nrfields = fields.length;

					String[] line = new String[nrfields];

					for (int f = 0; f < nrfields; f++) {

						line[f] = repository.getStepAttributeString(stepIdInRepository, i, fields[f]);

					}

					mapTable2.add(line);

				}

				break;

			default:

				setDefault();

			}

		} catch (Exception e) {

			throw new KettleException(

					"Unable to read step information from the repository for id_step=" + stepIdInRepository, e);

		}

	}

	@Override

	public void saveRep(Repository repository, ObjectId idOfTransformation, ObjectId idOfStep) throws KettleException {

		try {

			repository.saveStepAttribute(idOfTransformation, idOfStep, Field.VERSION.name(), 1);

			// TABLE 1

			int linhas = (int) mapTable.size();

			int colunas = mapTable.getHeader().size();

			repository.saveStepAttribute(idOfTransformation, idOfStep, "nr_lines", linhas);

			for (int i = 0; i < linhas; i++) {

				for (int f = 0; f < colunas; f++) {

					repository.saveStepAttribute(idOfTransformation, idOfStep, i, mapTable.getHeader().get(f),

							mapTable.getValue(i, f));

				}

			}

			// TABLE 2

			linhas = (int) mapTable1.size();

			colunas = mapTable1.getHeader().size();

			repository.saveStepAttribute(idOfTransformation, idOfStep, "nr_lines", linhas);

			for (int i = 0; i < linhas; i++) {

				for (int f = 0; f < colunas; f++) {

					repository.saveStepAttribute(idOfTransformation, idOfStep, i, mapTable1.getHeader().get(f),

							mapTable1.getValue(i, f));

				}

			}

			// Table 3

			repository.saveStepAttribute(idOfTransformation, idOfStep, Field.PREFIXES.name(),

					new XStream().toXML(prefixes));

			// Table 4

			repository.saveStepAttribute(idOfTransformation, idOfStep, Field.OUTPUT_UNITY_FIELD_NAME.name(), unity);

			// TABLE 5

			linhas = (int) mapTable2.size();

			colunas = mapTable2.getHeader().size();

			repository.saveStepAttribute(idOfTransformation, idOfStep, "nr_lines", linhas);

			for (int i = 0; i < linhas; i++) {

				for (int f = 0; f < colunas; f++) {

					repository.saveStepAttribute(idOfTransformation, idOfStep, i, mapTable2.getHeader().get(f),

							mapTable2.getValue(i, f));

				}

			}

		} catch (Exception e) {

			throw new KettleException("Unable to save step information to the repository for id_step= " + idOfStep, e);

		}

	}

	// Inicializacoes default

	@Override

	public void setDefault() {

		// TABLE 1

		mapTable = new DataTable<String>(Field.MAP_TABLE.name(), Field.MAP_TABLE_DIMENSIONS_FIELD_NAME.name(),

				Field.MAP_TABLE_LABELS_FIELD_NAME.name(), Field.MAP_TABLE_URI_FIELD_NAME.name(),

				Field.MAP_TABLE_URI_TYPE_FIELD_NAME.name());

		// TABLE 2

		mapTable1 = new DataTable<String>(Field.MAP_TABLE1.name(), Field.MAP_TABLE_MEASURE_FIELD_NAME.name(),

				Field.MAP_TABLE_MEASURE_LABEL_FIELD_NAME.name(), Field.MAP_TABLE_MEASURE_URI_FIELD_NAME.name(),

				Field.MAP_TABLE_MEASURE_URI_TYPE_FIELD_NAME.name());

		// TABLE 3

		prefixes = new ArrayList<List<String>>();

		// TABLE 4

		unity = "Descrição da unidade tratada no DataSet";

		// TABLE 5

		mapTable2 = new DataTable<String>(Field.MAP_TABLE2.name(), Field.MAP_TABLE_HIERARCHY_FIELD_NAME.name(),

				Field.MAP_TABLE_HIERARCHY_DE_FIELD_NAME.name(), Field.MAP_TABLE_HIERARCHY_LABEL_FIELD_NAME.name(),

				Field.MAP_TABLE_HIERARCHY_PARA_FIELD_NAME.name());

	}

	/**

	 * It describes what each output row is going to look like

	 */

	@Override

	public void getFields(RowMetaInterface inputRowMeta, String name, RowMetaInterface[] info, StepMeta nextStep,

			VariableSpace space) throws KettleStepException {

		inputRowMeta.clear();

		ValueMetaInterface field = new ValueMetaString("Turtle_File");

		field.setOrigin(name);

		inputRowMeta.addValueMeta(field);

	}

	// Getters & Setters

	public List<List<String>> getPrefixes() {

		return prefixes;

	}

	public void setPrefixes(List<List<String>> prefixes) {

		this.prefixes = prefixes;

	}

	public DataTable<String> getMapTable() {

		return mapTable;

	}

	public DataTable<String> getMapTable1() {

		return mapTable1;

	}

	public DataTable<String> getMapTable2() {

		return mapTable2;

	}

	public String getunity() {

		return unity;

	}

	public void setunity(String unity) {

		this.unity = unity;

	}

}