From 3b2a99530052c5e06b9179754a8dfe3e0d53e612 Mon Sep 17 00:00:00 2001
From: Wesley W. Terpstra
Date: Fri, 17 Feb 2017 20:20:28 +0100
Subject: Record: allow elements to start with a digit

This is necessary for user-defined Record-derived types to retain
the same signal name as they would using a Vec.
---
 chiselFrontend/src/main/scala/chisel3/internal/Builder.scala | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'chiselFrontend/src/main/scala/chisel3/internal')

diff --git a/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala b/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
index 147fa7d4..e0cbf302 100644
--- a/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
+++ b/chiselFrontend/src/main/scala/chisel3/internal/Builder.scala
@@ -21,18 +21,20 @@ private[chisel3] class Namespace(keywords: Set[String]) {
     if (this contains tryName) rename(n) else tryName
   }
 
-  private def sanitize(s: String): String = {
+  private def sanitize(s: String, leadingDigitOk: Boolean = false): String = {
     // TODO what character set does FIRRTL truly support? using ANSI C for now
     def legalStart(c: Char) = (c >= 'a' && c <= 'z') || (c >= 'A' && c <= 'Z') || c == '_'
     def legal(c: Char) = legalStart(c) || (c >= '0' && c <= '9')
     val res = s filter legal
-    if (res.isEmpty || !legalStart(res.head)) s"_$res" else res
+    val headOk = (!res.isEmpty) && (leadingDigitOk || legalStart(res.head))
+    if (headOk) res else s"_$res"
   }
 
   def contains(elem: String): Boolean = names.contains(elem)
 
-  def name(elem: String): String = {
-    val sanitized = sanitize(elem)
+  // leadingDigitOk is for use in fields of Records
+  def name(elem: String, leadingDigitOk: Boolean = false): String = {
+    val sanitized = sanitize(elem, leadingDigitOk)
     if (this contains sanitized) {
       name(rename(sanitized))
     } else {
-- 
cgit v1.2.3