Added "techmap -map %{design-name}"

4 files changed, 29 insertions, 10 deletions

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index 83bbd7b1..f864d88c 100644
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -219,6 +219,11 @@ void RTLIL::Selection::optimize(RTLIL::Design *design)
 	}
 }
 
+RTLIL::Design::Design()
+{
+	refcount_modules_ = 0;
+}
+
 RTLIL::Design::~Design()
 {
 	for (auto it = modules_.begin(); it != modules_.end(); it++)
diff --git a/kernel/rtlil.h b/kernel/rtlil.h
index e8d05e7e..1f25542f 100644
--- a/kernel/rtlil.h
+++ b/kernel/rtlil.h
@@ -352,11 +352,16 @@ struct RTLIL::Design
 	std::map<RTLIL::IdString, RTLIL::Selection> selection_vars;
 	std::string selected_active_module;
 
+	Design();
 	~Design();
 
 	RTLIL::ObjRange<RTLIL::Module*> modules();
 	RTLIL::Module *module(RTLIL::IdString name);
 
+	bool has(RTLIL::IdString id) const {
+		return modules_.count(id) != 0;
+	}
+
 	void add(RTLIL::Module *module);
 	RTLIL::Module *addModule(RTLIL::IdString name);
 	void remove(RTLIL::Module *module);
diff --git a/passes/techmap/extract.cc b/passes/techmap/extract.cc
index 9c5fa7f7..19d32334 100644
--- a/passes/techmap/extract.cc
+++ b/passes/techmap/extract.cc
@@ -603,9 +603,9 @@ struct ExtractPass : public Pass {
 						delete map;
 						log_cmd_error("Can't saved design `%s'.\n", filename.c_str()+1);
 					}
-					for (auto &it : saved_designs.at(filename.substr(1))->modules_)
-						if (!map->modules_.count(it.first))
-							map->modules_[it.first] = it.second->clone();
+					for (auto mod : saved_designs.at(filename.substr(1))->modules())
+						if (!map->has(mod->name))
+							map->add(mod->clone());
 				}
 				else
 				{
diff --git a/passes/techmap/techmap.cc b/passes/techmap/techmap.cc
index 5a69baca..0ae5220e 100644
--- a/passes/techmap/techmap.cc
+++ b/passes/techmap/techmap.cc
@@ -656,13 +656,22 @@ struct TechmapPass : public Pass {
 			Frontend::frontend_call(map, f, "<stdcells.v>", verilog_frontend);
 			fclose(f);
 		} else
-			for (auto &fn : map_files) {
-				FILE *f = fopen(fn.c_str(), "rt");
-				if (f == NULL)
-					log_cmd_error("Can't open map file `%s'\n", fn.c_str());
-				Frontend::frontend_call(map, f, fn, (fn.size() > 3 && fn.substr(fn.size()-3) == ".il") ? "ilang" : verilog_frontend);
-				fclose(f);
-			}
+			for (auto &fn : map_files)
+				if (fn.substr(0, 1) == "%") {
+					if (!saved_designs.count(fn.substr(1))) {
+						delete map;
+						log_cmd_error("Can't saved design `%s'.\n", fn.c_str()+1);
+					}
+					for (auto mod : saved_designs.at(fn.substr(1))->modules())
+						if (!map->has(mod->name))
+							map->add(mod->clone());
+				} else {
+					FILE *f = fopen(fn.c_str(), "rt");
+					if (f == NULL)
+						log_cmd_error("Can't open map file `%s'\n", fn.c_str());
+					Frontend::frontend_call(map, f, fn, (fn.size() > 3 && fn.substr(fn.size()-3) == ".il") ? "ilang" : verilog_frontend);
+					fclose(f);
+				}
 
 		std::map<RTLIL::IdString, RTLIL::Module*> modules_new;
 		for (auto &it : map->modules_) {