#define NULL ((void *)0)
typedef unsigned long size_t; // Customize by platform.
typedef long intptr_t;
typedef unsigned long uintptr_t;
typedef long scalar_t__; // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */

/* Variables and functions */
double RandomStandardUniform();
int /*<<< orphan*/ assert(int);
double pow(double, int);
int RandomDirichlet_rejection(double *a, double *p, int n) {
  int b;
  int reject = -1;
  double t;
  for (b = 0; b < n; b++)
    assert(a[b] >= 0.);
  do {
    reject++;
    t = 0.;
    for (b = 0; b < n; b++) {
      p[b] = RandomStandardUniform();
      p[b] = pow(p[b], 1. / a[b]);
      t += p[b];
    }
    t += RandomStandardUniform();
  } while (t > 1.);
  for (b = 0; b < n; b++) {
    assert(p[b] >= 0.);
    p[b] /= t;
  }
  return 0;
}


x86 assembly (clang, -O0):
<RandomDirichlet_rejection>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
movl   $0xffffffff,-0x1c(%rbp)
movl   $0x0,-0x18(%rbp)
mov    -0x18(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    118c <RandomDirichlet_rejection+0x5c>
mov    -0x8(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
xorps  %xmm1,%xmm1
ucomisd %xmm1,%xmm0
setae  %al
and    $0x1,%al
movzbl %al,%edi
call   1040 <assert@plt>
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    1151 <RandomDirichlet_rejection+0x21>
jmp    1191 <RandomDirichlet_rejection+0x61>
mov    -0x1c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x1c(%rbp)
xorps  %xmm0,%xmm0
movsd  %xmm0,-0x28(%rbp)
movl   $0x0,-0x18(%rbp)
mov    -0x18(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    1226 <RandomDirichlet_rejection+0xf6>
mov    $0x0,%al
call   1030 <RandomStandardUniform@plt>
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
mov    -0x8(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  0xe1a(%rip),%xmm1
divsd  (%rax,%rcx,8),%xmm1
cvttsd2si %xmm1,%edi
call   1050 <pow@plt>
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
addsd  -0x28(%rbp),%xmm0
movsd  %xmm0,-0x28(%rbp)
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    11a9 <RandomDirichlet_rejection+0x79>
mov    $0x0,%al
call   1030 <RandomStandardUniform@plt>
addsd  -0x28(%rbp),%xmm0
movsd  %xmm0,-0x28(%rbp)
movsd  -0x28(%rbp),%xmm0
movsd  0xdbc(%rip),%xmm1
ucomisd %xmm1,%xmm0
ja     1191 <RandomDirichlet_rejection+0x61>
movl   $0x0,-0x18(%rbp)
mov    -0x18(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    12ab <RandomDirichlet_rejection+0x17b>
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
xorps  %xmm1,%xmm1
ucomisd %xmm1,%xmm0
setae  %al
and    $0x1,%al
movzbl %al,%edi
call   1040 <assert@plt>
movsd  -0x28(%rbp),%xmm1
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
divsd  %xmm1,%xmm0
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    1255 <RandomDirichlet_rejection+0x125>
xor    %eax,%eax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<RandomDirichlet_rejection>:
push   %rbp
push   %r15
push   %r14
push   %r12
push   %rbx
sub    $0x10,%rsp
mov    %edx,%r14d
mov    %rsi,%rbx
mov    %rdi,%r12
mov    %edx,%r15d
test   %edx,%edx
jle    119d <RandomDirichlet_rejection+0x6d>
xor    %ebp,%ebp
xchg   %ax,%ax
movsd  (%r12,%rbp,8),%xmm0
xor    %edi,%edi
ucomisd 0xea8(%rip),%xmm0
setae  %dil
call   1040 <assert@plt>
add    $0x1,%rbp
cmp    %rbp,%r15
jne    1150 <RandomDirichlet_rejection+0x20>
jmp    119d <RandomDirichlet_rejection+0x6d>
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
xor    %eax,%eax
call   1030 <RandomStandardUniform@plt>
addsd  0xe79(%rip),%xmm0
ucomisd 0xe69(%rip),%xmm0
jbe    1255 <RandomDirichlet_rejection+0x125>
test   %r14d,%r14d
jle    1180 <RandomDirichlet_rejection+0x50>
xorps  %xmm1,%xmm1
movsd  %xmm1,0x8(%rsp)
xor    %ebp,%ebp
nopl   (%rax)
xor    %eax,%eax
call   1030 <RandomStandardUniform@plt>
movsd  %xmm0,(%rbx,%rbp,8)
movsd  0xe3c(%rip),%xmm1
divsd  (%r12,%rbp,8),%xmm1
cvttsd2si %xmm1,%edi
call   1050 <pow@plt>
movsd  %xmm0,(%rbx,%rbp,8)
movsd  0x8(%rsp),%xmm1
addsd  %xmm0,%xmm1
movsd  %xmm1,0x8(%rsp)
add    $0x1,%rbp
cmp    %rbp,%r15
jne    11b0 <RandomDirichlet_rejection+0x80>
xor    %eax,%eax
call   1030 <RandomStandardUniform@plt>
movsd  0x8(%rsp),%xmm1
addsd  %xmm0,%xmm1
movsd  0xdf6(%rip),%xmm0
ucomisd %xmm0,%xmm1
ja     119d <RandomDirichlet_rejection+0x6d>
test   %r14d,%r14d
jle    1255 <RandomDirichlet_rejection+0x125>
xor    %ebp,%ebp
movsd  %xmm1,0x8(%rsp)
nopl   (%rax)
movsd  (%rbx,%rbp,8),%xmm0
xor    %edi,%edi
ucomisd 0xdd9(%rip),%xmm0
setae  %dil
call   1040 <assert@plt>
movsd  0x8(%rsp),%xmm1
movsd  (%rbx,%rbp,8),%xmm0
divsd  %xmm1,%xmm0
movsd  %xmm0,(%rbx,%rbp,8)
add    $0x1,%rbp
cmp    %rbp,%r15
jne    1220 <RandomDirichlet_rejection+0xf0>
xor    %eax,%eax
add    $0x10,%rsp
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define i32 @RandomDirichlet_rejection(double* %0, double* %1, i32 %2) {
%4 = alloca double*
%5 = alloca double*
%6 = alloca i32
%7 = alloca i32
%8 = alloca i32
%9 = alloca double
store double* %0, double** %4
store double* %1, double** %5
store i32 %2, i32* %6
store i32 -1, i32* %8
store i32 0, i32* %7
br label %10
10: ; preds = %23, %3
%11 = load i32, i32* %7
%12 = load i32, i32* %6
%13 = icmp slt i32 %11, %12
br i1 %13, label %14, label %26
14: ; preds = %10
%15 = load double*, double** %4
%16 = load i32, i32* %7
%17 = sext i32 %16 to i64
%18 = getelementptr inbounds double, double* %15, i64 %17
%19 = load double, double* %18
%20 = fcmp oge double %19, 0.000000e+00
%21 = zext i1 %20 to i32
%22 = call i32 @assert(i32 %21)
br label %23
23: ; preds = %14
%24 = load i32, i32* %7
%25 = add nsw i32 %24, 1
store i32 %25, i32* %7
br label %10
26: ; preds = %10
br label %27
27: ; preds = %71, %26
%28 = load i32, i32* %8
%29 = add nsw i32 %28, 1
store i32 %29, i32* %8
store double 0.000000e+00, double* %9
store i32 0, i32* %7
br label %30
30: ; preds = %64, %27
%31 = load i32, i32* %7
%32 = load i32, i32* %6
%33 = icmp slt i32 %31, %32
br i1 %33, label %34, label %67
34: ; preds = %30
%35 = call double (...) @RandomStandardUniform()
%36 = load double*, double** %5
%37 = load i32, i32* %7
%38 = sext i32 %37 to i64
%39 = getelementptr inbounds double, double* %36, i64 %38
store double %35, double* %39
%40 = load double*, double** %5
%41 = load i32, i32* %7
%42 = sext i32 %41 to i64
%43 = getelementptr inbounds double, double* %40, i64 %42
%44 = load double, double* %43
%45 = load double*, double** %4
%46 = load i32, i32* %7
%47 = sext i32 %46 to i64
%48 = getelementptr inbounds double, double* %45, i64 %47
%49 = load double, double* %48
%50 = fdiv double 1.000000e+00, %49
%51 = fptosi double %50 to i32
%52 = call double @pow(double %44, i32 %51)
%53 = load double*, double** %5
%54 = load i32, i32* %7
%55 = sext i32 %54 to i64
%56 = getelementptr inbounds double, double* %53, i64 %55
store double %52, double* %56
%57 = load double*, double** %5
%58 = load i32, i32* %7
%59 = sext i32 %58 to i64
%60 = getelementptr inbounds double, double* %57, i64 %59
%61 = load double, double* %60
%62 = load double, double* %9
%63 = fadd double %62, %61
store double %63, double* %9
br label %64
64: ; preds = %34
%65 = load i32, i32* %7
%66 = add nsw i32 %65, 1
store i32 %66, i32* %7
br label %30
67: ; preds = %30
%68 = call double (...) @RandomStandardUniform()
%69 = load double, double* %9
%70 = fadd double %69, %68
store double %70, double* %9
br label %71
71: ; preds = %67
%72 = load double, double* %9
%73 = fcmp ogt double %72, 1.000000e+00
br i1 %73, label %27, label %74
74: ; preds = %71
store i32 0, i32* %7
br label %75
75: ; preds = %95, %74
%76 = load i32, i32* %7
%77 = load i32, i32* %6
%78 = icmp slt i32 %76, %77
br i1 %78, label %79, label %98
79: ; preds = %75
%80 = load double*, double** %5
%81 = load i32, i32* %7
%82 = sext i32 %81 to i64
%83 = getelementptr inbounds double, double* %80, i64 %82
%84 = load double, double* %83
%85 = fcmp oge double %84, 0.000000e+00
%86 = zext i1 %85 to i32
%87 = call i32 @assert(i32 %86)
%88 = load double, double* %9
%89 = load double*, double** %5
%90 = load i32, i32* %7
%91 = sext i32 %90 to i64
%92 = getelementptr inbounds double, double* %89, i64 %91
%93 = load double, double* %92
%94 = fdiv double %93, %88
store double %94, double* %92
br label %95
95: ; preds = %79
%96 = load i32, i32* %7
%97 = add nsw i32 %96, 1
store i32 %97, i32* %7
br label %75
98: ; preds = %75
ret i32 0
}

LLVM IR (-O2):
define i32 @RandomDirichlet_rejection(double* nocapture %0, double* nocapture %1, i32 %2) {
%4 = icmp sgt i32 %2, 0
br i1 %4, label %5, label %7
5: ; preds = %3
%6 = zext i32 %2 to i64
br label %9
7: ; preds = %9, %3
%8 = zext i32 %2 to i64
br label %18
9: ; preds = %5, %9
%10 = phi i64 [ 0, %5 ], [ %16, %9 ]
%11 = getelementptr inbounds double, double* %0, i64 %10
%12 = load double, double* %11
%13 = fcmp oge double %12, 0.000000e+00
%14 = zext i1 %13 to i32
%15 = tail call i32 @assert(i32 %14)
%16 = add nuw nsw i64 %10, 1
%17 = icmp eq i64 %16, %6
br i1 %17, label %7, label %9
18: ; preds = %36, %7
br i1 %4, label %19, label %37
19: ; preds = %18, %19
%20 = phi i64 [ %30, %19 ], [ 0, %18 ]
%21 = phi double [ %29, %19 ], [ 0.000000e+00, %18 ]
%22 = tail call double (...) @RandomStandardUniform()
%23 = getelementptr inbounds double, double* %1, i64 %20
store double %22, double* %23
%24 = getelementptr inbounds double, double* %0, i64 %20
%25 = load double, double* %24
%26 = fdiv double 1.000000e+00, %25
%27 = fptosi double %26 to i32
%28 = tail call double @pow(double %22, i32 %27)
store double %28, double* %23
%29 = fadd double %21, %28
%30 = add nuw nsw i64 %20, 1
%31 = icmp eq i64 %30, %8
br i1 %31, label %32, label %19
32: ; preds = %19
%33 = tail call double (...) @RandomStandardUniform()
%34 = fadd double %29, %33
%35 = fcmp ogt double %34, 1.000000e+00
br i1 %35, label %36, label %41
36: ; preds = %32, %37
br label %18
37: ; preds = %18
%38 = tail call double (...) @RandomStandardUniform()
%39 = fadd double %38, 0.000000e+00
%40 = fcmp ogt double %39, 1.000000e+00
br i1 %40, label %36, label %55
41: ; preds = %32
br i1 %4, label %42, label %55
42: ; preds = %41
%43 = zext i32 %2 to i64
br label %44
44: ; preds = %42, %44
%45 = phi i64 [ 0, %42 ], [ %53, %44 ]
%46 = getelementptr inbounds double, double* %1, i64 %45
%47 = load double, double* %46
%48 = fcmp oge double %47, 0.000000e+00
%49 = zext i1 %48 to i32
%50 = tail call i32 @assert(i32 %49)
%51 = load double, double* %46
%52 = fdiv double %51, %34
store double %52, double* %46
%53 = add nuw nsw i64 %45, 1
%54 = icmp eq i64 %53, %43
br i1 %54, label %55, label %44
55: ; preds = %37, %44, %41
ret i32 0
}