|
14 | 14 | import htsjdk.variant.variantcontext.writer.Options; |
15 | 15 | import htsjdk.variant.variantcontext.writer.VariantContextWriter; |
16 | 16 | import htsjdk.variant.variantcontext.writer.VariantContextWriterBuilder; |
| 17 | +import htsjdk.variant.vcf.VCFConstants; |
17 | 18 | import htsjdk.variant.vcf.VCFHeader; |
| 19 | +import htsjdk.variant.vcf.VCFStandardHeaderLines; |
18 | 20 | import org.apache.commons.lang3.StringUtils; |
19 | 21 | import org.json.JSONObject; |
20 | 22 | import org.labkey.api.pipeline.PipelineJobException; |
@@ -458,6 +460,9 @@ public static File createVcf(File pindelOutput, File vcfOutput, ReferenceGenome |
458 | 460 | VCFHeader header = new VCFHeader(); |
459 | 461 | header.setSequenceDictionary(dict); |
460 | 462 | LofreqAnalysis.addMetaLines(header); |
| 463 | + header.addMetaDataLine(VCFStandardHeaderLines.getInfoLine(VCFConstants.ALLELE_FREQUENCY_KEY)); |
| 464 | + header.addMetaDataLine(VCFStandardHeaderLines.getInfoLine(VCFConstants.DEPTH_KEY)); |
| 465 | + |
461 | 466 | writer.writeHeader(header); |
462 | 467 |
|
463 | 468 | String[] line; |
@@ -507,13 +512,13 @@ public static File createVcf(File pindelOutput, File vcfOutput, ReferenceGenome |
507 | 512 | vcb.stop(end); |
508 | 513 | vcb.chr(line[1]); |
509 | 514 | vcb.alleles(Arrays.asList(Allele.create(refAllele, true), Allele.create(altAllele))); |
510 | | - vcb.attribute("AF", Double.parseDouble(line[6])); |
| 515 | + vcb.attribute(VCFConstants.ALLELE_FREQUENCY_KEY, Double.parseDouble(line[6])); |
511 | 516 |
|
512 | 517 | vcb.attribute("IN_CONSENSUS", 1); |
513 | 518 | vcb.attribute("GATK_DP", (int)Double.parseDouble(line[7])); |
514 | 519 |
|
515 | 520 | int dp = "I".equals(line[0]) ? Integer.parseInt(line[4]) : (int) Double.parseDouble(line[10]); |
516 | | - vcb.attribute("DP", dp); |
| 521 | + vcb.attribute(VCFConstants.DEPTH_KEY, dp); |
517 | 522 |
|
518 | 523 | writer.add(vcb.make()); |
519 | 524 | } |
|
0 commit comments