asm_backend.rs 58 KB
Newer Older
1 2
#![allow(unused_variables)]

3
use compiler::backend;
4 5
use compiler::backend::AOT_EMIT_CONTEXT_FILE;
use compiler::backend::AOT_EMIT_DIR;
qinsoon's avatar
qinsoon committed
6
use compiler::backend::RegGroup;
qinsoon's avatar
qinsoon committed
7
use utils::ByteSize;
8
use compiler::backend::x86_64;
9
use compiler::backend::x86_64::CodeGenerator;
qinsoon's avatar
qinsoon committed
10
use compiler::machine_code::MachineCode;
qinsoon's avatar
qinsoon committed
11
use vm::VM;
qinsoon's avatar
qinsoon committed
12
use runtime::ValueLocation;
13

qinsoon's avatar
qinsoon committed
14
use utils::vec_utils;
15
use utils::string_utils;
16 17 18
use ast::ptr::P;
use ast::ir::*;

19 20
use std::collections::HashMap;
use std::str;
21
use std::usize;
22
use std::slice::Iter;
23
use std::ops;
24 25

struct ASMCode {
qinsoon's avatar
qinsoon committed
26
    name: MuName, 
qinsoon's avatar
qinsoon committed
27
    code: Vec<ASMInst>,
qinsoon's avatar
qinsoon committed
28 29

    blocks: HashMap<MuName, ASMBlock>
30 31
}

qinsoon's avatar
qinsoon committed
32 33 34
unsafe impl Send for ASMCode {} 
unsafe impl Sync for ASMCode {}

35
impl ASMCode {
qinsoon's avatar
qinsoon committed
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
    fn get_use_locations(&self, reg: MuID) -> Vec<ASMLocation> {
        let mut ret = vec![];

        for inst in self.code.iter() {
            match inst.uses.get(&reg) {
                Some(ref locs) => {
                    ret.append(&mut locs.to_vec());
                },
                None => {}
            }
        }

        ret
    }

    fn get_define_locations(&self, reg: MuID) -> Vec<ASMLocation> {
        let mut ret = vec![];

        for inst in self.code.iter() {
            match inst.defines.get(&reg) {
                Some(ref locs) => {
                    ret.append(&mut locs.to_vec());
                },
                None => {}
            }
        }

        ret
    }

qinsoon's avatar
qinsoon committed
66 67 68 69 70 71 72 73 74
    fn is_block_start(&self, inst: usize) -> bool {
        for block in self.blocks.values() {
            if block.start_inst == inst {
                return true;
            }
        }
        false
    }

qinsoon's avatar
qinsoon committed
75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
    fn is_block_end(&self, inst: usize) -> bool {
        for block in self.blocks.values() {
            if block.end_inst == inst + 1 {
                return true;
            }
        }
        false
    }

    fn get_block_by_inst(&self, inst: usize) -> (&String, &ASMBlock) {
        for (name, block) in self.blocks.iter() {
            if inst >= block.start_inst && inst < block.end_inst {
                return (name, block);
            }
        }

        panic!("didnt find any block for inst {}", inst)
    }

qinsoon's avatar
qinsoon committed
94 95 96 97 98 99 100 101 102 103
    fn get_block_by_start_inst(&self, inst: usize) -> Option<&ASMBlock> {
        for block in self.blocks.values() {
            if block.start_inst == inst {
                return Some(block);
            }
        }

        None
    }

104 105 106 107 108 109 110 111
    fn rewrite_insert(
        &self,
        insert_before: HashMap<usize, Vec<Box<ASMCode>>>,
        insert_after: HashMap<usize, Vec<Box<ASMCode>>>) -> Box<ASMCode>
    {
        let mut ret = ASMCode {
            name: self.name.clone(),
            code: vec![],
qinsoon's avatar
qinsoon committed
112
            blocks: hashmap!{},
113 114 115 116
        };

        // iterate through old machine code
        let mut inst_offset = 0;    // how many instructions has been inserted
qinsoon's avatar
qinsoon committed
117
        let mut cur_block_start = usize::MAX;
118 119

        for i in 0..self.number_of_insts() {
qinsoon's avatar
qinsoon committed
120 121 122 123
            if self.is_block_start(i) {
                cur_block_start = i + inst_offset;
            }

124 125 126 127 128 129 130 131 132
            // insert code before this instruction
            if insert_before.contains_key(&i) {
                for insert in insert_before.get(&i).unwrap() {
                    ret.append_code_sequence_all(insert);
                    inst_offset += insert.number_of_insts();
                }
            }

            // copy this instruction
qinsoon's avatar
qinsoon committed
133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
            let mut inst = self.code[i].clone();

            // this instruction has been offset by several instructions('inst_offset')
            // update its info
            // 1. fix defines and uses
            for locs in inst.defines.values_mut() {
                for loc in locs {
                    debug_assert!(loc.line == i);
                    loc.line += inst_offset;
                }
            }
            for locs in inst.uses.values_mut() {
                for loc in locs {
                    debug_assert!(loc.line == i);
                    loc.line += inst_offset;
                }
            }
            // 2. we can ignore preds/succs - CFA is required anyway
            // 3. add the inst
            ret.code.push(inst);
153 154 155 156


            // insert code after this instruction
            if insert_after.contains_key(&i) {
qinsoon's avatar
qinsoon committed
157 158 159 160 161 162 163 164
                for insert in insert_after.get(&i).unwrap() {
                    ret.append_code_sequence_all(insert);
                    inst_offset += insert.number_of_insts();
                }
            }

            if self.is_block_end(i) {
                let cur_block_end = i + inst_offset;
165

qinsoon's avatar
qinsoon committed
166 167 168 169 170 171 172 173 174 175
                // copy the block
                let (name, block) = self.get_block_by_inst(i);

                let mut new_block = block.clone();
                new_block.start_inst = cur_block_start;
                cur_block_start = usize::MAX;
                new_block.end_inst = cur_block_end;

                // add to the new code
                ret.blocks.insert(name.clone(), new_block);
176 177 178
            }
        }

qinsoon's avatar
qinsoon committed
179 180 181
        ret.control_flow_analysis();

        Box::new(ret)
182 183 184 185 186 187 188 189
    }

