blob: 5295668494ec6a094d965e7083cc6753a4a4419c [file] [log] [blame]
// Lucene version compatibility level 8.2.0
using Lucene.Net.Analysis.OpenNlp.Tools;
using Lucene.Net.Analysis.Util;
using System;
using System.Collections.Generic;
using System.IO;
namespace Lucene.Net.Analysis.OpenNlp
{
/*
* 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.
*/
/// <summary>
/// Factory for <see cref="OpenNLPPOSFilter"/>.
/// <code>
/// &lt;fieldType name="text_opennlp_pos" class="solr.TextField" positionIncrementGap="100"&gt;
/// &lt;analyzer&gt;
/// &lt;tokenizer class="solr.OpenNLPTokenizerFactory" sentenceModel="filename" tokenizerModel="filename"/&gt;
/// &lt;filter class="solr.OpenNLPPOSFilterFactory" posTaggerModel="filename"/&gt;
/// &lt;/analyzer&gt;
/// &lt;/fieldType&gt;
/// </code>
/// </summary>
/// <since>7.3.0</since>
public class OpenNLPPOSFilterFactory : TokenFilterFactory, IResourceLoaderAware
{
public const string POS_TAGGER_MODEL = "posTaggerModel";
private readonly string posTaggerModelFile;
public OpenNLPPOSFilterFactory(IDictionary<string, string> args)
: base(args)
{
posTaggerModelFile = Require(args, POS_TAGGER_MODEL);
if (args.Count > 0)
{
throw new ArgumentException("Unknown parameters: " + args);
}
}
public override TokenStream Create(TokenStream input)
{
try
{
return new OpenNLPPOSFilter(input, OpenNLPOpsFactory.GetPOSTagger(posTaggerModelFile));
}
catch (IOException e)
{
throw new ArgumentException(e.ToString(), e);
}
}
public virtual void Inform(IResourceLoader loader)
{
try
{ // load and register the read-only model in cache with file/resource name
OpenNLPOpsFactory.GetPOSTaggerModel(posTaggerModelFile, loader);
}
catch (IOException e)
{
throw new ArgumentException(e.ToString(), e);
}
}
}
}