File: rbit.ml

package info (click to toggle)
herdtools7 7.58-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 19,732 kB
  • sloc: ml: 128,583; ansic: 3,827; makefile: 670; python: 407; sh: 212; awk: 14
file content (69 lines) | stat: -rw-r--r-- 2,554 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
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
66
67
68
69
(****************************************************************************)
(*                           the diy toolsuite                              *)
(*                                                                          *)
(* Jade Alglave, University College London, UK.                             *)
(* Luc Maranget, INRIA Paris-Rocquencourt, France.                          *)
(*                                                                          *)
(* Copyright 2023-present Institut National de Recherche en Informatique et *)
(* en Automatique, ARM Ltd and the authors. All rights reserved.            *)
(*                                                                          *)
(* This software is governed by the CeCILL-B license under French law and   *)
(* abiding by the rules of distribution of free software. You can use,      *)
(* modify and/ or redistribute the software under the terms of the CeCILL-B *)
(* license as circulated by CEA, CNRS and INRIA at the following URL        *)
(* "http://www.cecill.info". We also give a copy in LICENSE.txt.            *)
(****************************************************************************)

(** Reverse bits and bytes *)

module type I = sig
    type t
    val zero : t
    val one : t
    val pp : bool -> t -> string
    val shift_left : t  -> int -> t
    val shift_right_logical : t -> int -> t
    val bit_at : int -> t -> t
    val logor : t -> t -> t
    val mask : MachSize.sz -> t -> t
end

module Make(I:I) :
sig
  val rbit : MachSize.sz -> I.t -> I.t
  val revbytes : MachSize.sz -> MachSize.sz -> I.t -> I.t
end =
  struct

    let rec rev_rec extract sz dst k src =
      if k <= 0 then dst
      else
        let b = extract src in
        let dst = I.shift_left dst sz in
        let dst = I.logor dst b in
        let src = I.shift_right_logical src sz in
        rev_rec extract sz dst (k-sz) src

    let rbit sz src = rev_rec (I.bit_at 0) 1 I.zero (MachSize.nbits sz)  src

    let rbytes sz src = rev_rec (I.mask MachSize.Byte)  8 I.zero sz src

    let revbytes csz sz x =
      let mask x = I.mask csz x in
      let csz = MachSize.nbits csz
      and sz = MachSize.nbits sz in
      assert (csz <= sz) ;
      let rec loop acc c =
        if c < 0 then acc
        else
          let elem =
            I.shift_right_logical x (c*csz)
            |> mask
            |> rbytes csz in
          let acc =
            I.shift_left elem (c*csz)
            |> I.logor acc in
          loop acc (c-1) in
      loop I.zero (sz/csz-1)

  end