    fn append_code_sequence(
        &mut self,
        another: &Box<ASMCode>,
        start_inst: usize,
        n_insts: usize)
    {
qinsoon's avatar
qinsoon committed
190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215
        let base_line = self.number_of_insts();

        for i in 0..n_insts {
            let cur_line_in_self = base_line + i;
            let cur_line_from_copy = start_inst + i;

            let mut inst = another.code[cur_line_from_copy].clone();

            // fix info
            for locs in inst.defines.values_mut() {
                for loc in locs {
                    debug_assert!(loc.line == i);
                    loc.line = cur_line_in_self;
                }
            }
            for locs in inst.uses.values_mut() {
                for loc in locs {
                    debug_assert!(loc.line == i);
                    loc.line = cur_line_in_self;
                }
            }
            // ignore preds/succs

            // add to self
            self.code.push(inst);
        }
216 217 218 219 220 221
    }

    fn append_code_sequence_all(&mut self, another: &Box<ASMCode>) {
        let n_insts = another.number_of_insts();
        self.append_code_sequence(another, 0, n_insts)
    }
qinsoon's avatar
qinsoon committed
222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323

    fn control_flow_analysis(&mut self) {
        const TRACE_CFA : bool = false;

        // control flow analysis
        let n_insts = self.number_of_insts();

        let ref blocks = self.blocks;
        let ref mut asm = self.code;

        let block_start = {
            let mut ret = vec![];
            for block in blocks.values() {
                ret.push(block.start_inst);
            }
            ret
        };

        for i in 0..n_insts {
            // determine predecessor - if cur is not block start, its predecessor is previous insts
            let is_block_start = block_start.contains(&i);
            if !is_block_start {
                if i > 0 {
                    if TRACE_CFA {
                        trace!("inst {}: not a block start", i);
                        trace!("inst {}: set PREDS as previous inst {}", i, i - 1);
                    }
                    asm[i].preds.push(i - 1);
                }
            } else {
                // if cur is a branch target, we already set its predecessor
                // if cur is a fall-through block, we set it in a sanity check pass
            }

            // determine successor
            let branch = asm[i].branch.clone();
            match branch {
                ASMBranchTarget::Unconditional(ref target) => {
                    // branch to target
                    let target_n = self.blocks.get(target).unwrap().start_inst;

                    // cur inst's succ is target
                    asm[i].succs.push(target_n);

                    // target's pred is cur
                    asm[target_n].preds.push(i);

                    if TRACE_CFA {
                        trace!("inst {}: is a branch to {}", i, target);
                        trace!("inst {}: branch target index is {}", i, target_n);
                        trace!("inst {}: set SUCCS as branch target {}", i, target_n);
                        trace!("inst {}: set PREDS as branch source {}", target_n, i);
                    }
                },
                ASMBranchTarget::Conditional(ref target) => {
                    // branch to target
                    let target_n = self.blocks.get(target).unwrap().start_inst;

                    // cur insts' succ is target and next inst
                    asm[i].succs.push(target_n);

                    if TRACE_CFA {
                        trace!("inst {}: is a cond branch to {}", i, target);
                        trace!("inst {}: branch target index is {}", i, target_n);
                        trace!("inst {}: set SUCCS as branch target {}", i, target_n);
                    }

                    if i < n_insts - 1 {
                        if TRACE_CFA {
                            trace!("inst {}: set SUCCS as next inst", i + 1);
                        }
                        asm[i].succs.push(i + 1);
                    }

                    // target's pred is cur
                    asm[target_n].preds.push(i);
                    if TRACE_CFA {
                        trace!("inst {}: set PREDS as {}", target_n, i);
                    }
                },
                ASMBranchTarget::None => {
                    // not branch nor cond branch, succ is next inst
                    if TRACE_CFA {
                        trace!("inst {}: not a branch inst", i);
                    }
                    if i < n_insts - 1 {
                        if TRACE_CFA {
                            trace!("inst {}: set SUCCS as next inst {}", i, i + 1);
                        }
                        asm[i].succs.push(i + 1);
                    }
                }
            }
        }

        // a sanity check for fallthrough blocks
        for i in 0..n_insts {
            if i != 0 && asm[i].preds.len() == 0 {
                asm[i].preds.push(i - 1);
            }
        }
    }
324 325 326 327
}

use std::any::Any;

328
impl MachineCode for ASMCode {
329 330 331
    fn as_any(&self) -> &Any {
        self
    }
332 333 334
    fn number_of_insts(&self) -> usize {
        self.code.len()
    }
335
    
336 337 338 339 340 341 342 343
    fn is_move(&self, index: usize) -> bool {
        let inst = self.code.get(index);
        match inst {
            Some(inst) => inst.code.starts_with("mov"),
            None => false
        }
    }
    
qinsoon's avatar
qinsoon committed
344
    fn is_using_mem_op(&self, index: usize) -> bool {
qinsoon's avatar
qinsoon committed
345
        self.code[index].is_mem_op_used
qinsoon's avatar
qinsoon committed
346 347
    }
    
qinsoon's avatar
qinsoon committed
348
    fn get_succs(&self, index: usize) -> &Vec<usize> {
qinsoon's avatar
qinsoon committed
349
        &self.code[index].succs
qinsoon's avatar
qinsoon committed
350 351 352
    }
    
