Rottor.cc

Rottor::Rottor()
{
	Key=0;
	TempCount=0;
	Count=0;
	//cout<<"Enter the Rottor Number "<<endl;
	//cin>>RottorNo;
    char a[26]={ 'w','r','y','i','p','k','h','f','s','z','c','b','m','q', 'n','e','v','t','x','u','a','o','d','l','g','j' };
    AlphaArray=new char[strlen(a)];                     //setting the address of the AlphaArray
    strcpy(AlphaArray,a);                               //copying the data from a to TempArray
    TempArray=new char[strlen(a)];                      //creating a new memory with length same as AlphaArray is
    strcpy(TempArray,a);                                //copying the data from a to TempArray
    SetRottor();                                        //calling SetRottor function to set TempArray according to the Key
}

Rottor::Rottor(int k,char* ta,int RNo, int FFF)
{
    RottorNo=RNo;
    Count=0;
    TempCount=0;
    Key=k;                                                //setting the Key
    char a[26]={'a','b','c','d','e','f','g','h','i','j','k','l','m','n','o','p','q','r','s','t','u','v','w','x','y','z'};
    char p[26]={ 'w','r','y','i','p','k','h','f','s','z','c','b','m','q', 'n','e','v','t','x','u','a','o','d','l','g','j' };
    AlphaArray=new char[strlen(a)];                       //setting the address of the AlphaArray
    strcpy(AlphaArray,a);                                 //copying the data from a to TempArray
    if( FFF == 1 )
       {
        TempArray=new char[strlen(ta)];                    //creating a new memory with length same as AlphaArray is
        strcpy(TempArray,ta);                               //copying the data from a to TempArray
       }
   else
       {
        TempArray=new char[strlen(p)];                      //creating a new memory with length same as AlphaArray is
        strcpy(TempArray,p);
       }

   SetRottor();                                             //calling SetRottor function to set TempArray according to the Key
}

void Rottor::Rotate(int time,bool Dir)
{
    int Size=strlen(TempArray);
   	if(Dir==true)                                           //if true rotate Clock Wise
      {                                                     //rotating clock wise for time 'time'
        for(int i=0;i<time;i++)
            {
                char temp=TempArray[0];
                for(int i=0;i<Size;i++)
                {
                    TempArray[i]=TempArray[i+1];
                }
                TempArray[25]=temp;
            }
       }

	 if(Dir==false)                                 //rotating Anti-ClockWise
        {
         for(int i=0;i<time;i++)
            {
                char temp=TempArray[25];
                for(int i=Size;i>=0;i--)
                {
                    TempArray[i]=TempArray[i-1];
                }
                TempArray[0]=temp;
            }
         }
}

char Rottor::Encrypt(char ch)
{

    if(RottorNo==1)                                       // 1st Rottor rotates every time it Encrypt
    {
        if(TempCount==0)                                  //first half cycle of encryption
        {
            TempCount++;
            Rotate(1,true);                                //rotating one time clock wise
            return GetCharT(GetPositionA(ch));
        }

        else if(TempCount==1)
        {
            TempCount--;                                      //in secod half cycle we doses not rotate the rottor
            return GetCharA(GetPositionT(ch));
        }
    }

    if(RottorNo==2)
    {
     if(TempCount==0)
       {
         if(Count==26)
         {
             Rotate(1,true);
             Count=0;
         }
         Count++;
         TempCount++;
         return GetCharT(GetPositionA(ch));
       }

       else if(TempCount==1)
       {
         TempCount--;
         return GetCharA(GetPositionT(ch));
       }
    }

    if(RottorNo==3)
    {
       if(TempCount==0)
        {
         if(Count==52)
         {
             Rotate(1,true);
             Count=0;
         }
         Count++;
         TempCount++;
         return GetCharT(GetPositionA(ch));
       }
       else if(TempCount==1)
       {
         TempCount--;
         return GetCharA(GetPositionT(ch));
       }
    }
}

int Rottor::GetPositionT(char ch)                              //getting the present position of char ah in TempArray
{
    for(int i=0;i<26;i++)
    {
        if(TempArray[i]==ch)                                    //matching
        {
            return i;                                           //returning the address of the matched char with ch
        }
    }
}

int Rottor::GetPositionA(char ch)                               //getting the present position of char ah in TempArray
{
    for(int i=0;i<26;i++)
    {
        if(AlphaArray[i]==ch)                                   //matching
        {
            return i;                                           //returning the address of the matched char with ch
        }
    }
}

char Rottor::GetCharA(int no)
{
    return AlphaArray[no];                                      //returning the char present at the index no in AlphaArray
}

char Rottor::GetCharT(int no)
{
    return TempArray[no];                                       //returning the char present at the index no in AlphaArray
}

void Rottor::SetRottor()
{

    int temp=GetPositionT('a');                                  //getting the present position of a char in the TempArray

    if(Key>25)                                                   //case1
    {
        int TempKey=Key%26;                                      //calculating the position according to the Key
          if(temp!=Key)
             {
                if(TempKey<temp)
                {
                    Rotate(temp-TempKey,true);                    //rotating Clock_wise by "temp-Key" time
                }
                else if(TempKey>temp)
                {
                    Rotate(TempKey-temp,false);                    //rotating AntiClock_wise by "Key-temp" time
                }
            }
    }

    if(Key<=25)                                                     //case 2
    {
        if(temp!=Key)                                              //position is not matching
        {
            if(Key<temp)                                           //present at behind the required position
            {
                Rotate(temp-Key,true);                             //rotating Clock_wise by "temp-Key" time
            }
            else if(Key>temp)                                      //present ahead from the the required position
            {
                Rotate(Key-temp,false);                            //rotating AntiClock_wise by "Key-temp" time
            }
        }
    }
}

Rottor::~Rottor()
{
delete[] TempArray;
delete[] AlphaArray;
}