@Override
public List<LocatedFeature> locateFeatures(
FeatureList featureList, AdaptedModel adaptedModel, IProgressMonitor monitor) {
List<LocatedFeature> locatedFeatures = new ArrayList<LocatedFeature>();
/*
* We gather all words for each blocks
*/
List<List<String>> listBlocksWords = new ArrayList<List<String>>();
for (Block b : adaptedModel.getOwnedBlocks()) {
List<String> l = new ArrayList<String>();
for (BlockElement e : b.getOwnedBlockElements()) {
for (ElementWrapper wr : e.getElementWrappers()) {
AbstractElement element = (AbstractElement) (wr.getElement());
for (String s : element.getWords()) {
l.add(s.trim());
}
}
}
listBlocksWords.add(l);
}
/*
* We gather all words for each features
*/
List<List<String>> listFeaturesWords = new ArrayList<List<String>>();
for (Feature f : featureList.getOwnedFeatures()) {
listFeaturesWords.add(FeatureWordCloudUtil.getFeatureWords(f));
}
// Word cloud IDF calculation for feature
List<Cloud> clouds_IDF_Features = new ArrayList<Cloud>();
for (int i = 0; i < listFeaturesWords.size(); i++) {
clouds_IDF_Features.add(WordCloudUtil.createWordCloudIDF(listFeaturesWords, i));
}
// Word cloud IDF calculation for blocks
List<Cloud> clouds_IDF_Blocks = new ArrayList<Cloud>();
for (int i = 0; i < listBlocksWords.size(); i++) {
clouds_IDF_Blocks.add(WordCloudUtil.createWordCloudIDF(listBlocksWords, i));
}
for (int i = 0; i < featureList.getOwnedFeatures().size(); i++) {
for (Cloud c : clouds_IDF_Blocks) {
Block b = adaptedModel.getOwnedBlocks().get(clouds_IDF_Blocks.indexOf(c));
Feature f = featureList.getOwnedFeatures().get(i);
double d = WordCloudUtil.cmpClouds(clouds_IDF_Features.get(i), c);
if (d > 0.0) {
locatedFeatures.add(new LocatedFeature(f, b, d));
}
}
}
return locatedFeatures;
}
