mirror of
https://github.com/LCPQ/quantum_package
synced 2024-11-03 20:54:00 +01:00
326 lines
9.2 KiB
Python
Executable File
326 lines
9.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sys
|
|
|
|
# If type in **kwargs
|
|
from ei_handler import Type
|
|
|
|
|
|
class EZFIO_ocaml(object):
|
|
|
|
def __init__(self, **kwargs):
|
|
|
|
for k, v in kwargs.iteritems():
|
|
|
|
try:
|
|
exec "self.{0} = {1}".format(k, v)
|
|
except NameError:
|
|
exec "self.{0} = '{1}'".format(k, v)
|
|
|
|
@property
|
|
def Ocaml_type(self):
|
|
return self.type.ocaml.capitalize()
|
|
|
|
@property
|
|
def ocaml_type(self):
|
|
return self.type.ocaml
|
|
|
|
@property
|
|
def fancy_type(self):
|
|
return self.type.fancy
|
|
|
|
def check_if_init(self, l_arg, name):
|
|
for i in l_arg:
|
|
try:
|
|
exec "self.{0}".format(i)
|
|
except AttributeError:
|
|
msg = "You need to provide a '{0}' for creating {1}"
|
|
raise KeyError(msg.format(i, name))
|
|
|
|
def create_read(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
ezfio_dir = str
|
|
ezfio_name = str
|
|
type = Named_tuple(fancy_type, ocaml_type, fortrant_type)
|
|
|
|
Return the read template
|
|
'''
|
|
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["ezfio_dir", "ezfio_name", "type"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = [
|
|
'(* Read snippet for {self.ezfio_name} *)',
|
|
'let read_{self.ezfio_name} () =',
|
|
' if not (Ezfio.has_{self.ezfio_dir}_{self.ezfio_name} ()) then',
|
|
' get_default "{self.ezfio_name}"',
|
|
' |> {self.Ocaml_type}.of_string',
|
|
' |> Ezfio.set_{self.ezfio_dir}_{self.ezfio_name}',
|
|
' ;',
|
|
' Ezfio.get_{self.ezfio_dir}_{self.ezfio_name} ()']
|
|
|
|
if self.fancy_type:
|
|
l_template += [" |> {self.fancy_type}.of_{self.ocaml_type}"]
|
|
|
|
l_template += [";;"]
|
|
|
|
template = "\n ".join(l_template)
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e n d e r #
|
|
# ~#~#~#~#~#~ #
|
|
|
|
template_rendered = template.format(**locals())
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return template_rendered
|
|
|
|
def create_write(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
ezfio_dir = str
|
|
ezfio_name = str
|
|
type = Named_tuple(fancy_type, ocaml_type, fortrant_type)
|
|
|
|
Return the read template
|
|
'''
|
|
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["ezfio_dir", "ezfio_name", "type"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = ['(* Write snippet for {self.ezfio_name} *)']
|
|
|
|
if self.fancy_type:
|
|
l_template += ['let write_{self.ezfio_name} var = ',
|
|
' {self.fancy_type}.to_{self.ocaml_type} var',
|
|
' |> Ezfio.set_{self.ezfio_dir}_{self.ezfio_name}']
|
|
else:
|
|
l_template += ['let write_{self.ezfio_name} =',
|
|
' Ezfio.set_{self.ezfio_dir}_{self.ezfio_name}']
|
|
|
|
l_template += [';;']
|
|
|
|
template = "\n ".join(l_template)
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e n d e r #
|
|
# ~#~#~#~#~#~ #
|
|
|
|
template_rendered = template.format(**locals())
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return template_rendered
|
|
|
|
def create_type(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
l_provider = [provider_name, ...]
|
|
l_type = [Named_tuple(fancy_type, ocaml_type, fortrant_type), ...]
|
|
|
|
Return the type template
|
|
'''
|
|
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["l_provider", "l_type"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = ["(* Generate type *)",
|
|
"type t = ",
|
|
" {"]
|
|
|
|
for p, t in zip(self.l_provider, self.l_type):
|
|
|
|
if t.fancy:
|
|
l_template += [" {0:<30} : {1}.t;".format(p, t.fancy)]
|
|
else:
|
|
l_template += [" {0:<30} : {1};".format(p, t.ocaml)]
|
|
l_template += [" } with sexp",
|
|
";;"]
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return "\n ".join(l_template)
|
|
|
|
def create_read_global(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
l_provider = [provider_name, ...]
|
|
|
|
Return the read_global template
|
|
'''
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["l_provider"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = ["(* Read all *)",
|
|
"let read() = ",
|
|
" Some",
|
|
" {"]
|
|
|
|
l_template += [" {0:<30} = read_{0} ();".format(p)
|
|
for p in self.l_provider]
|
|
|
|
l_template += [" }",
|
|
";;"]
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return "\n ".join(l_template)
|
|
|
|
def create_write_global(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
l_provider = [provider_name, ...]
|
|
|
|
Return the type template
|
|
'''
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["l_provider"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = ["(* Write all *)",
|
|
"let write{ "]
|
|
l_template += [" {0};".format(p) for p in self.l_provider]
|
|
l_template += [" } ="]
|
|
l_template += [" write_{0:<30} {0};".format(p) for p in self.l_provider]
|
|
l_template += [";;"]
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return "\n ".join(l_template)
|
|
|
|
def create_to_string(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
l_provider = [provider_name, ...]
|
|
l_type = [Named_tuple(fancy_type, ocaml_type, fortrant_type), ...]
|
|
|
|
Return the type template
|
|
'''
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["l_provider", "l_type"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = ['(* to_string*)',
|
|
'let to_string b =',
|
|
' Printf.sprintf "']
|
|
|
|
l_template += ["{0} = %s".format(p) for p in self.l_provider]
|
|
l_template += ['"']
|
|
|
|
for p, t in zip(self.l_provider, self.l_type):
|
|
|
|
if t.fancy:
|
|
str_ = t.fancy
|
|
else:
|
|
str_ = t.ocaml.capitalize()
|
|
|
|
l_template += [" ({0}.to_string b.{1})".format(str_, p)]
|
|
|
|
l_template += [";;"]
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return "\n ".join(l_template)
|
|
|
|
def create_to_rst(self):
|
|
'''
|
|
You need to instantiate the EZFIO_ocaml with this keyword argument
|
|
l_provider = [provider_name, ...]
|
|
l_type = [Named_tuple(fancy_type, ocaml_type, fortrant_type), ...]
|
|
|
|
Return the type template
|
|
'''
|
|
# ~#~#~#~#~#~#~#~ #
|
|
# C h e c k i n g #
|
|
# ~#~#~#~#~#~#~#~ #
|
|
|
|
self.check_if_init(["l_provider", "l_type", "l_doc"],
|
|
sys._getframe().f_code.co_name)
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
# C r e a t e _ t e m pl a t e #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
|
|
|
|
l_template = ['(* to_rst*)',
|
|
'let to_rst b =',
|
|
' Printf.sprintf "']
|
|
|
|
for p, d in zip(self.l_provider, self.l_doc):
|
|
|
|
l_template += ["{0} ::".format(d),
|
|
"",
|
|
" {0} = %s".format(p),
|
|
""]
|
|
l_template += ['"']
|
|
|
|
for p, t in zip(self.l_provider, self.l_type):
|
|
|
|
if t.fancy:
|
|
str_ = t.fancy
|
|
else:
|
|
str_ = t.ocaml.capitalize()
|
|
|
|
l_template += [" ({0}.to_string b.{1})".format(str_, p)]
|
|
|
|
l_template += ["|> Rst_string.of_string",
|
|
";;"]
|
|
|
|
# ~#~#~#~#~#~ #
|
|
# R e t u r n #
|
|
# ~#~#~#~#~#~ #
|
|
return "\n ".join(l_template)
|