Class DiveModule

lit_nlp/client/modules/dive_module.ts:129–754

 */
@customElement('dive-module')
export class DiveModule extends LitModule {
  // Compatible types are those for which the GroupService can create bins.
  static compatibleTypes = [BooleanLitType, CategoryLabel, Scalar];
  static override title = 'Dive';
  static override numCols = 3;
  static override shouldDisplayModule =
      (modelSpecs: ModelInfoMap, datasetSpec: Spec) => {
        // Dive is a two-dimensional matrix of histograms, so we should have at
        // least two binnable fields in order to be anlaytically useful.
        const compatiblefields: string[] = [];
        compatiblefields.push(
            ...findSpecKeys(datasetSpec, DiveModule.compatibleTypes));
        for (const {spec} of Object.values(modelSpecs)) {
          const {input, output} = spec;
          compatiblefields.push(
              ...findSpecKeys(input, DiveModule.compatibleTypes));
          compatiblefields.push(
              ...findSpecKeys(output, DiveModule.compatibleTypes));
        }
        return new Set(compatiblefields).size > 1;
      };

  static override get styles() {
    const styles = css`
      .scene {
        height: 100%;
        width: 100%;
        cursor: crosshair;
      }

      select.limit-width {
        width: 150px;
        max-width: 150px;
      }

      .dive-controls {
        width: 100%;
        height: 100%;
        display: flex;
        flex-direction: row;
        align-items: center;
        justify-content: space-between;
      }

      .dive-controls .feature-selectors {
        display: flex;
        flex-direction: row;
      }

      .dive-controls .icon-button {
        font-size: 16px;
      }
    `;

    return [sharedStyles, styles];
  }

  static override template =