ext/dsent/model/electrical/Multiplexer.cc - arm/gem5 - Git at Google

 /* Copyright (c) 2012 Massachusetts Institute of Technology
  *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to deal
  * in the Software without restriction, including without limitation the rights
  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  * copies of the Software, and to permit persons to whom the Software is
  * furnished to do so, subject to the following conditions:
  *
  * The above copyright notice and this permission notice shall be included in
  * all copies or substantial portions of the Software.
  *
  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  * THE SOFTWARE.
  */

 #include "model/electrical/Multiplexer.h"

 #include <cmath>

 #include "model/PortInfo.h"
 #include "model/TransitionInfo.h"
 #include "model/EventInfo.h"
 #include "model/timing_graph/ElectricalDriverMultiplier.h"
 #include "model/timing_graph/ElectricalNet.h"
 #include "model/std_cells/StdCell.h"
 #include "model/std_cells/StdCellLib.h"

 namespace DSENT
 {
     Multiplexer::Multiplexer(const String& instance_name_, const TechModel* tech_model_)
         : ElectricalModel(instance_name_, tech_model_)
     {
         initParameters();
         initProperties();
     }

     Multiplexer::~Multiplexer()
     {}

     void Multiplexer::initParameters()
     {
         addParameterName("NumberInputs");
         addParameterName("NumberBits");
         addParameterName("BitDuplicate", "TRUE");
         addParameterName("IsTopLevel", "TRUE");
         return;
     }

     void Multiplexer::initProperties()
     {
         return;
     }

     Multiplexer* Multiplexer::clone() const
     {
         return NULL;
     }

     void Multiplexer::constructModel()
     {
         // Get parameters
         unsigned int number_bits = (unsigned int) getParameter("NumberBits");
         unsigned int number_inputs = (unsigned int) getParameter("NumberInputs");
         unsigned int number_selects = (unsigned int) ceil(log2((double) number_inputs));
         bool bit_duplicate = (bool) getParameter("BitDuplicate");
         bool is_top_level = getParameter("IsTopLevel").toBool();

         ASSERT(number_inputs > 0, "[Error] " + getInstanceName() + " -> Number of inputs must be > 0!");
         ASSERT(number_bits > 0, "[Error] " + getInstanceName() + " -> Number of bits must be > 0!");

         //Construct electrical ports and nets
         //Create each input port
         for(unsigned int i = 0; i < number_inputs; ++i)
             createInputPort(    "In" + (String) i, makeNetIndex(0, number_bits-1));
         //Create select signals
         for(unsigned int i = 0; i < number_selects; ++i)
         {
             createInputPort(    "Sel" + (String)i);
         }
         //Create output
         createOutputPort(   "Out", makeNetIndex(0, number_bits-1));

         //Create energy, power, and area results
         createElectricalResults();
         getEventInfo("Idle")->setStaticTransitionInfos();
         createElectricalEventResult("Mux");

         //Number of inputs on the 0 side
         unsigned int inputs_0 = (unsigned int) ceil((double) number_inputs / 2.0);
         unsigned int selects_0 = (unsigned int) ceil(log2((double) inputs_0));
         //Number of inputs on the 1 side
         unsigned int inputs_1 = (unsigned int) floor((double) number_inputs / 2.0);
         unsigned int selects_1 = (unsigned int) ceil(log2((double) inputs_1));

         //Depending on whether we want to create a 1-bit instance and have it multiplied
         //up by number of bits or actually instantiate number_bits of 1-bit instances.
         //Recursively instantiates smaller multiplexers
         if (bit_duplicate || number_bits == 1)
         {
             //If it is just a 1-input multiplexer, just connect output to input and be done
             if (number_inputs == 1)
             {
                 assign("Out", "In0");
             }
             else
             {
                 //If it is more than 1 input, instantiate two sub multiplexers (Mux_way0 and Mux_way1)
                 //and create a final 2:1 mux (muxf) to select between them
                 String mux0_name = "Mux_way0";
                 String mux1_name = "Mux_way1";
                 String muxf_name = "Mux2_i" + (String)number_inputs;

                 Multiplexer* mux0 = new Multiplexer(mux0_name, getTechModel());
                 mux0->setParameter("NumberInputs", inputs_0);
                 mux0->setParameter("NumberBits", 1);
                 mux0->setParameter("BitDuplicate", "TRUE");
                 mux0->setParameter("IsTopLevel", "FALSE");
                 mux0->construct();

                 Multiplexer* mux1 = new Multiplexer(mux1_name, getTechModel());
                 mux1->setParameter("NumberInputs", inputs_1);
                 mux1->setParameter("NumberBits", 1);
                 mux1->setParameter("BitDuplicate", "TRUE");
                 mux1->setParameter("IsTopLevel", "FALSE");
                 mux1->construct();

                 StdCell* muxf = getTechModel()->getStdCellLib()->createStdCell("MUX2", muxf_name);
                 muxf->construct();

                 // TODO hack
                 // create selector driver at the top level
                 if(is_top_level)
                 {
                     for(unsigned int i = 0; i < number_selects; ++i)
                     {
                         StdCell* selinv0 = getTechModel()->getStdCellLib()->createStdCell("INV", String::format("Sel%dInv0", i));
                         StdCell* selinv1 = getTechModel()->getStdCellLib()->createStdCell("INV", String::format("Sel%dInv1", i));
                         selinv0->construct();
                         selinv1->construct();

                         addSubInstances(selinv0, 1.0);
                         addElectricalSubResults(selinv0, 1.0);
                         addSubInstances(selinv1, 1.0);
                         addElectricalSubResults(selinv1, 1.0);
                         getEventResult("Mux")->addSubResult(selinv0->getEventResult("INV"), String::format("Sel%dInv0", i), 1.0);
                         getEventResult("Mux")->addSubResult(selinv1->getEventResult("INV"), String::format("Sel%dInv1", i), 1.0);
                     }
                 }

                 //Create outputs of way0 and way1 multiplexers with final mux
                 createNet("way0Out");
                 createNet("way1Out");
                 portConnect(mux0, "Out", "way0Out");
                 portConnect(mux1, "Out", "way1Out");
                 portConnect(muxf, "A", "way0Out");
                 portConnect(muxf, "B", "way1Out");

                 // TODO hack
                 // Connect selector bits
                 if(is_top_level)
                 {
                     for(unsigned int i = 0; i < number_selects; ++i)
                     {
                         ElectricalModel* selinv0 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv0", i));
                         ElectricalModel* selinv1 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv1", i));
                         createNet("SelInv" + (String)i);
                         createNet("SelBuf" + (String)i);
                         portConnect(selinv0, "A", "Sel" + (String)i);
                         portConnect(selinv0, "Y", "SelInv" + (String)i);
                         portConnect(selinv1, "A", "SelInv" + (String)i);
                         portConnect(selinv1, "Y", "SelBuf" + (String)i);
                     }
                 }
                 //Connect inputs to the sub multiplexers.
                 //Note that multiple inputs are connected to the mux0 and mux1 input and the
                 //selector signals are connected multiple times. This is just so that everything
                 //is loaded appropriately since bit duplication is applied
                 for (unsigned int n = 0; n < number_bits; ++n)
                 {
                     //Connect inputs
                     for (unsigned int i = 0; i < inputs_0; ++i)
                         portConnect(mux0, "In" + (String) i, "In" + (String) i, makeNetIndex(n));
                     for (unsigned int i = 0; i < inputs_1; ++i)
                         portConnect(mux1, "In" + (String) i, "In" + (String) (i + inputs_0), makeNetIndex(n));
                     // TODO hack
                     if(is_top_level)
                     {
                         //Connect selector bits
                         for (unsigned int i = 0; i < selects_0; ++i)
                             portConnect(mux0, "Sel" + (String)i, "SelBuf" + (String)i);
                         for (unsigned int i = 0; i < selects_1; ++i)
                             portConnect(mux1, "Sel" + (String)i, "SelBuf" + (String)i);
                         portConnect(muxf, "S0", "SelBuf" + (String)(number_selects - 1));
                     }
                     else
                     {
                         //Connect selector bits
                         for (unsigned int i = 0; i < selects_0; ++i)
                             portConnect(mux0, "Sel" + (String)i, "Sel" + (String)i);
                         for (unsigned int i = 0; i < selects_1; ++i)
                             portConnect(mux1, "Sel" + (String)i, "Sel" + (String)i);
                         portConnect(muxf, "S0", "Sel" + (String)(number_selects - 1));
                     }
                 }

                 //Connect final mux to outputs
                 //Because we use bit duplication and so there is only only one multiplexer
                 //instance, we must use driver multiplier to drive each output appropriately
                 if (number_bits == 1)
                     portConnect(muxf, "Y", "Out");
                 else
                 {
                     createNet("OutTemp");
                     createDriverMultiplier("OutMult");
                     ElectricalDriverMultiplier* drive_mult = getDriverMultiplier("OutMult");
                     portConnect(muxf, "Y", "OutTemp");
                     getNet("OutTemp")->addDownstreamNode(drive_mult);
                     for (unsigned int n = 0; n < number_bits; ++n)
                         drive_mult->addDownstreamNode(getNet("Out", makeNetIndex(n)));
                 }

                 //Add area, power, and event results for each mux
                 addSubInstances(mux0, number_bits);
                 addElectricalSubResults(mux0, number_bits);
                 addSubInstances(mux1, number_bits);
                 addElectricalSubResults(mux1, number_bits);
                 addSubInstances(muxf, number_bits);
                 addElectricalSubResults(muxf, number_bits);
                 getEventResult("Mux")->addSubResult(mux0->getEventResult("Mux"), mux0_name, number_bits);
                 getEventResult("Mux")->addSubResult(mux1->getEventResult("Mux"), mux1_name, number_bits);
                 getEventResult("Mux")->addSubResult(muxf->getEventResult("MUX2"), muxf_name, number_bits);

             }

         }
         else
         {
             //Instantiate a bunch of 1-bit multiplexers
             for (unsigned int n = 0; n < number_bits; ++n)
             {
                 String mux_name = "Mux_bit" + (String) n;

                 Multiplexer* mux = new Multiplexer(mux_name, getTechModel());
                 mux->setParameter("NumberInputs", number_inputs);
                 mux->setParameter("NumberBits", 1);
                 mux->setParameter("BitDuplicate", "TRUE");
                 mux->construct();

                 // Connect inputs
                 for (unsigned int i = 0; i < number_inputs; ++i)
                     portConnect(mux, "In" + (String) i, "In" + (String) i, makeNetIndex(n));
                 for(unsigned int i = 0; i < number_selects; ++i)
                     portConnect(mux, "Sel" + (String)i, "Sel" + (String)i);
                 portConnect(mux, "Out", "Out", makeNetIndex(n));

                 //Add area, power, and event results for each mux
                 addSubInstances(mux, 1.0);
                 addElectricalSubResults(mux, 1.0);
                 getEventResult("Mux")->addSubResult(mux->getEventResult("Mux"), mux_name, 1.0);
             }
         }

         return;
     }

     void Multiplexer::propagateTransitionInfo()
     {
         // The only thing can be updated are the input probabilities...so we will update them
         unsigned int number_bits = (unsigned int) getParameter("NumberBits");
         unsigned int number_inputs = (unsigned int) getParameter("NumberInputs");
         unsigned int number_selects = (unsigned int) ceil(log2((double) number_inputs));
         bool bit_duplicate = (bool) getParameter("BitDuplicate");
         bool is_top_level = getParameter("IsTopLevel").toBool();

         //Number of inputs on the 0 side
         unsigned int inputs_0 = (unsigned int) ceil((double) number_inputs / 2.0);
         unsigned int selects_0 = (unsigned int) ceil(log2((double) inputs_0));

         //Number of inputs on the 1 side
         unsigned int inputs_1 = (unsigned int) floor((double) number_inputs / 2.0);
         unsigned int selects_1 = (unsigned int) ceil(log2((double) inputs_1));

         if (bit_duplicate || number_bits == 1)
         {
             if (number_inputs == 1)
             {
                 //If theres only 1 input, output transition = input transition
                 propagatePortTransitionInfo("Out", "In0");
             }
             else
             {
                 // Update sub muxes with appropriate probabilities
                 ElectricalModel* mux0 = (ElectricalModel*)getSubInstance("Mux_way0");
                 for(unsigned int i = 0; i < inputs_0; ++i)
                 {
                     propagatePortTransitionInfo(mux0, "In" + (String)i, "In" + (String)i);
                 }
                 for(unsigned int i = 0; i < selects_0; ++i)
                 {
                     propagatePortTransitionInfo(mux0, "Sel" + (String)i, "Sel" + (String)i);
                 }
                 mux0->use();
                 ElectricalModel* mux1 = (ElectricalModel*)getSubInstance("Mux_way1");
                 for(unsigned int i = 0; i < inputs_1; ++i)
                 {
                     propagatePortTransitionInfo(mux1, "In" + (String)i, "In" + (String)(i + inputs_0));
                 }
                 for(unsigned int i = 0; i < selects_1; ++i)
                 {
                     propagatePortTransitionInfo(mux1, "Sel" + (String)i, "Sel" + (String)i);
                 }
                 mux1->use();
                 ElectricalModel* muxf = (ElectricalModel*)getSubInstance("Mux2_i" + (String)number_inputs);
                 propagatePortTransitionInfo(muxf, "A", mux0, "Out");
                 propagatePortTransitionInfo(muxf, "B", mux1, "Out");
                 propagatePortTransitionInfo(muxf, "S0", "Sel" + (String)(number_selects-1));
                 muxf->use();

                 // TODO hack
                 if(is_top_level)
                 {
                     for(unsigned int i = 0; i < number_selects; ++i)
                     {
                         ElectricalModel* selinv0 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv0", i));
                         ElectricalModel* selinv1 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv1", i));
                         propagatePortTransitionInfo(selinv0, "A", "Sel" + (String)i);
                         selinv0->use();
                         propagatePortTransitionInfo(selinv1, "A", selinv0, "Y");
                         selinv1->use();
                     }
                 }

                 // Set output transition
                 propagatePortTransitionInfo("Out", muxf, "Y");
             }
         }
         else
         {
             // Go through each bit and set the appropriate probability
             for (unsigned int n = 0; n < number_bits; ++n)
             {
                 ElectricalModel* mux_bit = (ElectricalModel*)getSubInstance("Mux_bit" + (String) n);
                 for(unsigned int i = 0; i < number_inputs; ++i)
                 {
                     propagatePortTransitionInfo(mux_bit, "In" + (String)i, "In" + (String)i);
                 }
                 for(unsigned int i = 0; i < number_selects; ++i)
                 {
                     propagatePortTransitionInfo(mux_bit, "Sel" + (String)i, "Sel" + (String)i);
                 }
                 mux_bit->use();
             }

             // Set output probability to be average that of probabilties of each output bit
             ElectricalModel* mux_bit = (ElectricalModel*)getSubInstance("Mux_bit0");
             propagatePortTransitionInfo("Out", mux_bit, "Out");
         }
         return;
     }

 } // namespace DSENT
	/* Copyright (c) 2012 Massachusetts Institute of Technology
	*
	* Permission is hereby granted, free of charge, to any person obtaining a copy
	* of this software and associated documentation files (the "Software"), to deal
	* in the Software without restriction, including without limitation the rights
	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	* copies of the Software, and to permit persons to whom the Software is
	* furnished to do so, subject to the following conditions:
	*
	* The above copyright notice and this permission notice shall be included in
	* all copies or substantial portions of the Software.
	*
	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
	* THE SOFTWARE.
	*/

	#include "model/electrical/Multiplexer.h"

	#include <cmath>

	#include "model/PortInfo.h"
	#include "model/TransitionInfo.h"
	#include "model/EventInfo.h"
	#include "model/timing_graph/ElectricalDriverMultiplier.h"
	#include "model/timing_graph/ElectricalNet.h"
	#include "model/std_cells/StdCell.h"
	#include "model/std_cells/StdCellLib.h"

	namespace DSENT
	{
	Multiplexer::Multiplexer(const String& instance_name_, const TechModel* tech_model_)
	: ElectricalModel(instance_name_, tech_model_)
	{
	initParameters();
	initProperties();
	}

	Multiplexer::~Multiplexer()
	{}

	void Multiplexer::initParameters()
	{
	addParameterName("NumberInputs");
	addParameterName("NumberBits");
	addParameterName("BitDuplicate", "TRUE");
	addParameterName("IsTopLevel", "TRUE");
	return;
	}

	void Multiplexer::initProperties()
	{
	return;
	}

	Multiplexer* Multiplexer::clone() const
	{
	return NULL;
	}

	void Multiplexer::constructModel()
	{
	// Get parameters
	unsigned int number_bits = (unsigned int) getParameter("NumberBits");
	unsigned int number_inputs = (unsigned int) getParameter("NumberInputs");
	unsigned int number_selects = (unsigned int) ceil(log2((double) number_inputs));
	bool bit_duplicate = (bool) getParameter("BitDuplicate");
	bool is_top_level = getParameter("IsTopLevel").toBool();

	ASSERT(number_inputs > 0, "[Error] " + getInstanceName() + " -> Number of inputs must be > 0!");
	ASSERT(number_bits > 0, "[Error] " + getInstanceName() + " -> Number of bits must be > 0!");

	//Construct electrical ports and nets
	//Create each input port
	for(unsigned int i = 0; i < number_inputs; ++i)
	createInputPort( "In" + (String) i, makeNetIndex(0, number_bits-1));
	//Create select signals
	for(unsigned int i = 0; i < number_selects; ++i)
	{
	createInputPort( "Sel" + (String)i);
	}
	//Create output
	createOutputPort( "Out", makeNetIndex(0, number_bits-1));

	//Create energy, power, and area results
	createElectricalResults();
	getEventInfo("Idle")->setStaticTransitionInfos();
	createElectricalEventResult("Mux");

	//Number of inputs on the 0 side
	unsigned int inputs_0 = (unsigned int) ceil((double) number_inputs / 2.0);
	unsigned int selects_0 = (unsigned int) ceil(log2((double) inputs_0));
	//Number of inputs on the 1 side
	unsigned int inputs_1 = (unsigned int) floor((double) number_inputs / 2.0);
	unsigned int selects_1 = (unsigned int) ceil(log2((double) inputs_1));

	//Depending on whether we want to create a 1-bit instance and have it multiplied
	//up by number of bits or actually instantiate number_bits of 1-bit instances.
	//Recursively instantiates smaller multiplexers
	if (bit_duplicate \|\| number_bits == 1)
	{
	//If it is just a 1-input multiplexer, just connect output to input and be done
	if (number_inputs == 1)
	{
	assign("Out", "In0");
	}
	else
	{
	//If it is more than 1 input, instantiate two sub multiplexers (Mux_way0 and Mux_way1)
	//and create a final 2:1 mux (muxf) to select between them
	String mux0_name = "Mux_way0";
	String mux1_name = "Mux_way1";
	String muxf_name = "Mux2_i" + (String)number_inputs;

	Multiplexer* mux0 = new Multiplexer(mux0_name, getTechModel());
	mux0->setParameter("NumberInputs", inputs_0);
	mux0->setParameter("NumberBits", 1);
	mux0->setParameter("BitDuplicate", "TRUE");
	mux0->setParameter("IsTopLevel", "FALSE");
	mux0->construct();

	Multiplexer* mux1 = new Multiplexer(mux1_name, getTechModel());
	mux1->setParameter("NumberInputs", inputs_1);
	mux1->setParameter("NumberBits", 1);
	mux1->setParameter("BitDuplicate", "TRUE");
	mux1->setParameter("IsTopLevel", "FALSE");
	mux1->construct();

	StdCell* muxf = getTechModel()->getStdCellLib()->createStdCell("MUX2", muxf_name);
	muxf->construct();

	// TODO hack
	// create selector driver at the top level
	if(is_top_level)
	{
	for(unsigned int i = 0; i < number_selects; ++i)
	{
	StdCell* selinv0 = getTechModel()->getStdCellLib()->createStdCell("INV", String::format("Sel%dInv0", i));
	StdCell* selinv1 = getTechModel()->getStdCellLib()->createStdCell("INV", String::format("Sel%dInv1", i));
	selinv0->construct();
	selinv1->construct();

	addSubInstances(selinv0, 1.0);
	addElectricalSubResults(selinv0, 1.0);
	addSubInstances(selinv1, 1.0);
	addElectricalSubResults(selinv1, 1.0);
	getEventResult("Mux")->addSubResult(selinv0->getEventResult("INV"), String::format("Sel%dInv0", i), 1.0);
	getEventResult("Mux")->addSubResult(selinv1->getEventResult("INV"), String::format("Sel%dInv1", i), 1.0);
	}
	}

	//Create outputs of way0 and way1 multiplexers with final mux
	createNet("way0Out");
	createNet("way1Out");
	portConnect(mux0, "Out", "way0Out");
	portConnect(mux1, "Out", "way1Out");
	portConnect(muxf, "A", "way0Out");
	portConnect(muxf, "B", "way1Out");

	// TODO hack
	// Connect selector bits
	if(is_top_level)
	{
	for(unsigned int i = 0; i < number_selects; ++i)
	{
	ElectricalModel* selinv0 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv0", i));
	ElectricalModel* selinv1 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv1", i));
	createNet("SelInv" + (String)i);
	createNet("SelBuf" + (String)i);
	portConnect(selinv0, "A", "Sel" + (String)i);
	portConnect(selinv0, "Y", "SelInv" + (String)i);
	portConnect(selinv1, "A", "SelInv" + (String)i);
	portConnect(selinv1, "Y", "SelBuf" + (String)i);
	}
	}
	//Connect inputs to the sub multiplexers.
	//Note that multiple inputs are connected to the mux0 and mux1 input and the
	//selector signals are connected multiple times. This is just so that everything
	//is loaded appropriately since bit duplication is applied
	for (unsigned int n = 0; n < number_bits; ++n)
	{
	//Connect inputs
	for (unsigned int i = 0; i < inputs_0; ++i)
	portConnect(mux0, "In" + (String) i, "In" + (String) i, makeNetIndex(n));
	for (unsigned int i = 0; i < inputs_1; ++i)
	portConnect(mux1, "In" + (String) i, "In" + (String) (i + inputs_0), makeNetIndex(n));
	// TODO hack
	if(is_top_level)
	{
	//Connect selector bits
	for (unsigned int i = 0; i < selects_0; ++i)
	portConnect(mux0, "Sel" + (String)i, "SelBuf" + (String)i);
	for (unsigned int i = 0; i < selects_1; ++i)
	portConnect(mux1, "Sel" + (String)i, "SelBuf" + (String)i);
	portConnect(muxf, "S0", "SelBuf" + (String)(number_selects - 1));
	}
	else
	{
	//Connect selector bits
	for (unsigned int i = 0; i < selects_0; ++i)
	portConnect(mux0, "Sel" + (String)i, "Sel" + (String)i);
	for (unsigned int i = 0; i < selects_1; ++i)
	portConnect(mux1, "Sel" + (String)i, "Sel" + (String)i);
	portConnect(muxf, "S0", "Sel" + (String)(number_selects - 1));
	}
	}

	//Connect final mux to outputs
	//Because we use bit duplication and so there is only only one multiplexer
	//instance, we must use driver multiplier to drive each output appropriately
	if (number_bits == 1)
	portConnect(muxf, "Y", "Out");
	else
	{
	createNet("OutTemp");
	createDriverMultiplier("OutMult");
	ElectricalDriverMultiplier* drive_mult = getDriverMultiplier("OutMult");
	portConnect(muxf, "Y", "OutTemp");
	getNet("OutTemp")->addDownstreamNode(drive_mult);
	for (unsigned int n = 0; n < number_bits; ++n)
	drive_mult->addDownstreamNode(getNet("Out", makeNetIndex(n)));
	}

	//Add area, power, and event results for each mux
	addSubInstances(mux0, number_bits);
	addElectricalSubResults(mux0, number_bits);
	addSubInstances(mux1, number_bits);
	addElectricalSubResults(mux1, number_bits);
	addSubInstances(muxf, number_bits);
	addElectricalSubResults(muxf, number_bits);
	getEventResult("Mux")->addSubResult(mux0->getEventResult("Mux"), mux0_name, number_bits);
	getEventResult("Mux")->addSubResult(mux1->getEventResult("Mux"), mux1_name, number_bits);
	getEventResult("Mux")->addSubResult(muxf->getEventResult("MUX2"), muxf_name, number_bits);

	}

	}
	else
	{
	//Instantiate a bunch of 1-bit multiplexers
	for (unsigned int n = 0; n < number_bits; ++n)
	{
	String mux_name = "Mux_bit" + (String) n;

	Multiplexer* mux = new Multiplexer(mux_name, getTechModel());
	mux->setParameter("NumberInputs", number_inputs);
	mux->setParameter("NumberBits", 1);
	mux->setParameter("BitDuplicate", "TRUE");
	mux->construct();

	// Connect inputs
	for (unsigned int i = 0; i < number_inputs; ++i)
	portConnect(mux, "In" + (String) i, "In" + (String) i, makeNetIndex(n));
	for(unsigned int i = 0; i < number_selects; ++i)
	portConnect(mux, "Sel" + (String)i, "Sel" + (String)i);
	portConnect(mux, "Out", "Out", makeNetIndex(n));

	//Add area, power, and event results for each mux
	addSubInstances(mux, 1.0);
	addElectricalSubResults(mux, 1.0);
	getEventResult("Mux")->addSubResult(mux->getEventResult("Mux"), mux_name, 1.0);
	}
	}

	return;
	}

	void Multiplexer::propagateTransitionInfo()
	{
	// The only thing can be updated are the input probabilities...so we will update them
	unsigned int number_bits = (unsigned int) getParameter("NumberBits");
	unsigned int number_inputs = (unsigned int) getParameter("NumberInputs");
	unsigned int number_selects = (unsigned int) ceil(log2((double) number_inputs));
	bool bit_duplicate = (bool) getParameter("BitDuplicate");
	bool is_top_level = getParameter("IsTopLevel").toBool();

	//Number of inputs on the 0 side
	unsigned int inputs_0 = (unsigned int) ceil((double) number_inputs / 2.0);
	unsigned int selects_0 = (unsigned int) ceil(log2((double) inputs_0));

	//Number of inputs on the 1 side
	unsigned int inputs_1 = (unsigned int) floor((double) number_inputs / 2.0);
	unsigned int selects_1 = (unsigned int) ceil(log2((double) inputs_1));

	if (bit_duplicate \|\| number_bits == 1)
	{
	if (number_inputs == 1)
	{
	//If theres only 1 input, output transition = input transition
	propagatePortTransitionInfo("Out", "In0");
	}
	else
	{
	// Update sub muxes with appropriate probabilities
	ElectricalModel* mux0 = (ElectricalModel*)getSubInstance("Mux_way0");
	for(unsigned int i = 0; i < inputs_0; ++i)
	{
	propagatePortTransitionInfo(mux0, "In" + (String)i, "In" + (String)i);
	}
	for(unsigned int i = 0; i < selects_0; ++i)
	{
	propagatePortTransitionInfo(mux0, "Sel" + (String)i, "Sel" + (String)i);
	}
	mux0->use();
	ElectricalModel* mux1 = (ElectricalModel*)getSubInstance("Mux_way1");
	for(unsigned int i = 0; i < inputs_1; ++i)
	{
	propagatePortTransitionInfo(mux1, "In" + (String)i, "In" + (String)(i + inputs_0));
	}
	for(unsigned int i = 0; i < selects_1; ++i)
	{
	propagatePortTransitionInfo(mux1, "Sel" + (String)i, "Sel" + (String)i);
	}
	mux1->use();
	ElectricalModel* muxf = (ElectricalModel*)getSubInstance("Mux2_i" + (String)number_inputs);
	propagatePortTransitionInfo(muxf, "A", mux0, "Out");
	propagatePortTransitionInfo(muxf, "B", mux1, "Out");
	propagatePortTransitionInfo(muxf, "S0", "Sel" + (String)(number_selects-1));
	muxf->use();

	// TODO hack
	if(is_top_level)
	{
	for(unsigned int i = 0; i < number_selects; ++i)
	{
	ElectricalModel* selinv0 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv0", i));
	ElectricalModel* selinv1 = (ElectricalModel*)getSubInstance(String::format("Sel%dInv1", i));
	propagatePortTransitionInfo(selinv0, "A", "Sel" + (String)i);
	selinv0->use();
	propagatePortTransitionInfo(selinv1, "A", selinv0, "Y");
	selinv1->use();
	}
	}

	// Set output transition
	propagatePortTransitionInfo("Out", muxf, "Y");
	}
	}
	else
	{
	// Go through each bit and set the appropriate probability
	for (unsigned int n = 0; n < number_bits; ++n)
	{
	ElectricalModel* mux_bit = (ElectricalModel*)getSubInstance("Mux_bit" + (String) n);
	for(unsigned int i = 0; i < number_inputs; ++i)
	{
	propagatePortTransitionInfo(mux_bit, "In" + (String)i, "In" + (String)i);
	}
	for(unsigned int i = 0; i < number_selects; ++i)
	{
	propagatePortTransitionInfo(mux_bit, "Sel" + (String)i, "Sel" + (String)i);
	}
	mux_bit->use();
	}

	// Set output probability to be average that of probabilties of each output bit
	ElectricalModel* mux_bit = (ElectricalModel*)getSubInstance("Mux_bit0");
	propagatePortTransitionInfo("Out", mux_bit, "Out");
	}
	return;
	}

	} // namespace DSENT