UndoableChangeTripletFeel.java

package org.herac.tuxguitar.gui.undo.undoables.custom;

import java.util.ArrayList;
import java.util.List;

import org.herac.tuxguitar.gui.TuxGuitar;
import org.herac.tuxguitar.gui.editors.tab.Caret;
import org.herac.tuxguitar.gui.editors.tab.TGMeasureImpl;
import org.herac.tuxguitar.gui.undo.CannotRedoException;
import org.herac.tuxguitar.gui.undo.CannotUndoException;
import org.herac.tuxguitar.gui.undo.UndoableEdit;
import org.herac.tuxguitar.gui.undo.undoables.UndoableCaretHelper;
import org.herac.tuxguitar.song.models.TGMeasure;

public class UndoableChangeTripletFeel implements UndoableEdit {
  private class TripletFeelPosition {
    private long position;
    private int tripletFeel;

    public TripletFeelPosition(long position, int tripletFeel) {
      this.position = position;
      this.tripletFeel = tripletFeel;
    }

    public long getPosition() {
      return this.position;
    }

    public int getTripletFeel() {
      return this.tripletFeel;
    }
  }

  private static Caret getCaret() {
    return TuxGuitar.instance().getTablatureEditor().getTablature().getCaret();
  }

  public static UndoableChangeTripletFeel startUndo() {
    UndoableChangeTripletFeel undoable = new UndoableChangeTripletFeel();
    Caret caret = getCaret();
    undoable.doAction = UNDO_ACTION;
    undoable.undoCaret = new UndoableCaretHelper();
    undoable.position = caret.getPosition();
    undoable.undoableTripletFeel = caret.getMeasure().getTripletFeel();
    undoable.nextTripletFeelPositions = new ArrayList<TripletFeelPosition>();

    int prevTripletFeel = undoable.undoableTripletFeel;
    for (final TGMeasure meas : TuxGuitar.instance().getSongManager()
        .getFirstTrack().getMeasures()) {
      TGMeasureImpl measure = (TGMeasureImpl) meas;
      if (measure.getStart() > undoable.position) {
        int currTripletFeel = measure.getTripletFeel();
        if (prevTripletFeel != currTripletFeel) {
          TripletFeelPosition tfp = undoable.new TripletFeelPosition(measure
              .getStart(), currTripletFeel);
          undoable.nextTripletFeelPositions.add(tfp);
        }
        prevTripletFeel = currTripletFeel;
      }
    }
    return undoable;
  }

  private int doAction;
  private List<TripletFeelPosition> nextTripletFeelPositions;
  private long position;
  private int redoableTripletFeel;
  private UndoableCaretHelper redoCaret;

  private boolean toEnd;

  private int undoableTripletFeel;

  private UndoableCaretHelper undoCaret;

  private UndoableChangeTripletFeel() {
    super();
  }

  public boolean canRedo() {
    return (this.doAction == REDO_ACTION);
  }

  public boolean canUndo() {
    return (this.doAction == UNDO_ACTION);
  }

  public UndoableChangeTripletFeel endUndo(int tripletFeel, boolean toEnd) {
    this.redoCaret = new UndoableCaretHelper();
    this.redoableTripletFeel = tripletFeel;
    this.toEnd = toEnd;
    return this;
  }

  public void redo() throws CannotRedoException {
    if (!canRedo()) {
      throw new CannotRedoException();
    }
    TuxGuitar.instance().getSongManager().changeTripletFeel(this.position,
        this.redoableTripletFeel, this.toEnd);
    TuxGuitar.instance().fireUpdate();
    this.redoCaret.update();

    this.doAction = UNDO_ACTION;
  }

  public void undo() throws CannotUndoException {
    if (!canUndo()) {
      throw new CannotUndoException();
    }
    TuxGuitar.instance().getSongManager().changeTripletFeel(this.position,
        this.undoableTripletFeel, this.toEnd);
    if (this.toEnd) {

      for (final TripletFeelPosition tfp : this.nextTripletFeelPositions) {
        TuxGuitar.instance().getSongManager().changeTripletFeel(
            tfp.getPosition(), tfp.getTripletFeel(), true);
      }
    }
    TuxGuitar.instance().fireUpdate();
    this.undoCaret.update();

    this.doAction = REDO_ACTION;
  }
}