KNGeometricAtomFactory.cpp

Go to the documentation of this file.

#include "stdafx.h"
#include <string.h>
#include "KNGeometricAtomFactory.h"

std::vector<CKNGeometricAtom> CKNGeometricAtomFactory::m_vectAtomList_1;
std::vector<CKNGeometricAtom> CKNGeometricAtomFactory::m_vectAtomList_2;

std::vector<CKNGeometricCoordination> CKNGeometricAtomFactory::m_vectNeighborRelation1To2;
std::vector<CKNGeometricCoordination> CKNGeometricAtomFactory::m_vectNeighborRelation2To1;


CKNGeometricAtomFactory::CKNGeometricAtomFactory()
{
}


CKNGeometricAtomFactory::~CKNGeometricAtomFactory()
{
}

void CKNGeometricAtomFactory::SetAtomCoordination(CKNGeometricAtom::ATOM_TYPE type, double x, double y, double z)
{
    CKNGeometricAtom            atom;

    atom.SetType(type, ATOM_NUM_BASE);
    atom.SetCoordination(x, y, z);

    switch (type)
    {
    case CKNGeometricAtom::A:
        m_vectAtomList_1.push_back(atom);
        break;
    case CKNGeometricAtom::C:
        m_vectAtomList_2.push_back(atom);
        break;
    }
}

void CKNGeometricAtomFactory::SetNeighborCoordination(CKNGeometricAtom::NEIGHOR_RELATION type, double x, double y, double z)
{
    CKNGeometricCoordination        relationCoordnation;

    relationCoordnation.SetCoordination(x, y, z);
    
    switch (type)
    {
    case CKNGeometricAtom::A2C:
        m_vectNeighborRelation1To2.push_back(relationCoordnation);
        break;
    case CKNGeometricAtom::C2A:
        m_vectNeighborRelation2To1.push_back(relationCoordnation);
        break;
    }
}

void CKNGeometricAtomFactory::SetRatio(double fRatio[3])
{
    int             i, nSize;

    nSize = m_vectAtomList_1.size();
    for (i = 0; i < nSize; ++i)
        m_vectAtomList_1[i].SetRatio(fRatio);

    nSize = m_vectAtomList_2.size();
    for (i = 0; i < nSize; ++i)
        m_vectAtomList_2[i].SetRatio(fRatio);

    nSize = m_vectNeighborRelation1To2.size();
    for (i = 0; i < nSize; ++i)
        m_vectNeighborRelation1To2[i].SetRatio(fRatio);

    nSize = m_vectNeighborRelation2To1.size();
    for (i = 0; i < nSize; ++i)
        m_vectNeighborRelation2To1[i].SetRatio(fRatio);
}

void CKNGeometricAtomFactory::SetUnitCellSize(double fSize[3])
{
    SetRatio(fSize);
}

void CKNGeometricAtomFactory::FreeFactoryData()
{
    m_vectAtomList_1.clear();
    m_vectAtomList_2.clear();
    m_vectNeighborRelation1To2.clear();
    m_vectNeighborRelation2To1.clear();
}

MATERIAL_INDEX CKNGeometricAtomFactory::GetMaterialNumber(char *pszMaterialName, CKNGeometricAtom::ATOM_TYPE type)
{
    if (!strcmp(pszMaterialName, "Si"))
    {
        return Si;
    }
    else if (!strcmp(pszMaterialName, "GaAs"))
    {
        //return GaAs;
        switch (type)
        {
        case CKNGeometricAtom::A:
            return Ga;
            break;
        case CKNGeometricAtom::C:
            return As;
            break;
        }
    }
    else if (!strcmp(pszMaterialName, "InAs"))
    {
        //return InAs;
        switch (type)
        {
        case CKNGeometricAtom::A:
            return In;
            break;
        case CKNGeometricAtom::C:
            return As;
            break;
        }
    }

    return ATOM_NUM_BASE;
}

MATERIAL_INDEX CKNGeometricAtomFactory::GetMaterialNumber(char *pszMaterialName)
{
    if (!strcmp(pszMaterialName, "Si"))
    {
        return Si;
    }
    else if (!strcmp(pszMaterialName, "GaAs"))
    {
        return GaAs;
    }
    else if (!strcmp(pszMaterialName, "InAs"))
    {
        return InAs;
    }

    return ATOM_NUM_BASE;
}

void CKNGeometricAtomFactory::InitAtomList()
{
    m_vectAtomList_1.clear();
    m_vectAtomList_2.clear();
}

void CKNGeometricAtomFactory::InitNeighborList()
{
    m_vectNeighborRelation1To2.clear();
    m_vectNeighborRelation2To1.clear();
}