コード例 #1
1
  @Override
  protected Void doInBackground() throws Exception {
    // protein identification id
    Collection<Comparable> protIdentIds = controller.getIdentificationIds();

    // protein identification id and accession buffer
    Map<Comparable, String> accBuffer = new LinkedHashMap<Comparable, String>();

    // protein map
    Map<String, Protein> proteins = new HashMap<String, Protein>();

    try {
      // iterate over each protein
      for (Comparable protIdentId : protIdentIds) {
        // get mapped protein accession
        String protAcc = controller.getProteinAccession(protIdentId);
        String protAccVersion = controller.getProteinAccessionVersion(protIdentId);
        String database = controller.getSearchDatabase(protIdentId);
        AccessionResolver resolver = new AccessionResolver(protAcc, protAccVersion, database, true);
        String mappedProtAcc = resolver.isValidAccession() ? resolver.getAccession() : null;

        if (mappedProtAcc != null) {
          // get existing protein details
          Protein protDetails =
              PrideInspectorCacheManager.getInstance().getProteinDetails(mappedProtAcc);
          if (protDetails != null) {
            proteins.put(mappedProtAcc, protDetails);
          }

          accBuffer.put(protIdentId, mappedProtAcc);
          if (accBuffer.size() == MAX_BATCH_DOWNLOAD_SIZE) {
            // fetch and publish protein details
            fetchAndPublish(accBuffer, proteins);

            // clear accession buffer
            accBuffer.clear();

            // clear protein map
            proteins = new HashMap<String, Protein>();
          }
        }
      }

      // this is important for cancelling
      if (Thread.interrupted()) {
        throw new InterruptedException();
      }

      if (!accBuffer.isEmpty() || !proteins.isEmpty()) {
        fetchAndPublish(accBuffer, proteins);
      }
    } catch (InterruptedException e) {
      logger.warn("Protein name download has been cancelled");
    }

    return null;
  }
コード例 #2
0
  /**
   * Fetch then publish
   *
   * @param accs protein accessions
   * @param proteins protein map
   * @throws Exception exception while fetching the protein
   */
  private void fetchAndPublish(Map<Comparable, String> accs, Map<String, Protein> proteins)
      throws Exception {
    Collection<String> accsToFetch = new HashSet<String>(accs.values());
    accsToFetch.removeAll(proteins.keySet());
    // fetch protein details
    Map<String, Protein> results = fetcher.getProteinDetails(accsToFetch);
    // add results to cache
    PrideInspectorCacheManager.getInstance().addProteinDetails(results.values());
    // add results to protein map
    proteins.putAll(results);
    // public results
    publish(new Tuple<TableContentType, Object>(TableContentType.PROTEIN_DETAILS, proteins));

    // protein sequence coverage
    Map<Comparable, Double> coverageMap = new HashMap<Comparable, Double>();
    // peptide fits
    Map<Tuple<Comparable, Comparable>, Integer> peptideFits =
        new HashMap<Tuple<Comparable, Comparable>, Integer>();

    for (Comparable protIdentId : accs.keySet()) {
      // protein sequence coverage
      Double coverage =
          PrideInspectorCacheManager.getInstance()
              .getSequenceCoverage(controller.getUid(), protIdentId);
      if (coverage == null) {
        String mappedProtAcc = accs.get(protIdentId);

        // get protein details
        Protein protein = proteins.get(mappedProtAcc);
        if (protein != null) {
          AnnotatedProtein annotatedProtein = new AnnotatedProtein(protein);
          Collection<Comparable> peptideIds = controller.getPeptideIds(protIdentId);
          for (Comparable peptideId : peptideIds) {
            PeptideAnnotation peptide = new PeptideAnnotation();
            peptide.setSequence(controller.getPeptideSequence(protIdentId, peptideId));
            peptide.setStart(controller.getPeptideSequenceStart(protIdentId, peptideId));
            peptide.setEnd(controller.getPeptideSequenceEnd(protIdentId, peptideId));
            annotatedProtein.addAnnotation(peptide);
          }
          coverage = annotatedProtein.getSequenceCoverage();
          coverageMap.put(protIdentId, coverage);
          // cache the sequence coverage
          PrideInspectorCacheManager.getInstance()
              .addSequenceCoverage(controller.getUid(), protIdentId, coverage);
        }
      } else {
        coverageMap.put(protIdentId, coverage);
      }

      // peptide fits
      Collection<Comparable> peptideIdentIds = controller.getPeptideIds(protIdentId);
      for (Comparable peptideIdentId : peptideIdentIds) {
        Integer state =
            PrideInspectorCacheManager.getInstance()
                .getPeptideFitState(controller.getUid(), protIdentId, peptideIdentId);
        if (state == null) {
          String mappedProtAcc = accs.get(protIdentId);

          // get protein details
          Protein protein = proteins.get(mappedProtAcc);
          if (protein != null) {
            protein = new AnnotatedProtein(protein);
          }

          // get peptide sequence
          String sequence = controller.getPeptideSequence(protIdentId, peptideIdentId);

          // start and end position
          int start = controller.getPeptideSequenceStart(protIdentId, peptideIdentId);
          int end = controller.getPeptideSequenceEnd(protIdentId, peptideIdentId);

          // peptide present
          if (protein == null || sequence == null || protein.getSequenceString() == null) {
            state = PeptideFitState.UNKNOWN;
          } else {
            if (protein.hasSubSequenceString(sequence, start, end)) {
              state = PeptideFitState.STRICT_FIT;
            } else if (protein.hasSubSequenceString(sequence)) {
              state = PeptideFitState.FIT;
            } else {
              state = PeptideFitState.NOT_FIT;
            }
          }
          PrideInspectorCacheManager.getInstance()
              .addPeptideFitState(controller.getUid(), protIdentId, peptideIdentId, state);
        }
        peptideFits.put(new Tuple<Comparable, Comparable>(protIdentId, peptideIdentId), state);
      }
    }

    // publish sequence coverage results
    publish(
        new Tuple<TableContentType, Object>(
            TableContentType.PROTEIN_SEQUENCE_COVERAGE, coverageMap));

    // publish peptide fit result
    publish(new Tuple<TableContentType, Object>(TableContentType.PEPTIDE_FIT, peptideFits));
  }