    fn get_preds(&self, index: usize) -> &Vec<usize> {
qinsoon's avatar
qinsoon committed
353
        &self.code[index].preds
qinsoon's avatar
qinsoon committed
354 355
    }
    
qinsoon's avatar
qinsoon committed
356 357
    fn get_inst_reg_uses(&self, index: usize) -> Vec<MuID> {
        self.code[index].uses.keys().map(|x| *x).collect()
358 359
    }
    
qinsoon's avatar
qinsoon committed
360 361
    fn get_inst_reg_defines(&self, index: usize) -> Vec<MuID> {
        self.code[index].defines.keys().map(|x| *x).collect()
362
    }
363
    
364
    fn replace_reg(&mut self, from: MuID, to: MuID) {
365 366 367 368
        let to_reg_tag : MuName = match backend::all_regs().get(&to) {
            Some(reg) => reg.name().unwrap(),
            None => panic!("expecting a machine register, but we are required to replace to {}", to)
        };
369
        let to_reg_string = "%".to_string() + &to_reg_tag;
qinsoon's avatar
qinsoon committed
370 371 372 373 374 375 376

        for loc in self.get_define_locations(from) {
            let ref mut inst_to_patch = self.code[loc.line];
            for i in 0..loc.len {
                // FIXME: why loop here?
                string_utils::replace(&mut inst_to_patch.code, loc.index, &to_reg_string, to_reg_string.len());
            }
377
        }
378

qinsoon's avatar
qinsoon committed
379 380 381 382 383
        for loc in self.get_use_locations(from) {
            let ref mut inst_to_patch = self.code[loc.line];
            for i in 0..loc.len {
                string_utils::replace(&mut inst_to_patch.code, loc.index, &to_reg_string, to_reg_string.len());
            }
384 385
        }
    }
386

387
    fn replace_tmp_for_inst(&mut self, from: MuID, to: MuID, inst: usize) {
388 389 390 391 392 393 394 395
        let to_reg_string : MuName = match backend::all_regs().get(&to) {
            Some(ref machine_reg) => {
                let name = machine_reg.name().unwrap();
                "%".to_string() + &name
            },
            None => REG_PLACEHOLDER.clone()
        };

qinsoon's avatar
qinsoon committed
396 397 398 399 400 401 402 403 404
        let asm = &mut self.code[inst];
        // if this reg is defined, replace the define
        if asm.defines.contains_key(&from) {
            let define_locs = asm.defines.get(&from).unwrap().to_vec();
            // replace temps
            for loc in define_locs.iter() {
                for i in 0..loc.len {
                    string_utils::replace(&mut asm.code, loc.index, &to_reg_string, to_reg_string.len());
                }
405 406
            }

qinsoon's avatar
qinsoon committed
407 408 409
            // remove old key, insert new one
            asm.defines.remove(&from);
            asm.defines.insert(from, define_locs);
410
        }
qinsoon's avatar
qinsoon committed
411 412 413 414 415 416 417 418

        // if this reg is used, replace the use
        if asm.uses.contains_key(&from) {
            let use_locs = asm.uses.get(&from).unwrap().to_vec();
            // replace temps
            for loc in use_locs.iter() {
                for i in 0..loc.len {
                    string_utils::replace(&mut asm.code, loc.index, &to_reg_string, to_reg_string.len());
419 420 421
                }
            }

qinsoon's avatar
qinsoon committed
422 423 424
            // remove old key, insert new one
            asm.uses.remove(&from);
            asm.uses.insert(from, use_locs);
425 426
        }
    }
427
    
428 429
    fn set_inst_nop(&mut self, index: usize) {
        self.code.remove(index);
qinsoon's avatar
qinsoon committed
430
        self.code.insert(index, ASMInst::nop());
431 432
    }
    
433 434 435 436 437 438 439 440 441 442 443
    fn emit(&self) -> Vec<u8> {
        let mut ret = vec![];
        
        for inst in self.code.iter() {
            ret.append(&mut inst.code.clone().into_bytes());
            ret.append(&mut "\n".to_string().into_bytes());
        }
        
        ret
    }
    
444 445
    fn trace_mc(&self) {
        trace!("");
446

447 448
        trace!("code for {}: \n", self.name);
        
449 450
        let n_insts = self.code.len();
        for i in 0..n_insts {
451
            self.trace_inst(i);
452 453
        }
        
454 455 456 457 458 459
        trace!("")      
    }
    
    fn trace_inst(&self, i: usize) {
        trace!("#{}\t{:30}\t\tdefine: {:?}\tuses: {:?}\tpred: {:?}\tsucc: {:?}", 
            i, self.code[i].code, self.get_inst_reg_defines(i), self.get_inst_reg_uses(i),
qinsoon's avatar
qinsoon committed
460
            self.code[i].preds, self.code[i].succs);
461
    }
462
    
463
    fn get_ir_block_livein(&self, block: &str) -> Option<&Vec<MuID>> {
qinsoon's avatar
qinsoon committed
464 465 466 467
        match self.blocks.get(block) {
            Some(ref block) => Some(&block.livein),
            None => None
        }
468 469
    }
    
470
    fn get_ir_block_liveout(&self, block: &str) -> Option<&Vec<MuID>> {
qinsoon's avatar
qinsoon committed
471 472 473 474
        match self.blocks.get(block) {
            Some(ref block) => Some(&block.liveout),
            None => None
        }
475 476
    }
    
477
    fn set_ir_block_livein(&mut self, block: &str, set: Vec<MuID>) {
qinsoon's avatar
qinsoon committed
478 479
        let block = self.blocks.get_mut(block).unwrap();
        block.livein = set;
480 481 482
    }
    
    fn set_ir_block_liveout(&mut self, block: &str, set: Vec<MuID>) {
qinsoon's avatar
qinsoon committed
483 484
        let block = self.blocks.get_mut(block).unwrap();
        block.liveout = set;
485 486
    }
    
qinsoon's avatar
qinsoon committed
487 488
    fn get_all_blocks(&self) -> Vec<MuName> {
        self.blocks.keys().map(|x| x.clone()).collect()
489 490
    }
    
491
    fn get_block_range(&self, block: &str) -> Option<ops::Range<usize>> {
qinsoon's avatar
qinsoon committed
492 493
        match self.blocks.get(block) {
            Some(ref block) => Some(block.start_inst..block.end_inst),
494 495 496
            None => None
        }
    }
497 498
}

qinsoon's avatar
qinsoon committed
499 500 501 502 503 504 505
#[derive(Clone, Debug)]
enum ASMBranchTarget {
    None,
    Conditional(MuName),
    Unconditional(MuName)
}

qinsoon's avatar
qinsoon committed
506
#[derive(Clone, Debug)]
qinsoon's avatar
qinsoon committed
507
struct ASMInst {
508
    code: String,
qinsoon's avatar
qinsoon committed
509 510 511 512 513 514 515 516

    defines: HashMap<MuID, Vec<ASMLocation>>,
    uses: HashMap<MuID, Vec<ASMLocation>>,

    is_mem_op_used: bool,
    preds: Vec<usize>,
    succs: Vec<usize>,
    branch: ASMBranchTarget
517 518
}

