Tissevert/pandoc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Biblio: Use a Map for the lookup table.

Browse source
This commit is contained in:
John MacFarlane 2010-11-11 22:35:04 -08:00
parent 1bfd8110af
commit 5c6dc5767d
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/Text/Pandoc/Biblio.hs
View file

@ -32,6 +32,7 @@ module Text.Pandoc.Biblio ( processBiblio ) where
import Control.Monad ( when ) import Control.Monad ( when )
import Data.List import Data.List
import Data.Unique import Data.Unique
import qualified Data.Map as M
import Text.CSL hiding ( Cite(..), Citation(..) ) import Text.CSL hiding ( Cite(..), Citation(..) )
import qualified Text.CSL as CSL ( Cite(..) ) import qualified Text.CSL as CSL ( Cite(..) )
import Text.Pandoc.Definition import Text.Pandoc.Definition
@ -52,19 +53,19 @@ processBiblio cf r p
needNt = cits \\ concat ncits needNt = cits \\ concat ncits
in (,) needNt $ getNoteCitations needNt p' in (,) needNt $ getNoteCitations needNt p'
result = citeproc csl r (setNearNote csl $ map (map toCslCite) grps) result = citeproc csl r (setNearNote csl $ map (map toCslCite) grps)
cits_map = zip grps (citations result) cits_map = M.fromList $ zip grps (citations result)
biblioList = map (renderPandoc' csl) (bibliography result) biblioList = map (renderPandoc' csl) (bibliography result)
Pandoc m b = processWith (procInlines $ processCite csl cits_map) p' Pandoc m b = processWith (procInlines $ processCite csl cits_map) p'
return . generateNotes nts . Pandoc m $ b ++ biblioList return . generateNotes nts . Pandoc m $ b ++ biblioList
-- | Substitute 'Cite' elements with formatted citations. -- | Substitute 'Cite' elements with formatted citations.
processCite :: Style -> [([Citation],[FormattedOutput])] -> [Inline] -> [Inline] processCite :: Style -> M.Map [Citation] [FormattedOutput] -> [Inline] -> [Inline]
processCite _ _ [] = [] processCite _ _ [] = []
processCite s cs (i:is) processCite s cs (i:is)
| Cite t _ <- i = process t ++ processCite s cs is | Cite t _ <- i = process t ++ processCite s cs is
| otherwise = i : processCite s cs is | otherwise = i : processCite s cs is
where where
process t = case lookup t cs of process t = case M.lookup t cs of
Just x -> if isTextualCitation t && x /= [] Just x -> if isTextualCitation t && x /= []
then renderPandoc s [head x] ++ [Space] ++ then renderPandoc s [head x] ++ [Space] ++
[Cite t $ renderPandoc s $ tail x] [Cite t $ renderPandoc s $ tail x]