import java.util.ArrayList;
import java.util.List;

import eu.dnetlib.data.utility.featureextraction.FeatureExtractionException;
import eu.dnetlib.data.utility.featureextraction.dataprovider.SourceDataProvider;


public class MySourceDataProvider implements SourceDataProvider {
	
	ArrayList<String> DMFRecords;
	
	public MySourceDataProvider() {
		this.DMFRecords = new ArrayList<String>();
	}
	
	public void addDMFRecord(String record) {
		DMFRecords.add(record);
	}

	@Override
	public List<String> getRecords(int from, int to)
			throws FeatureExtractionException {
		try {
			if(to >= this.DMFRecords.size())
				to = this.DMFRecords.size();
			return DMFRecords.subList(from-1, to);
		}
		catch(Exception e) {
			throw new FeatureExtractionException(e);
		}
	}

	@Override
	public int getSize() throws FeatureExtractionException {
		return this.DMFRecords.size();
	}

	@Override
	public String getStatus() throws FeatureExtractionException {
		// TODO Auto-generated method stub
		return null;
	}

}