From 34b92b7574fba82b0959006ff233443545b973d5 Mon Sep 17 00:00:00 2001
From: "wonder.mice" <wonder.mice@gmail.com>
Date: Tue, 21 Apr 2015 12:42:17 -0700
Subject: [PATCH] CodeBlocks: Take into account file target when explicitly
 specified

Change-Id: I37556ef6c2c7a651709012daa9a84c66cf3af505
Reviewed-by: Daniel Teske <daniel.teske@theqtcompany.com>
---
 src/libs/utils/algorithm.h                    |  8 +++++++
 .../cmakeprojectmanager/cmakecbpparser.cpp    | 22 ++++++++++++++-----
 .../cmakeprojectmanager/cmakecbpparser.h      |  2 ++
 3 files changed, 27 insertions(+), 5 deletions(-)

diff --git a/src/libs/utils/algorithm.h b/src/libs/utils/algorithm.h
index 0fb7a35a418..c13a3a5bb61 100644
--- a/src/libs/utils/algorithm.h
+++ b/src/libs/utils/algorithm.h
@@ -137,6 +137,14 @@ auto equal(R (S::*function)() const, T value)
     return std::bind<bool>(std::equal_to<T>(), value, std::bind(function, std::placeholders::_1));
 }
 
+template<typename R, typename S, typename T>
+auto equal(R S::*member, T value)
+    -> decltype(std::bind<bool>(std::equal_to<T>(), value, std::bind(member, std::placeholders::_1)))
+{
+    return std::bind<bool>(std::equal_to<T>(), value, std::bind(member, std::placeholders::_1));
+}
+
+
 //////////////////
 // transform
 /////////////////
diff --git a/src/plugins/cmakeprojectmanager/cmakecbpparser.cpp b/src/plugins/cmakeprojectmanager/cmakecbpparser.cpp
index 16d3336cd0d..6aadd391699 100644
--- a/src/plugins/cmakeprojectmanager/cmakecbpparser.cpp
+++ b/src/plugins/cmakeprojectmanager/cmakecbpparser.cpp
@@ -104,10 +104,19 @@ void CMakeCbpParser::sortFiles()
 
     foreach (const FileName &fileName, fileNames) {
         qCDebug(log) << fileName;
+        const QString unitTarget = m_unitTargetMap[fileName];
+        if (!unitTarget.isEmpty()) { // target was explicitly specified for that file
+            int index = Utils::indexOf(m_buildTargets, Utils::equal(&CMakeBuildTarget::title, unitTarget));
+            if (index != -1) {
+                m_buildTargets[index].files.append(fileName.toString());
+                qCDebug(log) << "  into" << m_buildTargets[index].title << "(target attribute)";
+                continue;
+            }
+        }
         if (fileName.parentDir() == parentDirectory && last) {
             // easy case, same parent directory as last file
             last->files.append(fileName.toString());
-            qCDebug(log) << "  into" << last->title;
+            qCDebug(log) << "  into" << last->title << "(same parent)";
         } else {
             int bestDistance = std::numeric_limits<int>::max();
             int bestIndex = -1;
@@ -228,8 +237,7 @@ void CMakeCbpParser::parseBuildTarget()
         readNext();
         if (isEndElement()) {
             if (!m_buildTarget.title.endsWith(QLatin1String("/fast"))
-                    && !m_buildTarget.title.endsWith(QLatin1String("_automoc"))
-                    && !m_buildTarget.title.endsWith(QLatin1String("_unittest")))
+                    && !m_buildTarget.title.endsWith(QLatin1String("_automoc")))
                 m_buildTargets.append(m_buildTarget);
             return;
         } else if (name() == QLatin1String("Compiler")) {
@@ -409,6 +417,7 @@ void CMakeCbpParser::parseUnit()
     }
 
     m_parsingCmakeUnit = false;
+    m_unitTarget.clear();
     while (!atEnd()) {
         readNext();
         if (isEndElement()) {
@@ -429,6 +438,8 @@ void CMakeCbpParser::parseUnit()
                     else
                         m_fileList.append( new ProjectExplorer::FileNode(fileName, ProjectExplorer::SourceType, generated));
                 }
+                if (!m_unitTarget.isEmpty())
+                    m_unitTargetMap.insert(fileName, m_unitTarget);
                 m_processedUnits.insert(fileName);
             }
             return;
@@ -442,8 +453,9 @@ void CMakeCbpParser::parseUnit()
 
 void CMakeCbpParser::parseUnitOption()
 {
-    if (attributes().hasAttribute(QLatin1String("virtualFolder")))
-        m_parsingCmakeUnit = true;
+    const QXmlStreamAttributes optionAttributes = attributes();
+    m_parsingCmakeUnit = optionAttributes.hasAttribute(QLatin1String("virtualFolder"));
+    m_unitTarget = optionAttributes.value(QLatin1String("target")).toString();
 
     while (!atEnd()) {
         readNext();
diff --git a/src/plugins/cmakeprojectmanager/cmakecbpparser.h b/src/plugins/cmakeprojectmanager/cmakecbpparser.h
index f18ff846ffd..2aa979eb792 100644
--- a/src/plugins/cmakeprojectmanager/cmakecbpparser.h
+++ b/src/plugins/cmakeprojectmanager/cmakecbpparser.h
@@ -72,6 +72,7 @@ private:
     void parseUnknownElement();
     void sortFiles();
 
+    QMap<Utils::FileName, QString> m_unitTargetMap;
     ProjectExplorer::Kit *m_kit;
     QList<ProjectExplorer::FileNode *> m_fileList;
     QList<ProjectExplorer::FileNode *> m_cmakeFileList;
@@ -84,6 +85,7 @@ private:
     QString m_compiler;
     QString m_sourceDirectory;
     QString m_buildDirectory;
+    QString m_unitTarget;
 };
 
 } // namespace Internal