Adds feature to merge Instrument Track patterns

include/Pattern.h

@@ -176,6 +176,7 @@ class PatternView : public TrackContentObjectView
 	void setMutedNoteBorderColor(QColor const & color) { m_mutedNoteBorderColor = color; }
 
 public slots:
+	Pattern* getPattern();
 	void update() override;
 
 

include/TrackContentObjectView.h

@@ -111,6 +111,10 @@ class TrackContentObjectView : public selectableObject, public ModelView
 	// some metadata to be written to the clipboard.
 	static void remove( QVector<TrackContentObjectView *> tcovs );
 	static void toggleMute( QVector<TrackContentObjectView *> tcovs );
+	static void mergeTCOs(QVector<TrackContentObjectView*> tcovs);
+
+	// Returns true if selection can be merged and false if not
+	static bool canMergeSelection(QVector<TrackContentObjectView*> tcovs);
 
 	QColor getColorForDisplay( QColor );
 
@@ -129,7 +133,8 @@ public slots:
 		Cut,
 		Copy,
 		Paste,
-		Mute
+		Mute,
+		Merge
 	};
 
 	virtual void constructContextMenu( QMenu * )

src/gui/TrackContentObjectView.cpp

@@ -33,9 +33,14 @@
 #include "ColorChooser.h"
 #include "ComboBoxModel.h"
 #include "DataFile.h"
+#include "Engine.h"
 #include "embed.h"
 #include "GuiApplication.h"
+#include "InstrumentTrack.h"
+#include "Note.h"
+#include "Pattern.h"
 #include "SampleTrack.h"
+#include "Song.h"
 #include "SongEditor.h"
 #include "StringPairDrag.h"
 #include "TextFloat.h"
@@ -937,9 +942,11 @@ void TrackContentObjectView::mouseReleaseEvent( QMouseEvent * me )
  */
 void TrackContentObjectView::contextMenuEvent( QContextMenuEvent * cme )
 {
+	QVector<TrackContentObjectView*> selectedTCOs = getClickedTCOs();
+
 	// Depending on whether we right-clicked a selection or an individual TCO we will have
 	// different labels for the actions.
-	bool individualTCO = getClickedTCOs().size() <= 1;
+	bool individualTCO = selectedTCOs.size() <= 1;
 
 	if( cme->modifiers() )
 	{
@@ -948,6 +955,8 @@ void TrackContentObjectView::contextMenuEvent( QContextMenuEvent * cme )
 
 	QMenu contextMenu( this );
 
+	bool canMergeTCOs = individualTCO ? false : canMergeSelection(selectedTCOs);
+
 	if( fixedTCOs() == false )
 	{
 		contextMenu.addAction(
@@ -965,6 +974,15 @@ void TrackContentObjectView::contextMenuEvent( QContextMenuEvent * cme )
 				? tr("Cut")
 				: tr("Cut selection"),
 			[this](){ contextMenuAction( Cut ); } );
+
+		if (canMergeTCOs)
+		{
+			contextMenu.addAction(
+				embed::getIconPixmap("edit_merge"),
+				tr("Merge Selection"),
+				[this]() { contextMenuAction(Merge); }
+			);
+		}
 	}
 
 	contextMenu.addAction(
@@ -1023,6 +1041,9 @@ void TrackContentObjectView::contextMenuAction( ContextMenuAction action )
 		case Mute:
 			toggleMute( active );
 			break;
+		case Merge:
+			mergeTCOs(active);
+			break;
 	}
 }
 
@@ -1106,6 +1127,122 @@ void TrackContentObjectView::toggleMute( QVector<TrackContentObjectView *> tcovs
 	}
 }
 
