de.sekmi.histream.etl.config.WideTable Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of histream-import Show documentation

There is a newer version: 0.7

package de.sekmi.histream.etl.config;

import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlElementWrapper;

import de.sekmi.histream.Observation;
import de.sekmi.histream.ObservationFactory;
import de.sekmi.histream.etl.ColumnMap;
import de.sekmi.histream.etl.ConceptTable;
import de.sekmi.histream.etl.ParseException;
import de.sekmi.histream.etl.WideRow;

public class WideTable extends Table implements ConceptTable{

	@XmlElement
	DataTableIdat idat;
	
	@XmlElementWrapper(name="mdat")
	@XmlElement(name="concept")
	Concept[] concepts;

	@Override
	public ColumnMap getColumnMap(String[] headers) throws ParseException {
		ColumnMap map = new ColumnMap(headers);

		if( idat.patientId == null ){
			throw new ParseException("datasource/wide-table/idat/patient-id column not specified");
		}
		if( idat.visitId == null ){
			throw new ParseException("datasource/wide-table/idat/visit-id column not specified");
		}

		map.registerColumn(idat.patientId);
		map.registerColumn(idat.visitId);
		
		for( Concept c : concepts ){
			mapRegisterConcept(map, c);
		}

		// TODO allow option/parameter to force column checking for data tables
		//validateAllHeaders(headers, map, idat.ignore);

		return map;
	}

	@Override
	public WideRow fillRecord(ColumnMap map, Object[] row, ObservationFactory factory) throws ParseException {
		String patid = idat.patientId.valueOf(map, row);
		String visit = idat.visitId.valueOf(map, row);
		WideRow rec = new WideRow(patid,visit);
		for( Concept c : concepts ){
			Observation o = c.createObservation(patid, visit, factory, map, row);
			if( o != null ){
				rec.addFact(o);				
			}
		}
		return rec;
	}

	@Override
	public Concept[] getConcepts() {
		return concepts;
	}
}