Java SAMReadGroupRecord.setPlatformUnit方法代码示例

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
protected SAMReadGroupRecord createReadGroupRecord(
        String RGID, String RGLB, String RGPL, 
        String RGPU, String RGSM, String RGCN, 
        String RGDS, Iso8601Date RGDT, Integer RGPI) {
    SAMReadGroupRecord rg = new SAMReadGroupRecord(RGID);
    rg.setLibrary(RGLB);
    rg.setPlatform(RGPL);
    rg.setSample(RGSM);
    rg.setPlatformUnit(RGPU);
    if(RGCN != null)
        rg.setSequencingCenter(RGCN);
    if(RGDS != null)
        rg.setDescription(RGDS);
    if(RGDT != null)
        rg.setRunDate(RGDT);
    if(RGPI != null)
        rg.setPredictedMedianInsertSize(RGPI);
    return rg;
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
/** Creates a simple header with the values provided on the command line. */
public SAMFileHeader createSamFileHeader() {
    final SAMReadGroupRecord rgroup = new SAMReadGroupRecord(this.READ_GROUP_NAME);
    rgroup.setSample(this.SAMPLE_NAME);
    if (this.LIBRARY_NAME != null) rgroup.setLibrary(this.LIBRARY_NAME);
    if (this.PLATFORM != null) rgroup.setPlatform(this.PLATFORM);
    if (this.PLATFORM_UNIT != null) rgroup.setPlatformUnit(this.PLATFORM_UNIT);
    if (this.SEQUENCING_CENTER != null) rgroup.setSequencingCenter(SEQUENCING_CENTER);
    if (this.PREDICTED_INSERT_SIZE != null) rgroup.setPredictedMedianInsertSize(PREDICTED_INSERT_SIZE);
    if (this.DESCRIPTION != null) rgroup.setDescription(this.DESCRIPTION);
    if (this.RUN_DATE != null) rgroup.setRunDate(this.RUN_DATE);
    if (this.PLATFORM_MODEL != null) rgroup.setPlatformModel(this.PLATFORM_MODEL);
    if (this.PROGRAM_GROUP != null) rgroup.setProgramGroup(this.PROGRAM_GROUP);

    final SAMFileHeader header = new SAMFileHeader();
    header.addReadGroup(rgroup);

    for (final String comment : COMMENT) {
        header.addComment(comment);
    }

    header.setSortOrder(this.SORT_ORDER);
    return header ;
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
@DataProvider(name = "ReadsWithReadGroupData")
public Object[][] readsWithReadGroupData() {
    final SAMFileHeader header = ArtificialReadUtils.createArtificialSamHeader(2, 1, 1000000);
    final SAMReadGroupRecord readGroup = new SAMReadGroupRecord("FOO");
    readGroup.setPlatform("FOOPLATFORM");
    readGroup.setPlatformUnit("FOOPLATFORMUNIT");
    readGroup.setLibrary("FOOLIBRARY");
    readGroup.setSample("FOOSAMPLE");
    header.addReadGroup(readGroup);

    final GATKRead googleBackedRead = new GoogleGenomicsReadToGATKReadAdapter(ArtificialReadUtils.createArtificialGoogleGenomicsRead("google", "1", 5, new byte[]{'A', 'C', 'G', 'T'}, new byte[]{1, 2, 3, 4}, "4M"));
    googleBackedRead.setReadGroup("FOO");

    final GATKRead samBackedRead = new SAMRecordToGATKReadAdapter(ArtificialReadUtils.createArtificialSAMRecord(header, "sam", header.getSequenceIndex("1"), 5, new byte[]{'A', 'C', 'G', 'T'}, new byte[]{1, 2, 3, 4}, "4M"));
    samBackedRead.setReadGroup("FOO");

    return new Object[][] {
            { googleBackedRead, header, "FOO" },
            { samBackedRead, header, "FOO" }
    };
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
public static SAMReadGroupRecord transfer(ReadGroupInfo readGroupInfo) {
    SAMReadGroupRecord record = new SAMReadGroupRecord(readGroupInfo.id());
    record.setSample(readGroupInfo.sample());
    record.setLibrary(readGroupInfo.lib());
    record.setPlatform(readGroupInfo.platform());
    record.setPlatformUnit(readGroupInfo.platformUnit());
    return record;
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
/**
 * Gets a basic Read Group from the arguments.
 *
 * Note: the program group is set to {@link RTHelpConstants#PROGRAM_NAME}.
 *
 */
public SAMReadGroupRecord getReadGroupFromArguments(final String id, final String sampleName) {
    final SAMReadGroupRecord rg = new SAMReadGroupRecord(id);
    rg.setProgramGroup(RTHelpConstants.PROGRAM_NAME);
    rg.setSample(sampleName);
    // the program group is the one in the project properties
    rg.setProgramGroup(RTHelpConstants.PROGRAM_NAME);
    if (readGroupLibrary != null) {
        rg.setLibrary(readGroupLibrary);
    }
    if (readGroupPlatform != null) {
        rg.setPlatform(readGroupPlatform.toString());
    }
    if (readGroupPlatformUnit != null) {
        rg.setPlatformUnit(readGroupPlatformUnit);
    }
    if (readGroupSequencingCenter != null) {
        rg.setSequencingCenter(readGroupSequencingCenter);
    }
    if (readGroupRunDate != null) {
        rg.setRunDate(readGroupRunDate);
    }
    if (readGroupPredictedInsertSize != null) {
        rg.setPredictedMedianInsertSize(readGroupPredictedInsertSize);
    }
    if (readGroupPlatformModel != null) {
        rg.setPlatformModel(readGroupPlatformModel);
    }
    // return it
    return rg;
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
@Test
public void testGetReadGroupWithArguments() throws Exception {
    // setting expected RG
    final SAMReadGroupRecord expected = new SAMReadGroupRecord("RGID");
    expected.setProgramGroup("ReadTools");
    expected.setSample("sampleName");
    // setting args
    final ReadGroupArgumentCollection rgargs = new ReadGroupArgumentCollection();

    // starting setting params
    expected.setLibrary("LB");
    rgargs.readGroupLibrary = "LB";

    expected.setPlatform("ILLUMINA");
    rgargs.readGroupPlatform = SAMReadGroupRecord.PlatformValue.ILLUMINA;

    expected.setPlatformUnit("PU");
    rgargs.readGroupPlatformUnit = expected.getPlatformUnit();

    expected.setSequencingCenter("CN");
    rgargs.readGroupSequencingCenter = expected.getSequencingCenter();

    final Iso8601Date date = new Iso8601Date("2007-11-03");
    expected.setRunDate(date);
    rgargs.readGroupRunDate = date;

    expected.setPredictedMedianInsertSize(100);
    rgargs.readGroupPredictedInsertSize = expected.getPredictedMedianInsertSize();

    expected.setPlatformModel("PM");
    rgargs.readGroupPlatformModel = expected.getPlatformModel();

    // testing
    Assert.assertEquals(rgargs
                    .getReadGroupFromArguments(expected.getReadGroupId(), expected.getSample()),
            expected);
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
public void setupTest1(final int ID, final String readGroupId, final SAMReadGroupRecord readGroupRecord, final String sample,
                  final String library, final SAMFileHeader header, final SAMRecordSetBuilder setBuilder)
        throws IOException {

    final String separator = ":";
    final int contig1 = 0;
    final int contig2 = 1;
    readGroupRecord.setSample(sample);
    readGroupRecord.setPlatform(platform);
    readGroupRecord.setLibrary(library);
    readGroupRecord.setPlatformUnit(readGroupId);
    header.addReadGroup(readGroupRecord);
    setBuilder.setReadGroup(readGroupRecord);
    setBuilder.setUseNmFlag(true);

    setBuilder.setHeader(header);

    final int max = 800;
    final int min = 1;
    final Random rg = new Random(5);

    //add records that align to chrM and O but not N
    for (int i = 0; i < NUM_READS; i++) {
        final int start = rg.nextInt(max) + min;
        final String newReadName = READ_NAME + separator + ID + separator + i;

        if (i != NUM_READS - 1) {
            setBuilder.addPair(newReadName, contig1, start + ID, start + ID + LENGTH);
        } else {
            setBuilder.addPair(newReadName, contig2, start + ID, start + ID + LENGTH);
        }
    }
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
public void setupTest2(final int ID, final String readGroupId, final SAMReadGroupRecord readGroupRecord, final String sample,
                       final String library, final SAMFileHeader header, final SAMRecordSetBuilder setBuilder)
        throws IOException {

    final String separator = ":";
    final int contig1 = 0;
    final int contig2 = 1;
    final int contig3 = 2;
    readGroupRecord.setSample(sample);
    readGroupRecord.setPlatform(platform);
    readGroupRecord.setLibrary(library);
    readGroupRecord.setPlatformUnit(readGroupId);
    setBuilder.setReadGroup(readGroupRecord);
    setBuilder.setUseNmFlag(true);

    setBuilder.setHeader(header);

    final int max = 800;
    final int min = 1;
    final Random rg = new Random(5);

    //add records that align to all 3 chr in reference file
    for (int i = 0; i < NUM_READS; i++) {
        final int start = rg.nextInt(max) + min;
        final String newReadName = READ_NAME + separator + ID + separator + i;

        if (i<=NUM_READS/3) {
            setBuilder.addPair(newReadName, contig1, start + ID, start + ID + LENGTH);
        } else if (i< (NUM_READS - (NUM_READS/3))) {
            setBuilder.addPair(newReadName, contig2, start + ID, start + ID + LENGTH);
        } else {
            setBuilder.addPair(newReadName, contig3, start + ID, start + ID + LENGTH);
        }
    }
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
void setup(final int numReads,
           final String readName,
           final int ID,
           final String readGroupId,
           final SAMReadGroupRecord readGroupRecord,
           final String sample,
           final String library,
           final SAMFileHeader header,
           final SAMRecordSetBuilder setBuilder) throws IOException {

    final String separator = ":";
    readGroupRecord.setSample(sample);
    readGroupRecord.setPlatform(platform);
    readGroupRecord.setLibrary(library);
    readGroupRecord.setPlatformUnit(readGroupId);
    header.addReadGroup(readGroupRecord);
    setBuilder.setReadGroup(readGroupRecord);
    setBuilder.setUseNmFlag(true);

    setBuilder.setHeader(header);

    final int max = 15000;
    final int min = 1;
    final Random rg = new Random(5);

    for (int i = 0; i < numReads; i++) {
        final int start = rg.nextInt(max) + min;
        final String newReadName = readName + separator + ID + separator + i;
        setBuilder.addPair(newReadName, 0, start+ID, start+ID+99);
    }
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
@Test
public void testSingleRecord() {
    final String id = "MY.ID";
    final String expected = "SAMPLE.1";
    final ReadGroupCovariate covariate = new ReadGroupCovariate(new RecalibrationArgumentCollection(), Arrays.asList(expected));
    SAMReadGroupRecord rg = new SAMReadGroupRecord(id);
    rg.setPlatformUnit(expected);
    runTest(rg, expected, covariate);
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
@Test
public void testMaxValue() {
    final String id = "MY.ID";
    final String expected = "SAMPLE.1";
    final ReadGroupCovariate covariate = new ReadGroupCovariate(new RecalibrationArgumentCollection(), Arrays.asList(expected));
    SAMReadGroupRecord rg = new SAMReadGroupRecord(id);
    rg.setPlatformUnit(expected);
    Assert.assertEquals(covariate.maximumKeyValue(), 0);//there's just 1 read group, so 0 is the max value
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
protected int doWork() {
    IOUtil.assertInputIsValid(INPUT);
    IOUtil.assertFileIsWritable(OUTPUT);

    final SamReader in = SamReaderFactory.makeDefault()
        .referenceSequence(REFERENCE_SEQUENCE)
        .open(SamInputResource.of(INPUT));

    // create the read-group we'll be using
    final SAMReadGroupRecord rg = new SAMReadGroupRecord(RGID);
    rg.setLibrary(RGLB);
    rg.setPlatform(RGPL);
    rg.setSample(RGSM);
    rg.setPlatformUnit(RGPU);
    if (RGCN != null) rg.setSequencingCenter(RGCN);
    if (RGDS != null) rg.setDescription(RGDS);
    if (RGDT != null) rg.setRunDate(RGDT);
    if (RGPI != null) rg.setPredictedMedianInsertSize(RGPI);
    if (RGPG != null) rg.setProgramGroup(RGPG);
    if (RGPM != null) rg.setPlatformModel(RGPM);
    if (RGKS != null) rg.setKeySequence(RGKS);
    if (RGFO != null) rg.setFlowOrder(RGFO);

    log.info(String.format("Created read-group ID=%s PL=%s LB=%s SM=%s%n", rg.getId(), rg.getPlatform(), rg.getLibrary(), rg.getSample()));

    // create the new header and output file
    final SAMFileHeader inHeader = in.getFileHeader();
    final SAMFileHeader outHeader = inHeader.clone();
    outHeader.setReadGroups(Collections.singletonList(rg));
    if (SORT_ORDER != null) outHeader.setSortOrder(SORT_ORDER);

    final SAMFileWriter outWriter = new SAMFileWriterFactory().makeSAMOrBAMWriter(outHeader,
            outHeader.getSortOrder() == inHeader.getSortOrder(),
            OUTPUT);

    final ProgressLogger progress = new ProgressLogger(log);
    for (final SAMRecord read : in) {
        read.setAttribute(SAMTag.RG.name(), RGID);
        outWriter.addAlignment(read);
        progress.record(read);
    }

    // cleanup
    CloserUtil.close(in);
    outWriter.close();
    return 0;
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
@BeforeTest
void setupBuilder() throws IOException {
    final String readName = "TESTBARCODE";

    //Create Sam Files
    tempSamFile = File.createTempFile("CollectTargetedMetrics", ".bam", TEST_DIR);
    tempSamFileIndex = new File(tempSamFile.toString().replaceAll("\\.bam$",".bai"));
    final File tempSamFileUnsorted = File.createTempFile("CollectTargetedMetrics", ".bam", TEST_DIR);
    tempSamFileUnsorted.deleteOnExit();
    tempSamFile.deleteOnExit();
    tempSamFileIndex.deleteOnExit();
    final SAMFileHeader header = new SAMFileHeader();

    //Check that dictionary file is readable and then set header dictionary
    try {
        header.setSequenceDictionary(SAMSequenceDictionaryExtractor.extractDictionary(dict.toPath()));
        header.setSortOrder(SAMFileHeader.SortOrder.unsorted);
    } catch (final SAMException e) {
        e.printStackTrace();
    }

    //Set readGroupRecord
    final SAMReadGroupRecord readGroupRecord = new SAMReadGroupRecord(readGroupId);
    readGroupRecord.setSample(sample);
    readGroupRecord.setPlatform(platform);
    readGroupRecord.setLibrary(library);
    readGroupRecord.setPlatformUnit(readGroupId);
    header.addReadGroup(readGroupRecord);

    //Add to setBuilder
    final SAMRecordSetBuilder setBuilder = new SAMRecordSetBuilder(true, SAMFileHeader.SortOrder.coordinate);
    setBuilder.setReadGroup(readGroupRecord);
    setBuilder.setUseNmFlag(true);
    setBuilder.setHeader(header);

    //Read settings
    final String separator = ":";
    final int ID = 1;
    final int max = 15000;
    final int min = 1;
    final Random rg = new Random(5);

    for (int i = 0; i < numReads; i++) {
        final int start = rg.nextInt(max) + min;
        final String newReadName = readName + separator + ID + separator + i;
        setBuilder.addPair(newReadName, 0, start + ID, start + ID + LENGTH);
    }

    //Write SAM file
    final SAMFileWriter writer = new SAMFileWriterFactory()
            .setCreateIndex(true).makeBAMWriter(header, false, tempSamFileUnsorted);

    for (final SAMRecord record : setBuilder) {
        writer.addAlignment(record);
    }
    writer.close();

    //sort the temp file
    final SortSam sorter = new SortSam();
    final String[] args = new String[]{
            "INPUT=" + tempSamFileUnsorted.getAbsolutePath(),
            "OUTPUT=" + tempSamFile.getAbsolutePath(),
            "SORT_ORDER=coordinate"
    };

    sorter.instanceMain(args);

    //create output files for tests
    outfile = File.createTempFile("test", ".TargetedMetrics_Coverage");
    perTargetOutfile = File.createTempFile("perTarget", ".perTargetCoverage");
    outfile.deleteOnExit();
    perTargetOutfile.deleteOnExit();
}
 

import htsjdk.samtools.SAMReadGroupRecord; //导入方法依赖的package包/类
@BeforeTest
void setupBuilder() throws IOException {
    final String readName = "TESTBARCODE";

    //Create Sam Files
    tempSamFile = File.createTempFile("CollectWgsMetrics", ".bam", TEST_DIR);
    final File tempSamIndex = new File(tempSamFile.getAbsolutePath().replace("bam", "bai"));
    final File tempSamFileUnsorted = File.createTempFile("CollectWgsMetrics", ".bam", TEST_DIR);
    tempSamFileUnsorted.deleteOnExit();
    tempSamIndex.deleteOnExit();
    tempSamFile.deleteOnExit();
    final File sortedSamIdx = new File(TEST_DIR, tempSamFile.getName() + ".idx");
    sortedSamIdx.deleteOnExit();

    final SAMFileHeader header = new SAMFileHeader();

    //Check that dictionary file is readable and then set header dictionary
    try {
        header.setSequenceDictionary(SAMSequenceDictionaryExtractor.extractDictionary(referenceDict.toPath()));
        header.setSortOrder(SAMFileHeader.SortOrder.unsorted);
    } catch (final SAMException e) {
        e.printStackTrace();
    }

    //Set readGroupRecord
    final SAMReadGroupRecord readGroupRecord = new SAMReadGroupRecord(READ_GROUP_ID);
    readGroupRecord.setSample(SAMPLE);
    readGroupRecord.setPlatform(PLATFORM);
    readGroupRecord.setLibrary(LIBRARY);
    readGroupRecord.setPlatformUnit(READ_GROUP_ID);
    header.addReadGroup(readGroupRecord);

    //Add to setBuilder
    final SAMRecordSetBuilder setBuilder = new SAMRecordSetBuilder(true, SAMFileHeader.SortOrder.coordinate);
    setBuilder.setReadGroup(readGroupRecord);
    setBuilder.setUseNmFlag(true);
    setBuilder.setHeader(header);

    //Read settings
    final String separator = ":";
    final int ID = 1;
    final int maxReadStart = 800;
    final int minReadStart = 1;
    final Random rg = new Random(5);

    for (int i = 0; i < NUM_READS; i++) {
        final int start = rg.nextInt(maxReadStart) + minReadStart;
        final String newReadName = readName + separator + ID + separator + i;
        setBuilder.addPair(newReadName, 0, start + ID, start + ID + READ_PAIR_DISTANCE);
    }

    //Write SAM file
    final SAMFileWriter writer = new SAMFileWriterFactory()
            .setCreateIndex(true).makeBAMWriter(header, false, tempSamFileUnsorted);

    for (final SAMRecord record : setBuilder) {
        writer.addAlignment(record);
    }
    writer.close();

    //sort the temp file
    final SortSam sorter = new SortSam();
    final String[] args = new String[]{
            "INPUT=" + tempSamFileUnsorted.getAbsolutePath(),
            "OUTPUT=" + tempSamFile.getAbsolutePath(),
            "SORT_ORDER=coordinate"
    };

    sorter.instanceMain(args);

    //create output files for tests
    outfile = File.createTempFile("testWgsMetrics", ".txt", TEST_DIR);
    outfile.deleteOnExit();
}