Examples with ExplorerChain de.bioforscher.jstructure.si.explorer.ExplorerChain used on opensource projects

Example 1 with ExplorerChain

use of de.bioforscher.jstructure.si.explorer.ExplorerChain in project jstructure by JonStargaryen.

the class A02_WriteHeatmapFiles method handleChain.

private static void handleChain(ExplorerChain explorerChain) {
    try {
        String entryId = explorerChain.getStfId();
        System.out.println(entryId);
        Chain chain = explorerChain.getChain();
        List<ContactStructuralInformation> contactStructuralInformation = explorerChain.getContacts();
        StringJoiner stringJoiner = new StringJoiner(System.lineSeparator(), "res1,res2,avg" + System.lineSeparator(), "");
        List<AminoAcid> aminoAcids = chain.aminoAcids().collect(Collectors.toList());
        SetOperations.cartesianProductOf(aminoAcids, aminoAcids).forEach(pair -> {
            int residueIdentifier1 = pair.getLeft().getResidueIdentifier().getResidueNumber();
            int residueIdentifier2 = pair.getRight().getResidueIdentifier().getResidueNumber();
            Optional<ContactStructuralInformation> information = contactStructuralInformation.stream().filter(contact -> (contact.getResidueIdentifier1() == residueIdentifier1 && contact.getResidueIdentifier2() == residueIdentifier2) || (contact.getResidueIdentifier1() == residueIdentifier2 && contact.getResidueIdentifier2() == residueIdentifier1)).findFirst();
            if (information.isPresent()) {
                stringJoiner.add(residueIdentifier1 + "," + residueIdentifier2 + "," + information.get().getAverageRmsdIncrease());
            } else {
                stringJoiner.add(residueIdentifier1 + "," + residueIdentifier2 + "," + "0.0");
            }
        });
        Files.write(Start2FoldConstants.DATA_DIRECTORY.resolve("si").resolve("heatmaps").resolve(entryId + ".csv"), stringJoiner.toString().getBytes());
    } catch (Exception e) {
        e.printStackTrace();
    }
}

Example 2 with ExplorerChain

use of de.bioforscher.jstructure.si.explorer.ExplorerChain in project jstructure by JonStargaryen.

the class A04A_CreatePyMolRendering method handleChain.

private static void handleChain(ExplorerChain explorerChain) {
    logger.info("handling chain {}", explorerChain.getStfId());
    try {
        Chain nativeChain = explorerChain.getChain();
        Path nativeChainPath = Files.createTempFile("nativechain-", ".pdb");
        Files.write(nativeChainPath, nativeChain.getPdbRepresentation().getBytes());
        ReconstructionContactMap nativeContactMap = ReconstructionContactMap.createReconstructionContactMap(nativeChain, ContactDefinitionFactory.createAlphaCarbonContactDefinition(8.0));
        List<AminoAcid> aminoAcids = nativeChain.getAminoAcids();
        List<Pair<AminoAcid, AminoAcid>> contacts = nativeContactMap.getLongRangeContacts();
        int numberNativeLongRangeContacts = contacts.size();
        List<ReconstructionContactMap> reconstructionContactMaps = new ArrayList<>();
        IntStream.of(5, 30, 100).forEach(coverage -> {
            int numberOfContactsToSelect = (int) Math.round(0.01 * coverage * numberNativeLongRangeContacts);
            for (int run = 0; run < REDUNDANCY; run++) {
                Collections.shuffle(contacts);
                List<Pair<AminoAcid, AminoAcid>> selectedContacts = contacts.subList(0, numberOfContactsToSelect);
                ReconstructionContactMap contactMap = new ReconstructionContactMap(aminoAcids, selectedContacts, nativeContactMap.getContactDefinition());
                contactMap.setName("p" + coverage + "-" + (run + 1));
                reconstructionContactMaps.add(contactMap);
            }
        });
        Map<String, List<Future<ReconstructionResult>>> reconstructionFutures = new HashMap<>();
        for (ReconstructionContactMap contactMap : reconstructionContactMaps) {
            String name = contactMap.getName().split("-")[0];
            logger.info("handling contact map with coverage {}", name);
            if (!reconstructionFutures.containsKey(name)) {
                reconstructionFutures.put(name, new ArrayList<>());
            }
            List<Future<ReconstructionResult>> bin = reconstructionFutures.get(name);
            bin.add(executorService.submit(new ConfoldServiceWorker("/home/sb/programs/confold_v1.0/confold.pl", contactMap.getSequence(), contactMap.getSecondaryStructureElements(), contactMap.getCaspRRRepresentation(), nativeContactMap.getConfoldRRType())));
        }
        for (Map.Entry<String, List<Future<ReconstructionResult>>> reconstructionFuture : reconstructionFutures.entrySet()) {
            try {
                String name = reconstructionFuture.getKey();
                List<Chain> reconstructions = reconstructionFuture.getValue().stream().map(future -> {
                    try {
                        return future.get();
                    } catch (Exception e) {
                        throw new ComputationException(e);
                    }
                }).map(ReconstructionResult::getChains).flatMap(Collection::stream).collect(Collectors.toList());
                for (Chain reconstructedChain : reconstructions) {
                    Files.write(OUTPUT_PATH.resolve(name + ".pdb"), reconstructedChain.getPdbRepresentation().getBytes());
                }
            } catch (IOException e) {
                throw new ComputationException(e);
            }
        }
    } catch (IOException e) {
        throw new ComputationException(e);
    }
}