qinsoon's avatar
qinsoon committed
519 520 521
impl ASMInst {
    fn symbolic(line: String) -> ASMInst {
        ASMInst {
522
            code: line,
qinsoon's avatar
qinsoon committed
523 524 525 526 527 528
            defines: HashMap::new(),
            uses: HashMap::new(),
            is_mem_op_used: false,
            preds: vec![],
            succs: vec![],
            branch: ASMBranchTarget::None
529 530 531
        }
    }
    
qinsoon's avatar
qinsoon committed
532 533 534 535 536 537 538 539 540
    fn inst(
        inst: String,
        defines: HashMap<MuID, Vec<ASMLocation>>,
        uses: HashMap<MuID, Vec<ASMLocation>>,
        is_mem_op_used: bool,
        target: ASMBranchTarget
    ) -> ASMInst
    {
        ASMInst {
541 542
            code: inst,
            defines: defines,
qinsoon's avatar
qinsoon committed
543 544 545 546 547
            uses: uses,
            is_mem_op_used: is_mem_op_used,
            preds: vec![],
            succs: vec![],
            branch: target
548 549
        }
    }
550
    
qinsoon's avatar
qinsoon committed
551 552
    fn nop() -> ASMInst {
        ASMInst {
553
            code: "".to_string(),
qinsoon's avatar
qinsoon committed
554 555 556 557 558 559
            defines: HashMap::new(),
            uses: HashMap::new(),
            is_mem_op_used: false,
            preds: vec![],
            succs: vec![],
            branch: ASMBranchTarget::None
560 561
        }
    }
562 563
}

564
#[derive(Clone, Debug)]
565 566 567 568 569 570 571
struct ASMLocation {
    line: usize,
    index: usize,
    len: usize
}

impl ASMLocation {
qinsoon's avatar
qinsoon committed
572
    fn new(line: usize, index: usize, len: usize) -> ASMLocation {
573
        ASMLocation{
qinsoon's avatar
qinsoon committed
574
            line: line,
575 576 577 578 579 580
            index: index,
            len: len
        }
    }
}

qinsoon's avatar
qinsoon committed
581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601
#[derive(Clone, Debug)]
/// [start_inst, end_inst)
struct ASMBlock {
    start_inst: usize,
    end_inst: usize,

    livein: Vec<MuID>,
    liveout: Vec<MuID>
}

impl ASMBlock {
    fn new() -> ASMBlock {
        ASMBlock {
            start_inst: usize::MAX,
            end_inst: usize::MAX,
            livein: vec![],
            liveout: vec![]
        }
    }
}

602
pub struct ASMCodeGen {
603
    cur: Option<Box<ASMCode>>
604 605
}

606
const REG_PLACEHOLDER_LEN : usize = 5;
607 608 609 610 611 612
lazy_static! {
    pub static ref REG_PLACEHOLDER : String = {
        let blank_spaces = [' ' as u8; REG_PLACEHOLDER_LEN];
        
        format!("%{}", str::from_utf8(&blank_spaces).unwrap())
    };
613 614 615 616
}

impl ASMCodeGen {
    pub fn new() -> ASMCodeGen {
617
        ASMCodeGen {
618
            cur: None
619 620 621 622 623 624 625 626 627 628 629
        }
    }
    
    fn cur(&self) -> &ASMCode {
        self.cur.as_ref().unwrap()
    }
    
    fn cur_mut(&mut self) -> &mut ASMCode {
        self.cur.as_mut().unwrap()
    }
    
630 631 632 633
    fn line(&self) -> usize {
        self.cur().code.len()
    }
    
634 635
    fn add_asm_label(&mut self, code: String) {
        let l = self.line();
qinsoon's avatar
qinsoon committed
636
        self.cur_mut().code.push(ASMInst::symbolic(code));
637 638
    }
    
639
    fn add_asm_block_label(&mut self, code: String, block_name: MuName) {
640
        let l = self.line();
qinsoon's avatar
qinsoon committed
641
        self.cur_mut().code.push(ASMInst::symbolic(code));
642 643
    }
    
644
    fn add_asm_symbolic(&mut self, code: String){
qinsoon's avatar
qinsoon committed
645
        self.cur_mut().code.push(ASMInst::symbolic(code));
646 647
    }
    
648 649
    fn prepare_machine_regs(&self, regs: Iter<P<Value>>) -> Vec<MuID> {
        regs.map(|x| self.prepare_machine_reg(x)).collect()
qinsoon's avatar
qinsoon committed
650
    }
651
    
652
    fn add_asm_call(&mut self, code: String) {
qinsoon's avatar
qinsoon committed
653
        // a call instruction will use all the argument registers
qinsoon's avatar
qinsoon committed
654 655 656 657 658 659 660
        let mut uses : HashMap<MuID, Vec<ASMLocation>> = HashMap::new();
        for reg in x86_64::ARGUMENT_GPRs.iter() {
            uses.insert(reg.id(), vec![]);
        }
        for reg in x86_64::ARGUMENT_FPRs.iter() {
            uses.insert(reg.id(), vec![]);
        }
qinsoon's avatar
qinsoon committed
661 662

        // defines: return registers
qinsoon's avatar
qinsoon committed
663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679
        let mut defines : HashMap<MuID, Vec<ASMLocation>> = HashMap::new();
        for reg in x86_64::RETURN_GPRs.iter() {
            defines.insert(reg.id(), vec![]);
        }
        for reg in x86_64::RETURN_FPRs.iter() {
            defines.insert(reg.id(), vec![]);
        }
        for reg in x86_64::CALLER_SAVED_GPRs.iter() {
            if !defines.contains_key(&reg.id()) {
                defines.insert(reg.id(), vec![]);
            }
        }
        for reg in x86_64::CALLER_SAVED_FPRs.iter() {
            if !defines.contains_key(&reg.id()) {
                defines.insert(reg.id(), vec![]);
            }
        }
680
          
qinsoon's avatar
qinsoon committed
681
        self.add_asm_inst(code, defines, uses, false);
682 683 684
    }
    
