/***************************************************************************
 *   Copyright (C) 2003-2004 by David Saxton                               *
 *   david@bluehaze.org                                                    *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 ***************************************************************************/

#include "matrix.h"
#include "voltagesource.h"
#include "elementset.h"

VoltageSource::VoltageSource( const double voltage )
	: Element::Element()
{
	m_v = voltage;
	m_numCBranches = 1;
	m_numCNodes = 2;
}

VoltageSource::~VoltageSource()
{
}

void VoltageSource::setVoltage( const double v )
{
	if ( m_v == v ) return;
	
	if (p_eSet)
		p_eSet->setCacheInvalidated();
	
	m_v = v;
	add_initial_dc();
}


void VoltageSource::add_map()
{
	if (!b_status) return;
	
	if ( !p_cnode[0]->isGround )
	{
		p_A->setUse_b( p_cnode[0]->n(), p_cbranch[0]->n(), Map::et_constant, true );
		p_A->setUse_c( p_cbranch[0]->n(), p_cnode[0]->n(), Map::et_constant, true );
	}
	
	if ( !p_cnode[1]->isGround )
	{
		p_A->setUse_b( p_cnode[1]->n(), p_cbranch[0]->n(), Map::et_constant, true );
		p_A->setUse_c( p_cbranch[0]->n(), p_cnode[1]->n(), Map::et_constant, true );
	}
}


void VoltageSource::add_initial_dc()
{
	if (!b_status)
		return;
	
	A_b( 0, 0 ) = -1;
	A_c( 0, 0 ) = -1;
	A_b( 1, 0 ) = 1;
	A_c( 0, 1 ) = 1;
	
	b_v( 0 ) = m_v;
}

void VoltageSource::updateCurrents()
{
	if (!b_status) return;
	m_cnodeI[0] = p_cbranch[0]->i;
	m_cnodeI[1] = -m_cnodeI[0];
}