<?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd"> <html> <head> <!-- Generated by HsColour, http://www.cs.york.ac.uk/fp/darcs/hscolour/ --> <title>Hledger/Data/AccountName.hs</title> <link type='text/css' rel='stylesheet' href='hscolour.css' /> </head> <body> <pre><a name="line-1"></a><span class='hs-comment'>{-# LANGUAGE NoMonomorphismRestriction#-}</span> <a name="line-2"></a><span class='hs-comment'>{-| <a name="line-3"></a> <a name="line-4"></a>'AccountName's are strings like @assets:cash:petty@, with multiple <a name="line-5"></a>components separated by ':'. From a set of these we derive the account <a name="line-6"></a>hierarchy. <a name="line-7"></a> <a name="line-8"></a>-}</span> <a name="line-9"></a> <a name="line-10"></a><span class='hs-keyword'>module</span> <span class='hs-conid'>Hledger</span><span class='hs-varop'>.</span><span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>AccountName</span> <a name="line-11"></a><span class='hs-keyword'>where</span> <a name="line-12"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Hledger</span><span class='hs-varop'>.</span><span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Utils</span> <a name="line-13"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Hledger</span><span class='hs-varop'>.</span><span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Types</span> <a name="line-14"></a><span class='hs-keyword'>import</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Map</span> <span class='hs-layout'>(</span><span class='hs-conid'>Map</span><span class='hs-layout'>)</span> <a name="line-15"></a><span class='hs-keyword'>import</span> <span class='hs-keyword'>qualified</span> <span class='hs-conid'>Data</span><span class='hs-varop'>.</span><span class='hs-conid'>Map</span> <span class='hs-keyword'>as</span> <span class='hs-conid'>M</span> <a name="line-16"></a> <a name="line-17"></a> <a name="line-18"></a> <a name="line-19"></a><a name="acctsepchar"></a><span class='hs-comment'>-- change to use a different separator for nested accounts</span> <a name="line-20"></a><span class='hs-definition'>acctsepchar</span> <span class='hs-keyglyph'>=</span> <span class='hs-chr'>':'</span> <a name="line-21"></a> <a name="line-22"></a><a name="accountNameComponents"></a><span class='hs-definition'>accountNameComponents</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-23"></a><span class='hs-definition'>accountNameComponents</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>splitAtElement</span> <span class='hs-varid'>acctsepchar</span> <a name="line-24"></a> <a name="line-25"></a><a name="accountNameFromComponents"></a><span class='hs-definition'>accountNameFromComponents</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <a name="line-26"></a><span class='hs-definition'>accountNameFromComponents</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>concat</span> <span class='hs-varop'>.</span> <span class='hs-varid'>intersperse</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>acctsepchar</span><span class='hs-keyglyph'>]</span> <a name="line-27"></a> <a name="line-28"></a><a name="accountLeafName"></a><span class='hs-definition'>accountLeafName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>String</span> <a name="line-29"></a><span class='hs-definition'>accountLeafName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>last</span> <span class='hs-varop'>.</span> <span class='hs-varid'>accountNameComponents</span> <a name="line-30"></a> <a name="line-31"></a><a name="accountNameLevel"></a><span class='hs-definition'>accountNameLevel</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Int</span> <a name="line-32"></a><span class='hs-definition'>accountNameLevel</span> <span class='hs-str'>""</span> <span class='hs-keyglyph'>=</span> <span class='hs-num'>0</span> <a name="line-33"></a><span class='hs-definition'>accountNameLevel</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>length</span> <span class='hs-layout'>(</span><span class='hs-varid'>filter</span> <span class='hs-layout'>(</span><span class='hs-varop'>==</span><span class='hs-varid'>acctsepchar</span><span class='hs-layout'>)</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-varop'>+</span> <span class='hs-num'>1</span> <a name="line-34"></a> <a name="line-35"></a><a name="accountNameDrop"></a><span class='hs-definition'>accountNameDrop</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <a name="line-36"></a><span class='hs-definition'>accountNameDrop</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>accountNameFromComponents</span> <span class='hs-varop'>.</span> <span class='hs-varid'>drop</span> <span class='hs-varid'>n</span> <span class='hs-varop'>.</span> <span class='hs-varid'>accountNameComponents</span> <a name="line-37"></a> <a name="line-38"></a><a name="expandAccountNames"></a><span class='hs-comment'>-- | ["a:b:c","d:e"] -> ["a","a:b","a:b:c","d","d:e"]</span> <a name="line-39"></a><span class='hs-definition'>expandAccountNames</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <a name="line-40"></a><span class='hs-definition'>expandAccountNames</span> <span class='hs-keyword'>as</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>nub</span> <span class='hs-varop'>$</span> <span class='hs-varid'>concatMap</span> <span class='hs-varid'>expand</span> <span class='hs-keyword'>as</span> <a name="line-41"></a> <span class='hs-keyword'>where</span> <span class='hs-varid'>expand</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>map</span> <span class='hs-varid'>accountNameFromComponents</span> <span class='hs-varop'>.</span> <span class='hs-varid'>tail</span> <span class='hs-varop'>.</span> <span class='hs-varid'>inits</span> <span class='hs-varop'>.</span> <span class='hs-varid'>accountNameComponents</span> <a name="line-42"></a> <a name="line-43"></a><a name="topAccountNames"></a><span class='hs-comment'>-- | ["a:b:c","d:e"] -> ["a","d"]</span> <a name="line-44"></a><span class='hs-definition'>topAccountNames</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <a name="line-45"></a><span class='hs-definition'>topAccountNames</span> <span class='hs-keyword'>as</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>a</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>expandAccountNames</span> <span class='hs-keyword'>as</span><span class='hs-layout'>,</span> <span class='hs-varid'>accountNameLevel</span> <span class='hs-varid'>a</span> <span class='hs-varop'>==</span> <span class='hs-num'>1</span><span class='hs-keyglyph'>]</span> <a name="line-46"></a> <a name="line-47"></a><a name="parentAccountName"></a><span class='hs-definition'>parentAccountName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <a name="line-48"></a><span class='hs-definition'>parentAccountName</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>accountNameFromComponents</span> <span class='hs-varop'>.</span> <span class='hs-varid'>init</span> <span class='hs-varop'>.</span> <span class='hs-varid'>accountNameComponents</span> <a name="line-49"></a> <a name="line-50"></a><a name="parentAccountNames"></a><span class='hs-definition'>parentAccountNames</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <a name="line-51"></a><span class='hs-definition'>parentAccountNames</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>parentAccountNames'</span> <span class='hs-varop'>$</span> <span class='hs-varid'>parentAccountName</span> <span class='hs-varid'>a</span> <a name="line-52"></a> <span class='hs-keyword'>where</span> <a name="line-53"></a> <span class='hs-varid'>parentAccountNames'</span> <span class='hs-str'>""</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>[]</span> <a name="line-54"></a> <span class='hs-varid'>parentAccountNames'</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>a</span> <span class='hs-conop'>:</span> <span class='hs-varid'>parentAccountNames'</span> <span class='hs-layout'>(</span><span class='hs-varid'>parentAccountName</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <a name="line-55"></a> <a name="line-56"></a><a name="isAccountNamePrefixOf"></a><span class='hs-definition'>isAccountNamePrefixOf</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Bool</span> <a name="line-57"></a><span class='hs-definition'>isAccountNamePrefixOf</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>isPrefixOf</span> <span class='hs-varop'>.</span> <span class='hs-layout'>(</span><span class='hs-varop'>++</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>acctsepchar</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <a name="line-58"></a> <a name="line-59"></a><a name="isSubAccountNameOf"></a><span class='hs-definition'>isSubAccountNameOf</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Bool</span> <a name="line-60"></a><a name="isSubAccountNameOf"></a><span class='hs-definition'>s</span> <span class='hs-varop'>`isSubAccountNameOf`</span> <span class='hs-varid'>p</span> <span class='hs-keyglyph'>=</span> <a name="line-61"></a> <span class='hs-layout'>(</span><span class='hs-varid'>p</span> <span class='hs-varop'>`isAccountNamePrefixOf`</span> <span class='hs-varid'>s</span><span class='hs-layout'>)</span> <span class='hs-varop'>&&</span> <span class='hs-layout'>(</span><span class='hs-varid'>accountNameLevel</span> <span class='hs-varid'>s</span> <span class='hs-varop'>==</span> <span class='hs-layout'>(</span><span class='hs-varid'>accountNameLevel</span> <span class='hs-varid'>p</span> <span class='hs-varop'>+</span> <span class='hs-num'>1</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-62"></a> <a name="line-63"></a><a name="subAccountNamesFrom"></a><span class='hs-comment'>-- | From a list of account names, select those which are direct</span> <a name="line-64"></a><span class='hs-comment'>-- subaccounts of the given account name.</span> <a name="line-65"></a><span class='hs-definition'>subAccountNamesFrom</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <a name="line-66"></a><span class='hs-definition'>subAccountNamesFrom</span> <span class='hs-varid'>accts</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>filter</span> <span class='hs-layout'>(</span><span class='hs-varop'>`isSubAccountNameOf`</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-varid'>accts</span> <a name="line-67"></a> <a name="line-68"></a><a name="accountNameTreeFrom"></a><span class='hs-comment'>-- | Convert a list of account names to a tree.</span> <a name="line-69"></a><span class='hs-definition'>accountNameTreeFrom</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree</span> <span class='hs-conid'>AccountName</span> <a name="line-70"></a><span class='hs-definition'>accountNameTreeFrom</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>accountNameTreeFrom1</span> <a name="line-71"></a> <a name="line-72"></a><a name="accountNameTreeFrom1"></a><span class='hs-definition'>accountNameTreeFrom1</span> <span class='hs-varid'>accts</span> <span class='hs-keyglyph'>=</span> <a name="line-73"></a> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-layout'>(</span><span class='hs-varid'>accounttreesfrom</span> <span class='hs-layout'>(</span><span class='hs-varid'>topAccountNames</span> <span class='hs-varid'>accts</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-74"></a> <span class='hs-keyword'>where</span> <a name="line-75"></a> <span class='hs-varid'>accounttreesfrom</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Tree</span> <span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <a name="line-76"></a> <span class='hs-varid'>accounttreesfrom</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>[]</span> <a name="line-77"></a> <span class='hs-varid'>accounttreesfrom</span> <span class='hs-keyword'>as</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-varid'>a</span> <span class='hs-layout'>(</span><span class='hs-varid'>accounttreesfrom</span> <span class='hs-varop'>$</span> <span class='hs-varid'>subs</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-keyword'>as</span><span class='hs-keyglyph'>]</span> <a name="line-78"></a> <span class='hs-varid'>subs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>subAccountNamesFrom</span> <span class='hs-layout'>(</span><span class='hs-varid'>expandAccountNames</span> <span class='hs-varid'>accts</span><span class='hs-layout'>)</span> <a name="line-79"></a> <a name="line-80"></a><a name="nullaccountnametree"></a><span class='hs-definition'>nullaccountnametree</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-conid'>[]</span> <a name="line-81"></a> <a name="line-82"></a><a name="accountNameTreeFrom2"></a><span class='hs-definition'>accountNameTreeFrom2</span> <span class='hs-varid'>accts</span> <span class='hs-keyglyph'>=</span> <a name="line-83"></a> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-varop'>$</span> <span class='hs-varid'>unfoldForest</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span><span class='hs-layout'>,</span> <span class='hs-varid'>subs</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <span class='hs-varop'>$</span> <span class='hs-varid'>topAccountNames</span> <span class='hs-varid'>accts</span> <a name="line-84"></a> <span class='hs-keyword'>where</span> <a name="line-85"></a> <span class='hs-varid'>subs</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>subAccountNamesFrom</span> <span class='hs-varid'>allaccts</span> <a name="line-86"></a> <span class='hs-varid'>allaccts</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>expandAccountNames</span> <span class='hs-varid'>accts</span> <a name="line-87"></a> <span class='hs-comment'>-- subs' a = subsmap ! a</span> <a name="line-88"></a> <span class='hs-comment'>-- subsmap :: Map AccountName [AccountName]</span> <a name="line-89"></a> <span class='hs-comment'>-- subsmap = Data.Map.fromList [(a, subAccountNamesFrom allaccts a) | a <- allaccts]</span> <a name="line-90"></a> <a name="line-91"></a><a name="accountNameTreeFrom3"></a><span class='hs-definition'>accountNameTreeFrom3</span> <span class='hs-varid'>accts</span> <span class='hs-keyglyph'>=</span> <a name="line-92"></a> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-varop'>$</span> <span class='hs-varid'>forestfrom</span> <span class='hs-varid'>allaccts</span> <span class='hs-varop'>$</span> <span class='hs-varid'>topAccountNames</span> <span class='hs-varid'>accts</span> <a name="line-93"></a> <span class='hs-keyword'>where</span> <a name="line-94"></a> <span class='hs-comment'>-- drop accts from the list of potential subs as we add them to the tree</span> <a name="line-95"></a> <span class='hs-varid'>forestfrom</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Forest</span> <span class='hs-conid'>AccountName</span> <a name="line-96"></a> <span class='hs-varid'>forestfrom</span> <span class='hs-varid'>subaccts</span> <span class='hs-varid'>accts</span> <span class='hs-keyglyph'>=</span> <a name="line-97"></a> <span class='hs-keyglyph'>[</span><span class='hs-keyword'>let</span> <span class='hs-varid'>subaccts'</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>subaccts</span> <span class='hs-varop'>\\</span> <span class='hs-varid'>accts</span> <span class='hs-keyword'>in</span> <span class='hs-conid'>Node</span> <span class='hs-varid'>a</span> <span class='hs-varop'>$</span> <span class='hs-varid'>forestfrom</span> <span class='hs-varid'>subaccts'</span> <span class='hs-layout'>(</span><span class='hs-varid'>subAccountNamesFrom</span> <span class='hs-varid'>subaccts'</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'><-</span> <span class='hs-varid'>accts</span><span class='hs-keyglyph'>]</span> <a name="line-98"></a> <span class='hs-varid'>allaccts</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>expandAccountNames</span> <span class='hs-varid'>accts</span> <a name="line-99"></a> <a name="line-100"></a> <a name="line-101"></a><a name="Tree'"></a><span class='hs-comment'>-- a more efficient tree builder from Cale Gibbard</span> <a name="line-102"></a><a name="Tree'"></a><span class='hs-keyword'>newtype</span> <span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>T</span> <span class='hs-layout'>(</span><span class='hs-conid'>Map</span> <span class='hs-varid'>a</span> <span class='hs-layout'>(</span><span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-103"></a> <span class='hs-keyword'>deriving</span> <span class='hs-layout'>(</span><span class='hs-conid'>Show</span><span class='hs-layout'>,</span> <span class='hs-conid'>Eq</span><span class='hs-layout'>,</span> <span class='hs-conid'>Ord</span><span class='hs-layout'>)</span> <a name="line-104"></a> <a name="line-105"></a><a name="mergeTrees"></a><span class='hs-definition'>mergeTrees</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ord</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=></span> <span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span> <a name="line-106"></a><span class='hs-definition'>mergeTrees</span> <span class='hs-layout'>(</span><span class='hs-conid'>T</span> <span class='hs-varid'>m</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-conid'>T</span> <span class='hs-varid'>m'</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>T</span> <span class='hs-layout'>(</span><span class='hs-conid'>M</span><span class='hs-varop'>.</span><span class='hs-varid'>unionWith</span> <span class='hs-varid'>mergeTrees</span> <span class='hs-varid'>m</span> <span class='hs-varid'>m'</span><span class='hs-layout'>)</span> <a name="line-107"></a> <a name="line-108"></a><a name="emptyTree"></a><span class='hs-definition'>emptyTree</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>T</span> <span class='hs-conid'>M</span><span class='hs-varop'>.</span><span class='hs-varid'>empty</span> <a name="line-109"></a> <a name="line-110"></a><a name="pathtree"></a><span class='hs-definition'>pathtree</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span> <a name="line-111"></a><span class='hs-definition'>pathtree</span> <span class='hs-conid'>[]</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>T</span> <span class='hs-conid'>M</span><span class='hs-varop'>.</span><span class='hs-varid'>empty</span> <a name="line-112"></a><span class='hs-definition'>pathtree</span> <span class='hs-layout'>(</span><span class='hs-varid'>x</span><span class='hs-conop'>:</span><span class='hs-varid'>xs</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>T</span> <span class='hs-layout'>(</span><span class='hs-conid'>M</span><span class='hs-varop'>.</span><span class='hs-varid'>singleton</span> <span class='hs-varid'>x</span> <span class='hs-layout'>(</span><span class='hs-varid'>pathtree</span> <span class='hs-varid'>xs</span><span class='hs-layout'>)</span><span class='hs-layout'>)</span> <a name="line-113"></a> <a name="line-114"></a><a name="fromPaths"></a><span class='hs-definition'>fromPaths</span> <span class='hs-keyglyph'>::</span> <span class='hs-layout'>(</span><span class='hs-conid'>Ord</span> <span class='hs-varid'>a</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=></span> <span class='hs-keyglyph'>[</span><span class='hs-keyglyph'>[</span><span class='hs-varid'>a</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree'</span> <span class='hs-varid'>a</span> <a name="line-115"></a><span class='hs-definition'>fromPaths</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>foldl'</span> <span class='hs-varid'>mergeTrees</span> <span class='hs-varid'>emptyTree</span> <span class='hs-varop'>.</span> <span class='hs-varid'>map</span> <span class='hs-varid'>pathtree</span> <a name="line-116"></a> <a name="line-117"></a><span class='hs-comment'>-- the above, but trying to build Tree directly</span> <a name="line-118"></a> <a name="line-119"></a><span class='hs-comment'>-- mergeTrees' :: (Ord a) => Tree a -> Tree a -> Tree a</span> <a name="line-120"></a><span class='hs-comment'>-- mergeTrees' (Node m ms) (Node m' ms') = Node undefined (ms `union` ms')</span> <a name="line-121"></a> <a name="line-122"></a><span class='hs-comment'>-- emptyTree' = Node "top" []</span> <a name="line-123"></a> <a name="line-124"></a><span class='hs-comment'>-- pathtree' :: [a] -> Tree a</span> <a name="line-125"></a><span class='hs-comment'>-- pathtree' [] = Node undefined []</span> <a name="line-126"></a><span class='hs-comment'>-- pathtree' (x:xs) = Node x [pathtree' xs]</span> <a name="line-127"></a> <a name="line-128"></a><span class='hs-comment'>-- fromPaths' :: (Ord a) => [[a]] -> Tree a</span> <a name="line-129"></a><span class='hs-comment'>-- fromPaths' = foldl' mergeTrees' emptyTree' . map pathtree'</span> <a name="line-130"></a> <a name="line-131"></a> <a name="line-132"></a><span class='hs-comment'>-- converttree :: [AccountName] -> Tree' AccountName -> [Tree AccountName]</span> <a name="line-133"></a><span class='hs-comment'>-- converttree parents (T m) = [Node (accountNameFromComponents $ parents ++ [a]) (converttree (parents++[a]) b) | (a,b) <- M.toList m]</span> <a name="line-134"></a> <a name="line-135"></a><span class='hs-comment'>-- accountNameTreeFrom4 :: [AccountName] -> Tree AccountName</span> <a name="line-136"></a><span class='hs-comment'>-- accountNameTreeFrom4 accts = Node "top" (converttree [] $ fromPaths $ map accountNameComponents accts)</span> <a name="line-137"></a> <a name="line-138"></a><a name="converttree"></a><span class='hs-definition'>converttree</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Tree'</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Tree</span> <span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <a name="line-139"></a><span class='hs-definition'>converttree</span> <span class='hs-layout'>(</span><span class='hs-conid'>T</span> <span class='hs-varid'>m</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-varid'>a</span> <span class='hs-layout'>(</span><span class='hs-varid'>converttree</span> <span class='hs-varid'>b</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>|</span> <span class='hs-layout'>(</span><span class='hs-varid'>a</span><span class='hs-layout'>,</span><span class='hs-varid'>b</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'><-</span> <span class='hs-conid'>M</span><span class='hs-varop'>.</span><span class='hs-varid'>toList</span> <span class='hs-varid'>m</span><span class='hs-keyglyph'>]</span> <a name="line-140"></a> <a name="line-141"></a><a name="expandTreeNames"></a><span class='hs-definition'>expandTreeNames</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Tree</span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree</span> <span class='hs-conid'>AccountName</span> <a name="line-142"></a><span class='hs-definition'>expandTreeNames</span> <span class='hs-layout'>(</span><span class='hs-conid'>Node</span> <span class='hs-varid'>x</span> <span class='hs-varid'>ts</span><span class='hs-layout'>)</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Node</span> <span class='hs-varid'>x</span> <span class='hs-layout'>(</span><span class='hs-varid'>map</span> <span class='hs-layout'>(</span><span class='hs-varid'>treemap</span> <span class='hs-layout'>(</span><span class='hs-keyglyph'>\</span><span class='hs-varid'>n</span> <span class='hs-keyglyph'>-></span> <span class='hs-varid'>accountNameFromComponents</span> <span class='hs-keyglyph'>[</span><span class='hs-varid'>x</span><span class='hs-layout'>,</span><span class='hs-varid'>n</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <span class='hs-varop'>.</span> <span class='hs-varid'>expandTreeNames</span><span class='hs-layout'>)</span> <span class='hs-varid'>ts</span><span class='hs-layout'>)</span> <a name="line-143"></a> <a name="line-144"></a><a name="accountNameTreeFrom4"></a><span class='hs-definition'>accountNameTreeFrom4</span> <span class='hs-keyglyph'>::</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>AccountName</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>Tree</span> <span class='hs-conid'>AccountName</span> <a name="line-145"></a><span class='hs-definition'>accountNameTreeFrom4</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-varop'>.</span> <span class='hs-varid'>map</span> <span class='hs-varid'>expandTreeNames</span> <span class='hs-varop'>.</span> <span class='hs-varid'>converttree</span> <span class='hs-varop'>.</span> <span class='hs-varid'>fromPaths</span> <span class='hs-varop'>.</span> <span class='hs-varid'>map</span> <span class='hs-varid'>accountNameComponents</span> <a name="line-146"></a> <a name="line-147"></a> <a name="line-148"></a><a name="elideAccountName"></a><span class='hs-comment'>-- | Elide an account name to fit in the specified width.</span> <a name="line-149"></a><span class='hs-comment'>-- From the ledger 2.6 news:</span> <a name="line-150"></a><span class='hs-comment'>-- </span> <a name="line-151"></a><span class='hs-comment'>-- @</span> <a name="line-152"></a><span class='hs-comment'>-- What Ledger now does is that if an account name is too long, it will</span> <a name="line-153"></a><span class='hs-comment'>-- start abbreviating the first parts of the account name down to two</span> <a name="line-154"></a><span class='hs-comment'>-- letters in length. If this results in a string that is still too</span> <a name="line-155"></a><span class='hs-comment'>-- long, the front will be elided -- not the end. For example:</span> <a name="line-156"></a><span class='hs-comment'>--</span> <a name="line-157"></a><span class='hs-comment'>-- Expenses:Cash ; OK, not too long</span> <a name="line-158"></a><span class='hs-comment'>-- Ex:Wednesday:Cash ; "Expenses" was abbreviated to fit</span> <a name="line-159"></a><span class='hs-comment'>-- Ex:We:Afternoon:Cash ; "Expenses" and "Wednesday" abbreviated</span> <a name="line-160"></a><span class='hs-comment'>-- ; Expenses:Wednesday:Afternoon:Lunch:Snack:Candy:Chocolate:Cash</span> <a name="line-161"></a><span class='hs-comment'>-- ..:Af:Lu:Sn:Ca:Ch:Cash ; Abbreviated and elided!</span> <a name="line-162"></a><span class='hs-comment'>-- @</span> <a name="line-163"></a><span class='hs-definition'>elideAccountName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <a name="line-164"></a><span class='hs-definition'>elideAccountName</span> <span class='hs-varid'>width</span> <span class='hs-varid'>s</span> <span class='hs-keyglyph'>=</span> <a name="line-165"></a> <span class='hs-varid'>elideLeft</span> <span class='hs-varid'>width</span> <span class='hs-varop'>$</span> <span class='hs-varid'>accountNameFromComponents</span> <span class='hs-varop'>$</span> <span class='hs-varid'>elideparts</span> <span class='hs-varid'>width</span> <span class='hs-conid'>[]</span> <span class='hs-varop'>$</span> <span class='hs-varid'>accountNameComponents</span> <span class='hs-varid'>s</span> <a name="line-166"></a> <span class='hs-keyword'>where</span> <a name="line-167"></a> <span class='hs-varid'>elideparts</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <span class='hs-keyglyph'>-></span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>String</span><span class='hs-keyglyph'>]</span> <a name="line-168"></a> <span class='hs-varid'>elideparts</span> <span class='hs-varid'>width</span> <span class='hs-varid'>done</span> <span class='hs-varid'>ss</span> <a name="line-169"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>length</span> <span class='hs-layout'>(</span><span class='hs-varid'>accountNameFromComponents</span> <span class='hs-varop'>$</span> <span class='hs-varid'>done</span><span class='hs-varop'>++</span><span class='hs-varid'>ss</span><span class='hs-layout'>)</span> <span class='hs-varop'><=</span> <span class='hs-varid'>width</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>done</span><span class='hs-varop'>++</span><span class='hs-varid'>ss</span> <a name="line-170"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>length</span> <span class='hs-varid'>ss</span> <span class='hs-varop'>></span> <span class='hs-num'>1</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>elideparts</span> <span class='hs-varid'>width</span> <span class='hs-layout'>(</span><span class='hs-varid'>done</span><span class='hs-varop'>++</span><span class='hs-keyglyph'>[</span><span class='hs-varid'>take</span> <span class='hs-num'>2</span> <span class='hs-varop'>$</span> <span class='hs-varid'>head</span> <span class='hs-varid'>ss</span><span class='hs-keyglyph'>]</span><span class='hs-layout'>)</span> <span class='hs-layout'>(</span><span class='hs-varid'>tail</span> <span class='hs-varid'>ss</span><span class='hs-layout'>)</span> <a name="line-171"></a> <span class='hs-keyglyph'>|</span> <span class='hs-varid'>otherwise</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>done</span><span class='hs-varop'>++</span><span class='hs-varid'>ss</span> <a name="line-172"></a> <a name="line-173"></a><a name="clipAccountName"></a><span class='hs-definition'>clipAccountName</span> <span class='hs-keyglyph'>::</span> <span class='hs-conid'>Int</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <span class='hs-keyglyph'>-></span> <span class='hs-conid'>AccountName</span> <a name="line-174"></a><span class='hs-definition'>clipAccountName</span> <span class='hs-varid'>n</span> <span class='hs-keyglyph'>=</span> <span class='hs-varid'>accountNameFromComponents</span> <span class='hs-varop'>.</span> <span class='hs-varid'>take</span> <span class='hs-varid'>n</span> <span class='hs-varop'>.</span> <span class='hs-varid'>accountNameComponents</span> <a name="line-175"></a> <a name="line-176"></a><a name="tests_Hledger_Data_AccountName"></a><span class='hs-definition'>tests_Hledger_Data_AccountName</span> <span class='hs-keyglyph'>=</span> <span class='hs-conid'>TestList</span> <a name="line-177"></a> <span class='hs-keyglyph'>[</span> <a name="line-178"></a> <span class='hs-str'>"accountNameTreeFrom"</span> <span class='hs-varop'>~:</span> <span class='hs-keyword'>do</span> <a name="line-179"></a> <span class='hs-varid'>accountNameTreeFrom</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"a"</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a"</span> <span class='hs-conid'>[]</span><span class='hs-keyglyph'>]</span> <a name="line-180"></a> <span class='hs-varid'>accountNameTreeFrom</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"a"</span><span class='hs-layout'>,</span><span class='hs-str'>"b"</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a"</span> <span class='hs-conid'>[]</span><span class='hs-layout'>,</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"b"</span> <span class='hs-conid'>[]</span><span class='hs-keyglyph'>]</span> <a name="line-181"></a> <span class='hs-varid'>accountNameTreeFrom</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"a"</span><span class='hs-layout'>,</span><span class='hs-str'>"a:b"</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a:b"</span> <span class='hs-conid'>[]</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span> <a name="line-182"></a> <span class='hs-varid'>accountNameTreeFrom</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"a:b:c"</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>Node</span> <span class='hs-str'>"top"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a:b"</span> <span class='hs-keyglyph'>[</span><span class='hs-conid'>Node</span> <span class='hs-str'>"a:b:c"</span> <span class='hs-conid'>[]</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span><span class='hs-keyglyph'>]</span> <a name="line-183"></a> <a name="line-184"></a> <span class='hs-layout'>,</span><span class='hs-str'>"expandAccountNames"</span> <span class='hs-varop'>~:</span> <a name="line-185"></a> <span class='hs-varid'>expandAccountNames</span> <span class='hs-keyglyph'>[</span><span class='hs-str'>"assets:cash"</span><span class='hs-layout'>,</span><span class='hs-str'>"assets:checking"</span><span class='hs-layout'>,</span><span class='hs-str'>"expenses:vacation"</span><span class='hs-keyglyph'>]</span> <span class='hs-varop'>`is`</span> <a name="line-186"></a> <span class='hs-keyglyph'>[</span><span class='hs-str'>"assets"</span><span class='hs-layout'>,</span><span class='hs-str'>"assets:cash"</span><span class='hs-layout'>,</span><span class='hs-str'>"assets:checking"</span><span class='hs-layout'>,</span><span class='hs-str'>"expenses"</span><span class='hs-layout'>,</span><span class='hs-str'>"expenses:vacation"</span><span class='hs-keyglyph'>]</span> <a name="line-187"></a> <a name="line-188"></a> <span class='hs-layout'>,</span><span class='hs-str'>"isAccountNamePrefixOf"</span> <span class='hs-varop'>~:</span> <span class='hs-keyword'>do</span> <a name="line-189"></a> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`isAccountNamePrefixOf`</span> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>False</span> <a name="line-190"></a> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`isAccountNamePrefixOf`</span> <span class='hs-str'>"assets:bank"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>True</span> <a name="line-191"></a> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`isAccountNamePrefixOf`</span> <span class='hs-str'>"assets:bank:checking"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>True</span> <a name="line-192"></a> <span class='hs-str'>"my assets"</span> <span class='hs-varop'>`isAccountNamePrefixOf`</span> <span class='hs-str'>"assets:bank"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>False</span> <a name="line-193"></a> <a name="line-194"></a> <span class='hs-layout'>,</span><span class='hs-str'>"isSubAccountNameOf"</span> <span class='hs-varop'>~:</span> <span class='hs-keyword'>do</span> <a name="line-195"></a> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`isSubAccountNameOf`</span> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>False</span> <a name="line-196"></a> <span class='hs-str'>"assets:bank"</span> <span class='hs-varop'>`isSubAccountNameOf`</span> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>True</span> <a name="line-197"></a> <span class='hs-str'>"assets:bank:checking"</span> <span class='hs-varop'>`isSubAccountNameOf`</span> <span class='hs-str'>"assets"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>False</span> <a name="line-198"></a> <span class='hs-str'>"assets:bank"</span> <span class='hs-varop'>`isSubAccountNameOf`</span> <span class='hs-str'>"my assets"</span> <span class='hs-varop'>`is`</span> <span class='hs-conid'>False</span> <a name="line-199"></a> <a name="line-200"></a> <span class='hs-keyglyph'>]</span> <a name="line-201"></a> </pre></body> </html>