CROT.cpp

Go to the documentation of this file.

/*
Created on Fri Jun 26 14:13:26 2020
Copyright 2020 Peter Rakyta, Ph.D.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.

@author: Peter Rakyta, Ph.D.
*/
#include "CROT.h"
#include "apply_large_kernel_to_input.h"
#include "apply_large_kernel_to_input_AVX.h"

static double M_PIOver2 = M_PI/2;

//static tbb::spin_mutex my_mutex;
CROT::CROT(){

        // A string describing the type of the gate
        type = CROT_OPERATION;
        
         // number of qubits spanning the matrix of the gate
        qbit_num = -1;
        // the size of the matrix
        matrix_size = -1;
        
        target_qbit = -1;
        // The index of the qubit which acts as a control qubit (control_qbit >= 0) in controlled gates
        control_qbit = -1;
        parameter_num = 0;

        name = "CROT";

}



CROT::CROT(int qbit_num_in, int target_qbit_in, int control_qbit_in) {

        name = "CROT";

        // number of qubits spanning the matrix of the gate
        qbit_num = qbit_num_in;
        // the size of the matrix
        matrix_size = Power_of_2(qbit_num);
        // A string describing the type of the gate

        // A string describing the type of the gate
        type = CROT_OPERATION;
        



        if (control_qbit_in >= qbit_num) {
         std::stringstream sstream;
         sstream << "The index of the control qubit is larger than the number of qubits in CROT gate." << std::endl;
         print(sstream, 0);     
            throw sstream.str();
        }

        // The index of the qubit which acts as a control qubit (control_qbit >= 0) in controlled gates
        control_qbit = control_qbit_in;


        // The index of the qubit on which the gate acts (target_qbit >= 0)
        target_qbit = target_qbit_in;
        

        parameter_num=2;

        parameters = Matrix_real(1, parameter_num);
}

CROT::~CROT() {

}


void 
CROT::apply_to_list( Matrix_real& parameters_mtx, std::vector<Matrix>& input ) {


    for ( std::vector<Matrix>::iterator it=input.begin(); it != input.end(); it++ ) {
        apply_to( parameters_mtx, *it, 0);
    }

}

void 
CROT::apply_to_list( Matrix_real& parameters_mtx, std::vector<Matrix>& inputs, int parallel ) {

    int work_batch = 1;
    if ( parallel == 0 ) {
        work_batch = inputs.size();
    }
    else {
        work_batch = 1;
    }


    tbb::parallel_for( tbb::blocked_range<int>(0,inputs.size(),work_batch), [&](tbb::blocked_range<int> r) {
        for (int idx=r.begin(); idx<r.end(); ++idx) { 

            Matrix* input = &inputs[idx];

            apply_to( parameters_mtx, *input, parallel );

        }

    });

}

