FormatToken.cpp revision 332c67703f63fafd26cf88fc31faeb80c1523725
1d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper//===--- FormatToken.cpp - Format C++ code --------------------------------===// 2d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// 3d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// The LLVM Compiler Infrastructure 4d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// 5d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// This file is distributed under the University of Illinois Open Source 6d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// License. See LICENSE.TXT for details. 7d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// 8d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper//===----------------------------------------------------------------------===// 9d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper/// 10d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper/// \file 11d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper/// \brief This file implements specific functions of \c FormatTokens and their 12d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper/// roles. 13d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper/// 14d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper//===----------------------------------------------------------------------===// 15d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 16d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper#include "FormatToken.h" 17d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper#include "ContinuationIndenter.h" 18d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper#include "clang/Format/Format.h" 19d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper#include "llvm/ADT/SmallVector.h" 20d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper#include "llvm/Support/Debug.h" 21d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 22d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jaspernamespace clang { 23d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jaspernamespace format { 24d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 25d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel JasperTokenRole::~TokenRole() {} 26d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 27d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jaspervoid TokenRole::precomputeFormattingInfos(const FormatToken *Token) {} 28d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 29d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasperunsigned CommaSeparatedList::format(LineState &State, 30d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ContinuationIndenter *Indenter, 31d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper bool DryRun) { 32d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (!State.NextToken->Previous || !State.NextToken->Previous->Previous || 33d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Commas.size() <= 2) 34d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return 0; 35d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 36d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Ensure that we start on the opening brace. 37d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper const FormatToken *LBrace = State.NextToken->Previous->Previous; 38d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (LBrace->isNot(tok::l_brace) || 39d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper LBrace->Next->Type == TT_DesignatedInitializerPeriod) 40d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return 0; 41d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 42d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Find the best ColumnFormat, i.e. the best number of columns to use. 43d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned RemainingCharacters = Style.ColumnLimit - State.Stack.back().Indent; 44d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper const ColumnFormat *Format = getColumnFormat(RemainingCharacters); 45d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (!Format) 46d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return 0; 47d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 48d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Format the entire list. 49d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned Penalty = 0; 50d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned Column = 0; 51d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned Item = 0; 52d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper while (State.NextToken != LBrace->MatchingParen) { 53d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper bool NewLine = false; 54d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned ExtraSpaces = 0; 55d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 56d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // If the previous token was one of our commas, we are now on the next item. 57d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (Item < Commas.size() && State.NextToken->Previous == Commas[Item]) { 58d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (!State.NextToken->isTrailingComment()) { 59d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ExtraSpaces += Format->ColumnSizes[Column] - ItemLengths[Item]; 60d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ++Column; 61d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 62d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ++Item; 63d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 64d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 65d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (Column == Format->Columns || State.NextToken->MustBreakBefore) { 66d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Column = 0; 67d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper NewLine = true; 68d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 69d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 70d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Place token using the continuation indenter and store the penalty. 71d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Penalty += Indenter->addTokenToState(State, NewLine, DryRun, ExtraSpaces); 72d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 73d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return Penalty; 74d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper} 75d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 76d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// Returns the lengths in code points between Begin and End (both included), 77d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper// assuming that the entire sequence is put on a single line. 78d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasperstatic unsigned CodePointsBetween(const FormatToken *Begin, 79d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper const FormatToken *End) { 80332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper assert(End->TotalLength >= Begin->TotalLength); 81d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return End->TotalLength - Begin->TotalLength + Begin->CodePointCount; 82d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper} 83d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 84d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jaspervoid CommaSeparatedList::precomputeFormattingInfos(const FormatToken *Token) { 85332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper // FIXME: At some point we might want to do this for other lists, too. 86332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper if (!Token->MatchingParen || Token->isNot(tok::l_brace)) 87d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return; 88d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 89d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper FormatToken *ItemBegin = Token->Next; 90d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper SmallVector<bool, 8> MustBreakBeforeItem; 91d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 92d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // The lengths of an item if it is put at the end of the line. This includes 93d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // trailing comments which are otherwise ignored for column alignment. 94d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper SmallVector<unsigned, 8> EndOfLineItemLength; 95d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 96d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper for (unsigned i = 0, e = Commas.size() + 1; i != e; ++i) { 97d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Skip comments on their own line. 98d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper while (ItemBegin->HasUnescapedNewline && ItemBegin->isTrailingComment()) 99d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemBegin = ItemBegin->Next; 100d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 101d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper MustBreakBeforeItem.push_back(ItemBegin->MustBreakBefore); 102d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper const FormatToken *ItemEnd = NULL; 103d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (i == Commas.size()) { 104d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemEnd = Token->MatchingParen; 105d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper const FormatToken *NonCommentEnd = ItemEnd->getPreviousNonComment(); 106d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemLengths.push_back(CodePointsBetween(ItemBegin, NonCommentEnd)); 107d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (Style.Cpp11BracedListStyle) { 108d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // In Cpp11 braced list style, the } and possibly other subsequent 109d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // tokens will need to stay on a line with the last element. 110d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper while (ItemEnd->Next && !ItemEnd->Next->CanBreakBefore) 111d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemEnd = ItemEnd->Next; 112d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } else { 113d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // In other braced lists styles, the "}" can be wrapped to the new line. 114d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemEnd = Token->MatchingParen->Previous; 115d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 116d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } else { 117d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemEnd = Commas[i]; 118d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // The comma is counted as part of the item when calculating the length. 119332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper ItemLengths.push_back(CodePointsBetween(ItemBegin, ItemEnd)); 120d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Consume trailing comments so the are included in EndOfLineItemLength. 121d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (ItemEnd->Next && !ItemEnd->Next->HasUnescapedNewline && 122d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemEnd->Next->isTrailingComment()) 123d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemEnd = ItemEnd->Next; 124d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 125d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper EndOfLineItemLength.push_back(CodePointsBetween(ItemBegin, ItemEnd)); 126332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper // If there is a trailing comma in the list, the next item will start at the 127332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper // closing brace. Don't create an extra item for this. 128332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper if (ItemEnd->getNextNonComment() == Token->MatchingParen) 129332c67703f63fafd26cf88fc31faeb80c1523725Daniel Jasper break; 130d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ItemBegin = ItemEnd->Next; 131d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 132d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 133d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // We can never place more than ColumnLimit / 3 items in a row (because of the 134d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // spaces and the comma). 135d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper for (unsigned Columns = 1; Columns <= Style.ColumnLimit / 3; ++Columns) { 136d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ColumnFormat Format; 137d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Format.Columns = Columns; 138d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Format.ColumnSizes.resize(Columns); 139d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Format.LineCount = 0; 140d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper bool HasRowWithSufficientColumns = false; 141d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned Column = 0; 142d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper for (unsigned i = 0, e = ItemLengths.size(); i != e; ++i) { 143d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (MustBreakBeforeItem[i] || Column == Columns) { 144d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ++Format.LineCount; 145d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Column = 0; 146d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 147d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (Column == Columns - 1) 148d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper HasRowWithSufficientColumns = true; 149d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper unsigned length = 150d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper (Column == Columns - 1) ? EndOfLineItemLength[i] : ItemLengths[i]; 151d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Format.ColumnSizes[Column] = 152d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper std::max(Format.ColumnSizes[Column], length); 153d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper ++Column; 154d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 155d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // If all rows are terminated early (e.g. by trailing comments), we don't 156d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // need to look further. 157d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (!HasRowWithSufficientColumns) 158d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper break; 159d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Format.TotalWidth = Columns - 1; // Width of the N-1 spaces. 160d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper for (unsigned i = 0; i < Columns; ++i) { 161d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Format.TotalWidth += Format.ColumnSizes[i]; 162d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 163d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 164d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper // Ignore layouts that are bound to violate the column limit. 165d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (Format.TotalWidth > Style.ColumnLimit) 166d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper continue; 167d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 168d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper Formats.push_back(Format); 169d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 170d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper} 171d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 172d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasperconst CommaSeparatedList::ColumnFormat * 173d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel JasperCommaSeparatedList::getColumnFormat(unsigned RemainingCharacters) const { 174d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper const ColumnFormat *BestFormat = NULL; 175d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper for (SmallVector<ColumnFormat, 4>::const_reverse_iterator 176d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper I = Formats.rbegin(), 177d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper E = Formats.rend(); 178d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper I != E; ++I) { 179d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (I->TotalWidth <= RemainingCharacters) { 180d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper if (BestFormat && I->LineCount > BestFormat->LineCount) 181d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper break; 182d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper BestFormat = &*I; 183d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 184d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper } 185d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper return BestFormat; 186d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper} 187d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper 188d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper} // namespace format 189d4a03dbb0bdbd4908375306283cb61813bd8f454Daniel Jasper} // namespace clang 190