LibrePCB

/*
 * LibrePCB - Professional EDA for everyone!
 * Copyright (C) 2013 LibrePCB Developers, see AUTHORS.md for contributors.
 * https://librepcb.org/
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

/*******************************************************************************
 *  Includes
 ******************************************************************************/
#include "cmdboardnetsegmentaddelements.h"

#include <librepcb/core/project/board/items/bi_netline.h>
#include <librepcb/core/project/board/items/bi_netpoint.h>
#include <librepcb/core/project/board/items/bi_netsegment.h>

#include <QtCore>

/*******************************************************************************
 *  Namespace
 ******************************************************************************/
namespace librepcb {
namespace editor {

/*******************************************************************************
 *  Constructors / Destructor
 ******************************************************************************/

CmdBoardNetSegmentAddElements::CmdBoardNetSegmentAddElements(
    BI_NetSegment& segment) noexcept
  : UndoCommand(tr("Add Vias/Pads/Traces")), mNetSegment(segment) {
}

CmdBoardNetSegmentAddElements::~CmdBoardNetSegmentAddElements() noexcept {
}

/*******************************************************************************
 *  General Methods
 ******************************************************************************/

BI_Pad* CmdBoardNetSegmentAddElements::addPad(BI_Pad& pad) {
  mPads.append(&pad);
  return &pad;
}

BI_Pad* CmdBoardNetSegmentAddElements::addPad(const BoardPadData& pad) {
  BI_Pad* p = new BI_Pad(mNetSegment, pad);
  return addPad(*p);
}

BI_Via* CmdBoardNetSegmentAddElements::addVia(BI_Via& via) {
  mVias.append(&via);
  return &via;
}

BI_Via* CmdBoardNetSegmentAddElements::addVia(const Via& via) {
  BI_Via* v = new BI_Via(mNetSegment, via);
  return addVia(*v);
}

BI_NetPoint* CmdBoardNetSegmentAddElements::addNetPoint(BI_NetPoint& netpoint) {
  mNetPoints.append(&netpoint);
  return &netpoint;
}

BI_NetPoint* CmdBoardNetSegmentAddElements::addNetPoint(const Point& position) {
  BI_NetPoint* netpoint = new BI_NetPoint(mNetSegment, Uuid::createRandom(),
                                          position);  // can throw
  return addNetPoint(*netpoint);
}

BI_NetLine* CmdBoardNetSegmentAddElements::addNetLine(BI_NetLine& netline) {
  mNetLines.append(&netline);
  return &netline;
}

BI_NetLine* CmdBoardNetSegmentAddElements::addNetLine(
    BI_NetLineAnchor& a, BI_NetLineAnchor& b, const Layer& layer,
    const PositiveLength& width) {
  BI_NetLine* netline =
      new BI_NetLine(mNetSegment, Uuid::createRandom(), a, b, layer,
                     width);  // can throw
  return addNetLine(*netline);
}

/*******************************************************************************
 *  Inherited from UndoCommand
 ******************************************************************************/

bool CmdBoardNetSegmentAddElements::performExecute() {
  performRedo();  // can throw

  return true;
}

void CmdBoardNetSegmentAddElements::performUndo() {
  mNetSegment.removeElements(mPads, mVias, mNetPoints, mNetLines);  // can throw
}

void CmdBoardNetSegmentAddElements::performRedo() {
  mNetSegment.addElements(mPads, mVias, mNetPoints, mNetLines);  // can throw
}

/*******************************************************************************
 *  End of File
 ******************************************************************************/

}  // namespace editor
}  // namespace librepcb