tools: update ttfw to support ESP32-H2 and ESP32-C6 chips

tools/ci/python_packages/ttfw_idf/IDFDUT.py

@@ -1,16 +1,5 @@
-# Copyright 2015-2017 Espressif Systems (Shanghai) PTE LTD
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http:#www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
+# SPDX-FileCopyrightText: 2015-2021 Espressif Systems (Shanghai) CO LTD
+# SPDX-License-Identifier: Apache-2.0
 
 """ DUT for IDF applications """
 import collections
@@ -611,6 +600,24 @@ class ESP32C3DUT(IDFDUT):
         return esptool.ESP32C3ROM
 
 
+class ESP32C6DUT(IDFDUT):
+    TARGET = 'esp32c6'
+    TOOLCHAIN_PREFIX = 'riscv32-esp-elf-'
+
+    @classmethod
+    def get_rom(cls):
+        return esptool.ESP32C6BETAROM
+
+
+class ESP32H2DUT(IDFDUT):
+    TARGET = 'esp32h2'
+    TOOLCHAIN_PREFIX = 'riscv32-esp-elf-'
+
+    @classmethod
+    def get_rom(cls):
+        return esptool.ESP32H2ROM
+
+
 class ESP8266DUT(IDFDUT):
     TARGET = 'esp8266'
     TOOLCHAIN_PREFIX = 'xtensa-lx106-elf-'
@@ -621,7 +628,7 @@ class ESP8266DUT(IDFDUT):
 
 
 def get_target_by_rom_class(cls):
-    for c in [ESP32DUT, ESP32S2DUT, ESP32S3DUT, ESP32C3DUT, ESP8266DUT, IDFQEMUDUT]:
+    for c in [ESP32DUT, ESP32S2DUT, ESP32S3DUT, ESP32C3DUT, ESP32C6DUT, ESP32H2DUT, ESP8266DUT, IDFQEMUDUT]:
         if c.get_rom() == cls:
             return c.TARGET
     return None