Example 3 with ExplorerChain

use of de.bioforscher.jstructure.si.explorer.ExplorerChain in project jstructure by JonStargaryen.

the class A01_WriteBFactorFiles method handleChain.

private static void handleChain(ExplorerChain explorerChain) {
    try {
        String entryId = explorerChain.getStfId();
        System.out.println(entryId);
        Chain chain = explorerChain.getChain();
        List<ResidueStructuralInformation> residueStructuralInformation = explorerChain.getResidues();
        // assign baseline: all atoms (including hetatms) have bfactor 0
        chain.atoms().forEach(atom -> atom.setBfactor(0));
        chain.aminoAcids().forEach(aminoAcid -> {
            ResidueStructuralInformation information = residueStructuralInformation.get(aminoAcid.getAminoAcidIndex());
            aminoAcid.atoms().forEach(atom -> {
                atom.setBfactor((float) information.getAverageRmsdIncrease());
            });
        });
        Files.write(Start2FoldConstants.DATA_DIRECTORY.resolve("si").resolve("pymol").resolve(entryId + "-avg.pdb"), chain.getPdbRepresentation().getBytes());
        chain.aminoAcids().forEach(aminoAcid -> {
            ResidueStructuralInformation information = residueStructuralInformation.get(aminoAcid.getAminoAcidIndex());
            aminoAcid.atoms().forEach(atom -> {
                atom.setBfactor((float) information.getSumRmsdIncrease());
            });
        });
        Files.write(Start2FoldConstants.DATA_DIRECTORY.resolve("si").resolve("pymol").resolve(entryId + "-sum.pdb"), chain.getPdbRepresentation().getBytes());
    } catch (Exception e) {
        e.printStackTrace();
    }
}

Example 4 with ExplorerChain

use of de.bioforscher.jstructure.si.explorer.ExplorerChain in project jstructure by JonStargaryen.

the class A03_ReconstructByVariousStrategy method handleChain.

