zeppelin-web-angular/src/app/visualizations/bar-chart/bar-chart-visualization.component.ts

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *     http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import {
  AfterViewInit,
  ChangeDetectionStrategy,
  ChangeDetectorRef,
  Component,
  ElementRef,
  Inject,
  OnInit,
  ViewChild
} from '@angular/core';
import * as G2 from '@antv/g2';

import { get } from 'lodash';

import { GraphConfig, VisualizationMultiBarChart } from '@zeppelin/sdk';
import { G2VisualizationComponentBase, Visualization, VISUALIZATION } from '@zeppelin/visualization';

import { VisualizationPivotSettingComponent } from '../common/pivot-setting/pivot-setting.component';
import { calcTickCount } from '../common/util/calc-tick-count';
import { setChartXAxis } from '../common/util/set-x-axis';
import { VisualizationXAxisSettingComponent } from '../common/x-axis-setting/x-axis-setting.component';

@Component({
  selector: 'zeppelin-bar-chart-visualization',
  templateUrl: './bar-chart-visualization.component.html',
  styleUrls: ['./bar-chart-visualization.component.less'],
  changeDetection: ChangeDetectionStrategy.OnPush
})
export class BarChartVisualizationComponent extends G2VisualizationComponentBase implements OnInit, AfterViewInit {
  @ViewChild('container', { static: false }) container!: ElementRef<HTMLDivElement>;
  @ViewChild(VisualizationXAxisSettingComponent, { static: false })
  xAxisSettingComponent!: VisualizationXAxisSettingComponent;
  @ViewChild(VisualizationPivotSettingComponent, { static: false })
  pivotSettingComponent!: VisualizationPivotSettingComponent;
  stacked = false;

  viewChange() {
    if (!this.config) {
      throw new Error('config is not defined');
    }
    if (!this.config.setting.multiBarChart) {
      this.config.setting.multiBarChart = new VisualizationMultiBarChart();
    }
    this.config.setting.multiBarChart.stacked = this.stacked;
    if (!this.visualization.configChange$) {
      throw new Error('visualization.configChange$ is not defined');
    }
    this.visualization.configChange$.next(this.config);
  }

  constructor(
    @Inject(VISUALIZATION) public visualization: Visualization,
    private cdr: ChangeDetectorRef
  ) {
    super(visualization);
  }

  ngOnInit() {}

  ngAfterViewInit() {
    this.render();
  }

  refreshSetting(config: GraphConfig) {
    this.stacked = get(config.setting, 'multiBarChart.stacked', false);
    this.pivotSettingComponent.init();
    this.xAxisSettingComponent.init();
    this.cdr.markForCheck();
  }

  setScale(chart: G2.Chart) {
    const key = this.getKey();
    const tickCount = calcTickCount(this.container.nativeElement);
    chart.scale(key, {
      tickCount,
      type: 'cat'
    });
  }

  renderBefore(config: GraphConfig, chart: G2.Chart) {
    const key = this.getKey();
    this.setScale(chart);

    chart.tooltip({
      shared: false
    });
    if (get(config.setting, 'multiBarChart.stacked', false)) {
      chart.intervalStack().position(`${key}*__value__`).color('__key__').opacity(1);
    } else {
      chart
        .interval()
        .position(`${key}*__value__`)
        .color('__key__')
        .opacity(1)
        .adjust([
          {
            type: 'dodge',
            marginRatio: 0
          }
        ]);
    }
    setChartXAxis(this.visualization, 'multiBarChart', chart, key);
  }
}