teichmann@5831: package org.dive4elements.river.artifacts.model.fixings;
sascha@3415: 
sascha@3730: import java.util.Collection;
raimund@3610: import java.util.Date;
raimund@3610: import java.util.TreeSet;
raimund@3610: 
teichmann@5831: import org.dive4elements.river.artifacts.model.Parameters;
sascha@3415: 
teichmann@5831: import org.dive4elements.river.utils.KMIndex;
sascha@3415: 
sascha@3415: public class FixAnalysisResult
sascha@3450: extends      FixResult
sascha@3415: {
sascha@3415:     protected KMIndex<AnalysisPeriod []> analysisPeriods;
sascha@3415: 
sascha@3415:     public FixAnalysisResult() {
sascha@3415:     }
sascha@3415: 
sascha@3415:     public FixAnalysisResult(
sascha@3415:         Parameters                 parameters,
sascha@3415:         KMIndex<QWD []>            referenced,
sascha@3729:         KMIndex<QWI []>            outliers,
sascha@3415:         KMIndex<AnalysisPeriod []> analysisPeriods
sascha@3415:     ) {
sascha@3450:         super(parameters, referenced, outliers);
sascha@3415:         this.analysisPeriods = analysisPeriods;
sascha@3415:     }
sascha@3415: 
sascha@3415:     public int getUsedSectorsInAnalysisPeriods() {
sascha@3415:         int result = 0;
sascha@3415:         for (KMIndex.Entry<AnalysisPeriod []> entry: analysisPeriods) {
sascha@3415:             for (AnalysisPeriod period: entry.getValue()) {
sascha@3415:                 for (int i = 0; i < 4; ++i) {
sascha@3415:                     result |= period.getQSectorAverage(i) != null
sascha@3415:                         ? (1 << i)
sascha@3415:                         : 0;
sascha@3415:                 }
sascha@3415:                 // XXX: Stop early on result == ~(~0 << 4)) ?
sascha@3415:             }
sascha@3415:         }
sascha@3415:         return result;
sascha@3415:     }
sascha@3415: 
sascha@3730:     public Collection<Date> getReferenceEventsDates() {
raimund@3610:         TreeSet<Date> dates = new TreeSet<Date>();
raimund@3610:         for (KMIndex.Entry<QWD []> entry: referenced) {
sascha@3730:             QWD [] values = entry.getValue();
sascha@3730:             for (int i = 0; i < values.length; i++) {
sascha@3730:                 dates.add(values[i].date);
raimund@3610:             }
raimund@3610:         }
raimund@3610:         return dates;
raimund@3610:     }
raimund@3610: 
sascha@3730:     public Collection<Date> getAnalysisEventsDates(int analysisPeriod) {
raimund@3610:         TreeSet<Date> dates = new TreeSet<Date>();
raimund@3610:         for (KMIndex.Entry<AnalysisPeriod []> entry: analysisPeriods) {
sascha@3730:             QWD [] qwds = entry.getValue()[analysisPeriod].getQWDs();
sascha@3730:             if (qwds == null) {
sascha@3730:                 continue;
sascha@3730:             }
sascha@3730:             for (int i = 0; i < qwds.length; i++) {
sascha@3730:                 dates.add(qwds[i].date);
raimund@3610:             }
raimund@3610:         }
raimund@3610:         return dates;
raimund@3610:     }
raimund@3610: 
sascha@3415:     public KMIndex<AnalysisPeriod []> getAnalysisPeriods() {
sascha@3415:         return analysisPeriods;
sascha@3415:     }
sascha@3415: 
sascha@3415:     public void setAnalysisPeriods(KMIndex<AnalysisPeriod []> analysisPeriods) {
sascha@3415:         this.analysisPeriods = analysisPeriods;
sascha@3415:     }
sascha@3415: }
sascha@3415: // vim:set ts=4 sw=4 si et sta sts=4 fenc=utf8 :