Class WeightServiceManager

browser/src/service/WeightServiceManager.tsx:8–37

/** Manages currently selected weight service. */
export default class WeightServiceManager {

  private weightServices: { [key: string]: WeightService };
  private weightService: WeightService;

  constructor(graphManager: GraphManager) {
    // Add bindings count as the default service.
    this.weightServices = {};
    this.weightService = this.weightServices['bindings'] = new BindingsService(graphManager);

    // TODO: Separate services into a config file.
    if (graphManager.classInfo && Object.keys(graphManager.classInfo).length) { 
      const classSizeService = new ClassSizeService(graphManager.classInfo);
      this.weightServices['class memory'] = classSizeService;
      this.weightServices['graph memory'] = new GraphSizeService(graphManager, classSizeService);
    }
  }

  getWeightServiceNames(): Array<string> {
    return Object.keys(this.weightServices).sort();
  }

  getWeightService(): WeightService {
    return this.weightService;
  }

  selectWeightService(weight: string) {
    this.weightService = this.weightServices[weight];
  }
}