c4f147f5db
It'll now recurse into the data item you give it, which means it works on types like Actual where the Meta isn't an immediate child.
79 lines
2.4 KiB
Haskell
79 lines
2.4 KiB
Haskell
{-
|
|
Tock: a compiler for parallel languages
|
|
Copyright (C) 2007, 2008 University of Kent
|
|
|
|
This program is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by the
|
|
Free Software Foundation, either version 2 of the License, or (at your
|
|
option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License along
|
|
with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
-}
|
|
|
|
-- | Metadata -- i.e. source position.
|
|
module Metadata where
|
|
|
|
{-! global : Haskell2Xml !-}
|
|
|
|
import Data.Generics
|
|
import Numeric
|
|
import Text.Printf
|
|
import Text.Regex
|
|
|
|
data Meta = Meta {
|
|
metaFile :: Maybe String,
|
|
metaLine :: Int,
|
|
metaColumn :: Int
|
|
}
|
|
deriving (Typeable, Data, Ord, Eq)
|
|
|
|
emptyMeta :: Meta
|
|
emptyMeta = Meta {
|
|
metaFile = Nothing,
|
|
metaLine = 0,
|
|
metaColumn = 0
|
|
}
|
|
|
|
instance Show Meta where
|
|
show m =
|
|
case metaFile m of
|
|
Just s -> s ++ ":" ++ show (metaLine m) ++ ":" ++ show (metaColumn m)
|
|
Nothing -> "no source position"
|
|
|
|
-- | Encode a Meta as the prefix of a string.
|
|
packMeta :: Meta -> String -> String
|
|
packMeta m s
|
|
= case metaFile m of
|
|
Nothing -> s
|
|
Just fn -> printf "//pos:%d:%d:%s//%s"
|
|
(metaLine m) (metaColumn m) (unslash fn) s
|
|
where
|
|
-- | Remove doubled slashes from a string, so we can unambiguously encode it.
|
|
unslash :: String -> String
|
|
unslash s = subRegex (mkRegex "//+") s "/"
|
|
|
|
-- | Extract a Meta (encoded by packMeta) from a String.
|
|
unpackMeta :: String -> (Maybe Meta, String)
|
|
unpackMeta s
|
|
= case matchRegex metaRE s of
|
|
Just [before, line, col, file, after] ->
|
|
(Just $ Meta (Just file) (getInt line) (getInt col), before ++ after)
|
|
Nothing -> (Nothing, s)
|
|
where
|
|
metaRE = mkRegex "^(.*)//pos:([0-9]*):([0-9]*):(.*)//(.*)$"
|
|
getInt s = case readDec s of [(v, "")] -> v
|
|
|
|
-- | Find the first Meta value in some part of the AST.
|
|
-- Return 'emptyMeta' if it couldn't find one.
|
|
findMeta :: Data t => t -> Meta
|
|
findMeta e
|
|
= case listify (const True :: Meta -> Bool) e of
|
|
(m:_) -> m
|
|
[] -> emptyMeta
|