ViewVC Help
View File | Revision Log | Show Annotations | View Changeset | Root Listing
root/OpenMD/branches/development/src/visitors/AtomVisitor.cpp
(Generate patch)

Comparing trunk/src/visitors/AtomVisitor.cpp (file contents):
Revision 1008 by chrisfen, Wed Jul 19 12:35:31 2006 UTC vs.
Revision 1291 by gezelter, Thu Sep 11 19:40:59 2008 UTC

# Line 384 | Line 384 | namespace oopse {
384      AtomData* atomData;
385      GenericData* data;
386      bool haveAtomData;
387 –    AtomType* atomType;
387      //if atom is not linear atom, just skip it
388      if(!isLinearAtom(datom->getType()) || !datom->getAtomType()->isGayBerne())
389        return;
# Line 522 | Line 521 | namespace oopse {
521  
522    void GBLipidAtomVisitor::visit(DirectionalAtom* datom){
523      std::vector<AtomInfo*> atoms;
524 <    //we need to convert linear into 4 different atoms
525 <    Vector3d c1(0.0, 0.0, -6.25);
527 <    Vector3d c2(0.0, 0.0, -2.1);
528 <    Vector3d c3(0.0, 0.0,  2.1);
529 <    Vector3d c4(0.0, 0.0,  6.25);
524 >    Vector3d c1(0.0, 0.0, 0.0);
525 >    Vector3d c2(0.0, 0.0, 1.0);
526      RotMat3x3d rotMatrix;
527      RotMat3x3d rotTrans;
528      AtomInfo* atomInfo;
529      Vector3d pos;
530      Vector3d newVec;
531 +    Vector3d dVec;
532      Quat4d q;
533      AtomData* atomData;
534      GenericData* data;
# Line 539 | Line 536 | namespace oopse {
536  
537      //if atom is not GBlipid atom, just skip it
538      if(!isGBLipidAtom(datom->getType()))
539 +      return;
540 +
541 +    data = datom->getPropertyByName("ATOMDATA");
542 +    if(data != NULL){
543 +      atomData = dynamic_cast<AtomData*>(data);  
544 +      if(atomData == NULL){
545 +        std::cerr << "can not get Atom Data from " << datom->getType() << std::endl;
546 +        atomData = new AtomData;
547 +        haveAtomData = false;      
548 +      } else {
549 +        haveAtomData = true;
550 +      }
551 +    } else {
552 +      atomData = new AtomData;
553 +      haveAtomData = false;
554 +    }
555 +  
556 +  
557 +    pos = datom->getPos();
558 +    q = datom->getQ();
559 +    rotMatrix = datom->getA();
560 +
561 +    // We need A^T to convert from body-fixed to space-fixed:  
562 +    rotTrans = rotMatrix.transpose();
563 +
564 +    newVec = rotTrans * c1;
565 +    dVec = rotTrans * c2;
566 +    atomInfo = new AtomInfo;
567 +    atomInfo->atomTypeName = "GB";
568 +    atomInfo->pos[0] = pos[0] + newVec[0];
569 +    atomInfo->pos[1] = pos[1] + newVec[1];
570 +    atomInfo->pos[2] = pos[2] + newVec[2];
571 +    atomInfo->dipole[0] = dVec[0];
572 +    atomInfo->dipole[1] = dVec[1];
573 +    atomInfo->dipole[2] = dVec[2];
574 +    atomInfo->hasVector = true;
575 +    atomInfo->charge = 3.0;
576 +    atomInfo->hasCharge = true;
577 +    atomData->addAtomInfo(atomInfo);
578 +
579 +    //add atom data into atom's property
580 +
581 +    if(!haveAtomData){
582 +      atomData->setID("ATOMDATA");
583 +      datom->addProperty(atomData);
584 +    }
585 +
586 +    setVisited(datom);
587 +
588 +  }
589 +
590 +  const std::string GBLipidAtomVisitor::toString(){
591 +    char buffer[65535];
592 +    std::string result;
593 +  
594 +    sprintf(buffer ,"------------------------------------------------------------------\n");
595 +    result += buffer;
596 +
597 +    sprintf(buffer ,"Visitor name: %s\n", visitorName.c_str());
598 +    result += buffer;
599 +
600 +    sprintf(buffer , "Visitor Description: Convert GBlipid into xyz-formatted atom for use with xyz2pov\n");
601 +    result += buffer;
602 +
603 +    sprintf(buffer ,"------------------------------------------------------------------\n");
604 +    result += buffer;
605 +
606 +    return result;
607 +  }
608 +
609 +  bool Ring5gbAtomVisitor::isRing5gbAtom(const std::string& atomType){
610 +    std::set<std::string>::iterator strIter;
611 +    strIter = Ring5gbAtomType.find(atomType);
612 +
613 +    return strIter != Ring5gbAtomType.end() ? true : false;
614 +  }
615 +
616 +  void Ring5gbAtomVisitor::visit(DirectionalAtom* datom){
617 +    std::vector<AtomInfo*> atoms;
618 +    //we need to convert linear into 4 different atoms
619 +    Vector3d c1(0.0, 0.0, -5.5);
620 +    Vector3d c2(0.0, 0.0, -1.8);
621 +    Vector3d c3(0.0, 0.0,  1.8);
622 +    Vector3d c4(0.0, 0.0,  5.5);
623 +    RotMat3x3d rotMatrix;
624 +    RotMat3x3d rotTrans;
625 +    AtomInfo* atomInfo;
626 +    Vector3d pos;
627 +    Vector3d newVec;
628 +    Vector3d dVec;
629 +    Quat4d q;
630 +    AtomData* atomData;
631 +    GenericData* data;
632 +    bool haveAtomData;
633 +
634 +    //if atom is not Ring5GB atom, just skip it
635 +    if(!isRing5gbAtom(datom->getType()))
636        return;
637  
638      data = datom->getPropertyByName("ATOMDATA");
# Line 619 | Line 713 | namespace oopse {
713  
714    }
715  
716 <  const std::string GBLipidAtomVisitor::toString(){
716 >  const std::string Ring5gbAtomVisitor::toString(){
717      char buffer[65535];
718      std::string result;
719    
# Line 629 | Line 723 | namespace oopse {
723      sprintf(buffer ,"Visitor name: %s\n", visitorName.c_str());
724      result += buffer;
725  
726 <    sprintf(buffer , "Visitor Description: Convert GBlipid into 4 different K atoms\n");
726 >    sprintf(buffer , "Visitor Description: Convert Ring5GB into 4 different K atoms\n");
727      result += buffer;
728  
729      sprintf(buffer ,"------------------------------------------------------------------\n");
# Line 638 | Line 732 | namespace oopse {
732      return result;
733    }
734  
735 +  bool HeadAtomVisitor::isHeadAtom(const std::string& atomType){
736 +    std::set<std::string>::iterator strIter;
737 +    strIter = HeadAtomType.find(atomType);
738 +
739 +    return strIter != HeadAtomType.end() ? true : false;
740 +  }
741 +
742 +  void HeadAtomVisitor::visit(DirectionalAtom* datom){
743 +    std::vector<AtomInfo*> atoms;
744 +    //we need to convert linear into 2 different atoms
745 +    Vector3d c1(0.0, 0.0, -1.5);
746 +    Vector3d c2(0.0, 0.0, 1.5);
747 +    RotMat3x3d rotMatrix;
748 +    RotMat3x3d rotTrans;
749 +    AtomInfo* atomInfo;
750 +    Vector3d pos;
751 +    Vector3d newVec;
752 +    Vector3d dVec;
753 +    Quat4d q;
754 +    AtomData* atomData;
755 +    GenericData* data;
756 +    bool haveAtomData;
757 +
758 +    //if atom is not Head atom, just skip it
759 +    if(!isHeadAtom(datom->getType()))
760 +      return;
761 +
762 +    data = datom->getPropertyByName("ATOMDATA");
763 +    if(data != NULL){
764 +      atomData = dynamic_cast<AtomData*>(data);  
765 +      if(atomData == NULL){
766 +        std::cerr << "can not get Atom Data from " << datom->getType() << std::endl;
767 +        atomData = new AtomData;
768 +        haveAtomData = false;      
769 +      } else {
770 +        haveAtomData = true;
771 +      }
772 +    } else {
773 +      atomData = new AtomData;
774 +      haveAtomData = false;
775 +    }
776 +  
777 +  
778 +    pos = datom->getPos();
779 +    q = datom->getQ();
780 +    rotMatrix = datom->getA();
781 +
782 +    // We need A^T to convert from body-fixed to space-fixed:  
783 +    rotTrans = rotMatrix.transpose();
784 +
785 +    newVec = rotTrans * c1;
786 +    atomInfo = new AtomInfo;
787 +    atomInfo->atomTypeName = "C";
788 +    atomInfo->pos[0] = pos[0] + newVec[0];
789 +    atomInfo->pos[1] = pos[1] + newVec[1];
790 +    atomInfo->pos[2] = pos[2] + newVec[2];
791 +    atomInfo->dipole[0] = 0.0;
792 +    atomInfo->dipole[1] = 0.0;
793 +    atomInfo->dipole[2] = 0.0;
794 +    atomData->addAtomInfo(atomInfo);
795 +
796 +    newVec = rotTrans * c2;
797 +    atomInfo = new AtomInfo;
798 +    atomInfo->atomTypeName = "O";
799 +    atomInfo->pos[0] = pos[0] + newVec[0];
800 +    atomInfo->pos[1] = pos[1] + newVec[1];
801 +    atomInfo->pos[2] = pos[2] + newVec[2];
802 +    atomInfo->dipole[0] = 0.0;
803 +    atomInfo->dipole[1] = 0.0;
804 +    atomInfo->dipole[2] = 0.0;
805 +    atomData->addAtomInfo(atomInfo);
806 +
807 +    //add atom data into atom's property
808 +
809 +    if(!haveAtomData){
810 +      atomData->setID("ATOMDATA");
811 +      datom->addProperty(atomData);
812 +    }
813 +
814 +    setVisited(datom);
815 +
816 +  }
817 +
818 +  const std::string HeadAtomVisitor::toString(){
819 +    char buffer[65535];
820 +    std::string result;
821 +  
822 +    sprintf(buffer ,"------------------------------------------------------------------\n");
823 +    result += buffer;
824 +
825 +    sprintf(buffer ,"Visitor name: %s\n", visitorName.c_str());
826 +    result += buffer;
827 +
828 +    sprintf(buffer , "Visitor Description: Convert HEAD into C atom and O atom\n");
829 +    result += buffer;
830 +
831 +    sprintf(buffer ,"------------------------------------------------------------------\n");
832 +    result += buffer;
833 +
834 +    return result;
835 +  }
836 +
837 +
838    //----------------------------------------------------------------------------//
839  
840    void DefaultAtomVisitor::visit(Atom *atom) {

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines