int segment_tree_size(int input_size) {
  int i = 31;
  int x = 1;
  x = x << 30;
  if (input_size & (input_size - 1)) {
    while (!(x & input_size)) {
      x = x >> 1;
      --i;
    }
    return (1 << (i + 1)) - 1;
  } else {
    return (input_size << 1) - 1;
  }
}


x86 assembly (clang, -O0):
<segment_tree_size>:
push   %rbp
mov    %rsp,%rbp
mov    %edi,-0x8(%rbp)
movl   $0x1f,-0xc(%rbp)
movl   $0x1,-0x10(%rbp)
mov    -0x10(%rbp),%eax
shl    $0x1e,%eax
mov    %eax,-0x10(%rbp)
mov    -0x8(%rbp),%eax
mov    -0x8(%rbp),%ecx
sub    $0x1,%ecx
and    %ecx,%eax
cmp    $0x0,%eax
je     1181 <segment_tree_size+0x81>
jmp    1137 <segment_tree_size+0x37>
mov    -0x10(%rbp),%eax
and    -0x8(%rbp),%eax
cmp    $0x0,%eax
setne  %al
xor    $0xff,%al
test   $0x1,%al
jne    1152 <segment_tree_size+0x52>
jmp    1169 <segment_tree_size+0x69>
mov    -0x10(%rbp),%eax
sar    $0x1,%eax
mov    %eax,-0x10(%rbp)
mov    -0xc(%rbp),%eax
add    $0xffffffff,%eax
mov    %eax,-0xc(%rbp)
jmp    1137 <segment_tree_size+0x37>
mov    -0xc(%rbp),%ecx
add    $0x1,%ecx
mov    $0x1,%eax
shl    %cl,%eax
sub    $0x1,%eax
mov    %eax,-0x4(%rbp)
jmp    118d <segment_tree_size+0x8d>
mov    -0x8(%rbp),%eax
shl    $0x1,%eax
sub    $0x1,%eax
mov    %eax,-0x4(%rbp)
mov    -0x4(%rbp),%eax
pop    %rbp
ret

x86 assembly (clang, -O2):
<segment_tree_size>:
lea    -0x1(%rdi),%eax
test   %eax,%edi
jne    110e <segment_tree_size+0xe>
lea    (%rdi,%rdi,1),%eax
add    $0xffffffff,%eax
ret
mov    $0x20,%ecx
test   $0x40000000,%edi
jne    1139 <segment_tree_size+0x39>
mov    $0x40000000,%eax
mov    $0x20,%ecx
cs nopw 0x0(%rax,%rax,1)
nop
shr    %eax
add    $0xffffffff,%ecx
test   %edi,%eax
je     1130 <segment_tree_size+0x30>
mov    $0xffffffff,%eax
shl    %cl,%eax
not    %eax
ret

LLVM IR (-O0):
define i32 @segment_tree_size(i32 %0) {
%2 = alloca i32
%3 = alloca i32
%4 = alloca i32
%5 = alloca i32
store i32 %0, i32* %3
store i32 31, i32* %4
store i32 1, i32* %5
%6 = load i32, i32* %5
%7 = shl i32 %6, 30
store i32 %7, i32* %5
%8 = load i32, i32* %3
%9 = load i32, i32* %3
%10 = sub nsw i32 %9, 1
%11 = and i32 %8, %10
%12 = icmp ne i32 %11, 0
br i1 %12, label %13, label %30
13: ; preds = %1
br label %14
14: ; preds = %20, %13
%15 = load i32, i32* %5
%16 = load i32, i32* %3
%17 = and i32 %15, %16
%18 = icmp ne i32 %17, 0
%19 = xor i1 %18, true
br i1 %19, label %20, label %25
20: ; preds = %14
%21 = load i32, i32* %5
%22 = ashr i32 %21, 1
store i32 %22, i32* %5
%23 = load i32, i32* %4
%24 = add nsw i32 %23, -1
store i32 %24, i32* %4
br label %14
25: ; preds = %14
%26 = load i32, i32* %4
%27 = add nsw i32 %26, 1
%28 = shl i32 1, %27
%29 = sub nsw i32 %28, 1
store i32 %29, i32* %2
br label %34
30: ; preds = %1
%31 = load i32, i32* %3
%32 = shl i32 %31, 1
%33 = sub nsw i32 %32, 1
store i32 %33, i32* %2
br label %34
34: ; preds = %30, %25
%35 = load i32, i32* %2
ret i32 %35
}

LLVM IR (-O2):
define i32 @segment_tree_size(i32 %0) {
%2 = tail call i32 @llvm.ctpop.i32(i32 %0)
%3 = icmp ult i32 %2, 2
br i1 %3, label %18, label %4
4: ; preds = %1
%5 = and i32 %0, 1073741824
%6 = icmp eq i32 %5, 0
br i1 %6, label %7, label %14
7: ; preds = %4, %7
%8 = phi i32 [ %10, %7 ], [ 1073741824, %4 ]
%9 = phi i32 [ %11, %7 ], [ 31, %4 ]
%10 = lshr i32 %8, 1
%11 = add nsw i32 %9, -1
%12 = and i32 %10, %0
%13 = icmp eq i32 %12, 0
br i1 %13, label %7, label %14
14: ; preds = %7, %4
%15 = phi i32 [ 32, %4 ], [ %9, %7 ]
%16 = shl nsw i32 -1, %15
%17 = xor i32 %16, -1
br label %21
18: ; preds = %1
%19 = shl i32 %0, 1
%20 = add nsw i32 %19, -1
br label %21
21: ; preds = %18, %14
%22 = phi i32 [ %17, %14 ], [ %20, %18 ]
ret i32 %22
}