Move size calculation from Java IDE

Author: facchinm

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

@@ -202,6 +202,7 @@ 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"

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

@@ -0,0 +1,160 @@
+/*
+ * 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"
+	"os"
+	"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["upload.maximum_size"]
+	maxDataSizeString := properties["upload.maximum_data_size"]
+
+	if maxTextSizeString == "" {
+		return nil
+	}
+
+	logLevel := constants.LOG_LEVEL_INFO
+
+	maxTextSize, _ := strconv.Atoi(maxTextSizeString)
+	maxDataSize := -1
+
+	if maxDataSizeString != "" {
+		maxDataSize, _ = strconv.Atoi(maxDataSizeString)
+	}
+
+	var textSize, dataSize, eepromSize int
+
+	err := execSizeReceipe(properties, logger, &textSize, &dataSize, &eepromSize)
+	if err != nil {
+		return err
+	}
+
+	logger.Fprintln(os.Stdout, logLevel, "Sketch uses {0} bytes ({2}%%) of program storage space. Maximum is {1} bytes.", strconv.Itoa(textSize), strconv.Itoa(maxTextSize), strconv.Itoa(textSize*100/maxTextSize))
+	if dataSize > 0 {
+		if maxDataSize > 0 {
+			logger.Fprintln(os.Stdout, logLevel, "Global variables use {0} bytes ({2}%%) of dynamic memory, leaving {3} bytes for local variables. Maximum is {1} bytes.", strconv.Itoa(dataSize), strconv.Itoa(maxDataSize), strconv.Itoa(dataSize*100/maxDataSize), strconv.Itoa(maxDataSize-dataSize))
+		} else {
+			logger.Fprintln(os.Stdout, logLevel, "Global variables use {0} bytes of dynamic memory.", strconv.Itoa(dataSize))
+		}
+	}
+
+	if textSize > maxTextSize {
+		return errors.New("Sketch too big; see http://www.arduino.cc/en/Guide/Troubleshooting#size for tips on reducing it.")
+	}
+
+	if maxDataSize > 0 && dataSize > maxDataSize {
+		return errors.New("Not enough memory; see http://www.arduino.cc/en/Guide/Troubleshooting#size for tips on reducing your footprint.")
+	}
+
+	if properties["build.warn_data_percentage"] != "" {
+		warnDataPercentage, _ := strconv.Atoi(properties["build.warn_data_percentage"])
+		if maxDataSize > 0 && dataSize > maxDataSize*warnDataPercentage/100 {
+			logger.Fprintln(os.Stdout, logLevel, "Low memory available, stability problems may occur.")
+		}
+	}
+
+	return nil
+}
+
+func execSizeReceipe(properties properties.Map, logger i18n.Logger, textSize *int, dataSize *int, eepromSize *int) error {
+
+	out, err := builder_utils.ExecRecipe(properties, constants.RECIPE_SIZE_PATTERN, false, false, false, logger)
+
+	if err != nil {
+		return errors.New("Couldn't determine program size: {0}")
+	}
+
+	// force multiline match prepending "(?m)" to the actual regexp
+
+	if len(properties["recipe.size.regex"]) > 0 {
+		textRegexp := regexp.MustCompile("(?m)" + properties["recipe.size.regex"])
+		result := textRegexp.FindAllSubmatch(out, -1)
+		for _, b := range result {
+			for _, c := range b {
+				res, _ := strconv.Atoi(string(c))
+				*textSize += res
+			}
+		}
+	}
+
+	if len(properties["recipe.size.regex.data"]) > 0 {
+		dataRegexp := regexp.MustCompile("(?m)" + properties["recipe.size.regex.data"])
+		result := dataRegexp.FindAllSubmatch(out, -1)
+		for _, b := range result {
+			for _, c := range b {
+				res, _ := strconv.Atoi(string(c))
+				*dataSize += res
+			}
+		}
+	}
+
+	if len(properties["recipe.size.regex.eeprom"]) > 0 {
+		eepromRegexp := regexp.MustCompile("(?m)" + properties["recipe.size.regex.eeprom"])
+		result := eepromRegexp.FindAllSubmatch(out, -1)
+		for _, b := range result {
+			for _, c := range b {
+				res, _ := strconv.Atoi(string(c))
+				*eepromSize += res
+			}
+		}
+	}
+
+	return nil
+}