Added "dump" command (part ilang backend)

2 files changed, 103 insertions, 13 deletions

diff --git a/backends/ilang/ilang_backend.cc b/backends/ilang/ilang_backend.cc
index 8dcdb1288..9c74e61ca 100644
--- a/backends/ilang/ilang_backend.cc
+++ b/backends/ilang/ilang_backend.cc
@@ -27,6 +27,8 @@
 #include "kernel/log.h"
 #include <string>
 #include <assert.h>
+#include <string.h>
+#include <errno.h>
 
 using namespace ILANG_BACKEND;
 
@@ -257,7 +259,7 @@ void ILANG_BACKEND::dump_conn(FILE *f, std::string indent, const RTLIL::SigSpec
 	fprintf(f, "\n");
 }
 
-void ILANG_BACKEND::dump_module(FILE *f, std::string indent, const RTLIL::Module *module)
+void ILANG_BACKEND::dump_module(FILE *f, std::string indent, const RTLIL::Module *module, const RTLIL::Design *design, bool only_selected)
 {
 	for (auto it = module->attributes.begin(); it != module->attributes.end(); it++) {
 		fprintf(f, "%s" "attribute %s ", indent.c_str(), it->first.c_str());
@@ -268,29 +270,63 @@ void ILANG_BACKEND::dump_module(FILE *f, std::string indent, const RTLIL::Module
 	fprintf(f, "%s" "module %s\n", indent.c_str(), module->name.c_str());
 
 	for (auto it = module->wires.begin(); it != module->wires.end(); it++)
-		dump_wire(f, indent + "  ", it->second);
+		if (!only_selected || design->selected(module, it->second)) {
+			if (only_selected)
+				fprintf(f, "\n");
+			dump_wire(f, indent + "  ", it->second);
+		}
 
 	for (auto it = module->memories.begin(); it != module->memories.end(); it++)
-		dump_memory(f, indent + "  ", it->second);
+		if (!only_selected || design->selected(module, it->second)) {
+			if (only_selected)
+				fprintf(f, "\n");
+			dump_memory(f, indent + "  ", it->second);
+		}
 
 	for (auto it = module->cells.begin(); it != module->cells.end(); it++)
-		dump_cell(f, indent + "  ", it->second);
+		if (!only_selected || design->selected(module, it->second)) {
+			if (only_selected)
+				fprintf(f, "\n");
+			dump_cell(f, indent + "  ", it->second);
+		}
 
 	for (auto it = module->processes.begin(); it != module->processes.end(); it++)
-		dump_proc(f, indent + "  ", it->second);
+		if (!only_selected || design->selected(module, it->second)) {
+			if (only_selected)
+				fprintf(f, "\n");
+			dump_proc(f, indent + "  ", it->second);
+		}
 
-	for (auto it = module->connections.begin(); it != module->connections.end(); it++)
-		dump_conn(f, indent + "  ", it->first, it->second);
+	bool first_conn_line = true;
+	for (auto it = module->connections.begin(); it != module->connections.end(); it++) {
+		bool show_conn = !only_selected;
+		if (only_selected) {
+			RTLIL::SigSpec sigs = it->first;
+			sigs.append(it->second);
+			for (auto &c : sigs.chunks) {
+				if (c.wire == NULL || !design->selected(module, c.wire))
+					continue;
+				show_conn = true;
+			}
+		}
+		if (show_conn) {
+			if (only_selected && first_conn_line)
+				fprintf(f, "\n");
+			dump_conn(f, indent + "  ", it->first, it->second);
+			first_conn_line = false;
+		}
+	}
 
 	fprintf(f, "%s" "end\n", indent.c_str());
 }
 
-void ILANG_BACKEND::dump_design(FILE *f, const RTLIL::Design *design)
+void ILANG_BACKEND::dump_design(FILE *f, const RTLIL::Design *design, bool only_selected)
 {
 	for (auto it = design->modules.begin(); it != design->modules.end(); it++) {
-		if (it != design->modules.begin())
+		if (it != design->modules.begin() || only_selected)
 			fprintf(f, "\n");
-		dump_module(f, "", it->second);
+		if (!only_selected || design->selected(it->second))
+			dump_module(f, "", it->second, design, only_selected);
 	}
 }
 
@@ -310,7 +346,61 @@ struct IlangBackend : public Backend {
 		log_header("Executing ILANG backend.\n");
 		extra_args(f, filename, args, 1);
 		log("Output filename: %s\n", filename.c_str());
-		ILANG_BACKEND::dump_design(f, design);
+		ILANG_BACKEND::dump_design(f, design, false);
 	}
 } IlangBackend;
 
+struct DumpPass : public Pass {
+	DumpPass() : Pass("dump", "print parts of the design in ilang format") { }
+	virtual void help()
+	{
+		//   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
+		log("\n");
+		log("    dump [options] [selection]\n");
+		log("\n");
+		log("Write the selected parts of the design to the console or specified file in\n");
+		log("ilang format.\n");
+		log("\n");
+		log("    -outfile <filename>\n");
+		log("        Write to the specified file.\n");
+		log("\n");
+	}
+	virtual void execute(std::vector<std::string> args, RTLIL::Design *design)
+	{
+		std::string filename;
+
+		size_t argidx;
+		for (argidx = 1; argidx < args.size(); argidx++)
+		{
+			std::string arg = args[argidx];
+			if (arg == "-outfile" && argidx+1 < args.size()) {
+				filename = args[++argidx];
+				continue;
+			}
+			break;
+		}
+		extra_args(args, argidx, design);
+
+		FILE *f = NULL;
+		char *buf_ptr;
+		size_t buf_size;
+
+		if (!filename.empty()) {
+			f = fopen(filename.c_str(), "w");
+			if (f == NULL)
+				log_error("Can't open file `%s' for writing: %s\n", filename.c_str(), strerror(errno));
+		} else {
+			f = open_memstream(&buf_ptr, &buf_size);
+		}
+
+		ILANG_BACKEND::dump_design(f, design, true);
+
+		fclose(f);
+
+		if (filename.empty()) {
+			log("%s", buf_ptr);
+			free(buf_ptr);
+		}
+	}
+} DumpPass;
+ 
diff --git a/backends/ilang/ilang_backend.h b/backends/ilang/ilang_backend.h
index e34c4e675..b9700970f 100644
--- a/backends/ilang/ilang_backend.h
+++ b/backends/ilang/ilang_backend.h
@@ -40,8 +40,8 @@ namespace ILANG_BACKEND {
 	void dump_proc_sync(FILE *f, std::string indent, const RTLIL::SyncRule *sy);
 	void dump_proc(FILE *f, std::string indent, const RTLIL::Process *proc);
 	void dump_conn(FILE *f, std::string indent, const RTLIL::SigSpec &left, const RTLIL::SigSpec &right);
-	void dump_module(FILE *f, std::string indent, const RTLIL::Module *module);
-	void dump_design(FILE *f, const RTLIL::Design *design);
+	void dump_module(FILE *f, std::string indent, const RTLIL::Module *module, const RTLIL::Design *design, bool only_selected);
+	void dump_design(FILE *f, const RTLIL::Design *design, bool only_selected);
 }
 
 #endif