How to pass Register to CodeAssembler method?

[ultimaweapon]

I tried to write the following code:

fn transform_lea64(&mut self, i: Instruction) -> usize {
    if i.is_ip_rel_memory_operand() {
        let dst = i.op0_register();
        let src = i.ip_rel_memory_address();

        if let Some(label) = self.labels.get(&src) {
            self.assembler.lea(dst, label.clone()).unwrap();
        } else {
            let label = self.assembler.create_label();

            self.assembler.lea(dst, label).unwrap();
            self.jobs.push_back((src, self.offset(src), label));
        }

        15
    } else {
        self.assembler.add_instruction(i).unwrap();
        i.len()
    }
}

But lea method on the assembler does not accept dst, which type is Register. I was tried dst.into() but no luck.

[ultimaweapon]

Just found there is a get_gpr64 method to convert Register to AsmRegister64.