diff --git a/svreader/annotation.py b/svreader/annotation.py index c1aa1cf11170239ecbe6bb856e62c46048f0c4d2..1197d1431f4d542e1b3ebf87ab3d9f0ca365f4ff 100644 --- a/svreader/annotation.py +++ b/svreader/annotation.py @@ -145,6 +145,7 @@ class AnnotateRecord(VCFRecord): def add_supporting_infos(self): supp_reads = self.variant_read_support() num_supp_samples = self.num_variant_samples() + #print(supp_reads, num_supp_samples) try: self.record.info['MAX_SUPP_READS'] = supp_reads self.record.info['NUM_SUPP_SAMPLES'] = num_supp_samples diff --git a/svreader/lumpy.py b/svreader/lumpy.py index 618f6a402b2e640658c50a364749a411ea001558..20d2d6e87e8097ac9c4fb012d0b4fca8a6762893 100644 --- a/svreader/lumpy.py +++ b/svreader/lumpy.py @@ -181,10 +181,15 @@ class LumpyReader(SVReader): def AreSamplesSpecified(self): return 1 + # def getOrderedSamples(self): + # if not hasattr(self.vcf_reader, "samples"): + # return [] + # return self.vcf_reader.samples + def getOrderedSamples(self): - if not hasattr(self.vcf_reader, "samples"): - return [] - return self.vcf_reader.samples + samples = self.vcf_reader.header.samples + sample_names = [sample.rsplit('.')[0] for sample in samples] + return sample_names def SpecificFilterPass(self, record): # First try after sensibility analysis with Mathieu @@ -219,4 +224,5 @@ class LumpyWriter(SVWriter): if self._isopen: self.vcf_writer.close() else: # nothing was written + # Make sure appropriate samples are added to the vcf self._dumpemptyvcf()