ExtDictFSAPortugueseFactory.java

/**
 * Copyright (C) 2012 cogroo <cogroo@cogroo.org>
 *
 * Licensed 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.cogroo.tools.postag;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStream;
import java.util.Map;

import opennlp.tools.dictionary.Dictionary;
import opennlp.tools.postag.POSContextGenerator;
import opennlp.tools.postag.TagDictionary;
import opennlp.tools.util.InvalidFormatException;
import opennlp.tools.util.SequenceValidator;
import opennlp.tools.util.model.ArtifactSerializer;

import org.cogroo.dictionary.impl.FSADictionary;
import org.cogroo.util.serializers.ByteArraySerializer;

public class ExtDictFSAPortugueseFactory extends PortugueseExFactory {

  private static final String FSA_POSDICT_SUF = "fsa_data";
  private static final String FSA_DICT_INFO_SUF = "fsa_info";
  
  private static final String FSA_POSDICT = "dict." + FSA_POSDICT_SUF;
  private static final String FSA_DICT_INFO = "dict." + FSA_DICT_INFO_SUF;
  
  private TagDictionary dict;
  
  public ExtDictFSAPortugueseFactory() {
  }

  public ExtDictFSAPortugueseFactory(Dictionary ngramDictionary,
      TagDictionary posDictionary) {
    super(ngramDictionary, null);
  }
  
  @Override
  protected void init(Dictionary ngramDictionary, TagDictionary posDictionary) {
    super.init(ngramDictionary, null);
    this.dict = posDictionary;
    
    // get the dictionary path
    String path = System.getProperty("fsa.dict");
    if(path == null) {
      throw new IllegalArgumentException("The property fsa.dict is missing! -Dfsa.dict=path");
    }
    
    // now we try to load it...
    try {
      InputStream dict = getClass().getResourceAsStream(path + ".dict");
      InputStream info = getClass().getResourceAsStream(path + ".info");
      
      if(dict == null) {
        System.out.println("NULLL");
      }
      
      this.dict = FSADictionary.create(dict, info);
    } catch (IllegalArgumentException e) {
      throw new IllegalArgumentException("The file is not a FSA dictionary!", e);
    } catch (IOException e) {
      throw new IllegalArgumentException("Could not open the FSA dictionary or the .info file", e);
    }
  }

  @Override
  public TagDictionary getTagDictionary() {
    if(this.dict == null) {
      
//      if(artifactProvider != null) {
//        Object obj = artifactProvider.getArtifact(FSA_POSDICT);
//        if(obj != null) {
          String path = "/fsa_dictionaries/pos/pt_br_jspell_corpus";
          InputStream dict = getClass().getResourceAsStream(path + ".dict");
          InputStream info = getClass().getResourceAsStream(path + ".info");
          
          try {
            this.dict = FSADictionary.create(dict, info);
          } catch (IllegalArgumentException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
          } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
          }
        }
//      }
//    }
      
    return this.dict;
  }
  
  @Override
  public void setTagDictionary(TagDictionary dictionary) {
    throw new UnsupportedOperationException("FSA factory does not support this operation");
  }
  
  @Override
  public TagDictionary createEmptyTagDictionary() {
    throw new UnsupportedOperationException("FSA factory does not support this operation");
  }
  
  @Override
  public TagDictionary createTagDictionary(File dictionary)
      throws InvalidFormatException, FileNotFoundException, IOException {
    throw new UnsupportedOperationException("FSA factory does not support this operation");
  }
  
  @Override
  public TagDictionary createTagDictionary(InputStream in)
      throws InvalidFormatException, IOException {
    throw new UnsupportedOperationException("FSA factory does not support this operation");
  }
  
}