/*
 * Copyright (c) 2008 Antoine Levitt (smeuuh@gmail.com)
 * Copyright (c) 2008 Steven Herbst (herbst@mit.edu)
 * Copyright (c) 2009 Julien Nadeau (vedge@hypertriton.com)
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE
 * USE OF THIS SOFTWARE EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/*
 * Model for PNP transistor
 */

#include <core/core.h>
#include "generic.h"

enum {
	PORT_B = 1,
	PORT_E = 2,
	PORT_C = 3
};
const ES_Port esPNPPorts[] = {
	{ 0, "" },
	{ PORT_B, "B" },	/* Base */
	{ PORT_E, "E" },	/* Emitter */
	{ PORT_C, "C"},		/* Collector */
	{ -1 },
};

static M_Real
vEB(ES_PNP *u)
{
	return VPORT(u,PORT_E)-VPORT(u,PORT_B);
}
static M_Real
vCB(ES_PNP *u)
{
	return VPORT(u,PORT_C)-VPORT(u,PORT_B);
}

static M_Real
VebPrevStep(ES_PNP *u)
{
	return V_PREV_STEP(u,PORT_E,1)-V_PREV_STEP(u,PORT_B,1);
}
static M_Real
VcbPrevStep(ES_PNP *u)
{
	return V_PREV_STEP(u,PORT_C,1)-V_PREV_STEP(u,PORT_B,1);
}

static void
ResetModel(ES_PNP *u)
{
	u->gPiF=1.0;
	u->gPiR=1.0;
	u->gmF=0.0;
	u->gmR=0.0;
	u->go=1.0;

	u->Ibf_eq = 0.0;
	u->Ibr_eq = 0.0;
	u->Icc_eq = 0.0;

	u->VebPrevIter = 0.7;
	u->VcbPrevIter = 0.7;
}

static void
UpdateModel(ES_PNP *u, ES_SimDC *dc, M_Real vEB, M_Real vCB)
{
	M_Real vEC=vEB-vCB;

	M_Real VebDiff = vEB - u->VebPrevIter;
	M_Real VcbDiff = vCB - u->VcbPrevIter;

	if (M_Fabs(VebDiff) > u->Vt)
	{
		vEB = u->VebPrevIter + M_Fabs(VebDiff)/VebDiff*u->Vt;
		dc->isDamped = 1;
	}

	if (M_Fabs(VcbDiff) > u->Vt)
	{
		vCB = u->VcbPrevIter + M_Fabs(VcbDiff)/VcbDiff*u->Vt;
		dc->isDamped = 1;
	}

	u->VebPrevIter = vEB;
	u->VcbPrevIter = vCB;

	M_Real Ibf = (u->Ifs/u->betaF)*(M_Exp(vEB/u->Vt)-1);
	M_Real Ibr = (u->Irs/u->betaR)*(M_Exp(vCB/u->Vt)-1);

	M_Real Icc = (u->betaF*Ibf-u->betaR*Ibr)*(1+vEC/u->Va);

	u->gPiF = Ibf/u->Vt;
	u->gPiR = Ibr/u->Vt;

	u->go = Icc/u->Va;

	u->gmF = u->betaF*u->gPiF;
	u->gmR = u->betaR*u->gPiR;

    	u->Ibf_eq = Ibf - u->gPiF*vEB;
	u->Ibr_eq = Ibr - u->gPiR*vCB;
	u->Icc_eq = Icc - u->gmF*vEB + u->gmR*vCB - u->go*vEC;
}

static __inline__ void
Stamp(ES_PNP *u, ES_SimDC *dc)
{
	StampConductance(u->gPiF,u->sc_be);
	StampConductance(u->gPiR,u->sc_bc);
	StampConductance(u->go,u->sc_ec);
	StampVCCS(u->gmF,u->sv_ebec);
	StampVCCS(u->gmR,u->sv_cbce);
	StampCurrentSource(u->Ibf_eq, u->si_be);
	StampCurrentSource(u->Ibr_eq, u->si_bc);
	StampCurrentSource(u->Icc_eq, u->si_ce);
}