private static void handleChain(ExplorerChain explorerChain) {
    logger.info("[{}] starting job", explorerChain.getStfId());
    try {
        Chain nativeChain = explorerChain.getChain();
        Path nativeChainPath = Files.createTempFile("nativechain-", ".pdb");
        Files.write(nativeChainPath, nativeChain.getPdbRepresentation().getBytes());
        List<ContactStructuralInformation> contactStructuralInformation = explorerChain.getContacts();
        // annotate with PLIP data
        PLIPInteractionContainer plipInteractionContainer = nativeChain.getFeature(PLIPInteractionContainer.class);
        for (ContactStructuralInformation csi : contactStructuralInformation) {
            AminoAcid aminoAcid1 = nativeChain.select().residueNumber(csi.getResidueIdentifier1()).asAminoAcid();
            AminoAcid aminoAcid2 = nativeChain.select().residueNumber(csi.getResidueIdentifier2()).asAminoAcid();
            if (plipInteractionContainer.getHydrogenBonds().stream().anyMatch(hydrogenBond -> isContact(hydrogenBond, aminoAcid1, aminoAcid2))) {
                csi.markAsHydrogenBond();
            }
            if (plipInteractionContainer.getHydrophobicInteractions().stream().anyMatch(hydrophobicInteraction -> isContact(hydrophobicInteraction, aminoAcid1, aminoAcid2))) {
                csi.markAsHydrophobicInteraction();
            }
        }
        int numberOfNativeContacts = contactStructuralInformation.size();
        int numberOfContactsToSelect = (int) (numberOfNativeContacts * DEFAULT_COVERAGE);
        List<ReconstructionContactMap> contactMaps = Stream.of(ReconstructionStrategyDefinition.values()).map(ReconstructionStrategyDefinition::getReconstructionStrategy).flatMap(reconstructionStrategy -> IntStream.range(0, REDUNDANCY).boxed().flatMap(i -> {
            if (!reconstructionStrategy.isNegatable()) {
                ReconstructionContactMap contactMap = reconstructionStrategy.composeReconstructionContactMap(nativeChain, contactStructuralInformation, numberOfContactsToSelect);
                contactMap.setName(reconstructionStrategy.getName() + "-" + (i + 1));
                return Stream.of(contactMap);
            } else {
                // short, long, hydrogen, and hydrophobic bins have to be negated explicitly to get comparable results
                Pair<ReconstructionContactMap, ReconstructionContactMap> contactMapPair = reconstructionStrategy.composeReconstructionAndNegatedReconstructionContactMap(nativeChain, contactStructuralInformation, numberOfContactsToSelect);
                contactMapPair.getLeft().setName(reconstructionStrategy.getName() + "-" + (i + 1));
                contactMapPair.getRight().setName(reconstructionStrategy.getNegatedName() + "-" + (i + 1));
                return Stream.of(contactMapPair.getLeft(), contactMapPair.getRight());
            }
        })).filter(reconstructionContactMap -> reconstructionContactMap.getNumberOfContacts() > 0).collect(Collectors.toList());
        Map<String, List<Future<ReconstructionResult>>> reconstructionFutures = new HashMap<>();
        for (ReconstructionContactMap contactMap : contactMaps) {
            String name = contactMap.getName().split("-")[0];
            logger.info("[{}] handling contact map definition {}", explorerChain.getStfId(), name);
            if (!reconstructionFutures.containsKey(name)) {
                reconstructionFutures.put(name, new ArrayList<>());
            }
            List<Future<ReconstructionResult>> bin = reconstructionFutures.get(name);
            bin.add(executorService.submit(new ConfoldServiceWorker("/home/sb/programs/confold_v1.0/confold.pl", contactMap.getSequence(), contactMap.getSecondaryStructureElements(), contactMap.getCaspRRRepresentation(), contactMap.getConfoldRRType())));
        }
        for (Map.Entry<String, List<Future<ReconstructionResult>>> reconstructionFuture : reconstructionFutures.entrySet()) {
            try {
                String name = reconstructionFuture.getKey();
                List<Chain> reconstructions = reconstructionFuture.getValue().stream().map(future -> {
                    try {
                        return future.get();
                    } catch (Exception e) {
                        throw new ComputationException(e);
                    }
                }).map(ReconstructionResult::getChains).flatMap(Collection::stream).collect(Collectors.toList());
                logger.info("[{}][{}] {} reconstructs in bin", explorerChain.getStfId(), name, reconstructions.size());
                List<TMAlignAlignmentResult> alignmentResults = new ArrayList<>();
                List<Path> tmpFiles = new ArrayList<>();
                if (reconstructions.isEmpty()) {
                    throw new ComputationException("reconstruction did not yield any reconstructs");
                }
                for (Chain reconstructedChain : reconstructions) {
                    Path reconstructPath = Files.createTempFile("confoldservice-recon", ".pdb");
                    tmpFiles.add(reconstructPath);
                    Files.write(reconstructPath, reconstructedChain.getPdbRepresentation().getBytes());
                    alignmentResults.add(TM_ALIGN_SERVICE.process(new String[] { "/home/sb/programs/tmalign", nativeChainPath.toFile().getAbsolutePath(), reconstructPath.toFile().getAbsolutePath() }));
                }
                logger.info("[{}][{}] {} alignments in bin", explorerChain.getStfId(), name, alignmentResults.size());
                if (alignmentResults.isEmpty()) {
                    throw new ComputationException("tmalign did not yield any alignments");
                }
                for (TMAlignAlignmentResult alignmentResult : alignmentResults) {
                    double rmsd = alignmentResult.getRootMeanSquareDeviation().getScore();
                    String line = explorerChain.getStfId() + "," + name + "," + rmsd;
                    logger.info("[{}][{}] {}", explorerChain.getStfId(), name, line);
                    fileWriter.write(line + System.lineSeparator());
                    fileWriter.flush();
                }
                // cleanup
                for (Path tmpFile : tmpFiles) {
                    Files.delete(tmpFile);
                }
            } catch (IOException e) {
                throw new ComputationException(e);
            }
        }
    } catch (IOException | AlignmentException e) {
        throw new ComputationException(e);
    }
}

