#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 */
int /*<<< orphan*/ assert(int /*<<< orphan*/);
unsigned int get_var_mask(const char *basemask, const int bits, const int bp) {
  const char *const end = basemask + bits;
  unsigned int curbit = 1;
  unsigned int var_mask = 0;
  int count2s = 0;
  for (; basemask < end; basemask++, curbit <<= 1) {
    if (*basemask == 2) {
      count2s++;
      var_mask |= curbit;
      if (count2s == bp)
        return var_mask;
    }
  }
  assert(0);
}


x86 assembly (clang, -O0):
<get_var_mask>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x10(%rbp)
mov    %esi,-0x14(%rbp)
mov    %edx,-0x18(%rbp)
mov    -0x10(%rbp),%rax
movslq -0x14(%rbp),%rcx
add    %rcx,%rax
mov    %rax,-0x20(%rbp)
movl   $0x1,-0x24(%rbp)
movl   $0x0,-0x28(%rbp)
movl   $0x0,-0x2c(%rbp)
mov    -0x10(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    11b1 <get_var_mask+0xa1>
mov    -0x10(%rbp),%rax
movsbl (%rax),%eax
cmp    $0x2,%eax
jne    1192 <get_var_mask+0x82>
mov    -0x2c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x2c(%rbp)
mov    -0x24(%rbp),%eax
or     -0x28(%rbp),%eax
mov    %eax,-0x28(%rbp)
mov    -0x2c(%rbp),%eax
cmp    -0x18(%rbp),%eax
jne    118d <get_var_mask+0x7d>
mov    -0x28(%rbp),%eax
mov    %eax,-0x4(%rbp)
jmp    11b8 <get_var_mask+0xa8>
jmp    1192 <get_var_mask+0x82>
jmp    1197 <get_var_mask+0x87>
mov    -0x10(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x10(%rbp)
mov    -0x24(%rbp),%eax
shl    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    1146 <get_var_mask+0x36>
xor    %edi,%edi
call   1030 <assert@plt>
mov    -0x4(%rbp),%eax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_var_mask>:
push   %rax
test   %esi,%esi
jle    114b <get_var_mask+0x3b>
movslq %esi,%r8
add    %rdi,%r8
mov    $0x1,%esi
xor    %ecx,%ecx
xor    %eax,%eax
jmp    113b <get_var_mask+0x2b>
cs nopw 0x0(%rax,%rax,1)
add    $0x1,%rdi
add    %esi,%esi
cmp    %r8,%rdi
jae    114b <get_var_mask+0x3b>
cmpb   $0x2,(%rdi)
jne    1130 <get_var_mask+0x20>
add    $0x1,%ecx
or     %esi,%eax
cmp    %edx,%ecx
jne    1130 <get_var_mask+0x20>
pop    %rcx
ret
xor    %edi,%edi
call   1030 <assert@plt>
pop    %rcx
ret

LLVM IR (-O0):
define i32 @get_var_mask(i8* %0, i32 %1, i32 %2) {
%4 = alloca i32
%5 = alloca i8*
%6 = alloca i32
%7 = alloca i32
%8 = alloca i8*
%9 = alloca i32
%10 = alloca i32
%11 = alloca i32
store i8* %0, i8** %5
store i32 %1, i32* %6
store i32 %2, i32* %7
%12 = load i8*, i8** %5
%13 = load i32, i32* %6
%14 = sext i32 %13 to i64
%15 = getelementptr inbounds i8, i8* %12, i64 %14
store i8* %15, i8** %8
store i32 1, i32* %9
store i32 0, i32* %10
store i32 0, i32* %11
br label %16
16: ; preds = %38, %3
%17 = load i8*, i8** %5
%18 = load i8*, i8** %8
%19 = icmp ult i8* %17, %18
br i1 %19, label %20, label %43
20: ; preds = %16
%21 = load i8*, i8** %5
%22 = load i8, i8* %21
%23 = sext i8 %22 to i32
%24 = icmp eq i32 %23, 2
br i1 %24, label %25, label %37
25: ; preds = %20
%26 = load i32, i32* %11
%27 = add nsw i32 %26, 1
store i32 %27, i32* %11
%28 = load i32, i32* %9
%29 = load i32, i32* %10
%30 = or i32 %29, %28
store i32 %30, i32* %10
%31 = load i32, i32* %11
%32 = load i32, i32* %7
%33 = icmp eq i32 %31, %32
br i1 %33, label %34, label %36
34: ; preds = %25
%35 = load i32, i32* %10
store i32 %35, i32* %4
br label %45
36: ; preds = %25
br label %37
37: ; preds = %36, %20
br label %38
38: ; preds = %37
%39 = load i8*, i8** %5
%40 = getelementptr inbounds i8, i8* %39, i32 1
store i8* %40, i8** %5
%41 = load i32, i32* %9
%42 = shl i32 %41, 1
store i32 %42, i32* %9
br label %16
43: ; preds = %16
%44 = call i32 @assert(i32 0)
br label %45
45: ; preds = %43, %34
%46 = load i32, i32* %4
ret i32 %46
}

LLVM IR (-O2):
define i32 @get_var_mask(i8* %0, i32 %1, i32 %2) {
%4 = sext i32 %1 to i64
%5 = getelementptr inbounds i8, i8* %0, i64 %4
%6 = icmp sgt i32 %1, 0
br i1 %6, label %7, label %24
7: ; preds = %3, %18
%8 = phi i32 [ %20, %18 ], [ 0, %3 ]
%9 = phi i32 [ %19, %18 ], [ 0, %3 ]
%10 = phi i32 [ %22, %18 ], [ 1, %3 ]
%11 = phi i8* [ %21, %18 ], [ %0, %3 ]
%12 = load i8, i8* %11
%13 = icmp eq i8 %12, 2
br i1 %13, label %14, label %18
14: ; preds = %7
%15 = add nsw i32 %8, 1
%16 = or i32 %9, %10
%17 = icmp eq i32 %15, %2
br i1 %17, label %26, label %18
18: ; preds = %7, %14
%19 = phi i32 [ %16, %14 ], [ %9, %7 ]
%20 = phi i32 [ %15, %14 ], [ %8, %7 ]
%21 = getelementptr inbounds i8, i8* %11, i64 1
%22 = shl i32 %10, 1
%23 = icmp ult i8* %21, %5
br i1 %23, label %7, label %24
24: ; preds = %18, %3
%25 = tail call i32 @assert(i32 0)
br label %26
26: ; preds = %14, %24
%27 = phi i32 [ undef, %24 ], [ %16, %14 ]
ret i32 %27
}