/*
 *  yosys -- Yosys Open SYnthesis Suite
 *
 *  Copyright (C) 2020  The Symbiflow Authors
 *
 *  Permission to use, copy, modify, and/or distribute this software for any
 *  purpose with or without fee is hereby granted, provided that the above
 *  copyright notice and this permission notice appear in all copies.
 *
 *  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 *  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 *  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 *  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 *  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 *  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 *  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */
#include "set_clock_groups.h"
#include <regex>
#include "kernel/log.h"

USING_YOSYS_NAMESPACE

void SetClockGroups::help() {
    //   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
    log("\n");
    log("   set_clock_groups [-quiet] [-group <args>] [-asynchronous] \n");
    log("\n");
    log("Set exclusive or asynchronous clock groups\n");
    log("\n");
    log("Print the output to stdout too. This is useful when all Yosys is "
        "executed.\n");
    log("\n");
    log("    -quiet\n");
    log("        Don't print the result of the execution to stdout.\n");
    log("\n");
    log("    -group\n");
    log("        List of clocks to be included in the clock group.\n");
    log("\n");
    log("    -asynchronous\n");
    log("       The specified clocks are asynchronous to each other.\n");
    log("\n");
}

void SetClockGroups::execute(std::vector<std::string> args,
                             RTLIL::Design* design) {
    RTLIL::Module* top_module = design->top_module();
    if (top_module == nullptr) {
	log_cmd_error("No top module detected\n");
    }

    size_t argidx;
    bool is_quiet = false;
    std::vector<ClockGroups::ClockGroup> clock_groups;
    auto clock_groups_relation = ClockGroups::NONE;

    // Parse command arguments
    for (argidx = 1; argidx < args.size(); argidx++) {
	std::string arg = args[argidx];
	if (arg == "-quiet") {
	    is_quiet = true;
	    continue;
	}

	// Parse clock groups relation: asynchronous, logically_exclusive, physically_exclusive
	auto is_relation_arg =
	    [arg](std::pair<ClockGroups::ClockGroupRelation, std::string>
	            relation) {
	        if (arg.substr(1) == relation.second) {
		    return true;
	        }
	        return false;
	    };
	auto relation_map_it =
	    std::find_if(ClockGroups::relation_name_map.begin(),
	                 ClockGroups::relation_name_map.end(), is_relation_arg);
	if (relation_map_it != ClockGroups::relation_name_map.end()) {
	    clock_groups_relation = relation_map_it->first;
	    continue;
	}

	if (arg == "-group" and argidx + 1 < args.size()) {
	    ClockGroups::ClockGroup clock_group;
	    while (argidx + 1 < args.size() and args[argidx+1][0] != '-') {
		clock_group.push_back(args[++argidx]);
	    }
	    clock_groups.push_back(clock_group);
	    continue;
	}

	if (arg.size() > 0 and arg[0] == '-') {
	    log_cmd_error("Unknown option %s.\n", arg.c_str());
	}

	break;
    }

    if (clock_groups.size()) {
	if (!is_quiet) {
	    std::string msg = ClockGroups::relation_name_map.at(clock_groups_relation);
	    msg += (!msg.empty()) ? " " : "";
	    log("Adding %sclock group with following clocks:\n", msg.c_str());
	}
	size_t count(0);
	for (auto& group : clock_groups) {
	    sdc_writer_.AddClockGroup(group, clock_groups_relation);
	    if (!is_quiet) {
		log("%zu: ", count++);
		for (auto clk : group) {
		    log("%s ", clk.c_str());
		}
		log("\n");
	    }
	}
    }
}