Example 5 with ExplorerChain

use of de.bioforscher.jstructure.si.explorer.ExplorerChain in project jstructure by JonStargaryen.

the class A04_CreateRmsdVsCoveragePlot method handleChain.

private static void handleChain(ExplorerChain explorerChain) {
    logger.info("handling chain {}", explorerChain.getStfId());
    try {
        Chain nativeChain = explorerChain.getChain();
        Path nativeChainPath = Files.createTempFile("nativechain-", ".pdb");
        Files.write(nativeChainPath, nativeChain.getPdbRepresentation().getBytes());
        ReconstructionContactMap nativeContactMap = ReconstructionContactMap.createReconstructionContactMap(nativeChain, ContactDefinitionFactory.createAlphaCarbonContactDefinition(8.0));
        List<AminoAcid> aminoAcids = nativeChain.getAminoAcids();
        List<Pair<AminoAcid, AminoAcid>> contacts = nativeContactMap.getLongRangeContacts();
        int numberNativeLongRangeContacts = contacts.size();
        List<ReconstructionContactMap> reconstructionContactMaps = new ArrayList<>();
        for (int coverage = 5; coverage <= 100; coverage = coverage + 5) {
            int numberOfContactsToSelect = (int) Math.round(0.01 * coverage * numberNativeLongRangeContacts);
            for (int run = 0; run < REDUNDANCY; run++) {
                Collections.shuffle(contacts);
                List<Pair<AminoAcid, AminoAcid>> selectedContacts = contacts.subList(0, numberOfContactsToSelect);
                ReconstructionContactMap contactMap = new ReconstructionContactMap(aminoAcids, selectedContacts, nativeContactMap.getContactDefinition());
                contactMap.setName("p" + coverage + "-" + (run + 1));
                reconstructionContactMaps.add(contactMap);
            }
        }
        Map<String, List<Future<ReconstructionResult>>> reconstructionFutures = new HashMap<>();
        for (ReconstructionContactMap contactMap : reconstructionContactMaps) {
            String name = contactMap.getName().split("-")[0];
            logger.info("handling contact map with coverage {}", name);
            if (!reconstructionFutures.containsKey(name)) {
                reconstructionFutures.put(name, new ArrayList<>());
            }
            List<Future<ReconstructionResult>> bin = reconstructionFutures.get(name);
            bin.add(executorService.submit(new ConfoldServiceWorker("/home/sb/programs/confold_v1.0/confold.pl", contactMap.getSequence(), contactMap.getSecondaryStructureElements(), contactMap.getCaspRRRepresentation(), nativeContactMap.getConfoldRRType())));
        }
        for (Map.Entry<String, List<Future<ReconstructionResult>>> reconstructionFuture : reconstructionFutures.entrySet()) {
            try {
                String name = reconstructionFuture.getKey();
                List<Chain> reconstructions = reconstructionFuture.getValue().stream().map(future -> {
                    try {
                        return future.get();
                    } catch (Exception e) {
                        throw new ComputationException(e);
                    }
                }).map(ReconstructionResult::getChains).flatMap(Collection::stream).collect(Collectors.toList());
                List<TMAlignAlignmentResult> alignmentResults = new ArrayList<>();
                List<Path> tmpFiles = new ArrayList<>();
                for (Chain reconstructedChain : reconstructions) {
                    Path reconstructPath = Files.createTempFile("confoldservice-recon", ".pdb");
                    tmpFiles.add(reconstructPath);
                    Files.write(reconstructPath, reconstructedChain.getPdbRepresentation().getBytes());
                    alignmentResults.add(TM_ALIGN_SERVICE.process(new String[] { "/home/sb/programs/tmalign", nativeChainPath.toFile().getAbsolutePath(), reconstructPath.toFile().getAbsolutePath() }));
                }
                if (alignmentResults.isEmpty()) {
                    throw new ComputationException("tmalign did not yield any alignments");
                }
                for (TMAlignAlignmentResult alignmentResult : alignmentResults) {
                    double rmsd = alignmentResult.getRootMeanSquareDeviation().getScore();
                    String line = explorerChain.getStfId() + "," + name.replace("p", "") + "," + rmsd;
                    logger.info(line);
                    fileWriter.write(line + System.lineSeparator());
                    fileWriter.flush();
                }
                // cleanup
                for (Path tmpFile : tmpFiles) {
                    Files.delete(tmpFile);
                }
            } catch (IOException e) {
                throw new ComputationException(e);
            }
        }
    } catch (IOException | AlignmentException e) {
        throw new ComputationException(e);
    }
}