tanghaibao / goatools

        """Set colors of GO terms."""

        goid2color = {}

        # 1. colors based on p-value override colors based on source GO

        if self.go2res is not None:

            alpha2col = self.pltvars.alpha2col

            pval_name = self.pval_name

            for goid, res in self.go2res.items():

                pval = getattr(res, pval_name, None)

                if pval is not None:

                    for alpha, color in alpha2col.items():

                        if pval <= alpha and res.study_count != 0:

                            if goid not in goid2color:

                                goid2color[goid] = color

        # 2. GO source color

        color = self.pltvars.key2col['go_sources']

        for goid in self.godag.go_sources:

    def set_goid2color_pval(self, goid2color):

        """Fill missing colors based on p-value of an enriched GO term."""

        alpha2col = self.alpha2col

        if self.pval_name is not None:

            pval_name = self.pval_name

            for goid, res in self.go2res.items():

                pval = getattr(res, pval_name, None)

                if pval is not None:

                    for alpha, color in alpha2col.items():

                        if pval <= alpha and res.study_count != 0:

                            if goid not in goid2color:

                                goid2color[goid] = color

    def get_goid2color_pval(self):

        """Return a go2color dict containing GO colors determined by P-value."""