void 
CROT::apply_to( Matrix_real& parameters_mtx, Matrix& input, int parallel ) {


    if (input.rows != matrix_size ) {
     std::stringstream sstream;
     sstream << "Wrong matrix size in CROT gate apply" << std::endl;
        print(sstream, 0);    
        exit(-1);
    }


    double ThetaOver2, Phi;
    

   ThetaOver2 = parameters_mtx[0];
   Phi = parameters_mtx[1];
   
    if (input.cols==1){

    Matrix U_2qbit(4,4);
    memset(U_2qbit.get_data(),0.0,(U_2qbit.size()*2)*sizeof(double));
    U_2qbit[0].real = std::cos(ThetaOver2);
    U_2qbit[2].real = std::sin(ThetaOver2)*std::sin(Phi);
    U_2qbit[2].imag = std::sin(ThetaOver2)*std::cos(Phi);
    U_2qbit[1*4+3].real = -1.*std::sin(-1.*ThetaOver2)*std::sin(Phi);
    U_2qbit[1*4+3].imag = -1.*std::sin(-1.*ThetaOver2)*std::cos(Phi);
    U_2qbit[1*4+1].real = std::cos(-1.*ThetaOver2);
    U_2qbit[2*4+2].real = std::cos(ThetaOver2);
    U_2qbit[2*4].real = -1.*std::sin(ThetaOver2)*std::sin(Phi);
    U_2qbit[2*4].imag = std::sin(ThetaOver2)*std::cos(Phi);
    U_2qbit[3*4+3].real = std::cos(-1.*ThetaOver2);
    U_2qbit[3*4+1].real = std::sin(-1.*ThetaOver2)*std::sin(Phi);
    U_2qbit[3*4+1].imag = -1.*std::sin(-1.*ThetaOver2)*std::cos(Phi);

    // apply the computing kernel on the matrix
    std::vector<int> involved_qbits = {control_qbit,target_qbit};
    if (parallel){
      apply_large_kernel_to_input_AVX(U_2qbit,input,involved_qbits,input.size());
    }
    else{
        apply_large_kernel_to_input(U_2qbit,input,involved_qbits,input.size());
    }
    
    }

    else{
    
      Matrix U3_matrix = calc_one_qubit_u3(ThetaOver2, Phi-M_PIOver2, -1*Phi+M_PIOver2 );
      Matrix U3_matrix2 = calc_one_qubit_u3(-1.*ThetaOver2, Phi-M_PIOver2, -1*Phi+M_PIOver2 );
      if(parallel){
        apply_crot_kernel_to_matrix_input_AVX_parallel(U3_matrix2,U3_matrix, input, target_qbit, control_qbit, input.rows);
      }
      else{
        apply_crot_kernel_to_matrix_input_AVX(U3_matrix2,U3_matrix, input, target_qbit, control_qbit, input.rows);
      }
  }

    

}



void 
CROT::apply_from_right( Matrix_real& parameters, Matrix& input ) {



}