    fn add_asm_ret(&mut self, code: String) {
qinsoon's avatar
qinsoon committed
685
        let uses : HashMap<MuID, Vec<ASMLocation>> = {
qinsoon's avatar
qinsoon committed
686 687 688 689 690 691 692 693 694
            let mut ret = HashMap::new();
            for reg in x86_64::RETURN_GPRs.iter() {
                ret.insert(reg.id(), vec![]);
            }
            for reg in x86_64::RETURN_FPRs.iter() {
                ret.insert(reg.id(), vec![]);
            }
            ret
        };
695
        
qinsoon's avatar
qinsoon committed
696
        self.add_asm_inst(code, hashmap!{}, uses, false);
697 698
    }
    
699
    fn add_asm_branch(&mut self, code: String, target: MuName) {
qinsoon's avatar
qinsoon committed
700
        self.add_asm_inst_internal(code, hashmap!{}, hashmap!{}, false, ASMBranchTarget::Unconditional(target));
701 702
    }
    
703
    fn add_asm_branch2(&mut self, code: String, target: MuName) {
qinsoon's avatar
qinsoon committed
704
        self.add_asm_inst_internal(code, hashmap!{}, hashmap!{}, false, ASMBranchTarget::Conditional(target));
705 706 707 708 709
    }
    
    fn add_asm_inst(
        &mut self, 
        code: String, 
qinsoon's avatar
qinsoon committed
710 711 712 713 714 715 716 717 718 719 720 721 722 723
        defines: HashMap<MuID, Vec<ASMLocation>>,
        uses: HashMap<MuID, Vec<ASMLocation>>,
        is_using_mem_op: bool)
    {
        self.add_asm_inst_internal(code, defines, uses, is_using_mem_op, ASMBranchTarget::None)
    }

    fn add_asm_inst_internal(
        &mut self,
        code: String,
        defines: HashMap<MuID, Vec<ASMLocation>>,
        uses: HashMap<MuID, Vec<ASMLocation>>,
        is_using_mem_op: bool,
        target: ASMBranchTarget)
724
    {
725 726
        let line = self.line();
        trace!("asm: {}", code);
qinsoon's avatar
qinsoon committed
727 728
        trace!("     defines: {:?}", defines);
        trace!("     uses: {:?}", uses);
729
        let mc = self.cur_mut();
qinsoon's avatar
qinsoon committed
730

731
        // put the instruction
qinsoon's avatar
qinsoon committed
732
        mc.code.push(ASMInst::inst(code, defines, uses, is_using_mem_op, target));
733 734
    }
    
735
    fn prepare_reg(&self, op: &P<Value>, loc: usize) -> (String, MuID, ASMLocation) {
736 737 738 739 740 741 742
        if cfg!(debug_assertions) {
            match op.v {
                Value_::SSAVar(_) => {},
                _ => panic!("expecting register op")
            }
        }
        
743 744
        let str = self.asm_reg_op(op);
        let len = str.len();
qinsoon's avatar
qinsoon committed
745
        (str, op.extract_ssa_id().unwrap(), ASMLocation::new(self.line(), loc, len))
746 747 748
    }
    
    fn prepare_machine_reg(&self, op: &P<Value>) -> MuID {
749 750 751 752 753 754 755
        if cfg!(debug_assertions) {
            match op.v {
                Value_::SSAVar(_) => {},
                _ => panic!("expecting machine register op")
            }
        }        
        
756
        op.extract_ssa_id().unwrap()
757
    }
758
    
759
    #[allow(unused_assignments)]
qinsoon's avatar
qinsoon committed
760
    fn prepare_mem(&self, op: &P<Value>, loc: usize) -> (String, HashMap<MuID, Vec<ASMLocation>>) {
761 762 763 764 765 766
        if cfg!(debug_assertions) {
            match op.v {
                Value_::Memory(_) => {},
                _ => panic!("expecting register op")
            }
        }        
qinsoon's avatar
qinsoon committed
767

768 769 770 771
        let mut ids : Vec<MuID> = vec![];
        let mut locs : Vec<ASMLocation> = vec![];
        let mut result_str : String = "".to_string();
        
772
        let mut loc_cursor : usize = loc;
773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854
        
        match op.v {
            // offset(base,index,scale)
            Value_::Memory(MemoryLocation::Address{ref base, ref offset, ref index, scale}) => {
                // deal with offset
                if offset.is_some() {
                    let offset = offset.as_ref().unwrap();
                    
                    match offset.v {
                        Value_::SSAVar(id) => {
                            // temp as offset
                            let (str, id, loc) = self.prepare_reg(offset, 0);
                            
                            result_str.push_str(&str);
                            ids.push(id);
                            locs.push(loc);
                            
                            loc_cursor += str.len();
                        },
                        Value_::Constant(Constant::Int(val)) => {
                            let str = val.to_string();
                            
                            result_str.push_str(&str);
                            loc_cursor += str.len();
                        },
                        _ => panic!("unexpected offset type: {:?}", offset)
                    }
                }
                
                result_str.push('(');
                loc_cursor += 1; 
                
                // deal with base, base is ssa
                let (str, id, loc) = self.prepare_reg(base, loc_cursor);
                result_str.push_str(&str);
                ids.push(id);
                locs.push(loc);
                loc_cursor += str.len();
                
                // deal with index (ssa or constant)
                if index.is_some() {
                    result_str.push(',');
                    loc_cursor += 1; // plus 1 for ,                    
                    
                    let index = index.as_ref().unwrap();
                    
                    match index.v {
                        Value_::SSAVar(id) => {
                            // temp as offset
                            let (str, id, loc) = self.prepare_reg(index, loc_cursor);
                            
                            result_str.push_str(&str);
                            ids.push(id);
                            locs.push(loc);
                            
                            loc_cursor += str.len();
                        },
                        Value_::Constant(Constant::Int(val)) => {
                            let str = val.to_string();
                            
                            result_str.push_str(&str);
                            loc_cursor += str.len();
                        },
                        _ => panic!("unexpected index type: {:?}", index)
                    }
                    
                    // scale
                    if scale.is_some() {
                        result_str.push(',');
                        loc_cursor += 1;
                        
                        let scale = scale.unwrap();
                        let str = scale.to_string();
                        
                        result_str.push_str(&str);
                        loc_cursor += str.len();
                    }
                }
                
                result_str.push(')');
                loc_cursor += 1;
            },
855 856
            Value_::Memory(MemoryLocation::Symbolic{ref base, ref label}) => {
                result_str.push_str(&symbol(label.clone()));
857 858 859 860 861 862 863 864 865 866 867
                loc_cursor += label.len();
                
                if base.is_some() {
                    result_str.push('(');
                    loc_cursor += 1;
                    
                    let (str, id, loc) = self.prepare_reg(base.as_ref().unwrap(), loc_cursor);
                    result_str.push_str(&str);
                    ids.push(id);
                    locs.push(loc);
                    loc_cursor += str.len();
qinsoon's avatar
qinsoon committed
868

869
                    result_str.push(')');
qinsoon's avatar
qinsoon committed
870
                    loc_cursor += 1;
871 872 873 874
                }
            },
            _ => panic!("expect mem location as value")
        }
qinsoon's avatar
qinsoon committed
875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892

        let uses : HashMap<MuID, Vec<ASMLocation>> = {
            let mut map : HashMap<MuID, Vec<ASMLocation>> = hashmap!{};
            for i in 0..ids.len() {
                let id = ids[i];
                let loc = locs[i].clone();

                if map.contains_key(&id) {
                    map.get_mut(&id).unwrap().push(loc);
                } else {
                    map.insert(id, vec![loc]);
                }
            }
            map
        };


        (result_str, uses)
893 894
    }
    
895 896
    fn asm_reg_op(&self, op: &P<Value>) -> String {
        let id = op.extract_ssa_id().unwrap();
qinsoon's avatar
qinsoon committed
897
        if id < MACHINE_ID_END {
898
            // machine reg
899
            format!("%{}", op.name().unwrap())
900 901 902 903 904 905
        } else {
            // virtual register, use place holder
            REG_PLACEHOLDER.clone()
        }
    }
    
qinsoon's avatar
qinsoon committed
906
    fn asm_block_label(&self, label: MuName) -> String {
907
        symbol(format!("{}_{}", self.cur().name, label))
908
    }
909 910 911 912
    
