private void loadFiles() throws IOException { File rawDir = new File(ioConf.getRawDir()); if (!rawDir.exists() || !rawDir.isDirectory()) { throw new IOException(rawDir.getAbsolutePath() + " is not a valid directory"); } Doc doc; File parentDir; File anotFile; for (File d : rawDir.listFiles()) { if (d.isDirectory()) { for (File f : d.listFiles()) { doc = XMLDoc.readXML(f); parentDir = new File(ioConf.getParsedDir(), doc.getParentDirName()); anotFile = new File(parentDir, doc.getAnotFileName()); doc.anotFile = anotFile; documents.add(doc); // addAnnotation(doc); } } } }
public void addAnnotation(Doc doc) throws IOException { File parentDir = new File(ioConf.getParsedDir(), doc.getParentDirName()); File anotFile = new File(parentDir, doc.getAnotFileName()); if (!anotFile.exists()) { System.err.println("Generating annotation for " + doc.f.getName()); genAnnotation(doc); saveAnnotation(doc, anotFile); } else { try { doc.setAno(ParsedDocReader.read(anotFile)); } catch (IOException e) { System.err.println("Generating annotation for " + doc.f.getName()); genAnnotation(doc); } } }
private Map<Doc, String[]> linkModels() { // Mapping of doc names to models Map<Doc, String[]> map = new HashMap<Doc, String[]>(); File modelDir = new File(ioConf.getModelDir()); String prefix, suffix; for (Doc d : documents) { // Set model prefix/suffix prefix = d.getParentDirName(); prefix = prefix.substring(0, prefix.length() - 1).toUpperCase(); suffix = d.f.getName(); suffix = suffix.substring(suffix.lastIndexOf(".") + 1); // Add doc/models to the map map.put(d, modelDir.list(new ModelFileFilter(prefix, suffix))); } return map; }