module Squeeze.Data.CommandOptions(
CommandOptions(
getIncludeEmpty,
getMaybeRandomSeed,
getMaximumBytes,
getMinimumUsageRatio,
getVerbosity
),
solutionSizeBounds,
deriveMinimumBytes,
subtractFile,
mkCommandOptions
) where
import Control.Arrow((&&&))
import qualified Data.Default
import qualified Distribution.Verbosity
import qualified Factory.Data.Interval
import qualified Squeeze.Data.File as Data.File
import qualified ToolShed.SelfValidate
data CommandOptions ratio = MkCommandOptions {
getIncludeEmpty :: Bool,
getMaximumBytes :: Data.File.FileSize,
getMaybeRandomSeed :: Maybe Int,
getMinimumUsageRatio :: ratio,
getVerbosity :: Distribution.Verbosity.Verbosity
} deriving (Eq, Show)
instance Fractional f => Data.Default.Default (CommandOptions f) where
def = MkCommandOptions {
getIncludeEmpty = False,
getMaximumBytes = 4700000000,
getMaybeRandomSeed = Nothing,
getMinimumUsageRatio = 9 / 10,
getVerbosity = Distribution.Verbosity.normal
}
instance Real ratio => ToolShed.SelfValidate.SelfValidator (CommandOptions ratio) where
getErrors commandOptions = map snd $ filter (($ commandOptions) . fst) [
((< 0) . getMaximumBytes, "invalid maximumBytes; " ++ show (getMaximumBytes commandOptions)),
((< 0) . getMinimumUsageRatio, "invalid minimumUsageRatio; " ++ show (realToFrac $ getMinimumUsageRatio commandOptions :: Double)),
((> 1) . getMinimumUsageRatio, "invalid minimumUsageRatio; " ++ show (realToFrac $ getMinimumUsageRatio commandOptions :: Double))
]
mkCommandOptions
:: Real ratio
=> Bool
-> Data.File.FileSize
-> Maybe Int
-> ratio
-> Distribution.Verbosity.Verbosity
-> CommandOptions ratio
mkCommandOptions includeEmpty maximumBytes maybeRandomSeed minimumUsageRatio verbosity
| ToolShed.SelfValidate.isValid commandOptions = commandOptions
| otherwise = error $ "Squeeze.Data.CommandOptions.mkCommandOptions:\t" ++ ToolShed.SelfValidate.getFirstError commandOptions
where
commandOptions = MkCommandOptions includeEmpty maximumBytes maybeRandomSeed minimumUsageRatio verbosity
deriveMinimumBytes :: RealFrac ratio => CommandOptions ratio -> Data.File.FileSize
deriveMinimumBytes = floor . uncurry (*) . (getMinimumUsageRatio &&& realToFrac . getMaximumBytes)
subtractFile
:: RealFrac ratio
=> Data.File.FileSize
-> CommandOptions ratio
-> CommandOptions ratio
subtractFile fileSize commandOptions
| fileSize < 0 = error $ "Squeeze.Data.CommandOptions.subtractFile:\tnegative file-size=" ++ show fileSize
| maximumBytes' < 0 = error $ "Squeeze.Data.CommandOptions.subtractFile:\tfile-size=" ++ show fileSize ++ " > maximum=" ++ show maximumBytes
| otherwise = commandOptions {
getMaximumBytes = maximumBytes',
getMinimumUsageRatio = if maximumBytes' == 0
then 0
else fromIntegral (max 0 $ deriveMinimumBytes commandOptions - fileSize) / fromIntegral maximumBytes'
} where
maximumBytes, maximumBytes' :: Data.File.FileSize
maximumBytes = getMaximumBytes commandOptions
maximumBytes' = maximumBytes - fileSize
solutionSizeBounds :: RealFrac f => CommandOptions f -> Factory.Data.Interval.Interval Data.File.FileSize
solutionSizeBounds = deriveMinimumBytes &&& getMaximumBytes