qiskitsim.py

import numpy as np
import qiskit 
import qiskit.quantum_info as qiskit_quantum_info
from abstractQregister import AbstractQuantumRegister

#***************************************************
#* to install qiskit with conda forge:
#* conda install pip-tools ply  mpmath  charset-normalizer  websockets  websocket-client  urllib3  sympy  symengine  six  pycparser  psutil  pbr  numpy  ntlm-auth  idna  dill  stevedore  scipy  rustworkx  requests  python-dateutil  cffi  cryptography symengine
#* pip install -U qiskit
#***************************************************

class QuantumRegister(AbstractQuantumRegister):
    batchcapable = False
    def __init__(self,nbqubit:int, batchsize:int=1) -> None:
        self.nbqubit = nbqubit
        self.qc=qiskit.QuantumCircuit(nbqubit)
        self.batchcapable = False
    
    def rz(self,q:int,theta:float):
        self.qc.rz(theta,self.nbqubit-1-q)
        
    def sx(self,q:int):
        self.qc.sx(self.nbqubit-1-q)
        
    def cz(self,q0:int,q1:int):
        self.qc.cz(self.nbqubit-1-q0,self.nbqubit-1-q1)
        
    def measureAll(self):
        self.mstate = qiskit_quantum_info.Statevector.from_instruction(self.qc).probabilities()
        return self.mstate
    
    def makeShots(self,nbshots:int):
        return np.random.multinomial(nbshots, self.mstate) #type:ignore
        # return qiskit_quantum_info.Statevector.from_instruction(self.qc).sample_memory(nbshots)
        
#********************************************************************
#* Notes: when running on actual hardware 
#* measureAll and makeShots has to be written to match the hardware actual functions
#*********************************************************************

    
if __name__=='__main__':
    # basic test
    qr = QuantumRegister(2)
    qr.sx(0)
    print(qr.measureAll())
    print(qr.makeShots(10))