    fn control_flow_analysis(&mut self) {
        // control flow analysis
        let n_insts = self.line();
qinsoon's avatar
qinsoon committed
913

914
        let code = self.cur_mut();
qinsoon's avatar
qinsoon committed
915
        let ref blocks = code.blocks;
qinsoon's avatar
qinsoon committed
916
        let ref mut asm = code.code;
qinsoon's avatar
qinsoon committed
917 918 919 920 921 922 923 924

        let block_start = {
            let mut ret = vec![];
            for block in blocks.values() {
                ret.push(block.start_inst);
            }
            ret
        };
925 926 927
        
        for i in 0..n_insts {
            // determine predecessor - if cur is not block start, its predecessor is previous insts
qinsoon's avatar
qinsoon committed
928 929
            let is_block_start = block_start.contains(&i);
            if !is_block_start {
930
                if i > 0 {
931 932
                    trace!("inst {}: not a block start", i);
                    trace!("inst {}: set PREDS as previous inst {}", i, i-1);
qinsoon's avatar
qinsoon committed
933
                    asm[i].preds.push(i - 1);
934 935 936 937 938 939 940
                }
            } else {
                // if cur is a branch target, we already set its predecessor
                // if cur is a fall-through block, we set it in a sanity check pass
            }
            
            // determine successor
qinsoon's avatar
qinsoon committed
941 942 943 944 945 946
            let branch = asm[i].branch.clone();
            match branch {
                ASMBranchTarget::Unconditional(ref target) => {
                    // branch to target
                    trace!("inst {}: is a branch to {}", i, target);

qinsoon's avatar
qinsoon committed
947
                    let target_n = code.blocks.get(target).unwrap().start_inst;
qinsoon's avatar
qinsoon committed
948 949 950 951
                    trace!("inst {}: branch target index is {}", i, target_n);

                    // cur inst's succ is target
                    trace!("inst {}: set SUCCS as branch target {}", i, target_n);
qinsoon's avatar
qinsoon committed
952
                    asm[i].succs.push(target_n);
qinsoon's avatar
qinsoon committed
953 954 955

                    // target's pred is cur
                    trace!("inst {}: set PREDS as branch source {}", target_n, i);
qinsoon's avatar
qinsoon committed
956
                    asm[target_n].preds.push(i);
qinsoon's avatar
qinsoon committed
957 958
                },
                ASMBranchTarget::Conditional(ref target) => {
959
                    // branch to target
960
                    trace!("inst {}: is a cond branch to {}", i, target);
qinsoon's avatar
qinsoon committed
961

qinsoon's avatar
qinsoon committed
962
                    let target_n = code.blocks.get(target).unwrap().start_inst;
963
                    trace!("inst {}: branch target index is {}", i, target_n);
qinsoon's avatar
qinsoon committed
964

965
                    // cur insts' succ is target and next inst
qinsoon's avatar
qinsoon committed
966
                    asm[i].succs.push(target_n);
967
                    trace!("inst {}: set SUCCS as branch target {}", i, target_n);
968
                    if i < n_insts - 1 {
qinsoon's avatar
qinsoon committed
969 970
                        trace!("inst {}: set SUCCS as next inst", i + 1);
                        asm[i].succs.push(i + 1);
971
                    }
qinsoon's avatar
qinsoon committed
972

973
                    // target's pred is cur
qinsoon's avatar
qinsoon committed
974 975
                    asm[target_n].preds.push(i);
                    trace!("inst {}: set PREDS as {}", target_n, i);
qinsoon's avatar
qinsoon committed
976 977
                },
                ASMBranchTarget::None => {
978
                    // not branch nor cond branch, succ is next inst
979
                    trace!("inst {}: not a branch inst", i);
980
                    if i < n_insts - 1 {
981
                        trace!("inst {}: set SUCCS as next inst {}", i, i + 1);
qinsoon's avatar
qinsoon committed
982
                        asm[i].succs.push(i + 1);
983 984
                    }
                }
qinsoon's avatar
qinsoon committed
985
            }
986 987 988 989
        }
        
        // a sanity check for fallthrough blocks
        for i in 0..n_insts {
qinsoon's avatar
qinsoon committed
990 991
            if i != 0 && asm[i].preds.len() == 0 {
                asm[i].preds.push(i - 1);
992 993 994
            }
        }        
    }
995 996 997 998

    fn finish_code_sequence_asm(&mut self) -> Box<ASMCode> {
        self.cur.take().unwrap()
    }
999 1000 1001
}

