Source code for cytoflowgui.workflow.views.kde_2d

#!/usr/bin/env python3.8
# coding: latin-1

# (c) Massachusetts Institute of Technology 2015-2018
# (c) Brian Teague 2018-2022
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 2 of the License, or
# (at your option) any later version.
# 
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
# 
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

"""
cytoflowgui.workflow.views.kde_2d
---------------------------------

"""

from textwrap import dedent

from traits.api import provides, Bool, Instance, Enum

from cytoflow import Kde2DView
import cytoflow.utility as util

from cytoflowgui.workflow.serialization import camel_registry, traits_repr, traits_str
from .view_base import IWorkflowView, WorkflowFacetView, Data2DPlotParams

Kde2DView.__repr__ = traits_repr

[docs]class Kde2DPlotParams(Data2DPlotParams): shade = Bool(False) min_alpha = util.PositiveCFloat(0.2, allow_zero = False) max_alpha = util.PositiveCFloat(0.9, allow_zero = False) n_levels = util.PositiveCInt(10, allow_zero = False) bw = Enum(['scott', 'silverman']) gridsize = util.PositiveCInt(50, allow_zero = False)
[docs]@provides(IWorkflowView) class Kde2DWorkflowView(WorkflowFacetView, Kde2DView): plot_params = Instance(Kde2DPlotParams, ())
[docs] def get_notebook_code(self, idx): view = Kde2DView() view.copy_traits(self, view.copyable_trait_names()) plot_params_str = traits_str(self.plot_params) return dedent(""" {repr}.plot(ex_{idx}{plot}{plot_params}) """ .format(repr = repr(view), idx = idx, plot = ", plot_name = " + repr(self.current_plot) if self.current_plot else "", plot_params = ", " + plot_params_str if plot_params_str else ""))
### Serialization @camel_registry.dumper(Kde2DWorkflowView, 'kde-2d', version = 2) def _dump(view): return dict(xchannel = view.xchannel, xscale = view.xscale, ychannel = view.ychannel, yscale = view.yscale, xfacet = view.xfacet, yfacet = view.yfacet, huefacet = view.huefacet, huescale = view.huescale, plotfacet = view.plotfacet, subset_list = view.subset_list, plot_params = view.plot_params, current_plot = view.current_plot) @camel_registry.dumper(Kde2DWorkflowView, 'kde-2d', version = 1) def _dump_v1(view): return dict(xchannel = view.xchannel, xscale = view.xscale, ychannel = view.ychannel, yscale = view.yscale, xfacet = view.xfacet, yfacet = view.yfacet, huefacet = view.huefacet, huescale = view.huescale, plotfacet = view.plot_facet, subset_list = view.subset_list) @camel_registry.dumper(Kde2DPlotParams, 'kde-2d-params', version = 1) def _dump_params(params): return dict( # BasePlotParams title = params.title, xlabel = params.xlabel, ylabel = params.ylabel, huelabel = params.huelabel, col_wrap = params.col_wrap, sns_style = params.sns_style, sns_context = params.sns_context, legend = params.legend, sharex = params.sharex, sharey = params.sharey, despine = params.despine, # DataplotParams min_quantile = params.min_quantile, max_quantile = params.max_quantile, # Data2DPlotParams xlim = params.xlim, ylim = params.ylim, # 2D KDE params shade = params.shade, min_alpha = params.min_alpha, max_alpha = params.max_alpha, n_levels = params.n_levels, bw = params.bw, gridsize = params.gridsize) @camel_registry.loader('kde-2d', version = any) def _load(data, version): return Kde2DWorkflowView(**data) @camel_registry.loader('kde-2d-params', version = any) def _load_params(data, version): return Kde2DPlotParams(**data)