LCPQ/quantum_package
10
0
Fork 0
mirror of https://github.com/LCPQ/quantum_package synced 2024-12-23 04:43:50 +01:00
Code Issues Releases Wiki Activity

Cleaning ei_handler.py

Browse Source
This commit is contained in:
Thomas Applencourt 2015-03-30 09:42:12 +02:00
parent d9e6c9f708
commit 6bdbe6a78c
2 changed files with 146 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
scripts/ezfio_interface/ei_handler.py
View File

@ -439,7 +439,8 @@ def save_ezfio_default(module_lower, str_ezfio_default):
"$QPACKAGE_ROOT/data/ezfio_defaults/{0}.ezfio_interface_default".format(module_lower) "$QPACKAGE_ROOT/data/ezfio_defaults/{0}.ezfio_interface_default".format(module_lower)
""" """
root_ezfio_default = "{0}/data/ezfio_defaults/".format(os.environ['QPACKAGE_ROOT']) root_ezfio_default = "{0}/data/ezfio_defaults/".format(
os.environ['QPACKAGE_ROOT'])
path = "{0}/{1}.ezfio_interface_default".format(root_ezfio_default, path = "{0}/{1}.ezfio_interface_default".format(root_ezfio_default,
module_lower) module_lower)
@ -458,6 +459,12 @@ def save_ezfio_default(module_lower, str_ezfio_default):
def create_ocaml_input(dict_ezfio_cfg): def create_ocaml_input(dict_ezfio_cfg):
# ~#~#~#~# #
# I n i t #
# ~#~#~#~# #
from ezfio_generate_ocaml import EZFIO_ocaml
l_provider = [] l_provider = []
l_type = [] l_type = []
l_doc = [] l_doc = []
@ -468,12 +475,14 @@ def create_ocaml_input(dict_ezfio_cfg):
l_type.append(v["type"]) l_type.append(v["type"])
l_doc.append(v["doc"]) l_doc.append(v["doc"])
e_glob = EZFIO_ocaml(l_provider=l_provider,
l_type=l_type,
l_doc=l_doc)
# ~#~#~#~#~#~#~#~# # # ~#~#~#~#~#~#~#~# #
# C r e a t i o n # # C r e a t i o n #
# ~#~#~#~#~#~#~#~# # # ~#~#~#~#~#~#~#~# #
from ezfio_generate_ocaml import EZFIO_ocaml
template = ['(* =~=~ *)', template = ['(* =~=~ *)',
'(* Init *)', '(* Init *)',
'(* =~=~ *)', '(* =~=~ *)',
@ -485,7 +494,7 @@ def create_ocaml_input(dict_ezfio_cfg):
"", "",
"module Full_ci : sig"] "module Full_ci : sig"]
template += [EZFIO_ocaml.create_type(l_provider, l_type)] template += [e_glob.create_type()]
template += [" val read : unit -> t option", template += [" val read : unit -> t option",
" val write : t-> unit", " val write : t-> unit",
@ -494,7 +503,7 @@ def create_ocaml_input(dict_ezfio_cfg):
" val of_rst : Rst_string.t -> t option", " val of_rst : Rst_string.t -> t option",
"end = struct"] "end = struct"]
template += [EZFIO_ocaml.create_type(l_provider, l_type)] template += [e_glob.create_type()]
template += ['', template += ['',
' let get_default = Qpackage.get_ezfio_default "full_ci";;', ' let get_default = Qpackage.get_ezfio_default "full_ci";;',
@ -513,11 +522,9 @@ def create_ocaml_input(dict_ezfio_cfg):
ezfio_dir = d_val["ezfio_dir"] ezfio_dir = d_val["ezfio_dir"]
ezfio_name = d_val["ezfio_name"] ezfio_name = d_val["ezfio_name"]
d = {"ezfio_dir": ezfio_dir, e = EZFIO_ocaml(ezfio_dir=ezfio_dir,
"ezfio_name": ezfio_name, ezfio_name=ezfio_name,
"type": d_val["type"]} type=d_val["type"])
e = EZFIO_ocaml(**d)
template += [e.create_read(), template += [e.create_read(),
e.create_write(), e.create_write(),
@ -528,10 +535,10 @@ def create_ocaml_input(dict_ezfio_cfg):
'(* =~=~=~=~=~=~=~=~=~=~=~=~ *)', '(* =~=~=~=~=~=~=~=~=~=~=~=~ *)',
""] ""]
template += [EZFIO_ocaml.create_read_global(l_provider), template += [e_glob.create_read_global(),
EZFIO_ocaml.create_write_global(l_provider), e_glob.create_write_global(),
EZFIO_ocaml.create_to_string(l_provider, l_type), e_glob.create_to_string(),
EZFIO_ocaml.create_to_rst(l_provider, l_type, l_doc)] e_glob.create_to_rst()]
template += [" include Generic_input_of_rst;;", template += [" include Generic_input_of_rst;;",
" let of_rst = of_rst t_of_sexp;;", " let of_rst = of_rst t_of_sexp;;",

158
scripts/ezfio_interface/ezfio_generate_ocaml.py
View File

@ -2,15 +2,19 @@
import sys import sys
# If type in **kwargs
from ei_handler import Type
class EZFIO_ocaml(object): class EZFIO_ocaml(object):
def __init__(self, **kwargs): def __init__(self, **kwargs):
for k, v in kwargs.iteritems(): for k, v in kwargs.iteritems():
if k == "type":
self.type = kwargs["type"] try:
else: exec "self.{0} = {1}".format(k, v)
except NameError:
exec "self.{0} = '{1}'".format(k, v) exec "self.{0} = '{1}'".format(k, v)
@property @property
@ -25,6 +29,14 @@ class EZFIO_ocaml(object):
def fancy_type(self): def fancy_type(self):
return self.type.fancy 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): def create_read(self):
''' '''
You need to instantiate the EZFIO_ocaml with this keyword argument You need to instantiate the EZFIO_ocaml with this keyword argument
@ -35,12 +47,12 @@ class EZFIO_ocaml(object):
Return the read template Return the read template
''' '''
for i in ["ezfio_dir", "ezfio_name", "type"]: # ~#~#~#~#~#~#~#~ #
try: # C h e c k i n g #
"exec self.{0}".format(i) # ~#~#~#~#~#~#~#~ #
except NameError:
msg = "You need to provide a '{0}' for creating read function" self.check_if_init(["ezfio_dir", "ezfio_name", "type"],
raise KeyError(msg.format(i)) sys._getframe().f_code.co_name)
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# # # ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
# C r e a t e _ t e m pl a t e # # C r e a t e _ t e m pl a t e #
@ -84,12 +96,12 @@ class EZFIO_ocaml(object):
Return the read template Return the read template
''' '''
for i in ["ezfio_dir", "ezfio_name", "type"]: # ~#~#~#~#~#~#~#~ #
try: # C h e c k i n g #
"exec self.{0}".format(i) # ~#~#~#~#~#~#~#~ #
except NameError:
msg = "You need to provide a '{0}' for creating read function" self.check_if_init(["ezfio_dir", "ezfio_name", "type"],
raise KeyError(msg.format(i)) sys._getframe().f_code.co_name)
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~# # # ~#~#~#~#~#~#~#~#~#~#~#~#~#~# #
# C r e a t e _ t e m pl a t e # # C r e a t e _ t e m pl a t e #
@ -120,14 +132,31 @@ class EZFIO_ocaml(object):
# ~#~#~#~#~#~ # # ~#~#~#~#~#~ #
return template_rendered return template_rendered
@staticmethod def create_type(self):
def create_type(l_provider, l_type): '''
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 *)", l_template = ["(* Generate type *)",
"type t = ", "type t = ",
" {"] " {"]
for p, t in zip(l_provider, l_type): for p, t in zip(self.l_provider, self.l_type):
if t.fancy: if t.fancy:
l_template += [" {0:<30} : {1}.t;".format(p, t.fancy)] l_template += [" {0:<30} : {1}.t;".format(p, t.fancy)]
@ -141,15 +170,31 @@ class EZFIO_ocaml(object):
# ~#~#~#~#~#~ # # ~#~#~#~#~#~ #
return "\n ".join(l_template) return "\n ".join(l_template)
@staticmethod def create_read_global(self):
def create_read_global(l_provider): '''
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 *)", l_template = ["(* Read all *)",
"let read() = ", "let read() = ",
" Some", " Some",
" {"] " {"]
l_template += [" {0:<30} = read_{0} ();".format(p) l_template += [" {0:<30} = read_{0} ();".format(p)
for p in l_provider] for p in self.l_provider]
l_template += [" }", l_template += [" }",
";;"] ";;"]
@ -159,14 +204,29 @@ class EZFIO_ocaml(object):
# ~#~#~#~#~#~ # # ~#~#~#~#~#~ #
return "\n ".join(l_template) return "\n ".join(l_template)
@staticmethod def create_write_global(self):
def create_write_global(l_provider): '''
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 *)", l_template = ["(* Write all *)",
"let write{ "] "let write{ "]
l_template += [" {0};".format(p) for p in l_provider] l_template += [" {0};".format(p) for p in self.l_provider]
l_template += [" } ="] l_template += [" } ="]
l_template += [" write_{0:<30} {0};".format(p) for p in l_provider] l_template += [" write_{0:<30} {0};".format(p) for p in self.l_provider]
l_template += [";;"] l_template += [";;"]
# ~#~#~#~#~#~ # # ~#~#~#~#~#~ #
@ -174,17 +234,33 @@ class EZFIO_ocaml(object):
# ~#~#~#~#~#~ # # ~#~#~#~#~#~ #
return "\n ".join(l_template) return "\n ".join(l_template)
@staticmethod def create_to_string(self):
def create_to_string(l_provider, l_type): '''
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*)', l_template = ['(* to_string*)',
'let to_string b =', 'let to_string b =',
' Printf.sprintf "'] ' Printf.sprintf "']
l_template += ["{0} = %s".format(p) for p in l_provider] l_template += ["{0} = %s".format(p) for p in self.l_provider]
l_template += ['"'] l_template += ['"']
for p, t in zip(l_provider, l_type): for p, t in zip(self.l_provider, self.l_type):
if t.fancy: if t.fancy:
str_ = t.fancy str_ = t.fancy
@ -200,14 +276,30 @@ class EZFIO_ocaml(object):
# ~#~#~#~#~#~ # # ~#~#~#~#~#~ #
return "\n ".join(l_template) return "\n ".join(l_template)
@staticmethod def create_to_rst(self):
def create_to_rst(l_provider, l_type, l_doc): '''
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*)', l_template = ['(* to_rst*)',
'let to_rst b =', 'let to_rst b =',
' Printf.sprintf "'] ' Printf.sprintf "']
for p, d in zip(l_provider, l_doc): for p, d in zip(self.l_provider, self.l_doc):
l_template += ["{0} ::".format(d), l_template += ["{0} ::".format(d),
"", "",
@ -215,7 +307,7 @@ class EZFIO_ocaml(object):
""] ""]
l_template += ['"'] l_template += ['"']
for p, t in zip(l_provider, l_type): for p, t in zip(self.l_provider, self.l_type):
if t.fancy: if t.fancy:
str_ = t.fancy str_ = t.fancy