IpnbCodeCell.java

package org.jetbrains.plugins.ipnb.format.cells;

import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import org.jetbrains.plugins.ipnb.format.cells.output.IpnbOutputCell;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class IpnbCodeCell extends IpnbEditableCell {
  @NotNull private final String myLanguage;
  @Nullable private Integer myPromptNumber;
  @NotNull private final List<IpnbOutputCell> myCellOutputs;

  public IpnbCodeCell(@NotNull final String language,
                      @NotNull final List<String> input,
                      @Nullable final Integer number,
                      @NotNull final List<IpnbOutputCell> cellOutputs,
                      @NotNull Map<String, Object> metadata) {
    super(input, metadata);
    myLanguage = language;
    myPromptNumber = number;
    myCellOutputs = cellOutputs;
  }

  @NotNull
  public String getLanguage() {
    return myLanguage;
  }

  @Nullable
  public Integer getPromptNumber() {
    return myPromptNumber;
  }

  public void setPromptNumber(@Nullable Integer promptNumber) {
    myPromptNumber = promptNumber;
  }

  @NotNull
  public List<IpnbOutputCell> getCellOutputs() {
    return myCellOutputs;
  }

  public void removeCellOutputs() {
    myCellOutputs.clear();
  }

  public void addCellOutput(@NotNull final IpnbOutputCell cellOutput) {
    myCellOutputs.add(cellOutput);
  }

  @SuppressWarnings({"CloneDoesntCallSuperClone", "CloneDoesntDeclareCloneNotSupportedException"})
  @Override
  public Object clone() {
    return new IpnbCodeCell(myLanguage, new ArrayList<>(getSource()), myPromptNumber, new ArrayList<>(myCellOutputs),
                            myMetadata);
  }

  @NotNull
  public static IpnbCodeCell createEmptyCodeCell() {
    return new IpnbCodeCell("python", new ArrayList<>(), null, new ArrayList<>(),
                            new HashMap<>());
  }
}