+bool TrackContentObjectView::canMergeSelection(QVector<TrackContentObjectView*> tcovs)
+{
+	// We can only merge InstrumentTrack's TCOs, so check if we only have those in the selection,
+	// and also if they all belong to the same track
+
+	// We start assuming we can merge and if any condition is broken we set it to false
+	bool canMerge = true;
+
+	// Variable to check if all TCOs belong to the same track
+	TrackView* previousOwnerTrackView = nullptr;
+
+	// Then we check every selected TCO to see if all of them are InstrumentTrack's TCOs.
+	// If any isn't, we set canMerge to false and quit the loop.
+	for (auto tcov: tcovs)
+	{
+		TrackView* ownerTrackView = tcov->getTrackView();
+
+		// Set the previousOwnerTrackView to the first TrackView
+		if (!previousOwnerTrackView)
+		{
+			previousOwnerTrackView = ownerTrackView;
+		}
+
+		// If there are TCOs from different tracks or TCOs from tracks
+		// other than an InstrumentTrack, can't merge them
+		if (ownerTrackView != previousOwnerTrackView
+			|| !dynamic_cast<InstrumentTrackView*>(ownerTrackView))
+		{
+			canMerge = false;
+			break;
+		}
+	}
+
+	return canMerge;
+}
+
+void TrackContentObjectView::mergeTCOs(QVector<TrackContentObjectView*> tcovs)
+{
+	// Get the track that we are merging TCOs in
+	InstrumentTrack* track =
+		dynamic_cast<InstrumentTrack*>(tcovs.at(0)->getTrackView()->getTrack());
+
+	if (!track)
+	{
+		qWarning("Warning: Couldn't retrieve InstrumentTrack in mergeTCOs()");
+		return;
+	}
+
+	// For Undo/Redo
+	track->addJournalCheckPoint();
+	track->saveJournallingState(false);
+
+	// Find the earliest position of all the selected TCOVs
+	TimePos earliestPos = tcovs.at(0)->getTrackContentObject()->startPosition();
+	TimePos currentPos = earliestPos;
+
+	for (auto tcov: tcovs)
+	{
+		currentPos = tcov->getTrackContentObject()->startPosition();
+		if (currentPos < earliestPos)
+		{
+			earliestPos = currentPos;
+		}
+	}
+
+	// Create a pattern where all notes will be added
+	Pattern* newPattern = dynamic_cast<Pattern*>(track->createTCO(earliestPos));
+	if (!newPattern)
+	{
+		qWarning("Warning: Failed to convert TCO to Pattern on mergeTCOs");
+		return;
+	}
+
+	newPattern->saveJournallingState(false);
+
+	// Add the notes and remove the TCOs that are being merged
+	for (auto tcov: tcovs)
+	{
+		// Convert TCOV to PatternView
+		PatternView* pView = dynamic_cast<PatternView*>(tcov);
+
+		if (!pView)
+		{
+			qWarning("Warning: Non-pattern TCO on InstrumentTrack");
+			continue;
+		}
+
+		NoteVector currentTCONotes = pView->getPattern()->notes();
+		TimePos pViewPos = pView->getPattern()->startPosition();
+
+		for (Note* note: currentTCONotes)
+		{
+			Note* newNote = newPattern->addNote(*note, false);
+			TimePos originalNotePos = newNote->pos();
+			newNote->setPos(originalNotePos + (pViewPos - earliestPos));
+		}
+
+		// We disable the journalling system before removing, so the
+		// removal doesn't get added to the undo/redo history
+		tcov->getTrackContentObject()->saveJournallingState(false);
+		// No need to check for nullptr because we check while building the tcovs QVector
+		tcov->remove();
+	}
+
+	// Update length since we might have moved notes beyond the end of the pattern length
+	newPattern->updateLength();
+	// Rearrange notes because we might have moved them
+	newPattern->rearrangeAllNotes();
+	// Restore journalling states now that the operation is finished
+	track->restoreJournallingState();
+	newPattern->restoreJournallingState();
+	// Update engine
+	Engine::getSong()->setModified();
+	gui->songEditor()->update();
+}
+
 
 
 

src/tracks/Pattern.cpp

@@ -627,6 +627,17 @@ PatternView::PatternView( Pattern* pattern, TrackView* parent ) :
 	setStyle( QApplication::style() );
 }
 
+
+
+
+Pattern* PatternView::getPattern()
+{
+	return m_pat;
+}
+
+
+
+
 void PatternView::update()
 {
 	ToolTip::add(this, m_pat->name());