static int
DC_SimBegin(void *obj, ES_SimDC *dc)
{
	ES_PNP *u = obj;
	Uint b = PNODE(u,PORT_B);
	Uint e = PNODE(u,PORT_E);
	Uint c = PNODE(u,PORT_C);

	InitStampConductance(b,e, u->sc_be, dc);
	InitStampConductance(b,c, u->sc_bc, dc);
	InitStampConductance(e,c, u->sc_ec, dc);
	InitStampVCCS(e,b,e,c, u->sv_ebec, dc);
	InitStampVCCS(c,b,c,e, u->sv_cbce, dc);
	InitStampCurrentSource(b,e, u->si_be, dc);
	InitStampCurrentSource(b,c, u->si_bc, dc);
	InitStampCurrentSource(c,e, u->si_ce, dc);

	ResetModel(u);
	Stamp(u,dc);
	return (0);
}

static void
DC_StepBegin(void *obj, ES_SimDC *dc)
{
	ES_PNP *u = obj;

	if (dc->inputStep) {
		ResetModel(u);
	} else {
		u->VebPrevIter = VebPrevStep(u);
		u->VcbPrevIter = VcbPrevStep(u);
		UpdateModel(u,dc,VebPrevStep(u),VcbPrevStep(u));
	}
	Stamp(u,dc);
}

static void
DC_StepIter(void *obj, ES_SimDC *dc)
{
        ES_PNP *u = obj;

	UpdateModel(u,dc,vEB(u),vCB(u));
	Stamp(u,dc);
}

static void
Init(void *p)
{
	ES_PNP *u = p;

	ES_InitPorts(u, esPNPPorts);

	u->Vt = 0.025;
	u->Va = 10.0;
	u->betaF = 100.0;
	u->betaR = 1.0;
	u->Ifs = 1e-14;
	u->Irs = 1e-14;

	COMPONENT(u)->dcSimBegin = DC_SimBegin;
	COMPONENT(u)->dcStepBegin = DC_StepBegin;
	COMPONENT(u)->dcStepIter = DC_StepIter;

	M_BindReal(u, "Vt",	&u->Vt);
	M_BindReal(u, "Va",	&u->Va);
	M_BindReal(u, "betaF",	&u->betaF);
	M_BindReal(u, "betaR",	&u->betaR);
	M_BindReal(u, "Ifs",	&u->Ifs);
	M_BindReal(u, "Irs",	&u->Irs);
}

static void *
Edit(void *p)
{
	ES_PNP *u = p;
	AG_Box *box = AG_BoxNewVert(NULL, AG_BOX_EXPAND);

	M_NumericalNewRealPNZ(box, 0, "V", _("Thermal voltage: "), &u->Vt);
	M_NumericalNewRealPNZ(box, 0, "V", _("Early voltage: "), &u->Va);
	M_NumericalNewRealPNZ(box, 0, NULL, _("Forward Beta: "), &u->betaF);
	M_NumericalNewRealPNZ(box, 0, NULL, _("Reverse Beta: "), &u->betaR);
	M_NumericalNewRealPNZ(box, 0, "pA", _("Forward Sat. Current: "), &u->Ifs);
	M_NumericalNewRealPNZ(box, 0, "pA", _("Reverse Sat. Current: "), &u->Irs);

	return (box);
}

ES_ComponentClass esPNPClass = {
	{
		"Edacious(Circuit:Component:PNP)"
		"@generic",
		sizeof(ES_PNP),
		{ 0,0 },
		Init,
		NULL,		/* reinit */
		NULL,		/* destroy */
		NULL,		/* load */
		NULL,		/* save */
		Edit		/* edit */
	},
	N_("Transistor (PNP)"),
	"U",
	"Generic|Nonlinear",
	NULL,
	NULL,			/* draw */
	NULL,			/* instance_menu */
	NULL,			/* class_menu */
	NULL,			/* export */
	NULL			/* connect */
};