#!/usr/bin/env python3
# -*- coding: utf-8 -*-

from .statistic import StatisticHistogram
import singlecellmultiomics.pyutils as pyutils


class TrimmingStats(StatisticHistogram):
    def __init__(self, args):
        StatisticHistogram.__init__(self, args)
        self.totalFragmentsTrimmed = 0

    def processRead(self, R1,R2=None):

        for read in [R1,R2]:
            if read is None:
                continue

            if read.has_tag('a1') or read.has_tag(
                    'eB') or read.has_tag('A2') or read.has_tag('EB'):
                self.totalFragmentsTrimmed += 1

    def __repr__(self):
        return f'Trimmed fragments: {self.totalFragmentsTrimmed}'

    def __iter__(self):
        yield 'Trimmed fragments', self.totalFragmentsTrimmed