{-# LANGUAGE RecordWildCards #-}
{-# LANGUAGE DeriveDataTypeable #-}

-- | Convert cmsearch output to Browser Extensible Data (BED) format
--   Testcommand: cmsearchToBED -i /path/to/test.clustal
module Main where
import Prelude 
import System.Console.CmdArgs    
import Bio.RNAlienLibrary
import Data.Either.Unwrap 
import qualified Data.ByteString.Lazy.Char8 as L
import qualified Data.Text as T
import Data.List

data Bed = Bed
  { browserPostition :: T.Text,
    browserSettings :: T.Text,
    bedName :: T.Text,
    bedDescription :: T.Text,
    bedVisibility :: Int,
    bedItemRgb :: Bool,
    bedEntries :: [BedEntry]
  } deriving (Eq, Read)

instance Show Bed where
  show (Bed _browserPostition _browserSettings _bedName _bedDescription _bedVisibility _bedItemRgb _bedEntries) = a ++ b ++ c ++ d ++ e ++ f ++ g
    where a = "browser position " ++ (T.unpack _browserPostition) ++ "\n" 
          b = (T.unpack _browserSettings) ++ "\n" 
          c = "track name=\"" ++ T.unpack _bedName  ++ "\" "
          d = "description=\"" ++ T.unpack _bedDescription ++ "\" "
          e = "visibility=" ++  show _bedVisibility ++ " "
          f = "itemRgb=\"" ++ itemRbg ++ "\"\n"
          itemRbg = if _bedItemRgb then "On" else "Off"
          g = concatMap show _bedEntries
          
  
data BedEntry = BedEntry    
  { chrom :: T.Text,
    chromStart :: Int,
    chromEnd  :: Int,
    chromName :: Maybe T.Text,
    score :: Maybe Int,
    strand :: Maybe Char,
    thickStart :: Maybe Int,
    thickEnd :: Maybe Int,
    color :: Maybe T.Text,
    blockCount :: Maybe Int,
    blockSizes :: Maybe [Int],
    blockStarts :: Maybe [Int]
  } deriving (Eq, Read) 

instance Show BedEntry where
  show (BedEntry _chrom _chromStart _chromEnd _chromName _score _strand _thickStart _thickEnd _color _blockCount _blockSizes _blockStarts) = a ++ b ++ c ++ d ++ e ++ f ++ g ++ h ++ i ++ j ++ k ++ l
    where a = T.unpack _chrom ++ "\t" 
          b = show _chromStart ++ "\t" 
          c = show _chromEnd ++ "\t"
          d = maybe "" T.unpack _chromName ++ "\t"
          e = maybe "" show _score ++ "\t"
          f = maybe "" (\s -> [s])  _strand ++ "\t"
          g = maybe "" show _thickStart ++ "\t"
          h = maybe "" show _thickEnd ++ "\t"
          i = maybe "" T.unpack _color ++ "\t"
          j = maybe "" show _blockCount ++ "\t" 
          k = maybe "" (\sizes -> intercalate "," (map show sizes)) _blockSizes ++ "\t"
          l = maybe "" (\starts -> intercalate "," (map show starts)) _blockStarts ++ "\n"
          
data Options = Options            
  { cmsearchPath :: String,
    inputBrowserSettings :: String,
    inputBedVisibility :: Int,
    inputTrackName :: String,
    inputTrackDescription :: String,
    inputItemRgb :: Bool,
    inputTrackColor :: String,
    sortBed :: Bool,
    withHeader :: Bool
  } deriving (Show,Data,Typeable)

options :: Options
options = Options
  { cmsearchPath = def &= name "i" &= help "Path to input cmsearch file",
    inputBrowserSettings = "browser hide all" &= name "b" &= help "Browser settings. Default: browser hide all",
    inputBedVisibility = (2 :: Int) &= name "y" &= help "Visibility setting of track. Default: 2",
    inputTrackName = "PredictedRNA" &= name "n" &= help "Name of the track Default: PredictedRNA",
    inputTrackDescription = "RNA loci predicted by cmsearch" &= name "d" &= help "Description of the track. Default: RNA loci predicted by cmsearch",
    inputItemRgb = True &= name "r" &= help "RGB Color of the track. Default: True",
    inputTrackColor = "255,0,0" &= name "c" &= help "RGB Color of the track. Default: 255,0,0",
    sortBed = True &= name "s" &= help "Sort entries of Bed file by start end end cooridinates. Default: True",
    withHeader = True &= name "w" &= help "Output contains bed header. Default: True"
  } &= summary "cmsearchToBED - Converts cmsearch file hits to BED file entries" &= help "Florian Eggenhofer 2016" &= verbosity       
            
main :: IO ()
main = do
  Options{..} <- cmdArgs options
  parsedCmsearch <- readCMSearch cmsearchPath
  if (isRight parsedCmsearch)
     then do
       let outputBED = convertcmSearchToBED (fromRight parsedCmsearch) inputBrowserSettings inputTrackName inputTrackDescription inputTrackColor inputBedVisibility inputItemRgb sortBed
       if (isRight outputBED)
         then
           if withHeader
             then print (fromRight outputBED)
             else do
               let output = concatMap show (bedEntries (fromRight outputBED))
               putStr output
         else putStr (fromLeft outputBED)
     else (putStr ("A problem occured converting from cmsearch to BED format:\n " ++ show (fromLeft parsedCmsearch)))

--convertcmSearchToBED :: CMsearch -> String -> String -> Either String String
--convertcmSearchToBED inputcmsearch trackName trackColor
--  | null cmHits = Left "cmsearch file contains no hits" 
--  | otherwise = Right (bedHeader ++ bedEntries)
--  where cmHits = cmsearchHits inputcmsearch
--        bedHeader = "browser position " ++ browserPosition ++ "\nbrowser hide all\ntrack name=\"cmsearch hits\" description=\"cmsearch hits\" visibility=2 itemRgb=\"On\"\n"
--        bedEntries = concatMap (cmsearchHitToBEDentry trackName trackColor) cmHits
--        browserPosition = L.unpack (hitSequenceHeader firstHit) ++ ":" ++ entryStart firstHit ++ "-" ++ entryEnd firstHit
--        firstHit = (head cmHits)        

convertcmSearchToBED :: CMsearch -> String -> String -> String -> String -> Int -> Bool -> Bool -> Either String Bed
convertcmSearchToBED inputcmsearch inputBrowserSettings trackName trackDescription trackColor inputBedVisibility inputItemRgb sortBed
  | null cmHits = Left "cmsearch file contains no hits"
  | otherwise = Right bed
  where cmHits = cmsearchHits inputcmsearch
        --bedHeader = "browser position " ++ browserPosition ++ "\nbrowser hide all\ntrack name=\"cmsearch hits\" description=\"cmsearch hits\" visibility=2 itemRgb=\"On\"\n"
        bedEntries = map (cmsearchHitToBEDentry trackName trackColor) cmHits
        sortedBedEntries = if sortBed then sortBy orderBedEntry bedEntries else bedEntries
        currentBrowserPosition = T.unpack (chrom firstEntry) ++ ":" ++ show (chromStart firstEntry) ++ "-" ++ show (chromEnd firstEntry)
        firstEntry = (head sortedBedEntries)
        bed = Bed (T.pack currentBrowserPosition) (T.pack inputBrowserSettings) (T.pack trackName) (T.pack trackDescription) inputBedVisibility inputItemRgb sortedBedEntries

cmsearchHitToBEDentry :: String -> String -> CMsearchHit -> BedEntry
cmsearchHitToBEDentry hitName hitColor cmHit = entry
  where entry = BedEntry  chromosome entrystart entryend (Just (T.pack hitName)) entryscore entrystrand thickstart thickend entrycolor blocks blockSize blockStart
        chromosome = T.pack (L.unpack (hitSequenceHeader cmHit)) 
        --entryline = L.unpack (hitSequenceHeader cmHit) ++ "\t" ++ entryStart cmHit ++ "\t" ++ entryEnd cmHit++ "\t" ++ (hitName) ++ "\t" ++ "0" ++ "\t" ++ [(hitStrand cmHit)] ++ "\t" ++ show (hitStart cmHit) ++ "\t" ++ show (hitEnd cmHit) ++ "\t" ++ hitColor ++ "\n"
        entrystart = if hitStrand cmHit == '+' then hitStart cmHit else hitEnd cmHit
        entryend = if hitStrand cmHit == '+' then hitEnd cmHit else hitStart cmHit
        entryscore = Just (0 :: Int)
        entrystrand = Just (hitStrand cmHit)
        thickstart = Just entrystart
        thickend = Just entryend
        entrycolor = Just (T.pack hitColor)
        blocks = Just (1 :: Int)
        blockSize = Just [(entryend - entrystart)]
        blockStart = Just [(0 :: Int)]
        

--cmsearchHitToBEDentry :: String -> String -> CMsearchHit -> String
--cmsearchHitToBEDentry hitName hitColor cmHit = entryline
--  where entryline = L.unpack (hitSequenceHeader cmHit) ++ "\t" ++ entryStart cmHit ++ "\t" ++ entryEnd cmHit++ "\t" ++ (hitName) ++ "\t" ++ "0" ++ "\t" ++ [(hitStrand cmHit)] ++ "\t" ++ show (hitStart cmHit) ++ "\t" ++ show (hitEnd cmHit) ++ "\t" ++ hitColor ++ "\n"
        --entrystart = if (hitStrand cmHit) == '+' then show (hitStart cmHit) else show (hitEnd cmHit)
        --entryend = if (hitStrand cmHit) == '+' then show (hitEnd cmHit) else show (hitStart cmHit)

entryStart :: CMsearchHit -> String
entryStart cmHit
  | (hitStrand cmHit) == '+' = show (hitStart cmHit)
  | otherwise = show (hitEnd cmHit)

entryEnd :: CMsearchHit -> String
entryEnd cmHit
  | (hitStrand cmHit) == '+' = show (hitEnd cmHit)
  | otherwise = show (hitStart cmHit) 

orderBedEntry :: BedEntry -> BedEntry -> Ordering
orderBedEntry firstHit secondHit
  | start1 > start2 = GT
  | start1 < start2 = LT
  | otherwise = orderBedEntryEnd firstHit secondHit
    where start1 = chromStart firstHit
          start2 = chromStart secondHit

orderBedEntryEnd :: BedEntry -> BedEntry -> Ordering
orderBedEntryEnd firstHit secondHit
  | end1 > end2 = GT
  | end1 < end2 = LT
  | otherwise = EQ
    where end1 = chromEnd firstHit
          end2 = chromEnd secondHit