VectorValuesFrequencyAnalyzer.java

package org.regenstrief.linkage.analysis;

import java.util.Iterator;

import org.regenstrief.linkage.MatchResult;
import org.regenstrief.linkage.MatchVector;
import org.regenstrief.linkage.Record;
import org.regenstrief.linkage.analysis.MatchVectorRecordFrequencies.Side;
import org.regenstrief.linkage.util.LoggingObject;
import org.regenstrief.linkage.util.MatchingConfig;
import org.regenstrief.linkage.util.MatchingConfigRow;
import org.regenstrief.linkage.util.ScorePair;

public class VectorValuesFrequencyAnalyzer extends RecordPairAnalyzer
		implements LoggingObject {

	private ScorePair sp;
	private MatchVectorRecordFrequencies mvrf;
	
	public VectorValuesFrequencyAnalyzer(MatchingConfig mc){
		super(mc);
		sp = new ScorePair(mc);
		mvrf = new MatchVectorRecordFrequencies();
	}
	
	public void finishAnalysis() {
		// nothing to do after counting values

	}

	@Override
	public void analyzeRecordPair(Record[] pair) {
		MatchResult mr = sp.scorePair(pair[0], pair[1]);
		MatchVector mv = mr.getMatchVector();
		Iterator<String> it = mv.getDemographics().iterator();
		while(it.hasNext()){
			String demographic = it.next();
			String val1 = pair[0].getDemographic(demographic);
			RecordFrequencies rf = mvrf.getFrequencies(Side.LEFT, mv);
			if(rf == null){
				rf = new RecordFrequencies();
				mvrf.setFrequencies(Side.LEFT, mv, rf);
				// add dependent demographics
				MatchingConfigRow sex_mcr = mc.getMatchingConfigRowByName("sex");
				MatchingConfigRow fn_mcr = mc.getMatchingConfigRowByName("fn");
				if(sex_mcr != null && fn_mcr != null && sex_mcr.isIncluded() && fn_mcr.isIncluded()){
					rf.setDependency("sex", "fn");
				}
			}
			
			// increment count for record 1
			ValueFrequencyTable vft = null;
			if(rf.isDependent(demographic)){
				// get the context the demographic depends on, increment frequency
				DependentValueFrequencyTable dvft = rf.getDependentValueFrequencyTable(demographic);
				if(dvft == null){
					dvft = new DependentValueFrequencyTable();
					rf.setDependentDemographicFrequencies(demographic, dvft);
				}
				String context = pair[0].getDemographic(rf.getContext(demographic));
				if(context != null){
					vft = dvft.getValueFrequencyTable(context);
					if(vft == null){
						vft = new ValueFrequencyTable();
						dvft.setValueFrequencyTable(context, vft);
					}
				}
				
			}
			// is an independent demographic or there is no value for the dependent demographic context
			if(vft == null){
				vft = rf.getDemographicFrequencies(demographic);
			}
			if(vft == null){
				vft = new ValueFrequencyTable();
				rf.setDemographicFrequencies(demographic, vft);
			}
			long current = vft.getFrequency(val1);
			vft.setFrequency(val1, current+1);
			
			// increment count for record 2
			String val2 = pair[1].getDemographic(demographic);
			rf = mvrf.getFrequencies(Side.RIGHT, mv);
			if(rf == null){
				rf = new RecordFrequencies();
				mvrf.setFrequencies(Side.RIGHT, mv, rf);
				// add dependent demographics
				MatchingConfigRow sex_mcr = mc.getMatchingConfigRowByName("sex");
				MatchingConfigRow fn_mcr = mc.getMatchingConfigRowByName("fn");
				if(sex_mcr != null && fn_mcr != null && sex_mcr.isIncluded() && fn_mcr.isIncluded()){
					rf.setDependency("sex", "fn");
				}
			}
			vft = null;
			if(rf.isDependent(demographic)){
				// get the context the demographic depends on, increment frequency
				DependentValueFrequencyTable dvft = rf.getDependentValueFrequencyTable(demographic);
				if(dvft == null){
					dvft = new DependentValueFrequencyTable();
					rf.setDependentDemographicFrequencies(demographic, dvft);
				}
				String context = pair[1].getDemographic(rf.getContext(demographic));
				if(context != null){
					vft = dvft.getValueFrequencyTable(context);
					if(vft == null){
						vft = new ValueFrequencyTable();
						dvft.setValueFrequencyTable(context, vft);
					}
				}
				
			}
			// is an independent demographic or there is no value for the dependent demographic context
			if(vft == null){
				vft = rf.getDemographicFrequencies(demographic);
			}
			if(vft == null){
				vft = new ValueFrequencyTable();
				rf.setDemographicFrequencies(demographic, vft);
			}
			current = vft.getFrequency(val1);
			vft.setFrequency(val2, current+1);
			
		}

	}
	
	public MatchVectorRecordFrequencies getVectorFrequencies(){
		return mvrf;
	}

}