Anders and Briegel in Python
Vous ne pouvez pas sélectionner plus de 25 sujets Les noms de sujets doivent commencer par une lettre ou un nombre, peuvent contenir des tirets ('-') et peuvent comporter jusqu'à 35 caractères.

971 lignes
21KB

  1. // CHP: CNOT-Hadamard-Phase
  2. // Stabilizer Quantum Computer Simulator
  3. // by Scott Aaronson
  4. // Last modified June 30, 2004
  5. // Thanks to Simon Anders and Andrew Cross for bugfixes
  6. #include <stdio.h>
  7. #include <stdlib.h>
  8. #include <string.h>
  9. #include <Python.h>
  10. #include <time.h>
  11. #define CNOT 0
  12. #define HADAMARD 1
  13. #define PHASE 2
  14. #define MEASURE 3
  15. struct QProg
  16. // Quantum circuit
  17. {
  18. long n; // # of qubits
  19. long T; // # of gates
  20. char *a; // Instruction opcode
  21. long *b; // Qubit 1
  22. long *c; // Qubit 2 (target for CNOT)
  23. int DISPQSTATE; // whether to print the state (q for final state only, Q for every iteration)
  24. int DISPTIME; // whether to print the execution time
  25. int SILENT; // whether NOT to print measurement results
  26. int DISPPROG; // whether to print instructions being executed as they're executed
  27. int SUPPRESSM; // whether to suppress actual computation of determinate measurement results
  28. };
  29. struct QState
  30. // Quantum state
  31. {
  32. // To save memory and increase speed, the bits are packed 32 to an unsigned long
  33. long n; // # of qubits
  34. unsigned long **x; // (2n+1)*n matrix for stabilizer/destabilizer x bits (there's one "scratch row" at
  35. unsigned long **z; // (2n+1)*n matrix for z bits the bottom)
  36. int *r; // Phase bits: 0 for +1, 1 for i, 2 for -1, 3 for -i. Normally either 0 or 2.
  37. unsigned long pw[32]; // pw[i] = 2^i
  38. long over32; // floor(n/8)+1
  39. };
  40. // Globals
  41. struct QProg *h;
  42. struct QState *q;
  43. void error(int k)
  44. {
  45. if (k==0) printf("\nSyntax: chp [-options] <filename> [input]\n");
  46. if (k==1) printf("\nFile not found\n");
  47. exit(0);
  48. }
  49. void cnot(struct QState *q, long b, long c)
  50. // Apply a CNOT gate with control b and target c
  51. {
  52. long i;
  53. long b5;
  54. long c5;
  55. unsigned long pwb;
  56. unsigned long pwc;
  57. b5 = b>>5;
  58. c5 = c>>5;
  59. pwb = q->pw[b&31];
  60. pwc = q->pw[c&31];
  61. for (i = 0; i < 2*q->n; i++)
  62. {
  63. if (q->x[i][b5]&pwb) q->x[i][c5] ^= pwc;
  64. if (q->z[i][c5]&pwc) q->z[i][b5] ^= pwb;
  65. if ((q->x[i][b5]&pwb) && (q->z[i][c5]&pwc) &&
  66. (q->x[i][c5]&pwc) && (q->z[i][b5]&pwb))
  67. q->r[i] = (q->r[i]+2)%4;
  68. if ((q->x[i][b5]&pwb) && (q->z[i][c5]&pwc) &&
  69. !(q->x[i][c5]&pwc) && !(q->z[i][b5]&pwb))
  70. q->r[i] = (q->r[i]+2)%4;
  71. }
  72. return;
  73. }
  74. void hadamard(struct QState *q, long b)
  75. // Apply a Hadamard gate to qubit b
  76. {
  77. long i;
  78. unsigned long tmp;
  79. long b5;
  80. unsigned long pw;
  81. b5 = b>>5;
  82. pw = q->pw[b&31];
  83. for (i = 0; i < 2*q->n; i++)
  84. {
  85. tmp = q->x[i][b5];
  86. q->x[i][b5] ^= (q->x[i][b5] ^ q->z[i][b5]) & pw;
  87. q->z[i][b5] ^= (q->z[i][b5] ^ tmp) & pw;
  88. if ((q->x[i][b5]&pw) && (q->z[i][b5]&pw)) q->r[i] = (q->r[i]+2)%4;
  89. }
  90. return;
  91. }
  92. void phase(struct QState *q, long b)
  93. // Apply a phase gate (|0>->|0>, |1>->i|1>) to qubit b
  94. {
  95. long i;
  96. long b5;
  97. unsigned long pw;
  98. b5 = b>>5;
  99. pw = q->pw[b&31];
  100. for (i = 0; i < 2*q->n; i++)
  101. {
  102. if ((q->x[i][b5]&pw) && (q->z[i][b5]&pw)) q->r[i] = (q->r[i]+2)%4;
  103. q->z[i][b5] ^= q->x[i][b5]&pw;
  104. }
  105. return;
  106. }
  107. void rowcopy(struct QState *q, long i, long k)
  108. // Sets row i equal to row k
  109. {
  110. long j;
  111. for (j = 0; j < q->over32; j++)
  112. {
  113. q->x[i][j] = q->x[k][j];
  114. q->z[i][j] = q->z[k][j];
  115. }
  116. q->r[i] = q->r[k];
  117. return;
  118. }
  119. void rowswap(struct QState *q, long i, long k)
  120. // Swaps row i and row k
  121. {
  122. rowcopy(q, 2*q->n, k);
  123. rowcopy(q, k, i);
  124. rowcopy(q, i, 2*q->n);
  125. return;
  126. }
  127. void rowset(struct QState *q, long i, long b)
  128. // Sets row i equal to the bth observable (X_1,...X_n,Z_1,...,Z_n)
  129. {
  130. long j;
  131. long b5;
  132. unsigned long b31;
  133. for (j = 0; j < q->over32; j++)
  134. {
  135. q->x[i][j] = 0;
  136. q->z[i][j] = 0;
  137. }
  138. q->r[i] = 0;
  139. if (b < q->n)
  140. {
  141. b5 = b>>5;
  142. b31 = b&31;
  143. q->x[i][b5] = q->pw[b31];
  144. }
  145. else
  146. {
  147. b5 = (b - q->n)>>5;
  148. b31 = (b - q->n)&31;
  149. q->z[i][b5] = q->pw[b31];
  150. }
  151. return;
  152. }
  153. int clifford(struct QState *q, long i, long k)
  154. // Return the phase (0,1,2,3) when row i is LEFT-multiplied by row k
  155. {
  156. long j;
  157. long l;
  158. unsigned long pw;
  159. long e=0; // Power to which i is raised
  160. for (j = 0; j < q->over32; j++)
  161. for (l = 0; l < 32; l++)
  162. {
  163. pw = q->pw[l];
  164. if ((q->x[k][j]&pw) && (!(q->z[k][j]&pw))) // X
  165. {
  166. if ((q->x[i][j]&pw) && (q->z[i][j]&pw)) e++; // XY=iZ
  167. if ((!(q->x[i][j]&pw)) && (q->z[i][j]&pw)) e--; // XZ=-iY
  168. }
  169. if ((q->x[k][j]&pw) && (q->z[k][j]&pw)) // Y
  170. {
  171. if ((!(q->x[i][j]&pw)) && (q->z[i][j]&pw)) e++; // YZ=iX
  172. if ((q->x[i][j]&pw) && (!(q->z[i][j]&pw))) e--; // YX=-iZ
  173. }
  174. if ((!(q->x[k][j]&pw)) && (q->z[k][j]&pw)) // Z
  175. {
  176. if ((q->x[i][j]&pw) && (!(q->z[i][j]&pw))) e++; // ZX=iY
  177. if ((q->x[i][j]&pw) && (q->z[i][j]&pw)) e--; // ZY=-iX
  178. }
  179. }
  180. e = (e+q->r[i]+q->r[k])%4;
  181. if (e>=0) return e;
  182. else return e+4;
  183. }
  184. void rowmult(struct QState *q, long i, long k)
  185. // Left-multiply row i by row k
  186. {
  187. long j;
  188. q->r[i] = clifford(q,i,k);
  189. for (j = 0; j < q->over32; j++)
  190. {
  191. q->x[i][j] ^= q->x[k][j];
  192. q->z[i][j] ^= q->z[k][j];
  193. }
  194. return;
  195. }
  196. void printstate(struct QState *q)
  197. // Print the destabilizer and stabilizer for state q
  198. {
  199. long i;
  200. long j;
  201. long j5;
  202. unsigned long pw;
  203. for (i = 0; i < 2*q->n; i++)
  204. {
  205. if (i == q->n)
  206. {
  207. printf("\n");
  208. for (j = 0; j < q->n+1; j++)
  209. printf("-");
  210. }
  211. if (q->r[i]==2) printf("\n-");
  212. else printf("\n+");
  213. for (j = 0; j < q->n; j++)
  214. {
  215. j5 = j>>5;
  216. pw = q->pw[j&31];
  217. if ((!(q->x[i][j5]&pw)) && (!(q->z[i][j5]&pw))) printf("I");
  218. if ((q->x[i][j5]&pw) && (!(q->z[i][j5]&pw))) printf("X");
  219. if ((q->x[i][j5]&pw) && (q->z[i][j5]&pw)) printf("Y");
  220. if ((!(q->x[i][j5]&pw)) && (q->z[i][j5]&pw)) printf("Z");
  221. }
  222. }
  223. printf("\n");
  224. return;
  225. }
  226. int measure(struct QState *q, long b, int sup)
  227. // Measure qubit b
  228. // Return 0 if outcome would always be 0
  229. // 1 if outcome would always be 1
  230. // 2 if outcome was random and 0 was chosen
  231. // 3 if outcome was random and 1 was chosen
  232. // sup: 1 if determinate measurement results should be suppressed, 0 otherwise
  233. {
  234. int ran = 0;
  235. long i;
  236. long p; // pivot row in stabilizer
  237. long m; // pivot row in destabilizer
  238. long b5;
  239. unsigned long pw;
  240. b5 = b>>5;
  241. pw = q->pw[b&31];
  242. for (p = 0; p < q->n; p++) // loop over stabilizer generators
  243. {
  244. if (q->x[p+q->n][b5]&pw) ran = 1; // if a Zbar does NOT commute with Z_b (the
  245. if (ran) break; // operator being measured), then outcome is random
  246. }
  247. // If outcome is indeterminate
  248. if (ran)
  249. {
  250. rowcopy(q, p, p + q->n); // Set Xbar_p := Zbar_p
  251. rowset(q, p + q->n, b + q->n); // Set Zbar_p := Z_b
  252. q->r[p + q->n] = 2*(rand()%2); // moment of quantum randomness
  253. for (i = 0; i < 2*q->n; i++) // Now update the Xbar's and Zbar's that don't commute with
  254. if ((i!=p) && (q->x[i][b5]&pw)) // Z_b
  255. rowmult(q, i, p);
  256. if (q->r[p + q->n]) return 3;
  257. else return 2;
  258. }
  259. // If outcome is determinate
  260. if ((!ran) && (!sup))
  261. {
  262. for (m = 0; m < q->n; m++) // Before we were checking if stabilizer generators commute
  263. if (q->x[m][b5]&pw) break; // with Z_b; now we're checking destabilizer generators
  264. rowcopy(q, 2*q->n, m + q->n);
  265. for (i = m+1; i < q->n; i++)
  266. if (q->x[i][b5]&pw)
  267. rowmult(q, 2*q->n, i + q->n);
  268. if (q->r[2*q->n]) return 1;
  269. else return 0;
  270. /*for (i = m+1; i < q->n; i++)
  271. if (q->x[i][b5]&pw)
  272. {
  273. rowmult(q, m + q->n, i + q->n);
  274. rowmult(q, i, m);
  275. }
  276. return (int)q->r[m + q->n];*/
  277. }
  278. return 0;
  279. }
  280. long gaussian(struct QState *q)
  281. // Do Gaussian elimination to put the stabilizer generators in the following form:
  282. // At the top, a minimal set of generators containing X's and Y's, in "quasi-upper-triangular" form.
  283. // (Return value = number of such generators = log_2 of number of nonzero basis states)
  284. // At the bottom, generators containing Z's only in quasi-upper-triangular form.
  285. {
  286. long i = q->n;
  287. long k;
  288. long k2;
  289. long j;
  290. long j5;
  291. long g; // Return value
  292. unsigned long pw;
  293. for (j = 0; j < q->n; j++)
  294. {
  295. j5 = j>>5;
  296. pw = q->pw[j&31];
  297. for (k = i; k < 2*q->n; k++) // Find a generator containing X in jth column
  298. if (q->x[k][j5]&pw) break;
  299. if (k < 2*q->n)
  300. {
  301. rowswap(q, i, k);
  302. rowswap(q, i-q->n, k-q->n);
  303. for (k2 = i+1; k2 < 2*q->n; k2++)
  304. if (q->x[k2][j5]&pw)
  305. {
  306. rowmult(q, k2, i); // Gaussian elimination step
  307. rowmult(q, i-q->n, k2-q->n);
  308. }
  309. i++;
  310. }
  311. }
  312. g = i - q->n;
  313. for (j = 0; j < q->n; j++)
  314. {
  315. j5 = j>>5;
  316. pw = q->pw[j&31];
  317. for (k = i; k < 2*q->n; k++) // Find a generator containing Z in jth column
  318. if (q->z[k][j5]&pw) break;
  319. if (k < 2*q->n)
  320. {
  321. rowswap(q, i, k);
  322. rowswap(q, i-q->n, k-q->n);
  323. for (k2 = i+1; k2 < 2*q->n; k2++)
  324. if (q->z[k2][j5]&pw)
  325. {
  326. rowmult(q, k2, i);
  327. rowmult(q, i-q->n, k2-q->n);
  328. }
  329. i++;
  330. }
  331. }
  332. return g;
  333. }
  334. long innerprod(struct QState *q1, struct QState *q2)
  335. // Returns -1 if q1 and q2 are orthogonal
  336. // Otherwise, returns a nonnegative integer s such that the inner product is (1/sqrt(2))^s
  337. {
  338. return 0;
  339. }
  340. void printbasisstate(struct QState *q)
  341. // Prints the result of applying the Pauli operator in the "scratch space" of q to |0...0>
  342. {
  343. long j;
  344. long j5;
  345. unsigned long pw;
  346. int e = q->r[2*q->n];
  347. for (j = 0; j < q->n; j++)
  348. {
  349. j5 = j>>5;
  350. pw = q->pw[j&31];
  351. if ((q->x[2*q->n][j5]&pw) && (q->z[2*q->n][j5]&pw)) // Pauli operator is "Y"
  352. e = (e+1)%4;
  353. }
  354. if (e==0) printf("\n +|");
  355. if (e==1) printf("\n+i|");
  356. if (e==2) printf("\n -|");
  357. if (e==3) printf("\n-i|");
  358. for (j = 0; j < q->n; j++)
  359. {
  360. j5 = j>>5;
  361. pw = q->pw[j&31];
  362. if (q->x[2*q->n][j5]&pw) printf("1");
  363. else printf("0");
  364. }
  365. printf(">");
  366. return;
  367. }
  368. void dumpbasisstate(struct QState *q, PyObject *output, int index)
  369. // Dumps the result of applying the Pauli operator in the "scratch space" of q to |0...0>
  370. {
  371. long j;
  372. long j5;
  373. unsigned long pw;
  374. int e = q->r[2*q->n];
  375. char buffer[1024];
  376. strcpy(buffer, "");
  377. for (j = 0; j < q->n; j++)
  378. {
  379. j5 = j>>5;
  380. pw = q->pw[j&31];
  381. if ((q->x[2*q->n][j5]&pw) && (q->z[2*q->n][j5]&pw)) // Pauli operator is "Y"
  382. e = (e+1)%4;
  383. }
  384. if (e==0) strcat(buffer, "+|");
  385. if (e==1) strcat(buffer, "+i|");
  386. if (e==2) strcat(buffer, "-|");
  387. if (e==3) strcat(buffer, "-i|");
  388. int key;
  389. key = 0;
  390. for (j = 0; j < q->n; j++)
  391. {
  392. j5 = j>>5;
  393. pw = q->pw[j&31];
  394. if (q->x[2*q->n][j5]&pw) key = key ^ (1 << j);
  395. }
  396. PyDict_SetItem(output, Py_BuildValue("i", key), Py_BuildValue("i", e));
  397. return;
  398. }
  399. void seed(struct QState *q, long g)
  400. // Finds a Pauli operator P such that the basis state P|0...0> occurs with nonzero amplitude in q, and
  401. // writes P to the scratch space of q. For this to work, Gaussian elimination must already have been
  402. // performed on q. g is the return value from gaussian(q).
  403. {
  404. long i;
  405. long j;
  406. long j5;
  407. unsigned long pw;
  408. int f;
  409. long min=0;
  410. q->r[2*q->n] = 0;
  411. for (j = 0; j < q->over32; j++)
  412. {
  413. q->x[2*q->n][j] = 0; // Wipe the scratch space clean
  414. q->z[2*q->n][j] = 0;
  415. }
  416. for (i = 2*q->n - 1; i >= q->n + g; i--)
  417. {
  418. f = q->r[i];
  419. for (j = q->n - 1; j >= 0; j--)
  420. {
  421. j5 = j>>5;
  422. pw = q->pw[j&31];
  423. if (q->z[i][j5]&pw)
  424. {
  425. min = j;
  426. if (q->x[2*q->n][j5]&pw) f = (f+2)%4;
  427. }
  428. }
  429. if (f==2)
  430. {
  431. j5 = min>>5;
  432. pw = q->pw[min&31];
  433. q->x[2*q->n][j5] ^= pw; // Make the seed consistent with the ith equation
  434. }
  435. }
  436. return;
  437. }
  438. static char act_hadamard_docs[] = "act_hadamard(): Do a Hadamard";
  439. static PyObject* act_hadamard(PyObject* self, PyObject* args)
  440. {
  441. int which;
  442. if (!PyArg_ParseTuple(args, "i", &which)) { return NULL; }
  443. hadamard(q, which);
  444. Py_INCREF(Py_None);
  445. return Py_None;
  446. }
  447. static char act_phase_docs[] = "act_phase(): Do a phase";
  448. static PyObject* act_phase(PyObject* self, PyObject* args)
  449. {
  450. int which;
  451. if (!PyArg_ParseTuple(args, "i", &which)) { return NULL; }
  452. phase(q, which);
  453. Py_INCREF(Py_None);
  454. return Py_None;
  455. }
  456. static char act_cnot_docs[] = "act_cnot(): Do a CNOT";
  457. static PyObject* act_cnot(PyObject* self, PyObject* args)
  458. {
  459. int a, b;
  460. if (!PyArg_ParseTuple(args, "ii", &a, &b)) { return NULL; }
  461. cnot(q, a, b);
  462. Py_INCREF(Py_None);
  463. return Py_None;
  464. }
  465. static char get_ket_docs[] = "get_ket(): Get the state vector";
  466. static PyObject* get_ket(PyObject* self, PyObject* noarg)
  467. // Print the state in ket notation (warning: could be huge!)
  468. {
  469. long g; // log_2 of number of nonzero basis states
  470. unsigned long t;
  471. unsigned long t2;
  472. long i;
  473. g = gaussian(q);
  474. PyObject *output = PyDict_New();
  475. if (g > 30) { return output; }
  476. seed(q, g);
  477. dumpbasisstate(q, output, 0);
  478. for (t = 0; t < q->pw[g]-1; t++)
  479. {
  480. t2 = t ^ (t+1);
  481. for (i = 0; i < g; i++)
  482. if (t2 & q->pw[i])
  483. rowmult(q, 2*q->n, q->n + i);
  484. dumpbasisstate(q, output, t+1);
  485. }
  486. return output;
  487. }
  488. void printket(struct QState *q)
  489. // Print the state in ket notation (warning: could be huge!)
  490. {
  491. long g; // log_2 of number of nonzero basis states
  492. unsigned long t;
  493. unsigned long t2;
  494. long i;
  495. g = gaussian(q);
  496. printf("\n2^%ld nonzero basis states", g);
  497. if (g > 31)
  498. {
  499. printf("\nState is WAY too big to print");
  500. return;
  501. }
  502. seed(q, g);
  503. printbasisstate(q);
  504. for (t = 0; t < q->pw[g]-1; t++)
  505. {
  506. t2 = t ^ (t+1);
  507. for (i = 0; i < g; i++)
  508. if (t2 & q->pw[i])
  509. rowmult(q, 2*q->n, q->n + i);
  510. printbasisstate(q);
  511. }
  512. printf("\n");
  513. return;
  514. }
  515. void runprog(struct QProg *h, struct QState *q)
  516. // Simulate the quantum circuit
  517. {
  518. long t;
  519. int m; // measurement result
  520. time_t tp;
  521. double dt;
  522. char mvirgin = 1;
  523. time(&tp);
  524. for (t = 0; t < h->T; t++)
  525. {
  526. if (h->a[t]==CNOT) cnot(q,h->b[t],h->c[t]);
  527. if (h->a[t]==HADAMARD) hadamard(q,h->b[t]);
  528. if (h->a[t]==PHASE) phase(q,h->b[t]);
  529. if (h->a[t]==MEASURE)
  530. {
  531. if (mvirgin && h->DISPTIME)
  532. {
  533. dt = difftime(time(0),tp);
  534. printf("\nGate time: %lf seconds", dt);
  535. printf("\nTime per 10000 gates: %lf seconds", dt*10000.0f/(h->T - h->n));
  536. time(&tp);
  537. }
  538. mvirgin = 0;
  539. m = measure(q,h->b[t],h->SUPPRESSM);
  540. if (!h->SILENT)
  541. {
  542. printf("\nOutcome of measuring qubit %ld: ", h->b[t]);
  543. if (m>1) printf("%d (random)", m-2);
  544. else printf("%d", m);
  545. }
  546. }
  547. if (h->DISPPROG)
  548. {
  549. if (h->a[t]==CNOT) printf("\nCNOT %ld->%ld", h->b[t], h->c[t]);
  550. if (h->a[t]==HADAMARD) printf("\nHadamard %ld", h->b[t]);
  551. if (h->a[t]==PHASE) printf("\nPhase %ld", h->b[t]);
  552. }
  553. }
  554. printf("\n");
  555. if (h->DISPTIME)
  556. {
  557. dt = difftime(time(0),tp);
  558. printf("\nMeasurement time: %lf seconds", dt);
  559. printf("\nTime per 10000 measurements: %lf seconds\n", dt*10000.0f/h->n);
  560. }
  561. if (h->DISPQSTATE)
  562. {
  563. printf("\nFinal state:");
  564. printstate(q);
  565. gaussian(q);
  566. printstate(q);
  567. printket(q);
  568. }
  569. return;
  570. }
  571. void preparestate(struct QState *q, char *s)
  572. // Prepare the initial state's "input"
  573. {
  574. long l;
  575. long b;
  576. l = strlen(s);
  577. for (b = 0; b < l; b++)
  578. {
  579. if (s[b]=='Z')
  580. {
  581. hadamard(q,b);
  582. phase(q,b);
  583. phase(q,b);
  584. hadamard(q,b);
  585. }
  586. if (s[b]=='x') hadamard(q,b);
  587. if (s[b]=='X')
  588. {
  589. hadamard(q,b);
  590. phase(q,b);
  591. phase(q,b);
  592. }
  593. if (s[b]=='y')
  594. {
  595. hadamard(q,b);
  596. phase(q,b);
  597. }
  598. if (s[b]=='Y')
  599. {
  600. hadamard(q,b);
  601. phase(q,b);
  602. phase(q,b);
  603. phase(q,b);
  604. }
  605. }
  606. return;
  607. }
  608. void initstae_(struct QState *q, long n, char *s)
  609. // Initialize state q to have n qubits, and input specified by s
  610. {
  611. long i;
  612. long j;
  613. q->n = n;
  614. q->x = malloc((2*q->n + 1) * sizeof(unsigned long*));
  615. q->z = malloc((2*q->n + 1) * sizeof(unsigned long*));
  616. q->r = malloc((2*q->n + 1) * sizeof(int));
  617. q->over32 = (q->n>>5) + 1;
  618. q->pw[0] = 1;
  619. for (i = 1; i < 32; i++)
  620. q->pw[i] = 2*q->pw[i-1];
  621. for (i = 0; i < 2*q->n + 1; i++)
  622. {
  623. q->x[i] = malloc(q->over32 * sizeof(unsigned long));
  624. q->z[i] = malloc(q->over32 * sizeof(unsigned long));
  625. for (j = 0; j < q->over32; j++)
  626. {
  627. q->x[i][j] = 0;
  628. q->z[i][j] = 0;
  629. }
  630. if (i < q->n)
  631. q->x[i][i>>5] = q->pw[i&31];
  632. else if (i < 2*q->n)
  633. {
  634. j = i-q->n;
  635. q->z[i][j>>5] = q->pw[j&31];
  636. }
  637. q->r[i] = 0;
  638. }
  639. if (s) preparestate(q, s);
  640. return;
  641. }
  642. void readprog(struct QProg *h, char *fn, char *params)
  643. // Read a quantum circuit from filename fn, with optional parameters params
  644. {
  645. long t;
  646. char fn2[255];
  647. FILE *fp;
  648. char c=0;
  649. long val;
  650. long l;
  651. h->DISPQSTATE = 0;
  652. h->DISPTIME = 0;
  653. h->SILENT = 0;
  654. h->DISPPROG = 0;
  655. h->SUPPRESSM = 0;
  656. if (params)
  657. {
  658. l = strlen(params);
  659. for (t = 1; t < l; t++)
  660. {
  661. if ((params[t]=='q')||(params[t]=='Q')) h->DISPQSTATE = 1;
  662. if ((params[t]=='p')||(params[t]=='P')) h->DISPPROG = 1;
  663. if ((params[t]=='t')||(params[t]=='T')) h->DISPTIME = 1;
  664. if ((params[t]=='s')||(params[t]=='S')) h->SILENT = 1;
  665. if ((params[t]=='m')||(params[t]=='M')) h->SUPPRESSM = 1;
  666. }
  667. }
  668. sprintf(fn2, "%s", fn);
  669. fp = fopen(fn2, "r");
  670. if (!fp)
  671. {
  672. sprintf(fn2, "%s.chp", fn);
  673. fp = fopen(fn2, "r");
  674. if (!fp) error(1);
  675. }
  676. while (!feof(fp)&&(c!='#'))
  677. fscanf(fp, "%c", &c);
  678. if (c!='#') error(2);
  679. h->T = 0;
  680. h->n = 0;
  681. while (!feof(fp))
  682. {
  683. fscanf(fp, "%c", &c);
  684. if ((c=='\r')||(c=='\n'))
  685. continue;
  686. fscanf(fp, "%ld", &val);
  687. if (val+1 > h->n) h->n = val+1;
  688. if ((c=='c')||(c=='C'))
  689. {
  690. fscanf(fp, "%ld", &val);
  691. if (val+1 > h->n) h->n = val+1;
  692. }
  693. h->T++;
  694. }
  695. fclose(fp);
  696. h->a = malloc(h->T * sizeof(char));
  697. h->b = malloc(h->T * sizeof(long));
  698. h->c = malloc(h->T * sizeof(long));
  699. fp = fopen(fn2, "r");
  700. while (!feof(fp)&&(c!='#'))
  701. fscanf(fp, "%c", &c);
  702. t=0;
  703. while (!feof(fp))
  704. {
  705. fscanf(fp, "%c", &c);
  706. if ((c=='\r')||(c=='\n'))
  707. continue;
  708. if ((c=='c')||(c=='C')) h->a[t] = CNOT;
  709. if ((c=='h')||(c=='H')) h->a[t] = HADAMARD;
  710. if ((c=='p')||(c=='P')) h->a[t] = PHASE;
  711. if ((c=='m')||(c=='M')) h->a[t] = MEASURE;
  712. fscanf(fp, "%ld", &h->b[t]);
  713. if (h->a[t]==CNOT) fscanf(fp, "%ld", &h->c[t]);
  714. t++;
  715. }
  716. fclose(fp);
  717. return;
  718. }
  719. static char init_docs[] = "init(nqubits): Initialize with n qubits";
  720. static PyObject* init(PyObject* self, PyObject* args)
  721. {
  722. int nqubits;
  723. if (!PyArg_ParseTuple(args, "i", &nqubits)) { return NULL; }
  724. srand(time(0));
  725. h = malloc(sizeof(struct QProg));
  726. q = malloc(sizeof(struct QState));
  727. initstae_(q,nqubits,NULL);
  728. PyObject *response = Py_BuildValue("i", nqubits);
  729. return response;
  730. }
  731. static PyMethodDef chp_funcs[] = {
  732. {"init", (PyCFunction)init, METH_VARARGS, init_docs},
  733. {"get_ket", (PyCFunction)get_ket, METH_NOARGS, get_ket_docs},
  734. {"act_hadamard", (PyCFunction)act_hadamard, METH_VARARGS, act_hadamard_docs},
  735. {"act_cnot", (PyCFunction)act_cnot, METH_VARARGS, act_cnot_docs},
  736. {"act_phase", (PyCFunction)act_phase, METH_VARARGS, act_phase_docs},
  737. {NULL}
  738. };
  739. void initchp(int argc, char **argv)
  740. {
  741. Py_InitModule3("chp", chp_funcs, "CHP");
  742. }