def finalize_chart(self, chart: Chart) -> Chart:
# ok, horizontal bar charts are confusing because by default, cvz bar makes a COLUMN chart
# TK: make cvz column type
if self.is_horizontal:
w = chart.get_prop("width")
h = chart.get_prop("height")
chart.set_props({"width": h, "height": w})
return chart
def bchart(mydata, mychannels):
c = Chart(
viz_name=VIZ_NAME,
data=mydata,
channels=mychannels,
defaults=Vizkit.chart_defaults(),
)
return c
def setup_viz(options={}) -> Chart:
opts = SETUP_OPTS.copy()
opts.update(options)
dataframe = pd.read_csv(INPUT_PATH)
channels = ChannelGroup(options=opts,
data=dataframe,
color_channel_name="stroke")
return Chart(
viz_name=MOCK_COMMAND_NAME,
channels=channels,
data=dataframe,
defaults=MOCK_DEFAULTS,
options=opts,
)
def init_chart(self) -> Chart:
"""
instantiate and create a Chart object
prereqs:
- self.channels has been set up
- self.chart_defaults() is defined
Note: `options` is still just a sloppy grabbag from the command-line arg parser, consider
refactoring Chart to prevent leaky abstraction
"""
x = Chart(
viz_name=self.viz_commandname,
data=self.df,
channels=self.channels,
defaults=self.chart_defaults(),
options=self.options,
)
return x