- Add inteface for getting summary features for a single hit too.

- Produce summaryhits for grouping results too.

1 files changed, 13 insertions, 0 deletions

diff --git a/streamingvisitors/src/vespa/searchvisitor/rankprocessor.cpp b/streamingvisitors/src/vespa/searchvisitor/rankprocessor.cpp
index 167d5ecde4c..ac7a61e5888 100644
--- a/streamingvisitors/src/vespa/searchvisitor/rankprocessor.cpp
+++ b/streamingvisitors/src/vespa/searchvisitor/rankprocessor.cpp
@@ -256,6 +256,19 @@ RankProcessor::calculateFeatureSet()
     return sf;
 }
 
+FeatureSet::SP
+RankProcessor::calculateFeatureSet(search::DocumentIdT docId)
+{
+    LOG(debug, "Calculate feature set for docId = %d", docId);
+    RankProgram &rankProgram = *(_summaryProgram ? _summaryProgram : _rankProgram);
+    search::fef::FeatureResolver resolver(rankProgram.get_seeds(false));
+    LOG(debug, "Feature handles: numNames(%ld)", resolver.num_features());
+    RankProgramWrapper wrapper(*_match_data);
+    FeatureSet::SP sf = _hitCollector->getFeatureSet(wrapper, docId, resolver, _rankSetup.get_feature_rename_map());
+    LOG(debug, "Feature set: numFeatures(%u), numDocs(%u)", sf->numFeatures(), sf->numDocs());
+    return sf;
+}
+
 FeatureValues
 RankProcessor::calculate_match_features()
 {