Program Listing for File gate_counter.hpp

↰ Return to documentation for file (/home/docs/checkouts/readthedocs.org/user_builds/intel-qs/checkouts/docs/include/gate_counter.hpp)

#ifndef IQS_GATE_COUNTER_HPP
#define IQS_GATE_COUNTER_HPP

#include <vector>




class GateCounter
{
 private:
  int num_qubits;
  int total_gate_count=0;
  int one_qubit_gate_count=0;
  int two_qubit_gate_count=0;
  std::vector<int> parallel_depth;

 public:


  // Constructor
  GateCounter (int new_num_qubits)
    : num_qubits(new_num_qubits)
  {
    parallel_depth.assign(num_qubits,0);
  }

  // Destructor
  ~GateCounter()
  { }


  void Reset()
  {
    parallel_depth.assign(num_qubits,0);
  }


  //Get stats
  int GetTotalGateCount()    { return total_gate_count; }
  int GetOneQubitGateCount() { return one_qubit_gate_count; }
  int GetTwoQubitGateCount() { return two_qubit_gate_count; }
  int GetParallelDepth()
  { return *std::max_element(std::begin(parallel_depth), std::end(parallel_depth)); }


  void OneQubitIncrement (int qubit)
  {
    ++total_gate_count;
    ++one_qubit_gate_count;
    ++parallel_depth[qubit];
  }


  void TwoQubitIncrement (int qubit_0, int qubit_1)
  {
    ++total_gate_count;
    ++two_qubit_gate_count;
    int new_depth = std::max(parallel_depth[qubit_0], parallel_depth[qubit_1]) +1;
    parallel_depth[qubit_0] = new_depth;
    parallel_depth[qubit_1] = new_depth;
  }


  void Breakdown()
  {
    if (qhipster::mpi::Environment::GetStateRank() == 0)
    {
        printf("The quantum circuit is composed of %d one-qubit gates and "
               "%d two-qubitgates, for a total of %d gates.\nThe greedy depth "
               "(all gates lasting one clock cycle) is %d.\n",
               GetOneQubitGateCount(), GetTwoQubitGateCount(),
               GetTotalGateCount(), GetParallelDepth() );
    }
  }

};


#endif  // header guard IQS_GATE_COUNTER_HPP