use of de.tudarmstadt.ukp.clarin.webanno.model.SourceDocument in project webanno by webanno.
the class RemoteApiController2 method readAnnotation.
private ResponseEntity<byte[]> readAnnotation(long aProjectId, long aDocumentId, String aAnnotatorId, Mode aMode, Optional<String> aFormat) throws RemoteApiException, ClassNotFoundException, IOException, UIMAException {
// Get project (this also ensures that it exists and that the current user can access it
Project project = getProject(aProjectId);
SourceDocument doc = getDocument(project, aDocumentId);
// Check format
String format;
if (aFormat.isPresent()) {
if (VAL_ORIGINAL.equals(aFormat.get())) {
format = doc.getFormat();
} else {
format = aFormat.get();
}
} else {
format = doc.getFormat();
}
// Determine the format
Class<?> writer = importExportService.getWritableFormats().get(format);
if (writer == null) {
String msg = "[" + doc.getName() + "] No writer found for format [" + format + "] - exporting as WebAnno TSV instead.";
LOG.info(msg);
writer = WebannoTsv3XWriter.class;
}
// annotation document entry is actually properly set up in the database.
if (Mode.ANNOTATION.equals(aMode)) {
getAnnotation(doc, aAnnotatorId, false);
}
// Create a temporary export file from the annotations
File exportedAnnoFile = null;
byte[] resource;
try {
exportedAnnoFile = importExportService.exportAnnotationDocument(doc, aAnnotatorId, writer, doc.getName(), Mode.ANNOTATION);
resource = FileUtils.readFileToByteArray(exportedAnnoFile);
} finally {
if (exportedAnnoFile != null) {
FileUtils.forceDelete(exportedAnnoFile);
}
}
String filename = FilenameUtils.removeExtension(doc.getName());
filename += "-" + aAnnotatorId;
// Actually, exportedAnnoFile cannot be null here - the warning can be ignored.
filename += "." + FilenameUtils.getExtension(exportedAnnoFile.getName());
HttpHeaders httpHeaders = new HttpHeaders();
httpHeaders.setContentLength(resource.length);
httpHeaders.set("Content-Disposition", "attachment; filename=\"" + filename + "\"");
return new ResponseEntity<>(resource, httpHeaders, OK);
}
use of de.tudarmstadt.ukp.clarin.webanno.model.SourceDocument in project webanno by webanno.
the class RemoteApiController2 method documentRead.
@ApiOperation(value = "Get a document from a project", response = byte[].class)
@RequestMapping(value = "/" + PROJECTS + "/{" + PARAM_PROJECT_ID + "}/" + DOCUMENTS + "/{" + PARAM_DOCUMENT_ID + "}", method = RequestMethod.GET, produces = { APPLICATION_OCTET_STREAM_VALUE, APPLICATION_JSON_UTF8_VALUE })
public ResponseEntity documentRead(@PathVariable(PARAM_PROJECT_ID) long aProjectId, @PathVariable(PARAM_DOCUMENT_ID) long aDocumentId, @RequestParam(value = PARAM_FORMAT) Optional<String> aFormat) throws Exception {
// Get project (this also ensures that it exists and that the current user can access it
Project project = getProject(aProjectId);
SourceDocument doc = getDocument(project, aDocumentId);
boolean originalFile;
String format;
if (aFormat.isPresent()) {
if (VAL_ORIGINAL.equals(aFormat.get())) {
format = doc.getFormat();
originalFile = true;
} else {
format = aFormat.get();
originalFile = doc.getFormat().equals(format);
}
} else {
format = doc.getFormat();
originalFile = true;
}
if (originalFile) {
// Export the original file - no temporary file created here, we export directly from
// the file system
File docFile = documentService.getSourceDocumentFile(doc);
FileSystemResource resource = new FileSystemResource(docFile);
HttpHeaders httpHeaders = new HttpHeaders();
httpHeaders.setContentLength(resource.contentLength());
httpHeaders.set("Content-Disposition", "attachment; filename=\"" + doc.getName() + "\"");
return new ResponseEntity<org.springframework.core.io.Resource>(resource, httpHeaders, OK);
} else {
// Export a converted file - here we first export to a local temporary file and then
// send that back to the client
// Check if the format is supported
Map<String, Class<JCasAnnotator_ImplBase>> writableFormats = importExportService.getWritableFormats();
Class<JCasAnnotator_ImplBase> writer = writableFormats.get(format);
if (writer == null) {
throw new UnsupportedFormatException("Format [%s] cannot be exported. Exportable formats are %s.", aFormat, writableFormats.keySet());
}
// Create a temporary export file from the annotations
JCas jcas = documentService.createOrReadInitialCas(doc);
File exportedFile = null;
try {
// Load the converted file into memory
exportedFile = importExportService.exportCasToFile(jcas.getCas(), doc, doc.getName(), writer, true);
byte[] resource = FileUtils.readFileToByteArray(exportedFile);
// Send it back to the client
HttpHeaders httpHeaders = new HttpHeaders();
httpHeaders.setContentLength(resource.length);
httpHeaders.set("Content-Disposition", "attachment; filename=\"" + exportedFile.getName() + "\"");
return new ResponseEntity<>(resource, httpHeaders, OK);
} finally {
if (exportedFile != null) {
FileUtils.forceDelete(exportedFile);
}
}
}
}
use of de.tudarmstadt.ukp.clarin.webanno.model.SourceDocument in project webanno by webanno.
the class RemoteApiController2 method curationCreate.
@ApiOperation(value = "Create curation for a document in a project")
@RequestMapping(value = "/" + PROJECTS + "/{" + PARAM_PROJECT_ID + "}/" + DOCUMENTS + "/{" + PARAM_DOCUMENT_ID + "}/" + CURATION, method = RequestMethod.POST, consumes = MULTIPART_FORM_DATA_VALUE, produces = APPLICATION_JSON_UTF8_VALUE)
public ResponseEntity<RResponse<RAnnotation>> curationCreate(@PathVariable(PARAM_PROJECT_ID) long aProjectId, @PathVariable(PARAM_DOCUMENT_ID) long aDocumentId, @RequestParam(value = PARAM_CONTENT) MultipartFile aFile, @RequestParam(value = PARAM_FORMAT) Optional<String> aFormat, @RequestParam(value = PARAM_STATE) Optional<String> aState, UriComponentsBuilder aUcb) throws Exception {
Project project = getProject(aProjectId);
SourceDocument document = getDocument(project, aDocumentId);
JCas annotationCas = createCompatibleCas(aProjectId, aDocumentId, aFile, aFormat);
// If they are compatible, then we can store the new annotations
curationService.writeCurationCas(annotationCas, document, false);
AnnotationDocumentState resultState = AnnotationDocumentState.IN_PROGRESS;
if (aState.isPresent()) {
SourceDocumentState state = parseSourceDocumentState(aState.get());
switch(state) {
case CURATION_IN_PROGRESS:
resultState = AnnotationDocumentState.IN_PROGRESS;
document.setState(state);
documentService.createSourceDocument(document);
break;
case CURATION_FINISHED:
resultState = AnnotationDocumentState.FINISHED;
document.setState(state);
documentService.createSourceDocument(document);
break;
// fallthrough
case NEW:
// fallthrough
case ANNOTATION_IN_PROGRESS:
// fallthrough
case ANNOTATION_FINISHED:
default:
throw new IllegalObjectStateException("State [%s] not valid when uploading a curation.", aState.get());
}
} else {
document.setState(SourceDocumentState.CURATION_IN_PROGRESS);
documentService.createSourceDocument(document);
}
RResponse<RAnnotation> response = new RResponse<>(new RAnnotation(WebAnnoConst.CURATION_USER, resultState, new Date()));
return ResponseEntity.created(aUcb.path(API_BASE + "/" + PROJECTS + "/{pid}/" + DOCUMENTS + "/{did}/" + CURATION).buildAndExpand(project.getId(), document.getId()).toUri()).body(response);
}
use of de.tudarmstadt.ukp.clarin.webanno.model.SourceDocument in project webanno by webanno.
the class RemoteApiController2 method annotationsList.
@ApiOperation(value = "List annotations of a document in a project")
@RequestMapping(value = "/" + PROJECTS + "/{" + PARAM_PROJECT_ID + "}/" + DOCUMENTS + "/{" + PARAM_DOCUMENT_ID + "}/" + ANNOTATIONS, method = RequestMethod.GET, produces = APPLICATION_JSON_UTF8_VALUE)
public ResponseEntity<RResponse<List<RAnnotation>>> annotationsList(@PathVariable(PARAM_PROJECT_ID) long aProjectId, @PathVariable(PARAM_DOCUMENT_ID) long aDocumentId) throws Exception {
// Get project (this also ensures that it exists and that the current user can access it
Project project = getProject(aProjectId);
SourceDocument doc = getDocument(project, aDocumentId);
List<AnnotationDocument> annotations = documentService.listAnnotationDocuments(doc);
List<RAnnotation> annotationList = new ArrayList<>();
for (AnnotationDocument annotation : annotations) {
annotationList.add(new RAnnotation(annotation));
}
return ResponseEntity.ok(new RResponse<>(annotationList));
}
Aggregations