impl CodeGenerator for ASMCodeGen {
qinsoon's avatar
qinsoon committed
1002
    fn start_code(&mut self, func_name: MuName) -> ValueLocation {
1003
        self.cur = Some(Box::new(ASMCode {
1004
                name: func_name.clone(),
1005
                code: vec![],
qinsoon's avatar
qinsoon committed
1006
                blocks: hashmap!{},
1007
            }));
1008
        
qinsoon's avatar
qinsoon committed
1009
        // to link with C sources via gcc
qinsoon's avatar
qinsoon committed
1010 1011 1012 1013
        let func_symbol = symbol(func_name.clone());
        self.add_asm_symbolic(directive_globl(func_symbol.clone()));
        self.add_asm_symbolic(format!("{}:", func_symbol.clone()));
        
1014
        ValueLocation::Relocatable(RegGroup::GPR, func_name)
1015 1016
    }
    
1017
    fn finish_code(&mut self, func_name: MuName) -> (Box<MachineCode + Sync + Send>, ValueLocation) {
1018 1019
        let func_end = {
            let mut symbol = func_name.clone();
qinsoon's avatar
qinsoon committed
1020 1021 1022
            symbol.push_str("_end");
            symbol
        };
1023 1024
        self.add_asm_symbolic(directive_globl(symbol(func_end.clone())));
        self.add_asm_symbolic(format!("{}:", symbol(func_end.clone())));
qinsoon's avatar
qinsoon committed
1025
        
1026
        self.control_flow_analysis();
qinsoon's avatar
qinsoon committed
1027 1028 1029
        
        (
            self.cur.take().unwrap(),
1030
            ValueLocation::Relocatable(RegGroup::GPR, func_end)
qinsoon's avatar
qinsoon committed
1031
        )
1032
    }
1033 1034 1035 1036 1037

    fn start_code_sequence(&mut self) {
        self.cur = Some(Box::new(ASMCode {
            name: "snippet".to_string(),
            code: vec![],
qinsoon's avatar
qinsoon committed
1038
            blocks: hashmap!{}
1039 1040 1041 1042
        }));
    }

    fn finish_code_sequence(&mut self) -> Box<MachineCode + Sync + Send> {
1043
        self.finish_code_sequence_asm()
1044 1045
    }

1046 1047 1048 1049 1050 1051 1052
    fn print_cur_code(&self) {
        println!("");
        
        if self.cur.is_some() {
            let code = self.cur.as_ref().unwrap();
            
            println!("code for {}: ", code.name);
1053 1054 1055 1056
            let n_insts = code.code.len();
            for i in 0..n_insts {
                let ref line = code.code[i];
                println!("#{}\t{}", i, line.code);
1057 1058 1059 1060 1061 1062 1063 1064
            }
        } else {
            println!("no current code");
        }
        
        println!("");
    }
    
qinsoon's avatar
qinsoon committed
1065
    fn start_block(&mut self, block_name: MuName) {
1066 1067
        let label = format!("{}:", self.asm_block_label(block_name.clone()));        
        self.add_asm_block_label(label, block_name.clone());
qinsoon's avatar
qinsoon committed
1068 1069

        self.cur_mut().blocks.insert(block_name.clone(), ASMBlock::new());
1070
        let start = self.line();
qinsoon's avatar
qinsoon committed
1071
        self.cur_mut().blocks.get_mut(&block_name).unwrap().start_inst = start;
1072 1073
    }
    
qinsoon's avatar
qinsoon committed
1074
    fn start_exception_block(&mut self, block_name: MuName) -> ValueLocation {
1075 1076 1077
        let block = self.asm_block_label(block_name.clone());
        self.add_asm_symbolic(directive_globl(symbol(block.clone())));
        self.add_asm_symbolic(format!("{}:", symbol(block.clone())));
qinsoon's avatar
qinsoon committed
1078 1079 1080
        
        self.start_block(block_name);
        
1081
        ValueLocation::Relocatable(RegGroup::GPR, block)
qinsoon's avatar
qinsoon committed
1082 1083
    }
    
qinsoon's avatar
qinsoon committed
1084
    fn end_block(&mut self, block_name: MuName) {
qinsoon's avatar
qinsoon committed
1085 1086 1087 1088 1089 1090 1091
        let line = self.line();
        match self.cur_mut().blocks.get_mut(&block_name) {
            Some(ref mut block) => {
                block.end_inst = line;
            }
            None => panic!("trying to end block {} which hasnt been started", block_name)
        }
1092 1093
    }
    
qinsoon's avatar
qinsoon committed
1094
    fn set_block_livein(&mut self, block_name: MuName, live_in: &Vec<P<Value>>) {
1095
        let cur = self.cur_mut();
qinsoon's avatar
qinsoon committed
1096 1097 1098 1099 1100 1101 1102 1103 1104

        match cur.blocks.get_mut(&block_name) {
            Some(ref mut block) => {
                if block.livein.is_empty() {
                    let mut live_in = {
                        let mut ret = vec![];
                        for p in live_in {
                            match p.extract_ssa_id() {
                                Some(id) => ret.push(id),
qinsoon's avatar
qinsoon committed
1105 1106
                                // this should not happen
                                None => error!("{} as live-in of block {} is not SSA", p, block_name)
qinsoon's avatar
qinsoon committed
1107 1108 1109 1110 1111 1112 1113 1114
                            }
                        }
                        ret
                    };
                    block.livein.append(&mut live_in);
                } else {
                    panic!("seems we are inserting livein to block {} twice", block_name);
                }
1115
            }
qinsoon's avatar
qinsoon committed
1116
            None => panic!("haven't created ASMBlock for {}", block_name)
1117 1118 1119
        }
    }
    
qinsoon's avatar
qinsoon committed
1120
    fn set_block_liveout(&mut self, block_name: MuName, live_out: &Vec<P<Value>>) {
1121
        let cur = self.cur_mut();
qinsoon's avatar
qinsoon committed
1122 1123 1124 1125 1126 1127 1128 1129 1130

        match cur.blocks.get_mut(&block_name) {
            Some(ref mut block) => {
                if block.liveout.is_empty() {
                    let mut live_out = {
                        let mut ret = vec![];
                        for p in live_out {
                            match p.extract_ssa_id() {
                                Some(id) => ret.push(id),
qinsoon's avatar
qinsoon committed
1131 1132 1133
                                // the liveout are actually args out of this block
                                // (they can be constants)
                                None => trace!("{} as live-out of block {} is not SSA", p, block_name)
qinsoon's avatar
qinsoon committed
1134 1135 1136 1137 1138 1139 1140 1141
                            }
                        }
                        ret
                    };
                    block.liveout.append(&mut live_out);
                } else {
                    panic!("seems we are inserting liveout to block {} twice", block_name);
                }
1142
            }
qinsoon's avatar
qinsoon committed
1143 1144
            None => panic!("haven't created ASMBlock for {}", block_name)
        }
1145 1146
    }
    
1147 1148 1149 1150 1151 1152 1153
    fn emit_nop(&mut self, bytes: usize) {
        trace!("emit: nop ({} bytes)", bytes);
        
        let asm = String::from("nop");
        
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1154 1155
            hashmap!{},
            hashmap!{},
1156 1157 1158 1159
            false
        );
    }
    
