| 42 |  | * | 
| 43 |  | *  Created by Charles F. Vardeman II on 17 Feb 2006. | 
| 44 |  | *  @author  Charles F. Vardeman II | 
| 45 | < | *  @version $Id: shapedLattice.cpp,v 1.2 2006-03-27 16:03:50 chuckv Exp $ | 
| 45 | > | *  @version $Id: shapedLattice.cpp,v 1.7 2008-09-10 19:51:43 cli2 Exp $ | 
| 46 |  | * | 
| 47 |  | */ | 
| 48 |  |  | 
| 49 | < |  | 
| 49 | > | #include <cstdlib> | 
| 50 |  | #include "lattice/shapedLattice.hpp" | 
| 51 |  | #include "math/Vector3.hpp" | 
| 52 |  | #include "brains/Register.hpp" | 
| 53 |  | #include "lattice/LatticeFactory.hpp" | 
| 54 |  |  | 
| 55 |  | namespace oopse{ | 
| 56 | < | shapedLattice::shapedLattice(double latticeConstant, | 
| 56 | > | shapedLattice::shapedLattice(RealType latticeConstant, | 
| 57 |  | std::string latticeType) { | 
| 58 | < | latticeConstant_ = latticeConstant; | 
| 59 | < | latticeType_ = latticeType; | 
| 60 | < | registerLattice(); | 
| 61 | < | simpleLattice_ = LatticeFactory::getInstance()->createLattice(latticeType); | 
| 62 | < | if (simpleLattice_ == NULL){ | 
| 63 | < | std::cerr << "shapedLattice:: Error creating lattice" << std::endl; | 
| 64 | < | exit(1); | 
| 65 | < | } | 
| 58 | > | latticeConstant_ = latticeConstant; | 
| 59 | > | latticeType_ = latticeType; | 
| 60 | > | registerLattice(); | 
| 61 | > | simpleLattice_ = LatticeFactory::getInstance()->createLattice(latticeType); | 
| 62 | > | if (simpleLattice_ == NULL){ | 
| 63 | > | std::cerr << "shapedLattice:: Error creating lattice" << std::endl; | 
| 64 | > | exit(1); | 
| 65 | > | } | 
| 66 |  |  | 
| 67 | < | //Set the lattice constant | 
| 68 | < | std::vector<double> lc; | 
| 67 | > | //Set the lattice constant | 
| 68 | > | std::vector<RealType> lc; | 
| 69 |  | lc.push_back(latticeConstant_); | 
| 70 | < | simpleLattice_->setLatticeConstant(lc); | 
| 71 | < | } | 
| 72 | < |  | 
| 73 | < |  | 
| 74 | < |  | 
| 70 | > | simpleLattice_->setLatticeConstant(lc); | 
| 71 | > | sitesComputed_ = false; | 
| 72 | > | } | 
| 73 | > |  | 
| 74 |  | void shapedLattice::setGridDimension(Vector3d dimension){ | 
| 75 |  | dimension_ = dimension; | 
| 76 | < | // Find number of unit cells in each direction | 
| 77 | < | beginNx_ = -(int) ceil(0.5*dimension_[0]/latticeConstant_) ; | 
| 78 | < | beginNy_ = -(int) ceil(0.5*dimension_[1]/latticeConstant_) ; | 
| 79 | < | beginNz_ = -(int) ceil(0.5*dimension_[2]/latticeConstant_) ; | 
| 76 | > | // Find     number of unit cells in each direction | 
| 77 | > | beginNx_ = -(int) ceil(0.5*dimension_[0]/latticeConstant_) ; | 
| 78 | > | beginNy_ = -(int) ceil(0.5*dimension_[1]/latticeConstant_) ; | 
| 79 | > | beginNz_ = -(int) ceil(0.5*dimension_[2]/latticeConstant_) ; | 
| 80 |  | endNx_ = (int) ceil(0.5*dimension_[0]/latticeConstant_); | 
| 81 | < | endNy_ = (int) ceil(0.5*dimension_[1]/latticeConstant_); | 
| 82 | < | endNz_ = (int) ceil(0.5*dimension_[2]/latticeConstant_); | 
| 81 | > | endNy_ = (int) ceil(0.5*dimension_[1]/latticeConstant_); | 
| 82 | > | endNz_ = (int) ceil(0.5*dimension_[2]/latticeConstant_); | 
| 83 | > | sitesComputed_ = false; | 
| 84 |  | } | 
| 85 | + |  | 
| 86 | + | void shapedLattice::setOrigin(Vector3d origin){ | 
| 87 | + | origin_ = origin; | 
| 88 | + | simpleLattice_->setOrigin(origin_); | 
| 89 | + | sitesComputed_ = false; | 
| 90 | + | } | 
| 91 |  |  | 
| 92 | < |  | 
| 93 | < |  | 
| 94 | < | std::vector<Vector3d> shapedLattice::getPoints(){ | 
| 95 | < | std::vector<Vector3d> latticePos; | 
| 96 | < |  | 
| 97 | < | int numMolPerCell = simpleLattice_->getNumSitesPerCell(); | 
| 98 | < | for(int i = beginNx_; i < endNx_; i++) { | 
| 99 | < | for(int j = beginNy_; j < endNy_; j++) { | 
| 100 | < | for(int k = beginNz_; k < endNz_; k++) { | 
| 101 | < | //get the position of the cell sites | 
| 102 | < | simpleLattice_->getLatticePointsPos(latticePos, i, j, k); | 
| 103 | < |  | 
| 104 | < | for(int l = 0; l < numMolPerCell; l++) { | 
| 105 | < |  | 
| 106 | < |  | 
| 107 | < | if (isInterior(latticePos[l])){ | 
| 108 | < | Vector3d myPoint = latticePos[l]; | 
| 109 | < | coords_.push_back(myPoint); | 
| 110 | < | } | 
| 92 | > | void shapedLattice::findSites(){ | 
| 93 | > |  | 
| 94 | > | sites_.clear(); | 
| 95 | > | orientations_.clear(); | 
| 96 | > |  | 
| 97 | > | std::vector<Vector3d> latticePos; | 
| 98 | > | std::vector<Vector3d> pointsOrt =  simpleLattice_->getLatticePointsOrt(); | 
| 99 | > | int numMolPerCell = simpleLattice_->getNumSitesPerCell(); | 
| 100 | > |  | 
| 101 | > | for(int i = beginNx_; i < endNx_; i++) { | 
| 102 | > | for(int j = beginNy_; j < endNy_; j++) { | 
| 103 | > | for(int k = beginNz_; k < endNz_; k++) { | 
| 104 | > | //get the position of the cell sites | 
| 105 | > | simpleLattice_->getLatticePointsPos(latticePos, i, j, k); | 
| 106 | > | for(int l = 0; l < numMolPerCell; l++) { | 
| 107 | > | if (isInterior(latticePos[l])){ | 
| 108 | > | Vector3d myPoint = latticePos[l]; | 
| 109 | > | Vector3d myOrt = pointsOrt[l]; | 
| 110 | > | sites_.push_back(myPoint); | 
| 111 | > | orientations_.push_back(myOrt); | 
| 112 |  | } | 
| 113 |  | } | 
| 114 |  | } | 
| 115 |  | } | 
| 116 | < |  | 
| 117 | < | return coords_; | 
| 118 | < |  | 
| 119 | < | } | 
| 116 | > | } | 
| 117 | > | sitesComputed_ = true; | 
| 118 | > | } | 
| 119 | > |  | 
| 120 | > | std::vector<Vector3d> shapedLattice::getSites() { | 
| 121 | > | if (!sitesComputed_) { | 
| 122 | > | findSites(); | 
| 123 | > | } | 
| 124 | > | return sites_; | 
| 125 | > | } | 
| 126 | > |  | 
| 127 | > | std::vector<Vector3d> shapedLattice::getOrientations() { | 
| 128 | > | if (!sitesComputed_) { | 
| 129 | > | findSites(); | 
| 130 | > | } | 
| 131 | > | return orientations_; | 
| 132 | > | } | 
| 133 |  | } |