/*
* 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.
*/
using System;
using Token = Lucene.Net.Analysis.Token;
namespace Lucene.Net.Highlight
{
/// One, or several overlapping tokens, along with the score(s) and the
/// scope of the original text
///
/// MAHarwood
///
public class TokenGroup
{
private const int MAX_NUM_TOKENS_PER_GROUP = 50;
internal Token[] tokens = new Token[MAX_NUM_TOKENS_PER_GROUP];
internal float[] scores = new float[MAX_NUM_TOKENS_PER_GROUP];
internal int numTokens = 0;
internal int startOffset = 0;
internal int endOffset = 0;
internal float tot;
internal int matchStartOffset, matchEndOffset;
internal virtual void AddToken(Token token, float score)
{
if (numTokens < MAX_NUM_TOKENS_PER_GROUP)
{
if (numTokens == 0)
{
startOffset = matchStartOffset = token.StartOffset();
endOffset = matchEndOffset = token.EndOffset();
tot += score;
}
else
{
startOffset = Math.Min(startOffset, token.StartOffset());
endOffset = Math.Max(endOffset, token.EndOffset());
if (score > 0)
{
if (tot == 0)
{
matchStartOffset = token.StartOffset();
matchEndOffset = token.EndOffset();
}
else
{
matchStartOffset = Math.Min(matchStartOffset, token.StartOffset());
matchEndOffset = Math.Max(matchEndOffset, token.EndOffset());
}
tot += score;
}
}
tokens[numTokens] = token;
scores[numTokens] = score;
numTokens++;
}
}
internal virtual bool IsDistinct(Token token)
{
return token.StartOffset() >= endOffset;
}
internal virtual void Clear()
{
numTokens = 0;
tot = 0;
}
///
/// a value between 0 and numTokens -1
///
/// the "n"th token
///
public virtual Token GetToken(int index)
{
return tokens[index];
}
///
/// a value between 0 and numTokens -1
///
/// the "n"th score
///
public virtual float GetScore(int index)
{
return scores[index];
}
/// the end position in the original text
///
public virtual int GetEndOffset()
{
return endOffset;
}
/// the number of tokens in this group
///
public virtual int GetNumTokens()
{
return numTokens;
}
/// the start position in the original text
///
public virtual int GetStartOffset()
{
return startOffset;
}
/// all tokens' scores summed up
///
public virtual float GetTotalScore()
{
return tot;
}
}
}