1160
    fn emit_cmp_r64_r64(&mut self, op1: &P<Value>, op2: &P<Value>) {
qinsoon's avatar
qinsoon committed
1161
        trace!("emit: cmp {} {}", op1, op2);
1162
        
1163 1164
        let (reg1, id1, loc1) = self.prepare_reg(op1, 4 + 1);
        let (reg2, id2, loc2) = self.prepare_reg(op2, 4 + 1 + reg1.len() + 1);
1165
        
qinsoon's avatar
qinsoon committed
1166
        let asm = format!("cmpq {},{}", reg1, reg2);
1167
        
1168 1169
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1170 1171 1172 1173 1174
            hashmap!{},
            hashmap!{
                id1 => vec![loc1],
                id2 => vec![loc2]
            },
qinsoon's avatar
qinsoon committed
1175
            false
1176
        );
1177 1178
    }
    
1179
    fn emit_cmp_r64_imm32(&mut self, op1: &P<Value>, op2: i32) {
qinsoon's avatar
qinsoon committed
1180
        trace!("emit: cmp {} {}", op1, op2);
1181
        
1182
        let (reg1, id1, loc1) = self.prepare_reg(op1, 4 + 1 + 1 + op2.to_string().len() + 1);
1183
        
qinsoon's avatar
qinsoon committed
1184
        let asm = format!("cmpq ${},{}", op2, reg1);
1185
        
1186 1187
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1188 1189 1190 1191
            hashmap!{},
            hashmap!{
                id1 => vec![loc1]
            },
qinsoon's avatar
qinsoon committed
1192
            false
1193
        )
1194 1195 1196
    }
    
    fn emit_cmp_r64_mem64(&mut self, op1: &P<Value>, op2: &P<Value>) {
qinsoon's avatar
qinsoon committed
1197
        trace!("emit: cmp {} {}", op1, op2);
1198 1199
        
        let (reg, id1, loc1) = self.prepare_reg(op1, 4 + 1);
qinsoon's avatar
qinsoon committed
1200
        let (mem, mut uses) = self.prepare_mem(op2, 4 + 1 + reg.len() + 1);
1201 1202 1203 1204
        
        let asm = format!("cmpq {},{}", reg, mem);
        
        // merge use vec
qinsoon's avatar
qinsoon committed
1205 1206 1207 1208 1209
        if uses.contains_key(&id1) {
            uses.get_mut(&id1).unwrap().push(loc1);
        } else {
            uses.insert(id1, vec![loc1]);
        }
1210 1211 1212
        
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1213 1214
            hashmap!{},
            uses,
1215 1216
            true
        )
1217 1218
    }
    
1219
    fn emit_mov_r64_imm32(&mut self, dest: &P<Value>, src: i32) {
qinsoon's avatar
qinsoon committed
1220
        trace!("emit: mov {} -> {}", src, dest);
1221
        
1222
        let (reg1, id1, loc1) = self.prepare_reg(dest, 4 + 1 + 1 + src.to_string().len() + 1);
1223
        
qinsoon's avatar
qinsoon committed
1224
        let asm = format!("movq ${},{}", src, reg1);
1225
        
1226 1227
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1228 1229 1230 1231
            hashmap!{
                id1 => vec![loc1]
            },
            hashmap!{},
qinsoon's avatar
qinsoon committed
1232
            false
1233
        )
1234 1235
    }
    
1236
    // load
1237
    fn emit_mov_r64_mem64(&mut self, dest: &P<Value>, src: &P<Value>) {
qinsoon's avatar
qinsoon committed
1238
        trace!("emit: mov {} -> {}", src, dest);
1239
        
qinsoon's avatar
qinsoon committed
1240
        let (mem, uses) = self.prepare_mem(src, 4 + 1);
1241 1242 1243 1244 1245 1246
        let (reg, id2, loc2) = self.prepare_reg(dest, 4 + 1 + mem.len() + 1);
        
        let asm = format!("movq {},{}", mem, reg);
        
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1247 1248 1249 1250
            hashmap!{
                id2 => vec![loc2]
            },
            uses,
qinsoon's avatar
qinsoon committed
1251
            true
1252 1253 1254
        )
    }
    
1255
    // store
1256 1257 1258 1259
    fn emit_mov_mem64_r64(&mut self, dest: &P<Value>, src: &P<Value>) {
        trace!("emit: mov {} -> {}", src, dest);
        
        let (reg, id1, loc1) = self.prepare_reg(src, 4 + 1);
qinsoon's avatar
qinsoon committed
1260
        let (mem, mut uses) = self.prepare_mem(dest, 4 + 1 + reg.len() + 1);
1261 1262
        
        // the register we used for the memory location is counted as 'use'
1263
        // use the vec from mem as 'use' (push use reg from src to it)
qinsoon's avatar
qinsoon committed
1264 1265 1266 1267 1268
        if uses.contains_key(&id1) {
            uses.get_mut(&id1).unwrap().push(loc1);
        } else {
            uses.insert(id1, vec![loc1]);
        }
1269 1270 1271 1272 1273
        
        let asm = format!("movq {},{}", reg, mem);
        
        self.add_asm_inst(
            asm,
qinsoon's avatar
qinsoon committed
1274 1275
            hashmap!{},
            uses,
qinsoon's avatar
qinsoon committed
1276
            true
1277 1278 1279
        )
    }
    
qinsoon's avatar