opennlp-similarity/src/main/java/opennlp/tools/textsimilarity/chunker2matcher/ParserCacheSerializer.java

/*
 * 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.
 */
/*
 * 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.
 */

package opennlp.tools.textsimilarity.chunker2matcher;

import java.io.BufferedInputStream;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.StandardOpenOption;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Logger;

import au.com.bytecode.opencsv.CSVReader;
import au.com.bytecode.opencsv.CSVWriter;

public class ParserCacheSerializer {
  private static final Logger LOG = Logger
      .getLogger("opennlp.tools.textsimilarity.chunker2matcher.ParserCacheSerializer");
  private static final boolean JAVA_OBJECT_SERIALIZATION = false;
  private static final String RESOURCE_DIR = "src/test/resources/";
  private static final String PARSE_CACHE_FILE_NAME = "sentence_parseObject.dat";
  private static final String PARSE_CACHE_FILE_NAME_CSV = "sentence_parseObject.csv";

  public static void writeObject(Object objectToSerialize) {
    if (JAVA_OBJECT_SERIALIZATION) {
      String filename = RESOURCE_DIR + PARSE_CACHE_FILE_NAME;
      try(FileOutputStream fos = new FileOutputStream(filename);
          ObjectOutputStream out = new ObjectOutputStream(fos)) {
        
        out.writeObject(objectToSerialize);
      } catch (IOException ex) {
        ex.printStackTrace();
      }
    } else {

      Map<String, String[][]> sentence_parseObject = (Map<String, String[][]>) objectToSerialize;
      final List<String> keys = new ArrayList<>(sentence_parseObject.keySet());

      final Path p = Path.of(RESOURCE_DIR + PARSE_CACHE_FILE_NAME_CSV);
      try (CSVWriter writer = new CSVWriter(Files.newBufferedWriter(p, StandardCharsets.UTF_8,
              StandardOpenOption.WRITE, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING))) {
        for (String k : keys) {
          String[][] triplet = sentence_parseObject.get(k);
          writer.writeNext(new String[] { k });
          writer.writeNext(triplet[0]);
          writer.writeNext(triplet[1]);
          writer.writeNext(triplet[2]);
        }
      } catch (IOException e) {
        LOG.severe(e.getMessage());
      }
    }

  }

  public static Object readObject() {
    if (JAVA_OBJECT_SERIALIZATION) {
      String filename = RESOURCE_DIR + PARSE_CACHE_FILE_NAME;
      Object data = null;
      try (InputStream fis = new BufferedInputStream(new FileInputStream(filename));
           ObjectInputStream in = new ObjectInputStream(fis)) {

        data = in.readObject();
      } catch (IOException ex) {
        System.out.println("Cant find parsing cache file ");
      } catch (ClassNotFoundException ex) {
        ex.printStackTrace();
      }
      return data;
    } else {
      List<String[]> lines;

      try (CSVReader reader = new CSVReader(new FileReader(RESOURCE_DIR
              + PARSE_CACHE_FILE_NAME_CSV), ',')) {
        lines = reader.readAll();
      } catch (FileNotFoundException e) {
        if (JAVA_OBJECT_SERIALIZATION)
          System.err.println("Cannot find cache file");
        return null;
      } catch (IOException ioe) {
        ioe.printStackTrace();
        return null;
      }
      Map<String, String[][]> sentence_parseObject = new HashMap<>();
      for (int i = 0; i < lines.size() - 3; i += 4) {
        String key = lines.get(i)[0];
        String[][] value = new String[][] { lines.get(i + 1), lines.get(i + 2),
            lines.get(i + 3) };
        sentence_parseObject.put(key, value);
      }

      return sentence_parseObject;
    }

  }

}