/compiler/nativeGen/X86/RegInfo.hs

https://github.com/ezyang/ghc · Haskell · 69 lines · 55 code · 13 blank · 1 comment · 1 complexity · 64e579960bf762536662e5bb9137ee9c MD5 · raw file 

    

  1. {-# LANGUAGE CPP #-}
    2. 

  2. module X86.RegInfo (
    3. 

  3.         mkVirtualReg,
    4. 

  4.         regDotColor
    5. 

  5. )
    6. 

  6. 

  7. where
    8. 

  8. 

  9. #include "nativeGen/NCG.h"
    10. 

  10. #include "HsVersions.h"
    11. 

  11. 

  12. import GhcPrelude
    13. 

  13. 

  14. import Format
    15. 

  15. import Reg
    16. 

  16. 

  17. import Outputable
    18. 

  18. import Platform
    19. 

  19. import Unique
    20. 

  20. 

  21. import UniqFM
    22. 

  22. import X86.Regs
    23. 

  23. 

  24. 

  25. mkVirtualReg :: Unique -> Format -> VirtualReg
    26. 

  26. mkVirtualReg u format
    27. 

  27.    = case format of
    28. 

  28.         FF32    -> VirtualRegSSE u
    29. 

  29.         FF64    -> VirtualRegSSE u
    30. 

  30.         FF80    -> VirtualRegD   u
    31. 

  31.         _other  -> VirtualRegI   u
    32. 

  32. 

  33. regDotColor :: Platform -> RealReg -> SDoc
    34. 

  34. regDotColor platform reg
    35. 

  35.  = let Just str = lookupUFM (regColors platform) reg
    36. 

  36.    in text str
    37. 

  37. 

  38. regColors :: Platform -> UniqFM [Char]
    39. 

  39. regColors platform = listToUFM (normalRegColors platform ++ fpRegColors)
    40. 

  40. 

  41. normalRegColors :: Platform -> [(Reg,String)]
    42. 

  42. normalRegColors platform
    43. 

  43.  | target32Bit platform = [ (eax, "#00ff00")
    44. 

  44.                           , (ebx, "#0000ff")
    45. 

  45.                           , (ecx, "#00ffff")
    46. 

  46.                           , (edx, "#0080ff") ]
    47. 

  47.  | otherwise            = [ (rax, "#00ff00"), (eax, "#00ff00")
    48. 

  48.                           , (rbx, "#0000ff"), (ebx, "#0000ff")
    49. 

  49.                           , (rcx, "#00ffff"), (ecx, "#00ffff")
    50. 

  50.                           , (rdx, "#0080ff"), (edx, "#00ffff")
    51. 

  51.                           , (r8,  "#00ff80")
    52. 

  52.                           , (r9,  "#008080")
    53. 

  53.                           , (r10, "#0040ff")
    54. 

  54.                           , (r11, "#00ff40")
    55. 

  55.                           , (r12, "#008040")
    56. 

  56.                           , (r13, "#004080")
    57. 

  57.                           , (r14, "#004040")
    58. 

  58.                           , (r15, "#002080") ]
    59. 

  59. 

  60. fpRegColors :: [(Reg,String)]
    61. 

  61. fpRegColors =
    62. 

  62.         [ (fake0, "#ff00ff")
    63. 

  63.         , (fake1, "#ff00aa")
    64. 

  64.         , (fake2, "#aa00ff")
    65. 

  65.         , (fake3, "#aa00aa")
    66. 

  66.         , (fake4, "#ff0055")
    67. 

  67.         , (fake5, "#5500ff") ]
    68. 

  68. 

  69.         ++ zip (map regSingle [24..39]) (repeat "red")
    70.