/*
 * 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 PMOS transistor
 */

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

enum {
	PORT_G = 1,
	PORT_D = 2,
	PORT_S = 3
};
const ES_Port esPMOSPorts[] = {
	{ 0, "" },
	{ PORT_G, "G" },	/* Gate */
	{ PORT_D, "D" },	/* Drain */
	{ PORT_S, "S"},		/* Source */
	{ -1 },
};

static M_Real
vSD(ES_PMOS *u)
{
	return VPORT(u,PORT_S)-VPORT(u,PORT_D);
}
static M_Real
vSG(ES_PMOS *u)
{
	return VPORT(u,PORT_S)-VPORT(u,PORT_G);
}

static M_Real
VsdPrevStep(ES_PMOS *u)
{
	return V_PREV_STEP(u,PORT_S,1)-V_PREV_STEP(u,PORT_D,1);
}
static M_Real
VsgPrevStep(ES_PMOS *u)
{
	return V_PREV_STEP(u,PORT_S,1)-V_PREV_STEP(u,PORT_G,1);
}

static void
UpdateModel(ES_PMOS *u, M_Real vSG, M_Real vSD)
{
	M_Real I;

	if (vSG < u->Vt)
	{
		// Cutoff 

		I = 0;
		u->gm = 0;
		u->go = 0;

	}
	else if ((vSG-u->Vt) < vSD)
	{
		// Saturation

		M_Real vSat = vSG-u->Vt;

		I=(u->K)/2*vSat*vSat*(1+(vSD-vSat)/u->Va);
		u->gm=sqrt(2*(u->K)*I);
		u->go=I/(u->Va);

	}
	else
	{
		// Triode
		
		I=(u->K)*((vSG-(u->Vt))-vSD/2)*vSD;
		u->gm=(u->K)*vSD;
		u->go=(u->K)*((vSG-(u->Vt))-vSD);

	}

	u->Ieq = I - u->gm*vSG - u->go*vSD;

}

static __inline__ void
Stamp(ES_PMOS *u, ES_SimDC *dc)
{
	StampVCCS(u->gm, u->s_vccs);
	StampConductance(u->go, u->s_conductance);
	StampCurrentSource(u->Ieq, u->s_current);
}

static int
DC_SimBegin(void *obj, ES_SimDC *dc)
{
	ES_PMOS *u = obj;
	Uint g = PNODE(u,PORT_G);
	Uint d = PNODE(u,PORT_D);
	Uint s = PNODE(u,PORT_S);

	InitStampVCCS(s,g,s,d,u->s_vccs, dc);
	InitStampConductance(s,d,u->s_conductance, dc);
	InitStampCurrentSource(d,s, u->s_current, dc);

	u->gm=0.0;
	u->go=1.0;
	u->Ieq=0.0;
	Stamp(u,dc);
	return (0);
}

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

	UpdateModel(u,VsgPrevStep(u),VsdPrevStep(u));
	Stamp(u,dc);

}

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

	UpdateModel(u,vSG(u),vSD(u));
	Stamp(u,dc);
}

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

	ES_InitPorts(u, esPMOSPorts);
	u->Vt = 0.5;
	u->Va = 10;
	u->K = 1e-3;

	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, "K", &u->K);
}

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

	M_NumericalNewRealPNZ(box, 0, "V", _("Threshold voltage: "), &u->Vt);
	M_NumericalNewRealPNZ(box, 0, "V", _("Early voltage: "), &u->Va);
	M_NumericalNewRealPNZ(box, 0, NULL, _("K: "), &u->K);

	return (box);
}

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