std::vector<Matrix> 
CROT::apply_derivate_to( Matrix_real& parameters_mtx, Matrix& input, int parallel ) {

    if (input.rows != matrix_size ) {
     std::stringstream sstream;
     sstream << "Wrong matrix size in CROT gate apply" << std::endl;
        print(sstream, 0);       
        exit(-1);
    }

    std::vector<Matrix> ret;

    double ThetaOver2, Phi;


   ThetaOver2 = parameters_mtx[0];

   Phi = parameters_mtx[1];

    



    // the resulting matrix


    if (input.cols==1){

    double Theta0Over2_shifted = ThetaOver2 + M_PIOver2;
    double Theta1Over2_shifted;

    Theta1Over2_shifted = -1.*Theta0Over2_shifted;
  
    double Phi0_shifted = Phi + M_PIOver2;
    double Phi1_shifted = Phi + M_PIOver2; 
    
    //Theta derivative
    Matrix res_mtx = input.copy();   
    Matrix U_2qbit(4,4);
    memset(U_2qbit.get_data(),0.0,(U_2qbit.size()*2)*sizeof(double));
    
    U_2qbit[0].real = std::cos(Theta0Over2_shifted);
    U_2qbit[2].real = std::sin(Theta0Over2_shifted)*std::sin(Phi);
    U_2qbit[2].imag = std::sin(Theta0Over2_shifted)*std::cos(Phi);
    
    U_2qbit[2*4+2].real = std::cos(Theta0Over2_shifted);
    U_2qbit[2*4].real = -1.*std::sin(Theta0Over2_shifted)*std::sin(Phi);
    U_2qbit[2*4].imag = std::sin(Theta0Over2_shifted)*std::cos(Phi);
    
    U_2qbit[1*4+3].real = -1.*std::sin(Theta1Over2_shifted)*std::sin(Phi);
    U_2qbit[1*4+3].imag = -1.*std::sin(Theta1Over2_shifted)*std::cos(Phi);
    U_2qbit[1*4+1].real = std::cos(Theta1Over2_shifted);
    
    U_2qbit[3*4+3].real = std::cos(Theta1Over2_shifted);
    U_2qbit[3*4+1].real = std::sin(Theta1Over2_shifted)*std::sin(Phi);
    U_2qbit[3*4+1].imag = -1.*std::sin(Theta1Over2_shifted)*std::cos(Phi);


    std::vector<int> involved_qbits = {control_qbit,target_qbit};

    apply_large_kernel_to_input(U_2qbit,res_mtx,involved_qbits,res_mtx.size());
    ret.push_back(res_mtx);
    
    //Phi derivative
    Matrix res_mtx1 = input.copy();   
    memset(U_2qbit.get_data(),0.0,(U_2qbit.size()*2)*sizeof(double));

    U_2qbit[2].real = std::sin(ThetaOver2)*std::sin(Phi0_shifted);
    U_2qbit[2].imag = std::sin(ThetaOver2)*std::cos(Phi0_shifted);

    U_2qbit[2*4].real = -1.*std::sin(ThetaOver2)*std::sin(Phi0_shifted);
    U_2qbit[2*4].imag = std::sin(ThetaOver2)*std::cos(Phi0_shifted);
    
    U_2qbit[1*4+3].real = -1.*std::sin(-1.*ThetaOver2)*std::sin(Phi1_shifted);
    U_2qbit[1*4+3].imag = -1.*std::sin(-1.*ThetaOver2)*std::cos(Phi1_shifted);

    U_2qbit[3*4+1].real = std::sin(-1.*ThetaOver2)*std::sin(Phi1_shifted);
    U_2qbit[3*4+1].imag = -1.*std::sin(-1.*ThetaOver2)*std::cos(Phi1_shifted);

    apply_large_kernel_to_input(U_2qbit,res_mtx1,involved_qbits,res_mtx1.size());
    ret.push_back(res_mtx1);

   
    }
    else{
      
      //Theta derivative
      Matrix res_mtx = input.copy();   
      Matrix U3_matrix = calc_one_qubit_u3(ThetaOver2+M_PIOver2, Phi-M_PIOver2, -1*Phi+M_PIOver2 );
      Matrix U3_matrix2 = calc_one_qubit_u3(-1.*(ThetaOver2+M_PIOver2), Phi-M_PIOver2, -1*Phi+M_PIOver2 );
      
      apply_crot_kernel_to_matrix_input(U3_matrix2, U3_matrix, res_mtx, target_qbit, control_qbit, res_mtx.rows);
      ret.push_back(res_mtx);
      Matrix res_mtx1 = input.copy();   
      U3_matrix = calc_one_qubit_u3(ThetaOver2, Phi, -1*Phi );
      U3_matrix2 = calc_one_qubit_u3(-1.*ThetaOver2, Phi, -1*Phi );
      U3_matrix[0].real = 0; 
      U3_matrix[0].imag = 0;
      U3_matrix[3].real = 0; 
      U3_matrix[3].imag = 0;
      U3_matrix2[0].real = 0; 
      U3_matrix2[0].imag = 0;
      U3_matrix2[3].real = 0; 
      U3_matrix2[3].imag = 0;
      apply_crot_kernel_to_matrix_input(U3_matrix2, U3_matrix, res_mtx1, target_qbit, control_qbit, res_mtx1.rows);
      ret.push_back(res_mtx1); 
      
    }
    

    return ret;


}

void CROT::set_qbit_num(int qbit_num_in) {

        // setting the number of qubits
        Gate::set_qbit_num(qbit_num_in);
}



void CROT::reorder_qubits( std::vector<int> qbit_list) {

    Gate::reorder_qubits(qbit_list);

}



CROT* CROT::clone() {

    CROT* ret = new CROT(qbit_num, target_qbit, control_qbit);

    ret->set_parameter_start_idx( get_parameter_start_idx() );
    ret->set_parents( parents );
    ret->set_children( children );

    return ret;

}

Matrix_real 
CROT::extract_parameters( Matrix_real& parameters ) {

    if ( get_parameter_start_idx() + get_parameter_num() > parameters.size()  ) {
        std::string err("CROT::extract_parameters: Cant extract parameters, since the dinput arary has not enough elements.");
        throw err;     
    }

    Matrix_real extracted_parameters(1,2);

    extracted_parameters[0] = std::fmod( 2*parameters[ get_parameter_start_idx() ], 4*M_PI);
    extracted_parameters[1] = std::fmod( parameters[ get_parameter_start_idx() + 1 ], 2*M_PI);

    return extracted_parameters;

}