edu.harvard.med.screensaver.ui.screenresults.ScreenResultImporter.java Source code

Introduction

Here is the source code for edu.harvard.med.screensaver.ui.screenresults.ScreenResultImporter.java

Source

// $HeadURL$
// $Id$
//
// Copyright  2006, 2010, 2011, 2012 by the President and Fellows of Harvard College.
//
// Screensaver is an open-source project developed by the ICCB-L and NSRB labs
// at Harvard Medical School. This software is distributed under the terms of
// the GNU General Public License.

package edu.harvard.med.screensaver.ui.screenresults;

import java.io.IOException;
import java.util.List;

import javax.faces.model.DataModel;
import javax.faces.model.ListDataModel;

import com.google.common.collect.Sets;
import org.apache.log4j.Logger;
import org.apache.myfaces.custom.fileupload.UploadedFile;

import edu.harvard.med.screensaver.db.GenericEntityDAO;
import edu.harvard.med.screensaver.io.ParseError;
import edu.harvard.med.screensaver.io.ParseErrorsException;
import edu.harvard.med.screensaver.io.workbook2.Workbook;
import edu.harvard.med.screensaver.model.AuditedAbstractEntity;
import edu.harvard.med.screensaver.model.screenresults.ScreenResult;
import edu.harvard.med.screensaver.model.screens.Screen;
import edu.harvard.med.screensaver.model.users.AdministratorUser;
import edu.harvard.med.screensaver.service.screenresult.ScreenResultLoader;
import edu.harvard.med.screensaver.ui.arch.view.AbstractBackingBean;
import edu.harvard.med.screensaver.ui.arch.view.aspects.UICommand;
import edu.harvard.med.screensaver.ui.screens.ScreenViewer;

/**
 * The JSF backing bean for the screenResultImporter view.
 *
 * @author <a mailto="andrew_tolopko@hms.harvard.edu">Andrew Tolopko</a>
 * @author <a mailto="john_sullivan@hms.harvard.edu">John Sullivan</a>
 */
public class ScreenResultImporter extends AbstractBackingBean {

    // static data

    private static final String ERRORS_XLS_FILE_EXTENSION = ".errors.xls";
    private static final Logger log = Logger.getLogger(ScreenResultImporter.class);

    // instance data

    private GenericEntityDAO _dao;
    private ScreenViewer _screenViewer;
    private ScreenResultLoader _screenResultLoader;

    private UploadedFile _uploadedFile;
    private String _comments;
    private List<? extends ParseError> _lastParseErrors;

    // constructors

    /**
     * @motivation for CGLIB2
     */
    protected ScreenResultImporter() {
    }

    public ScreenResultImporter(GenericEntityDAO dao, ScreenViewer screenViewer,
            ScreenResultLoader screenResultLoader) {
        _dao = dao;
        _screenViewer = screenViewer;
        _screenResultLoader = screenResultLoader;
    }

    public Screen getScreen() {
        return _screenViewer.getEntity();
    }

    public void setUploadedFile(UploadedFile uploadedFile) {
        _uploadedFile = uploadedFile;
    }

    public UploadedFile getUploadedFile() {
        return _uploadedFile;
    }

    public String getComments() {
        return _comments;
    }

    public void setComments(String comments) {
        _comments = comments;
    }

    public boolean getHasErrors() {
        return _lastParseErrors != null && !_lastParseErrors.isEmpty();
    }

    public DataModel getImportErrors() {
        return new ListDataModel(_lastParseErrors);
    }

    // JSF application methods

    @UICommand
    public String cancel() {
        return _screenViewer.reload();
    }

    @UICommand
    public String doImport() throws IOException {
        Screen screen = _screenViewer.getEntity();
        try {
            ScreenResult screenResult = _screenResultLoader.parseAndLoad(screen,
                    new Workbook("Input Stream for screen: " + screen, _uploadedFile.getInputStream()),
                    (AdministratorUser) getScreensaverUser(), _comments, null, true);
            screenResult = _dao.reloadEntity(screenResult, true,
                    ScreenResult.screen.to(AuditedAbstractEntity.updateActivities.castToSubtype(Screen.class)));
            showMessage("screens.screenResultDataLoaded", screenResult.getLastDataLoadingActivity().getComments());
            screen = _dao.reloadEntity(screen, true, Screen.assayPlates);
            int assayPlatesCreated = Sets
                    .difference(screen.getAssayPlatesDataLoaded(), screen.getAssayPlatesScreened()).size();
            if (assayPlatesCreated > 0) {
                showMessage("screens.assayPlatesCreatedForLoadedData", assayPlatesCreated);
            }
        } catch (ParseErrorsException e) {
            log.info("parse errors encountered during import of ScreenResult for Screen " + screen);
            _lastParseErrors = e.getErrors();
            return REDISPLAY_PAGE_ACTION_RESULT;
        }
        return _screenViewer.viewEntity(screen);
    }

    @UICommand
    public String importScreenResultData() {
        return IMPORT_SCREEN_RESULT_DATA;
    }
}