Move size calculation from Java IDE

src/arduino.cc/builder/builder.go

@@ -116,6 +116,8 @@ func (s *Builder) Run(ctx *types.Context) error {
 		&RecipeByPrefixSuffixRunner{Prefix: "recipe.objcopy.", Suffix: constants.HOOKS_PATTERN_SUFFIX},
 		&RecipeByPrefixSuffixRunner{Prefix: constants.HOOKS_OBJCOPY_POSTOBJCOPY, Suffix: constants.HOOKS_PATTERN_SUFFIX},
 
+		&phases.Sizer{},
+
 		&MergeSketchWithBootloader{},
 
 		&RecipeByPrefixSuffixRunner{Prefix: constants.HOOKS_POSTBUILD, Suffix: constants.HOOKS_PATTERN_SUFFIX},

src/arduino.cc/builder/constants/constants.go

@@ -170,6 +170,13 @@ const MSG_PROP_IN_LIBRARY = "Missing '{0}' from library in {1}"
 const MSG_RUNNING_COMMAND = "Ts: {0} - Running: {1}"
 const MSG_RUNNING_RECIPE = "Running recipe: {0}"
 const MSG_SETTING_BUILD_PATH = "Setting build path to {0}"
+const MSG_SIZER_TEXT_FULL = "Sketch uses {0} bytes ({2}%%) of program storage space. Maximum is {1} bytes."
+const MSG_SIZER_DATA_FULL = "Global variables use {0} bytes ({2}%%) of dynamic memory, leaving {3} bytes for local variables. Maximum is {1} bytes."
+const MSG_SIZER_DATA = "Global variables use {0} bytes of dynamic memory."
+const MSG_SIZER_TEXT_TOO_BIG = "Sketch too big; see http://www.arduino.cc/en/Guide/Troubleshooting#size for tips on reducing it."
+const MSG_SIZER_DATA_TOO_BIG = "Not enough memory; see http://www.arduino.cc/en/Guide/Troubleshooting#size for tips on reducing your footprint."
+const MSG_SIZER_LOW_MEMORY = "Low memory available, stability problems may occur."
+const MSG_SIZER_ERROR_NO_RULE = "Couldn't determine program size"
 const MSG_SKETCH_CANT_BE_IN_BUILDPATH = "Sketch cannot be located in build path. Please specify a different build path"
 const MSG_SKIPPING_TAG_ALREADY_DEFINED = "Skipping tag {0} because prototype is already defined"
 const MSG_SKIPPING_TAG_BECAUSE_HAS_FIELD = "Skipping tag {0} because it has field {0}"
@@ -197,14 +204,21 @@ const PLATFORM_REWRITE_NEW = "new"
 const PLATFORM_REWRITE_OLD = "old"
 const PLATFORM_URL = "url"
 const PLATFORM_VERSION = "version"
+const PROPERTY_WARN_DATA_PERCENT = "build.warn_data_percentage"
+const PROPERTY_UPLOAD_MAX_SIZE = "upload.maximum_size"
+const PROPERTY_UPLOAD_MAX_DATA_SIZE = "upload.maximum_data_size"
 const PROGRAMMER_NAME = "name"
 const RECIPE_AR_PATTERN = "recipe.ar.pattern"
 const RECIPE_C_COMBINE_PATTERN = "recipe.c.combine.pattern"
 const RECIPE_C_PATTERN = "recipe.c.o.pattern"
 const RECIPE_CPP_PATTERN = "recipe.cpp.o.pattern"
+const RECIPE_SIZE_PATTERN = "recipe.size.pattern"
 const RECIPE_PREPROC_INCLUDES = "recipe.preproc.includes"
 const RECIPE_PREPROC_MACROS = "recipe.preproc.macros"
 const RECIPE_S_PATTERN = "recipe.S.o.pattern"
+const RECIPE_SIZE_REGEXP = "recipe.size.regex"
+const RECIPE_SIZE_REGEXP_DATA = "recipe.size.regex.data"
+const RECIPE_SIZE_REGEXP_EEPROM = "recipe.size.regex.eeprom"
 const REWRITING_DISABLED = "disabled"
 const REWRITING = "rewriting"
 const SPACE = " "

src/arduino.cc/builder/phases/sizer.go

@@ -0,0 +1,188 @@
+/*
+ * This file is part of Arduino Builder.
+ *
+ * Arduino Builder 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 2 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, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ * As a special exception, you may use this file as part of a free software
+ * library without restriction.  Specifically, if other files instantiate
+ * templates or use macros or inline functions from this file, or you compile
+ * this file and link it with other files to produce an executable, this
+ * file does not by itself cause the resulting executable to be covered by
+ * the GNU General Public License.  This exception does not however
+ * invalidate any other reasons why the executable file might be covered by
+ * the GNU General Public License.
+ *
+ * Copyright 2016 Arduino LLC (http://www.arduino.cc/)
+ */
+
+package phases
+
+import (
+	"errors"
+	"regexp"
+	"strconv"
+
+	"arduino.cc/builder/builder_utils"
+	"arduino.cc/builder/constants"
+	"arduino.cc/builder/i18n"
+	"arduino.cc/builder/types"
+	"arduino.cc/properties"
+)
+
+type Sizer struct{}
+
+func (s *Sizer) Run(ctx *types.Context) error {
+	buildProperties := ctx.BuildProperties
+	verbose := ctx.Verbose
+	warningsLevel := ctx.WarningsLevel
+	logger := ctx.GetLogger()
+
+	err := checkSize(buildProperties, verbose, warningsLevel, logger)
+	if err != nil {
+		return i18n.WrapError(err)
+	}
+
+	return nil
+}
+
+func checkSize(buildProperties properties.Map, verbose bool, warningsLevel string, logger i18n.Logger) error {
+
+	properties := buildProperties.Clone()
+	properties[constants.BUILD_PROPERTIES_COMPILER_WARNING_FLAGS] = properties[constants.BUILD_PROPERTIES_COMPILER_WARNING_FLAGS+"."+warningsLevel]
+
+	maxTextSizeString := properties[constants.PROPERTY_UPLOAD_MAX_SIZE]
+	maxDataSizeString := properties[constants.PROPERTY_UPLOAD_MAX_DATA_SIZE]
+
+	if maxTextSizeString == "" {
+		return nil
+	}
+
+	maxTextSize, err := strconv.Atoi(maxTextSizeString)
+	if err != nil {
+		return err
+	}
+
+	maxDataSize := -1
+	if maxDataSizeString != "" {
+		maxDataSize, err = strconv.Atoi(maxDataSizeString)
+		if err != nil {
+			return err
+		}
+	}
+
+	textSize, dataSize, _, err := execSizeReceipe(properties, logger)
+	if err != nil {
+		logger.Println(constants.LOG_LEVEL_WARN, constants.MSG_SIZER_ERROR_NO_RULE)
+		return nil
+	}
+
+	logger.Println(constants.LOG_LEVEL_INFO, constants.MSG_SIZER_TEXT_FULL, strconv.Itoa(textSize), strconv.Itoa(maxTextSize), strconv.Itoa(textSize*100/maxTextSize))
+	if dataSize >= 0 {
+		if maxDataSize > 0 {
+			logger.Println(constants.LOG_LEVEL_INFO, constants.MSG_SIZER_DATA_FULL, strconv.Itoa(dataSize), strconv.Itoa(maxDataSize), strconv.Itoa(dataSize*100/maxDataSize), strconv.Itoa(maxDataSize-dataSize))
+		} else {
+			logger.Println(constants.LOG_LEVEL_INFO, constants.MSG_SIZER_DATA, strconv.Itoa(dataSize))
+		}
+	}
+
+	if textSize > maxTextSize {
+		logger.Println(constants.LOG_LEVEL_ERROR, constants.MSG_SIZER_TEXT_TOO_BIG)
+		return errors.New("")
+	}
+
+	if maxDataSize > 0 && dataSize > maxDataSize {
+		logger.Println(constants.LOG_LEVEL_ERROR, constants.MSG_SIZER_DATA_TOO_BIG)
+		return errors.New("")
+	}
+
+	if properties[constants.PROPERTY_WARN_DATA_PERCENT] != "" {
+		warnDataPercentage, err := strconv.Atoi(properties[constants.PROPERTY_WARN_DATA_PERCENT])
+		if err != nil {
+			return err
+		}
+		if maxDataSize > 0 && dataSize > maxDataSize*warnDataPercentage/100 {
+			logger.Println(constants.LOG_LEVEL_WARN, constants.MSG_SIZER_LOW_MEMORY)
+		}
+	}
+
+	return nil
+}
+
+func execSizeReceipe(properties properties.Map, logger i18n.Logger) (textSize int, dataSize int, eepromSize int, resErr error) {
+	out, err := builder_utils.ExecRecipe(properties, constants.RECIPE_SIZE_PATTERN, false, false, false, logger)
+	if err != nil {
+		resErr = errors.New("Error while determining sketch size: " + err.Error())
+		return
+	}
+
+	// force multiline match prepending "(?m)" to the actual regexp
+	// return an error if RECIPE_SIZE_REGEXP doesn't exist
+
+	if len(properties[constants.RECIPE_SIZE_REGEXP]) > 0 {
+		textRegexp, err := regexp.Compile("(?m)" + properties[constants.RECIPE_SIZE_REGEXP])
+		if err != nil {
+			resErr = errors.New("Invalid size regexp: " + err.Error())
+			return
+		}
+		result := textRegexp.FindAllSubmatch(out, -1)
+		for _, b := range result {
+			for _, c := range b {
+				if res, err := strconv.Atoi(string(c)); err == nil {
+					textSize += res
+				}
+			}
+		}
+	} else {
+		resErr = errors.New("Missing size regexp")
+		return
+	}
+
+	if len(properties[constants.RECIPE_SIZE_REGEXP_DATA]) > 0 {
+		dataRegexp, err := regexp.Compile("(?m)" + properties[constants.RECIPE_SIZE_REGEXP_DATA])
+		if err != nil {
+			resErr = errors.New("Invalid data size regexp: " + err.Error())
+			return
+		}
+		result := dataRegexp.FindAllSubmatch(out, -1)
+		for _, b := range result {
+			for _, c := range b {
+				if res, err := strconv.Atoi(string(c)); err == nil {
+					dataSize += res
+				}
+			}
+		}
+	} else {
+		dataSize = -1
+	}
+
+	if len(properties[constants.RECIPE_SIZE_REGEXP_EEPROM]) > 0 {
+		eepromRegexp, err := regexp.Compile("(?m)" + properties[constants.RECIPE_SIZE_REGEXP_EEPROM])
+		if err != nil {
+			resErr = errors.New("Invalid eeprom size regexp: " + err.Error())
+			return
+		}
+		result := eepromRegexp.FindAllSubmatch(out, -1)
+		for _, b := range result {
+			for _, c := range b {
+				if res, err := strconv.Atoi(string(c)); err == nil {
+					eepromSize += res
+				}
+			}
+		}
+	} else {
+		eepromSize = -1
+	}
+	return
+}