/* | |
* Licensed to the Apache Software Foundation (ASF) under one | |
* or more contributor license agreements. See the NOTICE file | |
* distributed with this work for additional information | |
* regarding copyright ownership. The ASF licenses this file | |
* to you under the Apache License, Version 2.0 (the | |
* "License"); you may not use this file except in compliance | |
* with the License. You may obtain a copy of the License at | |
* | |
* http://www.apache.org/licenses/LICENSE-2.0 | |
* | |
* Unless required by applicable law or agreed to in writing, | |
* software distributed under the License is distributed on an | |
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY | |
* KIND, either express or implied. See the License for the | |
* specific language governing permissions and limitations | |
* under the License. | |
*/ | |
package org.apache.uima.ruta.statistics; | |
import java.util.ArrayList; | |
import java.util.List; | |
import org.apache.uima.cas.Feature; | |
import org.apache.uima.cas.FeatureStructure; | |
import org.apache.uima.cas.Type; | |
import org.apache.uima.cas.impl.DoubleArrayFSImpl; | |
import org.apache.uima.cas.impl.IntArrayFSImpl; | |
import org.apache.uima.cas.impl.StringArrayFSImpl; | |
public class StatisticsEntry { | |
private static final String PARTS = "part"; | |
private static final String AMOUNTS = "amount"; | |
private static final String TOTALS = "total"; | |
private static final String NAMES = "name"; | |
private static final String TYPE = "org.apache.uima.ruta.type.Statistics"; | |
private final String name; | |
private final double total; | |
private final int amount; | |
private final double part; | |
public StatisticsEntry(String name, double total, int amount, double part) { | |
super(); | |
this.name = name; | |
this.total = total; | |
this.amount = amount; | |
this.part = part; | |
} | |
public static List<StatisticsEntry> createEntries(FeatureStructure fs) { | |
List<StatisticsEntry> result = new ArrayList<StatisticsEntry>(); | |
Type type = fs.getCAS().getTypeSystem().getType(TYPE); | |
Feature fname = type.getFeatureByBaseName(NAMES); | |
Feature ftotal = type.getFeatureByBaseName(TOTALS); | |
Feature famount = type.getFeatureByBaseName(AMOUNTS); | |
Feature fparts = type.getFeatureByBaseName(PARTS); | |
StringArrayFSImpl vname = (StringArrayFSImpl) fs.getFeatureValue(fname); | |
DoubleArrayFSImpl vtotal = (DoubleArrayFSImpl) fs.getFeatureValue(ftotal); | |
IntArrayFSImpl vamount = (IntArrayFSImpl) fs.getFeatureValue(famount); | |
DoubleArrayFSImpl vpart = (DoubleArrayFSImpl) fs.getFeatureValue(fparts); | |
String[] narray = new String[0]; | |
if(vname != null) { | |
narray = vname.toArray(); | |
} | |
double[] tarray = vtotal.toArray(); | |
int[] aarray = vamount.toArray(); | |
double[] parray = vpart.toArray(); | |
int i = 0; | |
for (String each : narray) { | |
result.add(new StatisticsEntry(each, tarray[i], aarray[i], parray[i])); | |
i++; | |
} | |
return result; | |
} | |
public double getPart() { | |
return part; | |
} | |
public int getAmount() { | |
return amount; | |
} | |
public double getTotal() { | |
return total; | |
} | |
public String getName() { | |
return name; | |
} | |
@Override | |
public String toString() { | |
return getName() + " : " + getTotal() + " : " + getAmount() + " : " + getPart(); | |
} | |
} |